update prompt template

2025-12-13 16:53:24 +00:00 · 2024-05-31 18:27:48 +08:00
parent 4579528b9e
commit b3f8de2393
3 changed files with 14448 additions and 3 deletions
--- a/src/blackbox/chroma_query.py
+++ b/src/blackbox/chroma_query.py
@ -79,10 +79,11 @@ class ChromaQuery(Blackbox):
        # load collection
        collection = client.get_collection(chroma_collection_id, embedding_function=embedding_model)

+        print(usr_question)
        # query it
        results = collection.query(
            query_texts=[usr_question],
-            n_results=3,    
+            n_results=chroma_n_results,    
        )

        # response = str(results["documents"] + results["metadatas"])