From 18e4d7042fc0c51d453a2f975c34e5bb05fc7a0d Mon Sep 17 00:00:00 2001
From: "Ibrahim H." <bitsnaps@yahoo.fr>
Date: Thu, 1 Aug 2024 19:19:13 +0100
Subject: [PATCH] Improve GroqAgent

Use any model from groq, instead of hard coded model_id
---
 AstraOpenSourceSwarm/GroqAgent/GroqAgent.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/AstraOpenSourceSwarm/GroqAgent/GroqAgent.py b/AstraOpenSourceSwarm/GroqAgent/GroqAgent.py
index f93f9d1..6eb44e2 100644
--- a/AstraOpenSourceSwarm/GroqAgent/GroqAgent.py
+++ b/AstraOpenSourceSwarm/GroqAgent/GroqAgent.py
@@ -2,10 +2,10 @@
 
 
 class GroqAgent(Agent):
-    def __init__(self):
+    def __init__(self, model_id: str = 'llama-3.1-70b-versatile'):
         super().__init__(
             name="GroqAgent",
-            description="uses llama3 with groq",
+            description="uses any model available from groq",
             instructions="./instructions.md",
             files_folder="./files",
             schemas_folder="./schemas",
@@ -13,7 +13,7 @@ def __init__(self):
             tools_folder="./tools",
             temperature=0.3,
             max_prompt_tokens=25000,
-            model="groq/llama3-8b-8192"
+            model=f"groq/{model_id}"
         )
         
     def response_validator(self, message):