WladlenVK commited on
Commit
33accd6
·
unverified ·
1 Parent(s): 5b6867a

Update load_book.py

Browse files
Files changed (1) hide show
  1. 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
- # Добавлен префикс models/, чтобы избежать ошибки "unexpected model name format"
 
43
  embeddings = GoogleGenerativeAIEmbeddings(
44
- model="models/text-embedding-004",
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: Генерация векторов и загрузка в базу ---")