PradeepBodhi commited on
Commit
ccde62e
·
verified ·
1 Parent(s): c30e8fc

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -1
agent.py CHANGED
@@ -3,6 +3,7 @@ from dotenv import load_dotenv
3
  from langgraph.graph import START, StateGraph, MessagesState
4
  from langgraph.prebuilt import tools_condition, ToolNode
5
  from langchain_groq import ChatGroq
 
6
  from langchain_community.tools.tavily_search import TavilySearchResults
7
  from langchain_community.document_loaders import WikipediaLoader, ArxivLoader
8
  from langchain_community.vectorstores import SupabaseVectorStore
@@ -151,7 +152,7 @@ tools = [
151
  def build_graph():
152
  """Build the graph"""
153
  # llm = ChatGroq(model="compound-beta", temperature=0)
154
- llm = ChatGroq(model="compound-beta", temperature=0, api_key=os.getenv("GROQ_API_KEY"))
155
  # Bind tools to LLM
156
  llm_with_tools = llm.bind_tools(tools)
157
 
 
3
  from langgraph.graph import START, StateGraph, MessagesState
4
  from langgraph.prebuilt import tools_condition, ToolNode
5
  from langchain_groq import ChatGroq
6
+ from langchain_openai import ChatOpenAI
7
  from langchain_community.tools.tavily_search import TavilySearchResults
8
  from langchain_community.document_loaders import WikipediaLoader, ArxivLoader
9
  from langchain_community.vectorstores import SupabaseVectorStore
 
152
  def build_graph():
153
  """Build the graph"""
154
  # llm = ChatGroq(model="compound-beta", temperature=0)
155
+ llm = ChatOpenAI(model = "gpt-3.5-turbo", temperature=0)
156
  # Bind tools to LLM
157
  llm_with_tools = llm.bind_tools(tools)
158