sid385 commited on
Commit
fa5ffee
·
verified ·
1 Parent(s): 2ceea4d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -12,7 +12,7 @@ from langchain.agents import create_react_agent
12
  from langchain.agents.agent import RunnableAgent
13
  from langchain.agents.tools import Tool
14
  from langchain_core.messages import HumanMessage
15
- from langchain_openai import ChatOpenAI
16
 
17
 
18
  # (Keep Constants as is)
@@ -30,7 +30,8 @@ class BasicAgent:
30
  if not OPENAI_API_KEY:
31
  raise ValueError("OPENAI_API_KEY environment variable not found.")
32
 
33
- self.llm = ChatOpenAI(model="gpt-4o", api_key=OPENAI_API_KEY)
 
34
 
35
  # Example tools
36
  self.tools = [
 
12
  from langchain.agents.agent import RunnableAgent
13
  from langchain.agents.tools import Tool
14
  from langchain_core.messages import HumanMessage
15
+ from langchain_google_genai import ChatGoogleGenerativeAI
16
 
17
 
18
  # (Keep Constants as is)
 
30
  if not OPENAI_API_KEY:
31
  raise ValueError("OPENAI_API_KEY environment variable not found.")
32
 
33
+ self.llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash")
34
+
35
 
36
  # Example tools
37
  self.tools = [