Spaces:
Runtime error
Runtime error
GabrielJuan349
commited on
Commit
·
cf966ea
1
Parent(s):
659d842
Update agent and requirements
Browse files- agent.py +5 -4
- requirements.txt +2 -1
agent.py
CHANGED
@@ -4,7 +4,7 @@ from langgraph.graph import START, StateGraph, MessagesState
|
|
4 |
from langgraph.prebuilt import tools_condition, ToolNode
|
5 |
from langchain_core.messages import SystemMessage, HumanMessage
|
6 |
from langchain.tools.retriever import create_retriever_tool
|
7 |
-
from
|
8 |
from qdrant_client import QdrantClient
|
9 |
from langchain_google_genai import ChatGoogleGenerativeAI
|
10 |
from langchain_groq import ChatGroq
|
@@ -25,7 +25,7 @@ qdrant = QdrantClient(
|
|
25 |
api_key=os.environ.get("QDRANT_SERVICE_KEY")
|
26 |
)
|
27 |
|
28 |
-
vector_store =
|
29 |
client=qdrant,
|
30 |
embeddings=embeddings,
|
31 |
collection_name="documents",
|
@@ -77,9 +77,10 @@ def build_graph(provider: str = "groq"):
|
|
77 |
|
78 |
def retriever(state: MessagesState):
|
79 |
"""Retriever node"""
|
80 |
-
|
|
|
81 |
example_msg = HumanMessage(
|
82 |
-
content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].
|
83 |
)
|
84 |
return {"messages": [sys_msg] + state["messages"] + [example_msg]}
|
85 |
|
|
|
4 |
from langgraph.prebuilt import tools_condition, ToolNode
|
5 |
from langchain_core.messages import SystemMessage, HumanMessage
|
6 |
from langchain.tools.retriever import create_retriever_tool
|
7 |
+
from langchain_qdrant import QdrantVectorStore
|
8 |
from qdrant_client import QdrantClient
|
9 |
from langchain_google_genai import ChatGoogleGenerativeAI
|
10 |
from langchain_groq import ChatGroq
|
|
|
25 |
api_key=os.environ.get("QDRANT_SERVICE_KEY")
|
26 |
)
|
27 |
|
28 |
+
vector_store = QdrantVectorStore(
|
29 |
client=qdrant,
|
30 |
embeddings=embeddings,
|
31 |
collection_name="documents",
|
|
|
77 |
|
78 |
def retriever(state: MessagesState):
|
79 |
"""Retriever node"""
|
80 |
+
print(state["messages"][0])
|
81 |
+
similar_question = vector_store.similarity_search(state["messages"][0].page_content)
|
82 |
example_msg = HumanMessage(
|
83 |
+
content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].content}",
|
84 |
)
|
85 |
return {"messages": [sys_msg] + state["messages"] + [example_msg]}
|
86 |
|
requirements.txt
CHANGED
@@ -14,4 +14,5 @@ pymupdf
|
|
14 |
wikipedia
|
15 |
python-dotenv
|
16 |
duckduckgo-search
|
17 |
-
sentence-transformers
|
|
|
|
14 |
wikipedia
|
15 |
python-dotenv
|
16 |
duckduckgo-search
|
17 |
+
sentence-transformers
|
18 |
+
langchain_qdrant
|