- Started addition of Assets (to e.g. handle document templates).

- To be continued (Models, first views are ready)

eveai_chat_workers/retrievers/standard_rag.py

@@ -23,20 +23,12 @@ class StandardRAGRetriever(BaseRetriever):
     def __init__(self, tenant_id: int, retriever_id: int):
         super().__init__(tenant_id, retriever_id)
 
-        retriever = Retriever.query.get_or_404(retriever_id)
-        self.catalog_id = retriever.catalog_id
-        self.tenant_id = tenant_id
-        catalog = Catalog.query.get_or_404(self.catalog_id)
-        embedding_model = "mistral.mistral-embed"
-
-        self.embedding_model, self.embedding_model_class = get_embedding_model_and_class(self.tenant_id,
-                                                                                         self.catalog_id,
-                                                                                         embedding_model)
-        self.similarity_threshold = retriever.configuration.get('es_similarity_threshold', 0.3)
-        self.k = retriever.configuration.get('es_k', 8)
-        self.tuning = retriever.tuning
-
-        self.log_tuning("Standard RAG retriever initialized")
+        # Set up standard retrieval parameters
+        self.embedding_model, self.embedding_model_class, self.catalog_id, self.similarity_threshold, self.k = self.setup_standard_retrieval_params()
+        self.log_tuning("Standard RAG retriever initialized", {
+            "similarity_threshold": self.similarity_threshold,
+            "k": self.k
+        })
 
     @property
     def type(self) -> str:
@@ -167,4 +159,4 @@ class StandardRAGRetriever(BaseRetriever):
 
 
 # Register the retriever type
-RetrieverRegistry.register("STANDARD_RAG", StandardRAGRetriever)
+RetrieverRegistry.register("STANDARD_RAG", StandardRAGRetriever)