fix the tool

utils/__init__.py

@@ -0,0 +1,18 @@
+#!/usr/bin/env python3
+"""
+Utils package for CyberLegalAI
+"""
+
+from utils.lawyer_profile_formatter import (
+    format_lawyer_profile_from_api,
+    format_lawyer_profiles_from_response,
+    get_lawyer_profile_summary,
+    format_lawyers_as_string
+)
+
+__all__ = [
+    'format_lawyer_profile_from_api',
+    'format_lawyer_profiles_from_response',
+    'get_lawyer_profile_summary',
+    'format_lawyers_as_string'
+]

utils/tools.py

@@ -53,6 +53,28 @@ async def _query_knowledge_graph(
     conversation_history: List[Dict[str, str]],
     jurisdiction: Optional[str] = None
 ) -> str:
+    """
+    Query the legal knowledge graph for relevant information about cyber regulations and directives.
+    
+    This tool searches through a comprehensive knowledge graph containing legal documents,
+    regulations, and directives related to law
+    
+    The knowledge graph is dynamically selected based on jurisdiction:
+    - Romania: Romanian law documents
+    - Bahrain: Bahraini law documents
+    - Default: Falls back to default port if jurisdiction not specified
+    
+    Use this tool when answering legal questions to provide accurate, up-to-date information
+    from official legal sources specific to the user's jurisdiction.
+    
+    Args:
+        query: The legal question or topic to search for in the knowledge graph
+        conversation_history: Optional conversation history for context (automatically provided by the agent)
+        jurisdiction: The jurisdiction name (e.g., "romania", "bahrain") to query the appropriate graph
+    
+    Returns:
+        Relevant legal information from the knowledge graph with context and references
+    """
     try:
         # Validate jurisdiction if provided
         if jurisdiction: