prompt fix and fix to the tool use in doc editors

agent_api.py

@@ -31,7 +31,7 @@ from utils.lightrag_client import LightRAGClient
 from utils import tools
 from agents.lawyer_selector import LawyerSelectorAgent
 from agents.lawyer_messenger import LawyerMessengerAgent
-from prompts.main import SYSTEM_PROMPT_CLIENT, SYSTEM_PROMPT_LAWYER
+from prompts.chat_agent import SYSTEM_PROMPT_CLIENT, SYSTEM_PROMPT_LAWYER
 from agents.pdf_analyzer import PDFAnalyzerAgent
 from agents.doc_editor import DocumentEditorAgent
 from agents.doc_assistant import DocAssistant

agents/doc_editor.py

@@ -168,7 +168,8 @@ Use this plan as your guide for the editing steps to perform."""
         
         for tool_call in last_message.tool_calls:
             tool_name = tool_call['name']
-            tool_func = next((t for t in self.tools if t.name == tool_name), None)
+            # Look up implementation function with underscore prefix
+            tool_func = next((t for t in self.tools if t.name == "_" + tool_name), None)
             
             if not tool_func:
                 logger.warning(f"Tool function not found for {tool_name}")
@@ -204,6 +205,7 @@ Use this plan as your guide for the editing steps to perform."""
                         else:
                             logger.debug("No document_id/user_id, skipping update push")
                 
+
                 intermediate_steps.append(
                     ToolMessage(content=str(result), tool_call_id=tool_call['id'], name=tool_name)
                 )

prompts/{main.py → chat_agent.py}

@@ -65,12 +65,10 @@ The message_lawyer tool is exclusively for contacting lawyers after user explici
 
 # Lawyer specialist system prompt - designed for legal professionals
 SYSTEM_PROMPT_LAWYER = """### Role
-You are an expert legal assistant specializing in law, regulations and directives with deep knowledge of legal frameworks, precedents, and technical legal analysis.
+You are Hexiagon AI, an expert legal assistant specializing in law, regulations and directives with deep knowledge of legal frameworks, precedents, and technical legal analysis.
+Always answer at the first person and politely.
 Lawyer Jurisdiction: {jurisdiction}
 
-### Context
-You are one of the agent in Hexiagon AI platform, a juridical platform that helps lawyers to make their work easier and more efficient. You can help them for specific document research, legal analysis, and legal advice. If the user asks you to build a document redirect him to the "My documents" section.
-
 ### Available Tools
 1. **query_knowledge_graph**: Search legal documents to answer questions about law, regulations and directives.
 2. **search_web**: Search the web for current information, court decisions, or news that may not be in the knowledge graph.