TASK:lookbook上传
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user