Update document_generator.py

document_generator.py

@@ -153,6 +153,40 @@ class LLMObservabilityManager:
         return self.get_observations()
 
 
+# File: app.py
+import os
+import json
+import re
+import asyncio
+import time
+from typing import List, Dict, Optional, Any, Callable
+from openai import OpenAI
+import logging
+import functools
+from fastapi import APIRouter, HTTPException, Request
+from fastapi.responses import StreamingResponse
+from pydantic import BaseModel
+from fastapi_cache.decorator import cache
+import psycopg2
+from datetime import datetime
+
+
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
+
+def log_execution(func: Callable) -> Callable:
+    @functools.wraps(func)
+    def wrapper(*args: Any, **kwargs: Any) -> Any:
+        logger.info(f"Executing {func.__name__}")
+        try:
+            result = func(*args, **kwargs)
+            logger.info(f"{func.__name__} completed successfully")
+            return result
+        except Exception as e:
+            logger.error(f"Error in {func.__name__}: {e}")
+            raise
+    return wrapper
+
 # aiclient.py
 
 class AIClient:
@@ -232,40 +266,6 @@ class AIClient:
         return response.choices[0].message.content
 
 
-# File: app.py
-import os
-import json
-import re
-import asyncio
-import time
-from typing import List, Dict, Optional, Any, Callable
-from openai import OpenAI
-import logging
-import functools
-from fastapi import APIRouter, HTTPException, Request
-from fastapi.responses import StreamingResponse
-from pydantic import BaseModel
-from fastapi_cache.decorator import cache
-import psycopg2
-from datetime import datetime
-
-
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
-logger = logging.getLogger(__name__)
-
-def log_execution(func: Callable) -> Callable:
-    @functools.wraps(func)
-    def wrapper(*args: Any, **kwargs: Any) -> Any:
-        logger.info(f"Executing {func.__name__}")
-        try:
-            result = func(*args, **kwargs)
-            logger.info(f"{func.__name__} completed successfully")
-            return result
-        except Exception as e:
-            logger.error(f"Error in {func.__name__}: {e}")
-            raise
-    return wrapper
-
 class DatabaseManager:
     """Manages database operations."""
 
@@ -287,17 +287,6 @@ class DatabaseManager:
                 VALUES (%s, %s, %s);
                 """
                 cur.execute(insert_query, (user_id, user_query, response))
-
-def log_execution(func):
-    @wraps(func)
-    def wrapper(*args, **kwargs):
-        start_time = time.time()
-        result = func(*args, **kwargs)
-        end_time = time.time()
-        print(f"{func.__name__} executed in {end_time - start_time:.2f} seconds")
-        return result
-    return wrapper
-
         
 class DocumentGenerator:
     def __init__(self, ai_client: AIClient):