prompts change

app_one.py

@@ -44,11 +44,14 @@ from uuid import uuid4
 #os.environ["LANGCHAIN_API_KEY"] = getpass.getpass("LangSmith_API_Key: ")
 
 # PROMPTS IN USE
-duckduckgo_description = "Useful for when you need to answer questions about aml."
+duckduckgo_description = "Useful for when you need to answer questions about aml. "
 
-aml_act_retriever_description = "Searches and returns excerpts from the aml act."
+aml_act_retriever_description = "Searches and returns information from the aml act in Australia."
 
-agent_prompt = " Only conduct DuckDuckGo searches when asked about Anti Money Laundering (aml). "
+agent_prompt = """ When asked about Anti Money Laundering use the retriever tool.
+If the information is not there make DuckDuckGo searches and add this text to the query: site:austrac.gov.au
+
+Answer the follow question: """
 # If the question is not about aml answer with: I don't know.
      
 
@@ -75,7 +78,7 @@ retreiver_tool = create_retriever_tool(
 )
 
 tool_belt = [DuckDuckGoSearchRun(description=duckduckgo_description +
-    "Input should be a search query."), retreiver_tool] 
+    " Input should be a search query."), retreiver_tool] 
 
 tool_executor = ToolExecutor(tool_belt)
 
@@ -163,25 +166,24 @@ def print_messages(messages):
 @cl.on_chat_start
 async def on_chat_start():
 
-    runnable =  app #| output_parser    # | StrOutputParser()
+    runnable = app     #| output_parser    # | StrOutputParser()
     
     cl.user_session.set("runnable", runnable)
 
 
 @cl.on_message
 async def on_message(message: cl.Message):
-
-    print("Query content----------", message.content)
+    
+    runnable = cl.user_session.get("runnable")
 
     input_message = HumanMessage(content=(agent_prompt + message.content))
 
-    response = app.invoke({"messages": [input_message]})
+    response = runnable.invoke({"messages": [input_message]})
         
     await cl.Message(
         content=response["messages"][-1].content).send()
     
     print_messages(response)
 
-    print("Answer content----------", response["messages"][-1].content)