Upd OS import

helpers/diagram.py

@@ -4,6 +4,7 @@ helpers/diagram.py
 Mermaid diagram generation with NVIDIA_LARGE (gpt-oss). Includes a CoT retry
 mechanism that feeds back rendering errors to refine the diagram prompt.
 """
+import os
 
 from typing import Dict, Any
 from utils.logger import get_logger

helpers/models.py

@@ -1,5 +1,6 @@
 from typing import List, Dict, Any, Optional
 from pydantic import BaseModel
+import os
 
 
 # ────────────────────────────── Response Models ──────────────────────────────

helpers/namer.py

@@ -4,7 +4,7 @@ Session Auto-Naming Module
 This module handles automatic naming of chat sessions based on the first user query.
 Uses NVIDIA_SMALL API to generate concise, descriptive session names.
 """
-
+import os
 import asyncio
 from typing import Optional
 from utils.logger import get_logger

memo/consolidation.py

@@ -6,7 +6,7 @@ Handles memory consolidation, pruning, and optimization
 to prevent information overload and maintain performance.
 """
 
-import re
+import re, os
 from typing import List, Dict, Any, Optional
 
 from utils.logger import get_logger

memo/context.py

@@ -7,6 +7,7 @@ Functions for retrieving and managing conversation context.
 
 import numpy as np
 from typing import List, Dict, Any, Tuple, Optional
+import os
 
 from utils.logger import get_logger
 from utils.rag.embeddings import EmbeddingClient

memo/conversation.py

@@ -7,6 +7,7 @@ context retrieval, and memory consolidation for natural conversation flow.
 """
 
 from typing import List, Dict, Any, Tuple, Optional
+import os
 
 from utils.logger import get_logger
 from utils.rag.embeddings import EmbeddingClient

memo/history.py

@@ -6,6 +6,7 @@ Functions for managing conversation history and context.
 """
 
 from typing import List, Dict, Any, Tuple, Optional
+import os
 
 from utils.logger import get_logger
 from memo.nvidia import summarize_qa, files_relevance, related_recent_context

memo/legacy.py

@@ -7,6 +7,7 @@ In-memory LRU system for backward compatibility.
 
 from collections import deque, defaultdict
 from typing import List, Dict
+import os
 
 from utils.logger import get_logger
 

memo/plan/execution.py

@@ -6,6 +6,7 @@ Handles memory retrieval execution based on planned strategies.
 """
 
 from typing import List, Dict, Any, Tuple, Optional
+import os
 
 from utils.logger import get_logger
 from utils.rag.embeddings import EmbeddingClient

memo/plan/intent.py

@@ -5,7 +5,7 @@ Intent Detection
 Handles user intent detection for memory planning.
 """
 
-import re
+import re, os
 from typing import List, Dict, Any, Tuple, Optional
 from enum import Enum
 

memo/plan/strategy.py

@@ -7,6 +7,7 @@ Handles memory strategy planning and parameter optimization.
 
 from typing import List, Dict, Any, Tuple, Optional
 from enum import Enum
+import os
 
 from utils.logger import get_logger
 from memo.plan.intent import QueryIntent

memo/planning.py

@@ -7,6 +7,7 @@ intent detection, strategy planning, and execution.
 """
 
 from typing import List, Dict, Any, Tuple, Optional
+import os
 
 from utils.logger import get_logger
 from utils.rag.embeddings import EmbeddingClient

memo/retrieval.py

@@ -6,7 +6,7 @@ Handles intelligent context retrieval, enhancement decisions,
 and input optimization for natural conversation flow.
 """
 
-import re
+import re, os
 from typing import List, Dict, Any, Tuple, Optional
 
 from utils.logger import get_logger

memo/sessions.py

@@ -6,7 +6,7 @@ Handles conversation session tracking, context switching detection,
 and conversation insights.
 """
 
-import re
+import re, os
 import time
 from typing import Dict, Any, Tuple, Optional
 

routes/analytics.py

@@ -4,6 +4,7 @@ Analytics API Routes
 
 Provides endpoints for retrieving user and global analytics data.
 """
+import os
 
 from fastapi import HTTPException, Query
 from typing import Optional, Dict, Any, List

routes/auth.py

@@ -1,5 +1,5 @@
 # routes/auth.py
-import uuid, time, hashlib, secrets
+import uuid, time, hashlib, secrets, os
 from typing import Optional
 from fastapi import Form, HTTPException
 

routes/health.py

@@ -1,6 +1,6 @@
 from helpers.setup import app, rag, logger
 from helpers.models import HealthResponse
-
+import os
 
 @app.get("/healthz", response_model=HealthResponse)
 def health():

routes/projects.py

@@ -1,5 +1,5 @@
 # routes/projects.py
-import uuid
+import uuid, os
 import time
 from datetime import datetime, timezone
 from fastapi import Form, HTTPException

routes/search.py

@@ -1,5 +1,5 @@
 # routes/search.py
-import re, asyncio, time, json
+import re, asyncio, time, json, os
 from typing import List, Dict, Any, Tuple
 from helpers.setup import logger, embedder, gemini_rotator, nvidia_rotator
 from utils.api.router import select_model, generate_answer_with_model, qwen_chat_completion, nvidia_large_chat_completion

routes/sessions.py

@@ -1,5 +1,5 @@
 # routes/sessions.py
-import json, time, uuid
+import json, time, uuid, os
 from datetime import datetime, timezone
 from typing import Any, Dict, List, Optional
 from fastapi import Form, HTTPException