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

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -151,8 +151,8 @@ tools = [
151
  # Build graph function
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
 
 
151
  # Build graph function
152
  def build_graph():
153
  """Build the graph"""
154
+ llm = ChatGroq(model="llama3-8b-8192", 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