Spaces:
Sleeping
Sleeping
Anton Novoselov
commited on
Commit
·
9cbc6db
1
Parent(s):
3b510d0
change default model
Browse files
agent.py
CHANGED
|
@@ -212,7 +212,7 @@ tools = [
|
|
| 212 |
]
|
| 213 |
|
| 214 |
# Build graph function
|
| 215 |
-
def build_graph(provider: str = "
|
| 216 |
"""Build the graph"""
|
| 217 |
if provider == "google":
|
| 218 |
llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
|
|
|
|
| 212 |
]
|
| 213 |
|
| 214 |
# Build graph function
|
| 215 |
+
def build_graph(provider: str = "huggingface"):
|
| 216 |
"""Build the graph"""
|
| 217 |
if provider == "google":
|
| 218 |
llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
|