Update app.py

app.py

@@ -63,16 +63,15 @@ class GroqLLM(OpenAI):
 # مدل Groq را با API خود بارگذاری کنید
 llm = GroqLLM(api_key=groq_api_key, model_name="deepseek-r1-distill-llama-70b")
 
-# ----------------- ساخت SimpleRetriever -----------------
 from langchain_core.retrievers import BaseRetriever
 from langchain_core.documents import Document
 from typing import List
-from dataclasses import dataclass, field
 
-@dataclass
 class SimpleRetriever(BaseRetriever):
-    documents: List[Document] = field(default_factory=list)
-    embeddings: List = field(default_factory=list)
+    def __init__(self, documents: List[Document], embeddings: List):
+        super().__init__()
+        self.documents = documents
+        self.embeddings = embeddings
 
     def _get_relevant_documents(self, query: str) -> List[Document]:
         inputs = tokenizer(query, return_tensors="pt", padding=True, truncation=True)