Add Google providerOptions type assertions

ssdeanx · ssdeanx · commit 0519c0613c5e · 2026-01-16T10:41:43.000-05:00
Widen RequestContext generics to unknown in business/legal agents,
comment out pgQueryTool, and tidy imports/formatting in A2A
coordinators.
diff --git a/src/mastra/a2a/a2aCoordinatorAgent.ts b/src/mastra/a2a/a2aCoordinatorAgent.ts
@@ -1,26 +1,26 @@
+import { GoogleGenerativeAIProviderOptions } from '@ai-sdk/google'
 import { Agent } from '@mastra/core/agent'
 import {
-    createAnswerRelevancyScorer,
-    createToxicityScorer,
+  createAnswerRelevancyScorer,
+  createToxicityScorer,
 } from '@mastra/evals/scorers/prebuilt'
-import { copywriterAgent } from '../agents/copywriterAgent'
-import { editorAgent } from '../agents/editorAgent'
-import { researchAgent } from '../agents/researchAgent'
-import { knowledgeIndexingAgent } from '../agents/knowledgeIndexingAgent'
 import {
-    codeArchitectAgent,
-    codeReviewerAgent,
-    testEngineerAgent,
-    refactoringAgent,
+  codeArchitectAgent,
+  codeReviewerAgent,
+  refactoringAgent,
+  testEngineerAgent,
 } from '../agents/codingAgents'
 import { contentStrategistAgent } from '../agents/contentStrategistAgent'
+import { copywriterAgent } from '../agents/copywriterAgent'
+import { editorAgent } from '../agents/editorAgent'
+import { knowledgeIndexingAgent } from '../agents/knowledgeIndexingAgent'
 import { projectManagementAgent } from '../agents/projectManagementAgent'
-import { researchSynthesisWorkflow } from '../workflows/research-synthesis-workflow'
-import { repoIngestionWorkflow } from '../workflows/repo-ingestion-workflow'
-import { specGenerationWorkflow } from '../workflows/spec-generation-workflow'
+import { researchAgent } from '../agents/researchAgent'
 import { googleAI, googleAIFlashLite } from '../config/google'
 import { pgMemory } from '../config/pg-storage'
-import type { RequestContext } from '@mastra/core/request-context'
+import { repoIngestionWorkflow } from '../workflows/repo-ingestion-workflow'
+import { researchSynthesisWorkflow } from '../workflows/research-synthesis-workflow'
+import { specGenerationWorkflow } from '../workflows/spec-generation-workflow'
 // Import all agents
 
 // Import all workflows
@@ -35,15 +35,15 @@ import type { RequestContext } from '@mastra/core/request-context'
  */
 
 export const a2aCoordinatorAgent = new Agent({
-    id: 'a2aCoordinator',
-    name: 'a2aCoordinator',
-    description:
-        'A2A Coordinator that orchestrates multiple specialized agents in parallel. Routes tasks dynamically, coordinates workflows, and synthesizes results using the A2A protocol.',
-    instructions: ({ requestContext }) => {
-        const userId = requestContext.get('userId')
-        return {
-            role: 'system',
-            content: `You are an A2A (Agent-to-Agent) Coordinator that orchestrates multi-agent workflows.
+  id: 'a2aCoordinator',
+  name: 'a2aCoordinator',
+  description:
+    'A2A Coordinator that orchestrates multiple specialized agents in parallel. Routes tasks dynamically, coordinates workflows, and synthesizes results using the A2A protocol.',
+  instructions: ({ requestContext }) => {
+    const userId = requestContext.get('userId')
+    return {
+      role: 'system',
+      content: `You are an A2A (Agent-to-Agent) Coordinator that orchestrates multi-agent workflows.
 userId: ${userId}
 CORE CAPABILITIES:
 - Orchestrate multiple agents working in parallel
@@ -91,49 +91,49 @@ Use Promise.all() pattern for parallel execution.
 Maximize the use of E2B sandboxes via specialized agents for any code-related tasks.
 Use knowledgeIndexingAgent to provide semantic context for complex queries.
 `,
-            providerOptions: {
-                google: {
-                    thinkingConfig: {
-                        thinkingLevel: 'high',
-                        includeThoughts: true,
-                        thinkingBudget: -1,
-                    },
-                    mediaResolution: 'MEDIA_RESOLUTION_MEDIUM',
-                    responseModalities: ['TEXT', 'IMAGE'],
-                },
-            },
-        }
-    },
-    model: googleAI,
-    memory: pgMemory,
-    options: {},
-    agents: {
-        researchAgent,
-        knowledgeIndexingAgent,
-        editorAgent,
-        copywriterAgent,
-        codeArchitectAgent,
-        codeReviewerAgent,
-        testEngineerAgent,
-        refactoringAgent,
-        contentStrategistAgent,
-        projectManagementAgent,
-    },
-    workflows: {
-        researchSynthesisWorkflow,
-        repoIngestionWorkflow,
-        specGenerationWorkflow,
+      providerOptions: {
+        google: {
+          thinkingConfig: {
+            thinkingLevel: 'high',
+            includeThoughts: true,
+            thinkingBudget: -1,
+          },
+          mediaResolution: 'MEDIA_RESOLUTION_MEDIUM',
+          responseModalities: ['TEXT', 'IMAGE'],
+        } satisfies GoogleGenerativeAIProviderOptions,
+      },
+    }
+  },
+  model: googleAI,
+  memory: pgMemory,
+  options: {},
+  agents: {
+    researchAgent,
+    knowledgeIndexingAgent,
+    editorAgent,
+    copywriterAgent,
+    codeArchitectAgent,
+    codeReviewerAgent,
+    testEngineerAgent,
+    refactoringAgent,
+    contentStrategistAgent,
+    projectManagementAgent,
+  },
+  workflows: {
+    researchSynthesisWorkflow,
+    repoIngestionWorkflow,
+    specGenerationWorkflow,
+  },
+  maxRetries: 5,
+  tools: {},
+  scorers: {
+    relevancy: {
+      scorer: createAnswerRelevancyScorer({ model: googleAIFlashLite }),
+      sampling: { type: 'ratio', rate: 0.4 },
     },
-    maxRetries: 5,
-    tools: {},
-    scorers: {
-        relevancy: {
-            scorer: createAnswerRelevancyScorer({ model: googleAIFlashLite }),
-            sampling: { type: 'ratio', rate: 0.4 },
-        },
-        safety: {
-            scorer: createToxicityScorer({ model: googleAIFlashLite }),
-            sampling: { type: 'ratio', rate: 0.3 },
-        },
+    safety: {
+      scorer: createToxicityScorer({ model: googleAIFlashLite }),
+      sampling: { type: 'ratio', rate: 0.3 },
     },
+  },
 })
diff --git a/src/mastra/a2a/codingA2ACoordinator.ts b/src/mastra/a2a/codingA2ACoordinator.ts
@@ -8,6 +8,7 @@ import { googleAIFlashLite } from '../config/google'
 import { log } from '../config/logger'
 import { pgMemory } from '../config/pg-storage'
 
+import { GoogleGenerativeAIProviderOptions } from '@ai-sdk/google'
 import { InternalSpans } from '@mastra/core/observability'
 import {
   codeArchitectAgent,
@@ -197,7 +198,7 @@ When a user's request requires prolonged, structured work across multiple subtas
           },
           mediaResolution: 'MEDIA_RESOLUTION_MEDIUM',
           responseModalities: ['TEXT', 'IMAGE'],
-        },
+        } satisfies GoogleGenerativeAIProviderOptions,
       },
     }
   },
diff --git a/src/mastra/agents/businessLegalAgents.ts b/src/mastra/agents/businessLegalAgents.ts
@@ -71,12 +71,11 @@ export const legalResearchAgent = new Agent({
   instructions: ({
     requestContext,
   }: {
-    requestContext: RequestContext<BusinessRuntimeContext>
+    requestContext: RequestContext<unknown>
   }) => {
     // runtimeContext is read at invocation time
     const userTier = requestContext.get('user-tier') ?? 'free'
     const language = requestContext.get('language') ?? 'en'
-    const responseFormat = requestContext.get('responseFormat') ?? 'json'
     const research = requestContext.get('research') ?? {
       depth: 'extensive',
       scope: 'full',
@@ -88,37 +87,37 @@ export const legalResearchAgent = new Agent({
     return {
       role: 'system',
       content: `You are a Senior Legal Research Analyst. Your goal is to research legal topics thoroughly using authoritative sources.
-      Your working with:
-      - User: ${userTier}
-      - Language: ${language}
-      - Research: Depth ${research.depth}, Scope ${research.scope}
-      - Analysis: Depth ${analysis.depth}, Scope ${analysis.scope}
-
-**Key Guidelines:**
-- Focus on primary sources: statutes, case law, regulations
-- Evaluate authority and jurisdiction
-- Provide confidence assessments for findings
-- Cite sources properly
-
-**Rules:**
-- **Tool Efficiency:** Do NOT use the same tool repetitively or back-to-back for the same query.
-
-**Process:**
-1. Break down legal issues into specific queries
-2. Search authoritative databases
-3. Evaluate relevance and authority
-4. Extract key insights and follow-up questions
-5. Synthesize findings with confidence levels
-
-**Examples:**
-- Query: "Breach of contract remedies in California"
-  → Research California Civil Code, relevant case law, provide summary with citations
-- Query: "Data privacy regulations for EU businesses"
-  → Analyze GDPR requirements, enforcement cases, compliance implications
-
-**Output:** Return findings in JSON format with queries, results, summary, sources, and confidence level.
-${PGVECTOR_PROMPT}
-      `,
+                              Your working with:
+                              - User: ${userTier}
+                              - Language: ${language}
+                              - Research: Depth ${(research as any).depth ?? 'extensive'}, Scope ${(research as any).scope ?? 'full'}
+                              - Analysis: Depth ${(analysis as any).depth ?? 'extensive'}, Scope ${(analysis as any).scope ?? 'full'}
+
+                        **Key Guidelines:**
+                        - Focus on primary sources: statutes, case law, regulations
+                        - Evaluate authority and jurisdiction
+                        - Provide confidence assessments for findings
+                        - Cite sources properly
+
+                        **Rules:**
+                        - **Tool Efficiency:** Do NOT use the same tool repetitively or back-to-back for the same query.
+
+                        **Process:**
+                        1. Break down legal issues into specific queries
+                        2. Search authoritative databases
+                        3. Evaluate relevance and authority
+                        4. Extract key insights and follow-up questions
+                        5. Synthesize findings with confidence levels
+
+                        **Examples:**
+                        - Query: "Breach of contract remedies in California"
+                          → Research California Civil Code, relevant case law, provide summary with citations
+                        - Query: "Data privacy regulations for EU businesses"
+                          → Analyze GDPR requirements, enforcement cases, compliance implications
+
+                        **Output:** Return findings in JSON format with queries, results, summary, sources, and confidence level.
+                        ${PGVECTOR_PROMPT}
+                              `,
       providerOptions: {
         google: {
           thinkingConfig: {
@@ -134,7 +133,7 @@ ${PGVECTOR_PROMPT}
   model: ({
     requestContext,
   }: {
-    requestContext: RequestContext<BusinessRuntimeContext>
+    requestContext: RequestContext<unknown>
   }) => {
     const userTier = requestContext.get('user-tier') ?? 'free'
     if (userTier === 'enterprise') {
@@ -154,7 +153,7 @@ ${PGVECTOR_PROMPT}
     linkExtractorTool,
     htmlToMarkdownTool,
     contentCleanerTool,
-    pgQueryTool,
+    //  pgQueryTool,
     batchWebScraperTool,
     mdocumentChunker,
     evaluateResultTool,
@@ -241,13 +240,12 @@ You are a Senior Contract Analyst. Analyze legal documents for risks, obligation
       `,
       providerOptions: {
         google: {
-          structuredOutput: true,
           thinkingConfig: {
             includeThoughts: true,
             thinkingBudget: -1,
           },
           responseModalities: ['TEXT'],
-        },
+        } satisfies GoogleGenerativeAIProviderOptions,
       },
     }
   },
@@ -337,13 +335,13 @@ You are a Compliance Officer. Monitor regulatory compliance and identify risks a
       `,
       providerOptions: {
         google: {
-          structuredOutput: true,
+
           thinkingConfig: {
             includeThoughts: true,
             thinkingBudget: -1,
           },
           responseModalities: ['TEXT'],
-        },
+        } satisfies GoogleGenerativeAIProviderOptions,
       },
     }
   },
@@ -434,13 +432,12 @@ You are a Chief Strategy Officer with legal expertise. Align business strategy w
       `,
       providerOptions: {
         google: {
-          structuredOutput: true,
           thinkingConfig: {
             includeThoughts: true,
             thinkingBudget: -1,
           },
           responseModalities: ['TEXT'],
-        },
+        } satisfies GoogleGenerativeAIProviderOptions,
       },
     }
   },
