Spaces:
Sleeping
Sleeping
update
Browse files
agent.py
CHANGED
|
@@ -20,6 +20,7 @@ class State(TypedDict):
|
|
| 20 |
def get_llm():
|
| 21 |
os.getenv("GROQ_API_KEY")
|
| 22 |
# return init_chat_model("llama-3.3-70b-versatile", model_provider="groq")
|
|
|
|
| 23 |
return init_chat_model("gemini-2.0-flash", model_provider="google_genai")
|
| 24 |
|
| 25 |
def get_graph(llm):
|
|
|
|
| 20 |
def get_llm():
|
| 21 |
os.getenv("GROQ_API_KEY")
|
| 22 |
# return init_chat_model("llama-3.3-70b-versatile", model_provider="groq")
|
| 23 |
+
|
| 24 |
return init_chat_model("gemini-2.0-flash", model_provider="google_genai")
|
| 25 |
|
| 26 |
def get_graph(llm):
|