aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

从向量数据库中检索图片并集成到chat-robot

Browse Source
This commit is contained in:
xupei
2024-10-30 15:00:30 +08:00
parent 93b284721f
commit c508ddea48
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/service/search_image_with_text/service.py
View File

@@ -67,7 +67,7 @@ embedding_fn = OllamaEmbeddingFunction(url="http://10.1.1.240:11434/api/embeddin
def query(gender, content): def query(gender, content):
collection = client.get_collection("sub_sketches_description", embedding_function=embedding_fn) collection = client.get_collection("sub_sketches_description", embedding_function=embedding_fn)
# 6. 查询相似内容 # 6. 查询相似内容
user_gender = gender # 用户输入的性别 user_gender = gender.lower() # 用户输入的性别
user_content = content # 用户输入的内容 user_content = content # 用户输入的内容
results = collection.query( results = collection.query(