diff --git a/lib/experimental_agents/llm_agents/groq_agent.d.ts b/lib/experimental_agents/llm_agents/groq_agent.d.ts
index cbca0002b..c85c0f768 100644
--- a/lib/experimental_agents/llm_agents/groq_agent.d.ts
+++ b/lib/experimental_agents/llm_agents/groq_agent.d.ts
@@ -1,14 +1,15 @@
 import { AgentFunction } from "../../graphai";
+import { Groq } from "groq-sdk";
 export declare const groqAgent: AgentFunction<{
     model: string;
     query?: string;
     system?: string;
     verbose?: boolean;
-    tools?: Record<string, any>;
+    tools?: Record<string, Groq.Chat.CompletionCreateParams.ToolChoice>;
     temperature?: number;
     max_tokens?: number;
     tool_choice?: string | Record<string, any>;
-}, Record<string, any> | string, string | Array<Record<string, any>>>;
+}, Record<string, any> | string, string | Array<Groq.Chat.CompletionCreateParams.Message>>;
 declare const groqAgentInfo: {
     name: string;
     agent: AgentFunction<{
@@ -16,21 +17,21 @@ declare const groqAgentInfo: {
         query?: string | undefined;
         system?: string | undefined;
         verbose?: boolean | undefined;
-        tools?: Record<string, any> | undefined;
+        tools?: Record<string, Groq.Chat.Completions.CompletionCreateParams.ToolChoice> | undefined;
         temperature?: number | undefined;
         max_tokens?: number | undefined;
         tool_choice?: string | Record<string, any> | undefined;
-    }, string | Record<string, any>, string | Record<string, any>[]>;
+    }, string | Record<string, any>, string | Groq.Chat.Completions.CompletionCreateParams.Message[]>;
     mock: AgentFunction<{
         model: string;
         query?: string | undefined;
         system?: string | undefined;
         verbose?: boolean | undefined;
-        tools?: Record<string, any> | undefined;
+        tools?: Record<string, Groq.Chat.Completions.CompletionCreateParams.ToolChoice> | undefined;
         temperature?: number | undefined;
         max_tokens?: number | undefined;
         tool_choice?: string | Record<string, any> | undefined;
-    }, string | Record<string, any>, string | Record<string, any>[]>;
+    }, string | Record<string, any>, string | Groq.Chat.Completions.CompletionCreateParams.Message[]>;
     samples: never[];
     description: string;
     category: string[];