From 253da9f609c7163d8688c8c68d6480c6e7ac7aab Mon Sep 17 00:00:00 2001
From: isamu <isamu@to-kyo.to>
Date: Fri, 10 May 2024 05:45:29 +0900
Subject: [PATCH] add type

---
 src/experimental_agents/llm_agents/groq_agent.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/experimental_agents/llm_agents/groq_agent.ts b/src/experimental_agents/llm_agents/groq_agent.ts
index 01f2804a2..65174360d 100644
--- a/src/experimental_agents/llm_agents/groq_agent.ts
+++ b/src/experimental_agents/llm_agents/groq_agent.ts
@@ -28,13 +28,13 @@ export const groqAgent: AgentFunction<
     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>>
+  string | Array<Groq.Chat.CompletionCreateParams.Message>
 > = async ({ params, inputs }) => {
   assert(groq !== undefined, "The GROQ_API_KEY environment variable is missing.");
   const { verbose, query, system, tools, tool_choice, max_tokens, temperature } = params;