Spaces:
Runtime error
Runtime error
Update load_book.py
Browse files- load_book.py +4 -4
load_book.py
CHANGED
|
@@ -39,11 +39,11 @@ def load_book_to_supabase():
|
|
| 39 |
print(f"Текст успешно нарезан на {len(docs)} фрагментов.")
|
| 40 |
|
| 41 |
print("--- Шаг 4: Инициализация эмбеддингов Gemini ---")
|
| 42 |
-
#
|
|
|
|
| 43 |
embeddings = GoogleGenerativeAIEmbeddings(
|
| 44 |
-
model="models/
|
| 45 |
-
google_api_key=gemini_api_key
|
| 46 |
-
task_type="retrieval_document"
|
| 47 |
)
|
| 48 |
|
| 49 |
print("--- Шаг 5: Генерация векторов и загрузка в базу ---")
|
|
|
|
| 39 |
print(f"Текст успешно нарезан на {len(docs)} фрагментов.")
|
| 40 |
|
| 41 |
print("--- Шаг 4: Инициализация эмбеддингов Gemini ---")
|
| 42 |
+
# Используем проверенную модель embedding-001 без task_type (она его не требует так строго)
|
| 43 |
+
# и принудительно указываем версию API v1, чтобы уйти от проблемного v1beta
|
| 44 |
embeddings = GoogleGenerativeAIEmbeddings(
|
| 45 |
+
model="models/embedding-001",
|
| 46 |
+
google_api_key=gemini_api_key
|
|
|
|
| 47 |
)
|
| 48 |
|
| 49 |
print("--- Шаг 5: Генерация векторов и загрузка в базу ---")
|