Spaces:
Sleeping
Sleeping
update
Browse files- agent.py +2 -2
- app_playground.ipynb +0 -0
- requirements.txt +1 -0
agent.py
CHANGED
|
@@ -19,8 +19,8 @@ class State(TypedDict):
|
|
| 19 |
|
| 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 |
|
| 25 |
def get_graph(llm):
|
| 26 |
with open('prompts/system_prompt.md', 'r', encoding='utf-8') as markdown_file:
|
|
|
|
| 19 |
|
| 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):
|
| 26 |
with open('prompts/system_prompt.md', 'r', encoding='utf-8') as markdown_file:
|
app_playground.ipynb
CHANGED
|
The diff for this file is too large to render.
See raw diff
|
|
|
requirements.txt
CHANGED
|
@@ -10,3 +10,4 @@ langchain-core~=0.3.56
|
|
| 10 |
langchain-groq~=0.3.2
|
| 11 |
langchain-community ~=0.3.22
|
| 12 |
wikipedia ~= 1.4.0
|
|
|
|
|
|
| 10 |
langchain-groq~=0.3.2
|
| 11 |
langchain-community ~=0.3.22
|
| 12 |
wikipedia ~= 1.4.0
|
| 13 |
+
langchain-google-genai ~=2.1.3
|