agent prompts

martinohanlon · martinohanlon · commit b91b7faddd75 · 2024-06-13T13:12:44.000+01:00
diff --git a/agent.py b/agent.py
@@ -1,6 +1,5 @@
 from llm import llm
 from graph import graph
-from utils import get_session_id
 
 # Create a movie chat chain
 
diff --git a/solutions/agent.py b/solutions/agent.py
@@ -51,7 +51,6 @@
 # # from solutions.tools.fewshot import cypher_qa
 # from solutions.tools.finetuned import cypher_qa
 
-# tag::tools[]
 # tools = [
 #     Tool.from_function(
 #         name="General Chat",
@@ -72,15 +71,16 @@
 #         return_direct=True
 #     )
 # ]
-# end::tools[]
 
 # tag::get_memory[]
 def get_memory(session_id):
     return Neo4jChatMessageHistory(session_id=session_id, graph=graph)
 # end::get_memory[]
 
 # tag::agent[]
+# tag::agent_prompt[]
 agent_prompt = hub.pull("hwchase17/react-chat")
+# end::agent_prompt[]
 agent = create_react_agent(llm, tools, agent_prompt)
 agent_executor = AgentExecutor(
     agent=agent,
