From c4d8bf91dd97850ac15940277ab4e3fbba2d9e9d Mon Sep 17 00:00:00 2001 From: Abdullah Al Amin Date: Fri, 8 Sep 2023 09:27:58 +0000 Subject: [PATCH] chore: removed unused vars --- ai/question.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/ai/question.py b/ai/question.py index 1286eea..562116c 100644 --- a/ai/question.py +++ b/ai/question.py @@ -43,7 +43,6 @@ def ask(repo_id: int, chat_id: str, question: str): db = FAISS.load_local(os.path.join(repo_path, "vector_store"), embeddings) retriever = db.as_retriever() - end = time.time() retriever.search_kwargs["distance_metric"] = "cos" retriever.search_kwargs["fetch_k"] = 100 @@ -58,12 +57,10 @@ def ask(repo_id: int, chat_id: str, question: str): retriever=retriever, return_source_documents=True, ) - end = time.time() result = qa(question) print(f"Answer: {result['answer']}") print(f"Sources: {[x.metadata['source'] for x in result['source_documents']]}") - end = time.time() return result["answer"], [x.metadata["source"] for x in result["source_documents"]]