import { Agent } from "@mastra/core/agent";
import { openai } from "@ai-sdk/openai";
import retrievalInstructions from "../../prompts/retrieval-instructions";
import { Composio } from '@composio/core';
import { MastraProvider } from '@composio/mastra';


export const retrievalAgent = new Agent({
  name: "Retrieval",
  id: "retrieval-agent",
  description:
    "The Retrieval agent retrieves information from the database using semantic layer context.",
  instructions: retrievalInstructions,
  model: openai("gpt-4.1-mini"),
  tools: async ({ runtimeContext }) => {
    const composio = new Composio({
      provider: new MastraProvider(),
    });

    // retrieve userId and activeAccount from the runtimeContext
    const userId = runtimeContext.get<'userId', string>('userId');
    const activeAccount = runtimeContext.get<
    'activeAccount',
    Awaited<ReturnType<typeof composio.connectedAccounts.list>>['items'][number]
    >('activeAccount');
    
    // return empty set of tools if activeAccount isn't present
    if (!activeAccount) return {};

    // fetch composio tools and dynamically use them in the agent
    const composioTools = await composio.tools.get(userId, {
      toolkits: [activeAccount.toolkit.slug],
    });

    return composioTools;
  },

});

Error getting serialized tool {
  toolId: 'MICROSOFT_TEAMS_TEAMS_GET_MESSAGE',
  error: TypeError: Cannot read properties of undefined (reading 'def')
      at JSONSchemaGenerator.process (file:///Users/samjoshuva/Documents/Projects/deepspot/deepspot-query-executor/node_modules/zod/v4/core/to-json-schema.js:15:33)
      at Module.toJSONSchema (file:///Users/samjoshuva/Documents/Projects/deepspot/deepspot-query-executor/node_modules/zod/v4/core/to-json-schema.js:811:9)
      at zodToJsonSchema (file:///Users/samjoshuva/Documents/Projects/deepspot/deepspot-query-executor/node_modules/@mastra/schema-compat/dist/chunk-GWTUXMDD.js:8:17)
      at file:///Users/samjoshuva/Documents/Projects/deepspot/deepspot-query-executor/.mastra/output/index.mjs:32802:44
      at Array.reduce (<anonymous>)
      at getSerializedAgentTools (file:///Users/samjoshuva/Documents/Projects/deepspot/deepspot-query-executor/.mastra/output/index.mjs:32787:38)
      at formatAgent (file:///Users/samjoshuva/Documents/Projects/deepspot/deepspot-query-executor/.mastra/output/index.mjs:32981:38)
      at process.processTicksAndRejections (node:internal/process/task_queues:105:5)
      at async getAgentByIdHandler (file:///Users/samjoshuva/Documents/Projects/deepspot/deepspot-query-executor/.mastra/output/index.mjs:37580:18)
      at async dispatch (file:///Users/samjoshuva/Documents/Projects/deepspot/deepspot-query-executor/.mastra/output/index.mjs:5576:17)
}
Error getting serialized tool {
  toolId: 'MICROSOFT_TEAMS_TEAMS_LIST',
  error: TypeError: Cannot read properties of undefined (reading 'def')
      at JSONSchemaGenerator.process (file:///Users/samjoshuva/Documents/Projects/deepspot/deepspot-query-executor/node_modules/zod/v4/core/to-json-schema.js:15:33)
      at Module.toJSONSchema (file:///Users/samjoshuva/Documents/Projects/deepspot/deepspot-query-executor/node_modules/zod/v4/core/to-json-schema.js:811:9)
      at zodToJsonSchema (file:///Users/samjoshuva/Documents/Projects/deepspot/deepspot-query-executor/node_modules/@mastra/schema-compat/dist/chunk-GWTUXMDD.js:8:17)
      at file:///Users/samjoshuva/Documents/Projects/deepspot/deepspot-query-executor/.mastra/output/index.mjs:32802:44
      at Array.reduce (<anonymous>)
      at getSerializedAgentTools (file:///Users/samjoshuva/Documents/Projects/deepspot/deepspot-query-executor/.mastra/output/index.mjs:32787:38)
      at formatAgent (file:///Users/samjoshuva/Documents/Projects/deepspot/deepspot-query-executor/.mastra/output/index.mjs:32981:38)
      at process.processTicksAndRejections (node:internal/process/task_queues:105:5)
      at async getAgentByIdHandler (file:///Users/samjoshuva/Documents/Projects/deepspot/deepspot-query-executor/.mastra/output/index.mjs:37580:18)
      at async dispatch (file:///Users/samjoshuva/Documents/Projects/deepspot/deepspot-query-executor/.mastra/output/index.mjs:5576:17)
}