Peter Mutwiri commited on
Commit ·
ed1b3e8
1
Parent(s): f2757ec
fixed indentaion issue on ai services
Browse files- app/service/ai_service.py +13 -13
app/service/ai_service.py
CHANGED
|
@@ -7,21 +7,21 @@ from app.service.embedding_service import embedder
|
|
| 7 |
class AIService:
|
| 8 |
def __init__(self):
|
| 9 |
# Initialize vector DB with error handling
|
| 10 |
-
|
| 11 |
-
|
| 12 |
-
|
| 13 |
-
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
| 18 |
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
|
| 23 |
"ENABLED" if self.vss_available else "DISABLED"
|
| 24 |
-
|
| 25 |
def detect_entity_type(self, org_id: str, columns: list[str], filename: str) -> dict:
|
| 26 |
"""Detect entity type - per-org cache"""
|
| 27 |
columns_str = ",".join(columns)
|
|
|
|
| 7 |
class AIService:
|
| 8 |
def __init__(self):
|
| 9 |
# Initialize vector DB with error handling
|
| 10 |
+
try:
|
| 11 |
+
self.vector_db = get_vector_db()
|
| 12 |
+
self.vss_available = True
|
| 13 |
+
logger.info("✅ Vector DB initialized")
|
| 14 |
+
except Exception as e:
|
| 15 |
+
logger.warning(f"⚠️ Vector DB unavailable: {e}")
|
| 16 |
+
self.vector_db = None
|
| 17 |
+
self.vss_available = False
|
| 18 |
|
| 19 |
+
# Initialize LLM and embedder (always available)
|
| 20 |
+
self.llm = llm_service
|
| 21 |
+
self.embedder = embedder
|
| 22 |
+
logger.info("✅ AI Service initialized (VSS: {})".format(
|
| 23 |
"ENABLED" if self.vss_available else "DISABLED"
|
| 24 |
+
))
|
| 25 |
def detect_entity_type(self, org_id: str, columns: list[str], filename: str) -> dict:
|
| 26 |
"""Detect entity type - per-org cache"""
|
| 27 |
columns_str = ",".join(columns)
|