refactor code

towardsai · Jul 28, 2024 · 5a84661 · 5a84661
1 parent e9ec472
commit 5a84661
Show file tree

Hide file tree

Showing 8 changed files with 358 additions and 531 deletions.
diff --git a/README.md b/README.md
@@ -5,7 +5,7 @@ colorFrom: gray
 colorTo: pink
 sdk: gradio
 sdk_version: 4.39.0
-app_file: scripts/gradio-ui.py
+app_file: scripts/main.py
 pinned: false
 ---
 

diff --git a/requirements.txt b/requirements.txt
@@ -200,6 +200,7 @@ pydantic-core==2.20.1
 pydispatcher==2.0.7
 pydub==0.25.1
 pygments==2.18.0
+pymongo==4.8.0
 pyopenssl==24.2.1
 pyparsing==3.1.2
 pypdf==4.3.1

diff --git a/scripts/custom_retriever.py b/scripts/custom_retriever.py
@@ -1,4 +1,3 @@
-import logging
 import time
 from typing import List
 
@@ -8,9 +7,6 @@
 from llama_index.core.schema import NodeWithScore, TextNode
 from llama_index.postprocessor.cohere_rerank import CohereRerank
 
-logger = logging.getLogger(__name__)
-logging.basicConfig(level=logging.INFO)
-
 
 class CustomRetriever(BaseRetriever):
     """Custom retriever that performs both semantic search and hybrid search."""