HCho commited on
Commit
601d218
·
verified ·
1 Parent(s): c662dbe

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -100,7 +100,7 @@ tools = [
100
  ]
101
 
102
 
103
- def build_graph(provider: str):
104
  if provider == "google":
105
  # Google Gemini
106
  llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0,api_key=google_api_key)
 
100
  ]
101
 
102
 
103
+ def build_graph(provider: str = "huggingface"):
104
  if provider == "google":
105
  # Google Gemini
106
  llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0,api_key=google_api_key)