pieter/eveAI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

- Corrected old reference to catalog embedding model

Browse Source
This commit is contained in:
Josako
2025-03-18 14:45:03 +01:00
parent 09555ae8b0
commit dc64bbc257
1 changed files with 1 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
eveai_workers/tasks.py
View File

@@ -231,12 +231,7 @@ def summarize_chunk(tenant, model_variables, document_version, chunk):
def embed_chunks(tenant, catalog, document_version, chunks): def embed_chunks(tenant, catalog, document_version, chunks):
if catalog.embedding_model: embedding_model, embedding_model_class = get_embedding_model_and_class(tenant.id, catalog.id)
embedding_model, embedding_model_class = get_embedding_model_and_class(tenant.id, catalog.id,
catalog.embedding_model)
else:
raise EveAIInvalidEmbeddingModel(tenant.id, catalog.id)
# Actually embed # Actually embed
try: try:
embeddings = embedding_model.embed_documents(chunks) embeddings = embedding_model.embed_documents(chunks)