Update app.py

app.py

@@ -1,5 +1,6 @@
 import json
 import operator
+import uuid
 from operator import itemgetter
 from typing import Annotated, Sequence, TypedDict
 
@@ -37,6 +38,7 @@ from langgraph.checkpoint.aiosqlite import AsyncSqliteSaver
 from langgraph.prebuilt import ToolExecutor, ToolInvocation
 from qdrant_client import QdrantClient
 from qdrant_client.models import Distance, VectorParams
+from langchain_core.messages import HumanMessage
 
 # GLOBAL SCOPE - ENTIRE APPLICATION HAS ACCESS TO VALUES SET IN THIS SCOPE #
 # ---- ENV VARIABLES ---- #
@@ -139,7 +141,7 @@ def create_qa_chain(retriever):
 
 @tool
 async def rag_tool(question: str) -> str:
-    """Use this tool to retrieve relevant information from the knowledge base."""
+    """Only use this tool to retrieve research relevant information from the knowledge base."""
     # advanced_rag_prompt=ChatPromptTemplate.from_template(INSTRUCTION_PROMPT_TEMPLATE.format(user_query=question))
     parent_document_retriever_qa_chain = create_qa_chain(parent_document_retriever)
     response = await parent_document_retriever_qa_chain.ainvoke({"question": question})
@@ -391,11 +393,11 @@ def parse_output(input_state):
 
 # Run the agent chain with the input
 # messages=agent_chain.invoke({"question": mental_health_counseling_data['test'][14]['Context']})
-import uuid
+
 # ---------------------------------------------------------------------------------------------------------
 #                                       DEPLOYMENT
 # ---------------------------------------------------------------------------------------------------------
-from langchain_core.messages import HumanMessage
+
 
 @cl.author_rename
 def rename(original_author: str):