fix create draft

utils/tools.py

@@ -13,6 +13,8 @@ from agents.lawyer_messenger import LawyerMessengerAgent
 from utils.lightrag_client import LightRAGClient, get_lightrag_client, validate_jurisdiction, get_available_jurisdictions
 import resend
 import logging
+
+import httpx
 from bs4 import BeautifulSoup
 
 logger = logging.getLogger(__name__)
@@ -119,7 +121,6 @@ async def search_web(query: str) -> str:
             raise ValueError("TavilySearch not initialized in agent_api.py")
         
         result = await tavily_search.ainvoke({"query": query})
-        import json
         data = json.loads(result) if isinstance(result, str) else result
         
         output = ["🌐 WEB SEARCH RESULTS", "=" * 80]
@@ -297,8 +298,6 @@ async def _retrieve_lawyer_document(
         Document content including extracted_text, summary, actors, and key_details
     """
     try:
-        import httpx
-        
         # Check configuration from environment
         base_url = os.getenv("SUPABASE_BASE_URL")
         cyberlgl_api_key = os.getenv("CYBERLGL_API_KEY")
@@ -349,7 +348,7 @@ async def _retrieve_lawyer_document(
             
             return "\n".join(output)
             
-    except httpx.TimeoutError:
+    except httpx.TimeoutException:
         return "Error: Timeout while retrieving document"
     except httpx.RequestError as e:
         return f"Error: Failed to connect to document server: {str(e)}"
@@ -454,9 +453,8 @@ async def _create_draft_document(
         
         msg, document_id = await _create_draft_document_impl(
             user_id=user_id,
-            title=title,
             content=empty_html,
-            path=path
+            path=path_with_filename
         )
         
         if not document_id:
@@ -534,7 +532,6 @@ async def _create_draft_document(
 
 async def _create_draft_document_impl(
     user_id: str,
-    title: str,
     content: str,
     path: str
 ) -> tuple[str, str]:
@@ -543,31 +540,36 @@ async def _create_draft_document_impl(
     
     Args:
         user_id: User UUID
-        title: Document title
         content: Document HTML content
-        path: Folder path
+        path: Full path with filename (with or without extension)
     
     Returns:
         tuple: (message, document_id) - Success/failure message and document_id (or None on error)
     """
     try:
-        import httpx
-        
         # Check configuration from environment
         base_url = os.getenv("SUPABASE_BASE_URL")
         cyberlgl_api_key = os.getenv("CYBERLGL_API_KEY")
 
-        if path:
-            # Remove leading ./ if present
-            if path.startswith('./'):
-                path = path[2:]
-            # Ensure trailing /
-            if not path.endswith('/'):
-                path += '/'
+        # Extract filename (last part) and folder (everything before)
+        filename = os.path.basename(path)
+        folder = os.path.dirname(path)
+        
+        # Remove extension if present
+        if '.' in filename:
+            filename = filename.rsplit('.', 1)[0]
+        
+        # Build folder path with ./ prefix if needed
+        if folder and not folder.startswith('./'):
+            folder = f"./{folder}"
+        elif not folder:
+            folder = '.'
+        
+        # Build full path
+        if folder == '.':
+            full_path = f"./{filename}.pdf"
         else:
-            path = ''
-            
-        full_path = f"./{path}{title}.pdf"
+            full_path = f"{folder}/{filename}.pdf"
         
         # URL already has /functions/v1/ (correct)
         endpoint_url = f"{base_url}/functions/v1/create-document-from-html"
@@ -604,7 +606,7 @@ async def _create_draft_document_impl(
             else:
                 return f"❌ Error: HTTP {response.status_code} - {response.text}", None
                 
-    except httpx.TimeoutError:
+    except httpx.TimeoutException:
         return "❌ Error: Timeout while saving document", None
     except httpx.RequestError as e:
         return f"❌ Error: Failed to connect to document server: {str(e)}", None