fix: llamacpp

arc53 · Nov 13, 2024 · 9bbe756 · 9bbe756
1 parent bed4939
commit 9bbe756
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/application/api/answer/routes.py b/application/api/answer/routes.py
@@ -241,6 +241,7 @@ def complete_stream(
         yield f"data: {data}\n\n"
     except Exception as e:
         print("\033[91merr", str(e), file=sys.stderr)
+        traceback.print_exc()
         data = json.dumps(
             {
                 "type": "error",

diff --git a/application/retriever/classic_rag.py b/application/retriever/classic_rag.py
@@ -45,7 +45,6 @@ def _get_data(self):
                 settings.VECTOR_STORE, self.vectorstore, settings.EMBEDDINGS_KEY
             )
             docs_temp = docsearch.search(self.question, k=self.chunks)
-            print(docs_temp)
             docs = [
                 {
                     "title": i.metadata.get(
@@ -60,8 +59,6 @@ def _get_data(self):
                 }
                 for i in docs_temp
             ]
-        if settings.LLM_NAME == "llama.cpp":
-            docs = [docs[0]]
 
         return docs