Prasanthkumar commited on
Commit
a2868dc
·
verified ·
1 Parent(s): 166ba87

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -95,7 +95,7 @@ def build_graph(provider: str = "groq"):
95
  llm_with_tools = llm.bind_tools(tools)
96
 
97
  def assistant(state: MessagesState):
98
- return {"messages": [llm_with_tools.invoke(state["messages\])]}
99
 
100
  def retriever(state: MessagesState):
101
  similar_docs = vector_store.similarity_search(state["messages"][0].content)
 
95
  llm_with_tools = llm.bind_tools(tools)
96
 
97
  def assistant(state: MessagesState):
98
+ return {"messages": [llm_with_tools.invoke(state["messages"])]}
99
 
100
  def retriever(state: MessagesState):
101
  similar_docs = vector_store.similarity_search(state["messages"][0].content)