Baktabek commited on
Commit
e555409
·
verified ·
1 Parent(s): f4eadbe

Deploy RAG assistant

Browse files
Files changed (1) hide show
  1. app.py +12 -0
app.py CHANGED
@@ -106,9 +106,21 @@ async def startup_event():
106
  # Qdrant
107
  print("🔌 Подключение к Qdrant...")
108
  # In-memory Qdrant (для HF Spaces)
 
109
  qdrant_client = QdrantClient(":memory:")
110
  print("✅ Qdrant подключен (in-memory режим)")
111
 
 
 
 
 
 
 
 
 
 
 
 
112
  # Embedding model
113
  print("🧠 Загрузка модели эмбеддингов...")
114
  # Используем многоязычную модель для правильной работы с русским языком
 
106
  # Qdrant
107
  print("🔌 Подключение к Qdrant...")
108
  # In-memory Qdrant (для HF Spaces)
109
+ from qdrant_client.models import Distance, VectorParams
110
  qdrant_client = QdrantClient(":memory:")
111
  print("✅ Qdrant подключен (in-memory режим)")
112
 
113
+ # Создаем коллекцию если её нет
114
+ try:
115
+ qdrant_client.get_collection(COLLECTION_NAME)
116
+ print(f"✅ Коллекция {COLLECTION_NAME} уже существует")
117
+ except:
118
+ qdrant_client.create_collection(
119
+ collection_name=COLLECTION_NAME,
120
+ vectors_config=VectorParams(size=384, distance=Distance.COSINE)
121
+ )
122
+ print(f"✅ Коллекция {COLLECTION_NAME} создана (384 dimensions)")
123
+
124
  # Embedding model
125
  print("🧠 Загрузка модели эмбеддингов...")
126
  # Используем многоязычную модель для правильной работы с русским языком