Update src/rag.py

src/rag.py

@@ -18,7 +18,12 @@ from langchain.prompts import PromptTemplate
 from langchain_community.llms import Ollama 
 from langchain_openai import ChatOpenAI
 from transformers import AutoTokenizer, pipeline, AutoModelForCausalLM, BitsAndBytesConfig
-
+try:
+    from importlib.metadata import PackageNotFoundError
+except ImportError:
+    # Define a fallback for older Python versions
+    class PackageNotFoundError(Exception):
+        pass
 def setup_document_retriever(document_path):
     # Load documents with the AAC user's personal experiences
     loader = TextLoader(document_path)