TASK:lookbook上传

This commit is contained in:
shahaibo
2024-10-21 23:48:19 +08:00
parent b35885e2be
commit c81c91963f
3 changed files with 77 additions and 2 deletions

View File

@@ -7,7 +7,7 @@ import aiofiles
from openai import OpenAI
from app.service.lookbooks.utils.image_utils import base64_encode_image, generate_text_id
from app.service.lookbooks.utils.openai_utils import wait_for_job_completion
from app.service.lookbooks.config import DOCUMENT_COLLECTION
# 设置日志
logger = logging.getLogger()
@@ -195,7 +195,7 @@ def save_to_vector_db(image_description_results_file, tag, year):
})
# 将图像的描述和元数据添加到向量数据库中
collection.add_texts(texts=image_summaries, metadatas=image_metadatas, ids=list(image_ids))
DOCUMENT_COLLECTION.add_texts(texts=image_summaries, metadatas=image_metadatas, ids=list(image_ids))
logger.info("Successfully saved data to vector database")
except Exception as e: