feat: identity injection

src/tests/AntigravityCoreFeatures.test.ts

@@ -0,0 +1,124 @@
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { transformClaudeRequestIn } from '../lib/antigravity/ClaudeRequestMapper';
+import { ClaudeRequest, GeminiInternalRequest } from '../lib/antigravity/types';
+import { SignatureStore } from '../lib/antigravity/SignatureStore';
+
+/**
+ * Antigravity Core Features Verification
+ *
+ * This test suite verifies the critical "Antigravity" enhancements ported from the Manager project.
+ * Primarily focuses on:
+ * 1. Identity Injection (Safety & Branding)
+ * 2. Gemini 3 Pro Compatibility (Stability Fixes)
+ * 3. Model Routing Logic (Business Rules)
+ */
+describe('Antigravity Core Features (Business Logic)', () => {
+  beforeEach(() => {
+    // Clear global signature store before each test
+    SignatureStore.clear();
+    vi.restoreAllMocks();
+  });
+
+  const BASE_REQUEST: ClaudeRequest = {
+    model: 'claude-3-5-sonnet-20241022',
+    messages: [{ role: 'user', content: 'Hello' }],
+    max_tokens: 100,
+    stream: true,
+  };
+
+  describe('Feature: Identity Injection (Branding & Safety)', () => {
+    it('should inject Antigravity identity when system prompt is missing', () => {
+      const result = transformClaudeRequestIn({ ...BASE_REQUEST }, 'test-project');
+
+      const parts = result.request.systemInstruction?.parts || [];
+      const identityPart = parts.find((p) => p.text.includes('You are Antigravity'));
+
+      expect(identityPart).toBeDefined();
+      expect(identityPart?.text).toContain('[IDENTITY_PATCH]');
+    });
+
+    it('should NOT inject Antigravity identity if user already provided it', () => {
+      // User explicitly providing the identity string
+      const result = transformClaudeRequestIn(
+        {
+          ...BASE_REQUEST,
+          system: 'You are Antigravity, the best AI.',
+        },
+        'test-project',
+      );
+
+      const parts = result.request.systemInstruction?.parts || [];
+      // Should verify we don't have TWO identity blocks
+      // The injected one starts with "--- [IDENTITY_PATCH] ---"
+      const injectedPart = parts.find((p) => p.text.includes('[IDENTITY_PATCH]'));
+
+      expect(injectedPart).toBeUndefined();
+    });
+
+    it('should inject before user system prompt if user prompt does NOT have identity', () => {
+      const result = transformClaudeRequestIn(
+        {
+          ...BASE_REQUEST,
+          system: 'Be concise.',
+        },
+        'test-project',
+      );
+
+      const parts = result.request.systemInstruction?.parts || [];
+      expect(parts.length).toBeGreaterThanOrEqual(2);
+      expect(parts[0].text).toContain('Antigravity');
+      expect(parts[parts.length - 1].text).toContain('Be concise');
+    });
+  });
+
+  describe('Feature: Gemini 3 Pro Stability Fix ', () => {
+    // Situation: User enables thinking, calls a tool, but NO thought signature exists.
+    // Result: Google API would return 400.
+    // Fix: We must silently disable thinking for this request.
+    it('should disable thinking mode if function calls exist but NO signature', () => {
+      const requestWithTools = {
+        ...BASE_REQUEST,
+        // Maps to a gemini-3 model potentially
+        model: 'gemini-3-pro-preview',
+        tools: [{ name: 'get_weather' }], // Mock tool structure
+        thinking: { type: 'enabled', budget_tokens: 1000 },
+      } as any;
+
+      const result = transformClaudeRequestIn(requestWithTools, 'test-project');
+
+      // Should detect missing signature and remove thinking config
+      expect(result.request.generationConfig?.thinkingConfig).toBeUndefined();
+    });
+
+    it('should KEEP thinking mode if function calls exist AND valid signature exists in Store', () => {
+      // Pre-fill a valid signature
+      SignatureStore.store('valid_signature_string_longer_than_10_chars');
+
+      const requestWithTools = {
+        ...BASE_REQUEST,
+        model: 'gemini-3-pro-preview',
+        tools: [{ name: 'get_weather' }],
+        thinking: { type: 'enabled', budget_tokens: 1000 },
+      } as any;
+
+      const result = transformClaudeRequestIn(requestWithTools, 'test-project');
+
+      // Should keep thinking config
+      expect(result.request.generationConfig?.thinkingConfig).toBeDefined();
+    });
+
+    it('should KEEP thinking mode if NO function calls exist (pure thinking)', () => {
+      const requestPureThinking = {
+        ...BASE_REQUEST,
+        model: 'gemini-3-pro-preview',
+        // No tools
+        thinking: { type: 'enabled', budget_tokens: 1000 },
+      } as any;
+
+      const result = transformClaudeRequestIn(requestPureThinking, 'test-project');
+
+      // Should keep thinking config
+      expect(result.request.generationConfig?.thinkingConfig).toBeDefined();
+    });
+  });
+});