Merge branch 'main' into main

Author: cogniware-devops

CogniwareIms/backend/app/core/config.py

@@ -5,7 +5,7 @@
 
 import os
 from functools import lru_cache
-from typing import List
+from typing import List, Optional
 
 from pydantic_settings import BaseSettings
 

CogniwareIms/backend/app/init_knowledge_base.py

@@ -14,10 +14,10 @@
 # Add parent directory to path
 sys.path.insert(0, str(Path(__file__).parent.parent))
 
-from app.services.csv_processor import csv_processor  # noqa: E402
-from app.services.embedding_service import embedding_service  # noqa: E402
-from app.services.knowledge_manager import knowledge_manager  # noqa: E402
-from app.services.retrieval_service import retrieval_service  # noqa: E402
+from app.services.csv_processor import csv_processor
+from app.services.embedding_service import embedding_service
+from app.services.knowledge_manager import knowledge_manager
+from app.services.retrieval_service import retrieval_service
 
 logging.basicConfig(
     level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"

CogniwareIms/backend/app/main.py

@@ -9,8 +9,11 @@
 import logging
 import os
 from datetime import datetime
+from pathlib import Path
 from typing import Any, Dict, List, Optional
 
+import httpx
+from app.services.csv_processor import csv_processor
 from app.services.dbqna_service import dbqna_service
 from app.services.doc_summarization import doc_summarization
 
@@ -23,6 +26,7 @@
 from app.services.llm_service import llm_service
 from app.services.retrieval_service import retrieval_service
 from fastapi import (
+    Depends,
     FastAPI,
     File,
     HTTPException,
@@ -614,7 +618,7 @@ async def broadcast(self, message: dict):
         for connection in self.active_connections:
             try:
                 await connection.send_json(message)
-            except Exception:
+            except:
                 pass
 
 

CogniwareIms/backend/app/services/dbqna_service.py

@@ -6,11 +6,11 @@
 Converts natural language to SQL and executes against inventory database
 """
 
-import json
 import logging
 import os
-from typing import Any, Dict
+from typing import Any, Dict, List, Optional
 
+import sqlalchemy
 from sqlalchemy import create_engine, text
 
 from .llm_service import llm_service
@@ -224,7 +224,7 @@ async def health_check(self) -> bool:
             with engine.connect() as conn:
                 conn.execute(text("SELECT 1"))
             return True
-        except Exception:
+        except:
             return False
 
 

CogniwareIms/backend/app/services/embedding_service.py

@@ -100,7 +100,7 @@ async def embed_batch(
                     try:
                         emb = await self.embed_text(text)
                         embeddings.append(emb)
-                    except Exception:
+                    except:
                         embeddings.append([0.0] * 768)  # Zero vector as last resort
 
         return embeddings
@@ -167,7 +167,7 @@ async def health_check(self) -> bool:
             async with httpx.AsyncClient(timeout=httpx.Timeout(5.0)) as client:
                 response = await client.get(f"{self.base_url}/v1/health_check")
                 return response.status_code == 200
-        except Exception:
+        except:
             return False
 
 

CogniwareIms/backend/app/services/file_upload_service.py

@@ -29,7 +29,9 @@
 except ImportError:
     Document = None
 
+from .embedding_service import embedding_service
 from .knowledge_manager import knowledge_manager
+from .retrieval_service import retrieval_service
 
 logger = logging.getLogger(__name__)
 

CogniwareIms/backend/app/services/interactive_agent.py

@@ -126,10 +126,7 @@ async def chat(
             return {
                 "success": False,
                 "error": str(e),
-                "response": (
-                    "I apologize, but I encountered an error processing your request. "
-                    "Please try rephrasing your question."
-                ),
+                "response": "I apologize, but I encountered an error processing your request. Please try rephrasing your question.",
             }
 
     async def _is_database_query(self, message: str) -> bool:
@@ -200,19 +197,9 @@ def _build_chat_messages(
 
         # System prompt based on user role
         role_prompts = {
-            "Consumer": (
-                "You are a helpful AI assistant for product research and PC building. "
-                "Help users find products and make informed decisions."
-            ),
-            "Inventory Manager": (
-                "You are an AI assistant for inventory management. "
-                "Help with stock queries, warehouse operations, and data analysis. "
-                "Be precise with numbers and locations."
-            ),
-            "Super Admin": (
-                "You are an AI assistant for system administration. "
-                "Provide comprehensive insights and administrative support."
-            ),
+            "Consumer": "You are a helpful AI assistant for product research and PC building. Help users find products and make informed decisions.",
+            "Inventory Manager": "You are an AI assistant for inventory management. Help with stock queries, warehouse operations, and data analysis. Be precise with numbers and locations.",
+            "Super Admin": "You are an AI assistant for system administration. Provide comprehensive insights and administrative support.",
         }
 
         system_content = role_prompts.get(user_role, "You are a helpful AI assistant.")

CogniwareIms/backend/app/services/llm_service.py

@@ -107,9 +107,7 @@ async def generate_sql_query(
         """
         schema_str = json.dumps(schema, indent=2)
 
-        system_prompt = (
-            f"""You are an expert SQL generator. Given a database schema and a """
-            f"""natural language question, generate a valid SQL query.
+        system_prompt = f"""You are an expert SQL generator. Given a database schema and a natural language question, generate a valid SQL query.
 
 Database Schema:
 {schema_str}
@@ -139,9 +137,7 @@ async def generate_sql_query(
 
     async def summarize_text(self, text: str, max_length: int = 150) -> str:
         """Summarize long text using OPEA DocSummarization pattern."""
-        prompt = (
-            f"""Summarize the following text in {max_length} words or less. """
-            f"""Focus on key points and important details:
+        prompt = f"""Summarize the following text in {max_length} words or less. Focus on key points and important details:
 
 Text:
 {text}
@@ -152,9 +148,7 @@ async def summarize_text(self, text: str, max_length: int = 150) -> str:
 
     async def extract_entities(self, text: str) -> List[Dict[str, str]]:
         """Extract named entities from text Useful for inventory data extraction."""
-        prompt = (
-            f"""Extract all product names, SKUs, quantities, and locations """
-            f"""from the following text. Return as JSON list.
+        prompt = f"""Extract all product names, SKUs, quantities, and locations from the following text. Return as JSON list.
 
 Text: {text}
 
@@ -212,11 +206,7 @@ async def answer_inventory_question(
 
         context = "\n".join(context_parts)
 
-        system_prompt = (
-            "You are an AI assistant for inventory management. "
-            "Answer questions accurately based on the provided inventory data. "
-            "Be concise and specific."
-        )
+        system_prompt = """You are an AI assistant for inventory management. Answer questions accurately based on the provided inventory data. Be concise and specific."""
 
         return await self.query_with_context(question, context, system_prompt)
 
@@ -258,7 +248,7 @@ async def health_check(self) -> bool:
             async with httpx.AsyncClient(timeout=httpx.Timeout(5.0)) as client:
                 response = await client.get(f"{self.base_url}/v1/health_check")
                 return response.status_code == 200
-        except Exception:
+        except:
             return False
 
 

CogniwareIms/backend/app/services/opea_client.py

@@ -8,7 +8,6 @@
 from typing import Any, Dict, List, Optional
 
 import httpx
-from fastapi import HTTPException
 
 logger = logging.getLogger(__name__)
 
@@ -122,15 +121,6 @@ async def health_check_all(self) -> Dict[str, str]:
             "dbqna": self.dbqna_url,
         }
 
-        async def check_service(url: str) -> bool:
-            """Check if a service is available."""
-            try:
-                async with httpx.AsyncClient(timeout=httpx.Timeout(5.0)) as client:
-                    response = await client.get(f"{url}/v1/health_check")
-                    return response.status_code == 200
-            except Exception:
-                return False
-
         status = {}
         for name, url in services.items():
             status[name] = await check_service(url)

CogniwareIms/backend/app/services/retrieval_service.py

@@ -253,7 +253,7 @@ async def health_check(self) -> Dict[str, Any]:
             async with httpx.AsyncClient(timeout=httpx.Timeout(5.0)) as client:
                 response = await client.get(f"{self.base_url}/v1/health_check")
                 status["opea_service"] = response.status_code == 200
-        except Exception:
+        except:
             pass
 
         # Check Redis
@@ -262,7 +262,7 @@ async def health_check(self) -> Dict[str, Any]:
             await client.ping()
             status["redis"] = True
             status["document_count"] = await self.count_documents()
-        except Exception:
+        except:
             pass
 
         return status