Prasanthkumar commited on
Commit
280e958
·
verified ·
1 Parent(s): 2779b5b

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -115,7 +115,7 @@ def build_graph(provider: str = "groq"):
115
  llm_with_tools = llm.bind_tools(tools)
116
 
117
  def assistant(state: MessagesState):
118
- return {"messages": [llm_with_tools.invoke(state["messages"\])]}
119
 
120
  def retriever(state: MessagesState):
121
  docs = vector_store.similarity_search(state["messages"][0].content)
 
115
  llm_with_tools = llm.bind_tools(tools)
116
 
117
  def assistant(state: MessagesState):
118
+ return {"messages": [llm_with_tools.invoke(state["messages"])]}
119
 
120
  def retriever(state: MessagesState):
121
  docs = vector_store.similarity_search(state["messages"][0].content)