pieter/eveAI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

- Move to Mistral iso OpenAI as primary choice

Browse Source
This commit is contained in:
Josako
2025-03-06 14:19:35 +01:00
parent 55a89c11bb
commit c15cabc289
11 changed files with 74 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
eveai_chat_workers/retrievers/standard_rag.py
View File

@@ -26,6 +26,10 @@ class StandardRAGRetriever(BaseRetriever):
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)
self.embedding_model, self.embedding_model_class = get_embedding_model_and_class(self.tenant_id,
self.catalog_id,
catalog.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
@@ -77,10 +81,10 @@ class StandardRAGRetriever(BaseRetriever):
query = arguments.query
# Get query embedding
query_embedding = self._get_query_embedding(query)
query_embedding = self.embedding_model.embed_query(query)
# Get the appropriate embedding database model
db_class = self.model_variables.embedding_model_class
db_class = self.embedding_model_class
# Get current date for validity checks
current_date = dt.now(tz=tz.utc).date()
@@ -159,12 +163,6 @@ class StandardRAGRetriever(BaseRetriever):
current_app.logger.error(f'Unexpected error in RAG retrieval: {e}')
raise
def _get_query_embedding(self, query: str):
"""Get embedding for the query text"""
catalog = Catalog.query.get_or_404(self.catalog_id)
embedding_model, embedding_model_class = get_embedding_model_and_class(self.tenant_id, self.catalog_id,
catalog.embedding_model)
# Register the retriever type
RetrieverRegistry.register("STANDARD_RAG", StandardRAGRetriever)