QuentinL52 commited on
Commit
8c2922b
·
verified ·
1 Parent(s): 2d06d01

Create models.py

Browse files
Files changed (1) hide show
  1. src/models.py +37 -0
src/models.py ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import logging
2
+ from typing import Dict, Any, Optional
3
+
4
+ logger = logging.getLogger(__name__)
5
+
6
+ def load_all_models() -> Dict[str, Any]:
7
+ models = {
8
+ "status": False,
9
+ "deep_learning_analyzer": None,
10
+ "rag_handler": None,
11
+ "llm": None
12
+ }
13
+
14
+ try:
15
+ from src.core.deep_learning_analyzer import MultiModelInterviewAnalyzer
16
+ models["deep_learning_analyzer"] = MultiModelInterviewAnalyzer()
17
+ logger.info("✅ Deep Learning Analyzer chargé")
18
+ except Exception as e:
19
+ logger.error(f"❌ Erreur chargement Deep Learning Analyzer: {e}")
20
+
21
+ try:
22
+ from src.core.rag_handler import get_rag_handler
23
+ models["rag_handler"] = get_rag_handler()
24
+ logger.info("✅ RAG Handler chargé")
25
+ except Exception as e:
26
+ logger.error(f"❌ Erreur chargement RAG Handler: {e}")
27
+
28
+ try:
29
+ from src.config import crew_openai
30
+ models["llm"] = crew_openai()
31
+ logger.info("✅ LLM chargé")
32
+ except Exception as e:
33
+ logger.error(f"❌ Erreur chargement LLM: {e}")
34
+
35
+ models["status"] = all(v is not None for k, v in models.items() if k != "status")
36
+
37
+ return models