Update server.py

server.py

@@ -81,12 +81,12 @@ def chromadb_add_messages():
         for m in data["messages"]
     ]
 
-    collection.delete()
-    collection.upsert(
-        ids=ids,
-        documents=documents,
-        metadatas=metadatas,
-    )
+    if len(ids) > 0:
+        collection.upsert(
+            ids=ids,
+            documents=documents,
+            metadatas=metadatas,
+        )
 
     return jsonify({"count": len(ids)})
 
@@ -137,5 +137,21 @@ def chromadb_query():
         ]
 
     return jsonify(messages)
+    
+@app.route("/api/chromadb/purge", methods=["POST"])
+def chromadb_purge():
+    data = request.get_json()
+    if "chat_id" not in data or not isinstance(data["chat_id"], str):
+        abort(400, '"chat_id" is required')
+
+    ip = get_real_ip()
+    chat_id_md5 = hashlib.md5(f'{ip}-{data["chat_id"]}'.encode()).hexdigest()
+    collection = chromadb_client.get_or_create_collection(
+        name=f"chat-{chat_id_md5}", embedding_function=chromadb_embed_fn
+    )
+
+    deleted = collection.delete()
+    print("ChromaDB embeddings deleted", len(deleted))
 
+    return 'Ok', 200
 app.run(host=host, port=port)