HCho commited on
Commit
c0f4e2d
·
verified ·
1 Parent(s): 4003c5d

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -103,7 +103,7 @@ tools = [
103
  ]
104
 
105
 
106
- def build_graph(provider: str = "huggingface"):
107
  if provider == "google":
108
  # Google Gemini
109
  llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0,api_key=google_api_key)
@@ -139,7 +139,7 @@ def build_graph(provider: str = "huggingface"):
139
  if __name__ == "__main__":
140
  question = "When was a picture of St. Thomas Aquinas first added to the Wikipedia page on the Principle of double effect?"
141
  # Build the graph
142
- graph = build_graph(provider="huggingface")
143
  # Run the graph
144
  messages = [HumanMessage(content=question)]
145
  messages = graph.invoke({"messages": messages})
 
103
  ]
104
 
105
 
106
+ def build_graph(provider: str = "google"):
107
  if provider == "google":
108
  # Google Gemini
109
  llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0,api_key=google_api_key)
 
139
  if __name__ == "__main__":
140
  question = "When was a picture of St. Thomas Aquinas first added to the Wikipedia page on the Principle of double effect?"
141
  # Build the graph
142
+ graph = build_graph(provider="google")
143
  # Run the graph
144
  messages = [HumanMessage(content=question)]
145
  messages = graph.invoke({"messages": messages})