GabrielJuan349 commited on
Commit
cf966ea
·
1 Parent(s): 659d842

Update agent and requirements

Browse files
Files changed (2) hide show
  1. agent.py +5 -4
  2. 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 langchain_community.vectorstores import Qdrant
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 = Qdrant(
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
- similar_question = vector_store.similarity_search(state["messages"][0].content)
 
81
  example_msg = HumanMessage(
82
- content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].page_content}",
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