Update model.py
Browse files
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)
|