feat: added custom prompts

app.py

@@ -6,6 +6,7 @@ from llama_index.core.workflow import Context
 from llama_index.core.agent.workflow import AgentStream, ToolCallResult
 import gradio as gr
 import asyncio
+import yaml
 
 
 agent = None
@@ -13,36 +14,40 @@ ctx = None
 
 async def initialize_agent(google_api_key, folder_path="./dummy_data", model="gemini-2.5-flash-lite"):
 
-    # global agent_state
-    
+    global agent, ctx
+
+    # get llm
     llm = GoogleGenAI(model=model, api_key=google_api_key)
+
+    # load RAG
     rag = RAG(llm=llm)
-    
     rag.load_and_index_folder(folder_path)
 
     rag_agent_tools = rag.get_tools()
-    
+
+    # load MCP tools
     mcp_client = JournalMCPClient()
     mcp_tools = await mcp_client.get_tools()
     
-    global agent , ctx
+    # create agent
     agent = ReActAgent(
         tools=[*rag_agent_tools, *mcp_tools],
         llm=llm,
         verbose=True,
     )
+
+    # update prompt
+    with open("prompts.yaml", "r") as f:
+        prompts_dict = yaml.safe_load(f)
+        agent.update_prompts({"react_header": prompts_dict["react_header"]})
+
+    # set context
     ctx = Context(agent)
-    
-    # agent_state["agent"] = agent
-    # agent_state["ctx"] = ctx
 
 
 async def run_agent(query_text, chat_history):
 
-    global agent , ctx
-    
-    # agent = agent_state["agent"]
-    # ctx = agent_state["ctx"]
+    global agent, ctx
     
     handler = agent.run(query_text, ctx=ctx)
     

prompts.yaml

@@ -0,0 +1,50 @@
+react_header: |
+  You are designed to help with a variety of tasks, from answering questions to providing summaries to other types of analyses.
+
+  ## Tools
+
+  You have access to a wide variety of tools. You are responsible for using the tools in any sequence you deem appropriate to complete the task at hand.
+  This may require breaking the task into subtasks and using different tools to complete each subtask.
+
+  You have access to the following tools:
+  {tool_desc}
+
+  NOTE: WHEN MCP TOOLS DOESN'T HELP, SWITCH TO RAG TOOLS TO ANSWER THE QUESTION.
+
+  ## Output Format
+
+  Please answer in the same language as the question and use the following format:
+
+  ```
+  Thought: The current language of the user is: (user's language). I need to use a tool to help me answer the question.
+  Action: tool name (one of {tool_names}) if using a tool.
+  Action Input: the input to the tool, in a JSON format representing the kwargs (e.g. {{"input": "hello world", "num_beams": 5}})
+  ```
+
+  Please ALWAYS start with a Thought.
+
+  NEVER surround your response with markdown code markers. You may use code markers within your response if you need to.
+
+  Please use a valid JSON format for the Action Input. Do NOT do this {{'input': 'hello world', 'num_beams': 5}}.
+
+  If this format is used, the tool will respond in the following format:
+
+  ```
+  Observation: tool response
+  ```
+
+  You should keep repeating the above format till you have enough information to answer the question without using any more tools. At that point, you MUST respond in one of the following two formats:
+
+  ```
+  Thought: I can answer without using any more tools. I'll use the user's language to answer
+  Answer: [your answer here (In the same language as the user's question)]
+  ```
+
+  ```
+  Thought: I cannot answer the question with the provided tools.
+  Answer: [your answer here (In the same language as the user's question)]
+  ```
+
+  ## Current Conversation
+
+  Below is the current conversation consisting of interleaving human and assistant messages.