diff --git a/src/build-rag-index.py b/src/build-rag-index.py index ac77417db3ab7a1294f14caf8b1e6bf5a3cc906d..5357dcf7627e80b295a9111d159fa1781d96707c 100644 --- a/src/build-rag-index.py +++ b/src/build-rag-index.py @@ -1,5 +1,4 @@ # build-rag-index.py -import os from rag import encoder, simple_document_index, document_processor, fais_db from config import RAG_INDEX_FOLDER, RAG_DOCUMENTS_FOLDER diff --git a/src/model/__init__.py b/src/model/__init__.py index 99a652c9d0cd80b4bed291fa8c89c996354c0097..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 100644 --- a/src/model/__init__.py +++ b/src/model/__init__.py @@ -1,4 +0,0 @@ -from . import huggingface_llm_model -from . import azure_openai_llm_model -from . import prompt_builder -from . import llm_model_factory \ No newline at end of file diff --git a/src/model/huggingface_llm_model.py b/src/model/huggingface_llm_model.py index 700d2a3dc3c8ca4452f3ee3805a1f033eb608d8c..aedbb3d6bded9bccafc6ea7bfb3be9b650baeb6b 100644 --- a/src/model/huggingface_llm_model.py +++ b/src/model/huggingface_llm_model.py @@ -8,7 +8,6 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig from config import MODEL_CACHE_DIR, HF_ACCESS_TOKEN, MODEL_CONFIG from .llm_model_base import LlmModelBase from .prompt_builder import PromptBuilder -import asyncio class HuggingfaceLlmModel(LlmModelBase): """ @@ -110,7 +109,7 @@ class HuggingfaceLlmModel(LlmModelBase): self.streamer.text_queue.put(None) # Signal the streamer to stop except Exception as e: print(f"An unexpected error occurred: {e}") - self.stream_error_with_delay(f"Sorry, I ran into an unexpected error 🤯. I've output the error message to the console.") + self.stream_error_with_delay("Sorry, I ran into an unexpected error 🤯. I've output the error message to the console.") self.streamer.text_queue.put(None) # Signal the streamer to stop # Run the generation in a separate thread, so that we can fetch the generated text in a non-blocking way. diff --git a/src/rag/__init__.py b/src/rag/__init__.py index 7d599247a38afbde256562ad21d9706cc94e32fe..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 100644 --- a/src/rag/__init__.py +++ b/src/rag/__init__.py @@ -1,8 +0,0 @@ -from . import context_extender -from . import encoder -from . import fais_db -from . import document_processor -from . import rag_context_retriever -from . import reranker -from . import simple_document_index -from . import rag_utils diff --git a/src/utils/__init__.py b/src/utils/__init__.py index 0717303806dd38a97c31705146cdcd108a697e8a..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 100644 --- a/src/utils/__init__.py +++ b/src/utils/__init__.py @@ -1,2 +0,0 @@ -from . import audio_generator -from . import response_generator