HCho commited on
Commit
7c7b8a8
·
verified ·
1 Parent(s): 59976dd

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -103,9 +103,9 @@ tools = [
103
 
104
 
105
  def build_graph(provider: str = "google"):
106
- if provider == "google":
107
  # Google Gemini
108
- llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0,api_key=google_api_key)
109
  # Bind tools to LLM
110
  llm_with_tools = llm.bind_tools(tools)
111
 
 
103
 
104
 
105
  def build_graph(provider: str = "google"):
106
+ #if provider == "google":
107
  # Google Gemini
108
+ llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0,api_key=google_api_key)
109
  # Bind tools to LLM
110
  llm_with_tools = llm.bind_tools(tools)
111