Update agent.py

agent.py

@@ -126,6 +126,8 @@ with open("system_prompt.txt", "r", encoding="utf-8") as f:
 # System message
 sys_msg = SystemMessage(content=system_prompt)
 
+embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2") #  dim=768
+
 
 def create_retriever_tool(persist_directory="vector_store"):
     documents = TextLoader("state_of_the_union.txt").load()
@@ -133,7 +135,7 @@ def create_retriever_tool(persist_directory="vector_store"):
     texts = text_splitter.split_documents(documents)
     retriever = FAISS.from_documents(texts, OpenAIEmbeddings()).as_retriever()
     
-    #docs = retriever.invoke("What did the president say about Ketanji Brown Jackson")
+    docs = retriever.invoke(sys_msg)
     #pretty_print_docs(docs)
 
 
@@ -212,7 +214,6 @@ def build_graph(provider: str = "google"):
     #builder.add_edge("tools", "assistant")
 
     builder = StateGraph(MessagesState)
-    builder.add_node("assistant", assistant)
     builder.add_node("retriever", retriever)
     
     # Retriever ist Start und Endpunkt