modify prompt to return idrect answer

Agents/agent.py

@@ -46,10 +46,14 @@ if not GROQ_API_KEY:
 
 langfuse_handler = CallbackHandler()
 
-with open("Prompts/SystemPromptFinal.txt","r",encoding="utf-8") as f:
+with open("Prompts/SystemPrompt.txt","r",encoding="utf-8") as f:
     systemPrompt = f.read()
 print(f"systemPrompt  {systemPrompt}")
 
+with open("Prompts/SystemPromptFinal.txt","r",encoding="utf-8") as f:
+    systemPromptFinalAnswer = f.read()
+print(f"systemPromptFinalAnswer  {systemPromptFinalAnswer}")
+
 sysMsg = SystemMessage(content=systemPrompt)
 print(f"system message {sysMsg}")
 
@@ -213,7 +217,8 @@ def tools_condition1(
     if hasattr(ai_message, "tool_calls") and len(ai_message.tool_calls) > 0:
         print("tools")
         return "tools"
-    return "__end__"
+    return "ResponseParser"
+    #return "__end__"
 
 
 def build_graph(provider: str="groq"):
@@ -238,6 +243,12 @@ def build_graph(provider: str="groq"):
         raise ValueError("Invalid Provider. Choose 'groq or 'huggingface'")
     llmWithTools = llm.bind_tools(tools)
 
+    def parseResponse(state:MessagesState):
+        finalContent = str(state["messages"][-1]["content"])
+        formattedMessage = [{"role": "system", "content": systemPromptFinalAnswer}] + \
+                [{"role":"user","content":finalContent} ]
+        return {"messages": [llmWithTools.invoke(formattedMessage)]}
+
     def assistant(state: MessagesState):
         #msg
         #msg = {"messages": [sysMsg] + state["messages"]}
@@ -264,13 +275,16 @@ def build_graph(provider: str="groq"):
     #builder.add_node("retriever",retriever)
     builder.add_node("assistant",assistant)
     builder.add_node("tools", create_tool_node_with_fallback(tools))
+    #builder.add_node("ResponseParser",parseResponse)
     builder.add_edge(START,"assistant")
 
     builder.add_conditional_edges(
         "assistant",
-        tools_condition,
+        tools_condition
     )
     builder.add_edge("tools","assistant")
+    #builder.add_edge("ResponseParser",END)
+
     '''if(directPath):
         builder.add_edge("assistant","tools")
         builder.add_edge("tools",END)

app.py

@@ -75,12 +75,13 @@ class BasicAgent:
 
         #self.graph.get_graph().draw_mermaid_png()
         answer = messages['messages'][-1].content
+        print(f"ANWSER: {answer}")
         #tools_calls = messages[-1].tool_calls
         #print(f"tool calls : {str(tools_calls)}")
         #print(messages[-1].__dict__)
         #print(f"full answer : {answer}")
         #print("Assistant Output:", messages[-1])
-        return answer[50:]
+        return answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """