Adding an empty retriever. Next, we'll add more tools.

agent.py

@@ -12,17 +12,17 @@ from langchain_groq import ChatGroq
 
 HUGGINGFACEHUB_API_TOKEN = getenv("HUGGINGFACEHUB_API_TOKEN")
 
-# Making the agent
-#llm = HuggingFaceEndpoint(
-#    repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-#    huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
-#)
 
-#chat = ChatHuggingFace(llm=llm, verbose=True)
+# load the system prompt from the file
+with open("prompt.txt", "r", encoding="utf-8") as f:
+    system_prompt = f.read()
 
+# System message
+sys_msg = SystemMessage(content=system_prompt)
+
+# Loading the assistant
 chat = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
 
-#chat = ChatGroq(model="qwen-qwq-32b", temperature=0) 
 tools = [add, 
          substract, 
          multiply, 
@@ -40,15 +40,22 @@ def simple_graph():
     ## Defining our nodes
     def assistant(state: MessagesState):
         """Assistant node"""
-        return {"messages": state["messages"] + [chat_with_tools.invoke(state["messages"])]}
+        return {"messages": [chat_with_tools.invoke(state["messages"])]}
+
+    def retriever(state: MessagesState):
+        """Retriever node"""
+        # I don't want to use a Retriever / Using similar questions.
+        return {"messages": [sys_msg] + state["messages"]}
     
     # Build graph / nodes
     builder = StateGraph(MessagesState)
+    builder.add_node("retriever", retriever) # Assistant
     builder.add_node("assistant", assistant) # Assistant
     builder.add_node("tools", ToolNode(tools)) # Tools
     
     # Logic / edges
-    builder.add_edge(START, "assistant")
+    builder.add_edge(START, "retriever")
+    builder.add_edge("retriever", "assistant")
     builder.add_conditional_edges("assistant", tools_condition)
     builder.add_edge("tools", "assistant")