Spaces:
Runtime error
Runtime error
Upload agent.py
Browse files
agent.py
CHANGED
|
@@ -176,7 +176,7 @@ def build_graph(provider: str = "openai"):
|
|
| 176 |
elif provider == "anthropic":
|
| 177 |
from langchain.chat_models import ChatAnthropic
|
| 178 |
llm = ChatAnthropic(model="claude-v1", temperature=0)
|
| 179 |
-
|
| 180 |
# Google Gemini
|
| 181 |
llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
|
| 182 |
elif provider == "groq":
|
|
|
|
| 176 |
elif provider == "anthropic":
|
| 177 |
from langchain.chat_models import ChatAnthropic
|
| 178 |
llm = ChatAnthropic(model="claude-v1", temperature=0)
|
| 179 |
+
elif provider == "google":
|
| 180 |
# Google Gemini
|
| 181 |
llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
|
| 182 |
elif provider == "groq":
|