fixed FAISS initial document
Browse files
agent.py
CHANGED
|
@@ -15,6 +15,7 @@ from langchain_community.embeddings import HuggingFaceEmbeddings
|
|
| 15 |
from langchain_community.vectorstores import FAISS
|
| 16 |
from langchain_openai import ChatOpenAI
|
| 17 |
from langchain_core.messages import HumanMessage, SystemMessage
|
|
|
|
| 18 |
from langchain.tools.retriever import create_retriever_tool
|
| 19 |
from langgraph.graph import StateGraph, START, END, MessagesState
|
| 20 |
from langgraph.prebuilt import ToolNode, tools_condition
|
|
@@ -126,10 +127,10 @@ sys_msg = SystemMessage(content=system_prompt)
|
|
| 126 |
|
| 127 |
embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
|
| 128 |
|
| 129 |
-
|
| 130 |
-
|
| 131 |
-
|
| 132 |
-
|
| 133 |
vector_store = FAISS.from_documents(documents, embeddings)
|
| 134 |
|
| 135 |
retriever_tool = create_retriever_tool(
|
|
|
|
| 15 |
from langchain_community.vectorstores import FAISS
|
| 16 |
from langchain_openai import ChatOpenAI
|
| 17 |
from langchain_core.messages import HumanMessage, SystemMessage
|
| 18 |
+
from langchain_core.documents import Document
|
| 19 |
from langchain.tools.retriever import create_retriever_tool
|
| 20 |
from langgraph.graph import StateGraph, START, END, MessagesState
|
| 21 |
from langgraph.prebuilt import ToolNode, tools_condition
|
|
|
|
| 127 |
|
| 128 |
embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
|
| 129 |
|
| 130 |
+
documents = [
|
| 131 |
+
Document(page_content="What is the capital of France? Paris.", metadata={"source": "example"}),
|
| 132 |
+
Document(page_content="How many legs does a spider have? 8.", metadata={"source": "example"}),
|
| 133 |
+
]
|
| 134 |
vector_store = FAISS.from_documents(documents, embeddings)
|
| 135 |
|
| 136 |
retriever_tool = create_retriever_tool(
|