SillyTavern/SillyTavern-extras
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern-Extras.git synced 2026-04-21 15:09:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #74 from BlipRanger/Chroma-Error-Handling-Patch

Browse Source
This commit is contained in:
Cohee
2023-07-06 08:27:44 +03:00
committed by GitHub
parent b05598fbcb 4ada6d3c7f
commit 560e2eff29
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
server.py
View File

@@ -897,9 +897,14 @@ def chromadb_export():
abort(400, '"chat_id" is required')
chat_id_md5 = hashlib.md5(data["chat_id"].encode()).hexdigest()
collection = chromadb_client.get_collection(
name=f"chat-{chat_id_md5}", embedding_function=chromadb_embed_fn
)
try:
collection = chromadb_client.get_collection(
name=f"chat-{chat_id_md5}", embedding_function=chromadb_embed_fn
)
except Exception as e:
print(e)
abort(400, "Chat collection not found in chromadb")
collection_content = collection.get()
documents = collection_content.get('documents', [])
ids = collection_content.get('ids', [])