fix: llamacpp

2025-12-02 01:53:14 +00:00 · 2024-11-13 11:45:42 +00:00
parent bed4939652
commit 9bbe7564a9
2 changed files with 1 additions and 3 deletions
--- a/application/retriever/classic_rag.py
+++ b/application/retriever/classic_rag.py
@@ -45,7 +45,6 @@ class ClassicRAG(BaseRetriever):
                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 @@ class ClassicRAG(BaseRetriever):
                }
                for i in docs_temp
            ]
-        if settings.LLM_NAME == "llama.cpp":
-            docs = [docs[0]]

        return docs