Просмотр исходного кода

fix error weaviate vector (#1058)

Co-authored-by: jyong <jyong@dify.ai>
tags/0.3.19
Jyong 2 лет назад
Родитель
Сommit
7df56ed617
Аккаунт пользователя с таким Email не найден
1 измененных файлов: 17 добавлений и 10 удалений
  1. 17
    10
      api/commands.py

+ 17
- 10
api/commands.py Просмотреть файл

@@ -329,16 +329,23 @@ def create_qdrant_indexes():
model_name=dataset.embedding_model
)
except Exception:
provider = Provider(
id='provider_id',
tenant_id=dataset.tenant_id,
provider_name='openai',
provider_type=ProviderType.CUSTOM.value,
encrypted_config=json.dumps({'openai_api_key': 'TEST'}),
is_valid=True,
)
model_provider = OpenAIProvider(provider=provider)
embedding_model = OpenAIEmbedding(name="text-embedding-ada-002", model_provider=model_provider)
try:
embedding_model = ModelFactory.get_embedding_model(
tenant_id=dataset.tenant_id
)
dataset.embedding_model = embedding_model.name
dataset.embedding_model_provider = embedding_model.model_provider.provider_name
except Exception:
provider = Provider(
id='provider_id',
tenant_id=dataset.tenant_id,
provider_name='openai',
provider_type=ProviderType.SYSTEM.value,
encrypted_config=json.dumps({'openai_api_key': 'TEST'}),
is_valid=True,
)
model_provider = OpenAIProvider(provider=provider)
embedding_model = OpenAIEmbedding(name="text-embedding-ada-002", model_provider=model_provider)
embeddings = CacheEmbedding(embedding_model)

from core.index.vector_index.qdrant_vector_index import QdrantVectorIndex, QdrantConfig

Загрузка…
Отмена
Сохранить