Spaces:
Sleeping
Sleeping
Update multiagents.py
Browse files- multiagents.py +8 -1
multiagents.py
CHANGED
|
@@ -48,12 +48,19 @@ dotenv.load_dotenv()
|
|
| 48 |
|
| 49 |
# --- Agent wrappers ---
|
| 50 |
groq_model = LiteLLMModel(
|
| 51 |
-
model_id="
|
| 52 |
api_key = os.getenv("GROQ_API_KEY"),
|
| 53 |
temperature=0.1,
|
| 54 |
max_tokens=4000,
|
| 55 |
)
|
| 56 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 57 |
# Pick your agent here (just change one line to test different ones)
|
| 58 |
# My_Agent = BasicAgent
|
| 59 |
My_Agent = groq_model
|
|
|
|
| 48 |
|
| 49 |
# --- Agent wrappers ---
|
| 50 |
groq_model = LiteLLMModel(
|
| 51 |
+
model_id="groq/qwen3-32b", # or any other Groq model like groq/mixtral-8x7b-32768
|
| 52 |
api_key = os.getenv("GROQ_API_KEY"),
|
| 53 |
temperature=0.1,
|
| 54 |
max_tokens=4000,
|
| 55 |
)
|
| 56 |
|
| 57 |
+
# response = litellm.completion(
|
| 58 |
+
# model="groq/llama3-8b-8192",
|
| 59 |
+
# messages=messages,
|
| 60 |
+
# tools=tools,
|
| 61 |
+
# tool_choice="auto", # auto is default, but we'll be explicit
|
| 62 |
+
# )
|
| 63 |
+
|
| 64 |
# Pick your agent here (just change one line to test different ones)
|
| 65 |
# My_Agent = BasicAgent
|
| 66 |
My_Agent = groq_model
|