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