Spaces:
Sleeping
Sleeping
remove base retriever
Browse files- core/ai_provider.py +1 -2
- main.py +0 -1
- 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:
|
| 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-
|
| 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
|