alisamak commited on
Commit
e2a7e20
·
verified ·
1 Parent(s): 278dfd0

Update LG_agent.py

Browse files
Files changed (1) hide show
  1. LG_agent.py +9 -8
LG_agent.py CHANGED
@@ -3,20 +3,21 @@ from langgraph.graph.message import add_messages
3
  from langchain_core.messages import HumanMessage, AIMessage, AnyMessage
4
  from langgraph.prebuilt import ToolNode, tools_condition
5
  from langgraph.graph import START, StateGraph
6
- from langchain_google_genai import ChatGoogleGenerativeAI
7
  from tools import all_tools
8
  import os
9
 
10
  # 1. Setup once
11
- GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
12
- if not GOOGLE_API_KEY:
13
- raise ValueError("Missing GOOGLE_API_KEY environment variable.")
14
 
15
- chat = ChatGoogleGenerativeAI(
16
- model="gemini-1.5-flash", # or "gemini-1.5-pro"
17
- google_api_key=GOOGLE_API_KEY,
18
- temperature=0
19
  )
 
20
  chat_with_tools = chat.bind_tools(all_tools)
21
 
22
  # 2. Define the agent state
 
3
  from langchain_core.messages import HumanMessage, AIMessage, AnyMessage
4
  from langgraph.prebuilt import ToolNode, tools_condition
5
  from langgraph.graph import START, StateGraph
6
+ from langchain_openai import ChatOpenAI
7
  from tools import all_tools
8
  import os
9
 
10
  # 1. Setup once
11
+ OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
12
+ if not OPENAI_API_KEY:
13
+ raise ValueError("Missing OPENAI_API_KEY environment variable.")
14
 
15
+ chat = ChatOpenAI(
16
+ model="gpt-3.5-turbo",
17
+ openai_api_key=OPENAI_API_KEY,
18
+ temperature=0,
19
  )
20
+
21
  chat_with_tools = chat.bind_tools(all_tools)
22
 
23
  # 2. Define the agent state