rag4

app.py

@@ -32,7 +32,7 @@ llm = Llama(
 
 def respond(message, history):
     # 1. Retrieve Context
-    context = ask(message)
+    context = ask(message, retriver_moedel=retriever)
     print(context)
     
     # 2. Define System/Contextual Prompt

rag_db.py

@@ -82,8 +82,8 @@ def format_docs(docs):
 
 
 # This is what will be used! 
-def ask(query: str) -> str: 
+def ask(query: str, retriver_moedel) -> str: 
     """Simple call for external modules (like narrator)."""
-    docs = retriever.invoke(query)
+    docs = retriver_moedel.invoke(query)
     return docs