minh-4T commited on
Commit
4b27f50
·
1 Parent(s): 89c8b6a

remove base retriever

Browse files
Files changed (3) hide show
  1. core/ai_provider.py +1 -2
  2. main.py +0 -1
  3. requirements.txt +1 -1
core/ai_provider.py CHANGED
@@ -16,11 +16,10 @@ class AIProviderManager:
16
 
17
  def get_groq_client(self):
18
  if not self.groq_keys: return None
19
- # Chỉ lấy key, không thay đổi state nên không cần lock
20
  return groq.Groq(api_key=self.groq_keys[self.groq_idx])
21
 
22
  def rotate_groq(self):
23
- with self._lock: # Khóa luồng khi xoay tua để tránh xung đột
24
  if len(self.groq_keys) > 1:
25
  self.groq_idx = (self.groq_idx + 1) % len(self.groq_keys)
26
  logger.info(f"Đã xoay sang Groq Key thứ {self.groq_idx + 1}")
 
16
 
17
  def get_groq_client(self):
18
  if not self.groq_keys: return None
 
19
  return groq.Groq(api_key=self.groq_keys[self.groq_idx])
20
 
21
  def rotate_groq(self):
22
+ with self._lock:
23
  if len(self.groq_keys) > 1:
24
  self.groq_idx = (self.groq_idx + 1) % len(self.groq_keys)
25
  logger.info(f"Đã xoay sang Groq Key thứ {self.groq_idx + 1}")
main.py CHANGED
@@ -148,7 +148,6 @@ async def lifespan(app: FastAPI):
148
 
149
  logger.info("Đang khởi tạo retriever (Qdrant collection router)...")
150
  app.state.retriever = CollectionRouterRetriever(
151
- base_retriever=None,
152
  qdrant_client=client,
153
  embeddings_model=embeddings,
154
  top_n_collections=COLLECTION_ROUTER_TOP_N,
 
148
 
149
  logger.info("Đang khởi tạo retriever (Qdrant collection router)...")
150
  app.state.retriever = CollectionRouterRetriever(
 
151
  qdrant_client=client,
152
  embeddings_model=embeddings,
153
  top_n_collections=COLLECTION_ROUTER_TOP_N,
requirements.txt CHANGED
@@ -8,7 +8,7 @@ tenacity>=8.3.0
8
 
9
  # AI Providers
10
  groq>=0.9.0
11
- google-generativeai>=0.7.0
12
 
13
  # Database & Vector Store
14
  asyncpg>=0.29.0
 
8
 
9
  # AI Providers
10
  groq>=0.9.0
11
+ google-genai>=0.7.0
12
 
13
  # Database & Vector Store
14
  asyncpg>=0.29.0