wip

agent.py

@@ -1,13 +1,14 @@
 import time
+import os
 from typing import TypedDict, Annotated, Optional
 from langchain_core.messages import AnyMessage, SystemMessage, HumanMessage
-from langchain_openai import ChatOpenAI
+from langchain_openai import ChatOpenAI, OpenAIEmbeddings
 from langgraph.graph import StateGraph, START
 from langgraph.graph.message import add_messages, MessagesState
 from langgraph.prebuilt import ToolNode, tools_condition
 from langchain.vectorstores import Chroma
-from langchain.embeddings.openai import OpenAIEmbeddings
 from langchain.chains import RetrievalQA
+from dotenv import load_dotenv
 
 from agent_tools import *
 
@@ -45,8 +46,7 @@ class CUSTOM_AGENT:
         """
 
     def __init__(self):
-        self.llm = ChatOpenAI(name="gpt-4o",
-                              api_key=os.getenv("OPENAI_API_KEY"))
+        self.llm = ChatOpenAI(model="gpt-4o", api_key=os.getenv("OPENAI_API_KEY"), temperature=0)
         self.tools = TOOLS
         self.llm_with_tools = self.llm.bind_tools(self.tools)
         self.sys_msg = sys_msg
@@ -78,15 +78,16 @@ class CUSTOM_AGENT:
 
     def _retriever_node(self, state: MessagesState):
         """Retriever node"""
-        question = state["messages"][-1].content
+        question = state["messages"][ -1 ].content
         docs = self.retriever.get_relevant_documents(question)
 
         if docs:
-            content = "\n".join([d.page_content for d in docs])
+            context = "\n\n".join([d.page_content for d in docs])
         else:
-            content = "No relevant documents found"
+            context = "No relevant documents found"
 
-        return {"messages": [HumanMessage(content=content)]}
+        combined = f"Context:\n{context}\n\nQuestion:\n{question}"
+        return {"messages": [HumanMessage(content=combined)]}
 
     def _assistant(self, state: MessagesState):
         """Assistant node"""

requirements.txt

@@ -22,7 +22,6 @@ ddgs
 rank_bm25
 openai
 langchain_openrouter
-langchain_google_genai
 langchain_openai
 google-genai
 openpyxl

test.py

@@ -107,7 +107,7 @@ questions = [
 # Test
 if __name__ == "__main__":
     agent = CUSTOM_AGENT()
-    q = questions[0]
+    q = questions[4]
     print("Question:", q["question"])
     answer = agent.run(q)
     print("Answer:", answer)