style: rename main class to HybridSearchRetriever

Author: lpm0073

models/examples/load.py

@@ -2,10 +2,10 @@
 """Sales Support Model (SSM) Retrieval Augmented Generation (RAG)"""
 import argparse
 
-from ..ssm import SalesSupportModel
+from ..ssm import HybridSearchRetriever
 
 
-ssm = SalesSupportModel()
+ssm = HybridSearchRetriever()
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(description="RAG example")

models/examples/prompt.py

@@ -2,10 +2,10 @@
 """Sales Support Model (SSM)"""
 import argparse
 
-from ..ssm import SalesSupportModel
+from ..ssm import HybridSearchRetriever
 
 
-ssm = SalesSupportModel()
+ssm = HybridSearchRetriever()
 
 
 if __name__ == "__main__":

models/examples/rag.py

@@ -2,10 +2,10 @@
 """Sales Support Model (SSM) Retrieval Augmented Generation (RAG)"""
 import argparse
 
-from ..ssm import SalesSupportModel
+from ..ssm import HybridSearchRetriever
 
 
-ssm = SalesSupportModel()
+ssm = HybridSearchRetriever()
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(description="RAG example")

models/examples/training_services.py

@@ -3,10 +3,10 @@
 import argparse
 
 from ..prompt_templates import NetecPromptTemplates
-from ..ssm import SalesSupportModel
+from ..ssm import HybridSearchRetriever
 
 
-ssm = SalesSupportModel()
+ssm = HybridSearchRetriever()
 templates = NetecPromptTemplates()
 
 if __name__ == "__main__":

models/examples/training_services_oracle.py

@@ -3,10 +3,10 @@
 import argparse
 
 from ..prompt_templates import NetecPromptTemplates
-from ..ssm import SalesSupportModel
+from ..ssm import HybridSearchRetriever
 
 
-ssm = SalesSupportModel()
+ssm = HybridSearchRetriever()
 templates = NetecPromptTemplates()
 
 if __name__ == "__main__":

models/ssm.py

@@ -64,7 +64,7 @@ def create_documents(self, texts):
         return documents
 
 
-class SalesSupportModel:
+class HybridSearchRetriever:
     """Sales Support Model (SSM)."""
 
     # prompting wrapper

models/tests/test_openai.py

@@ -6,16 +6,16 @@
 """
 import pytest  # pylint: disable=unused-import
 
-from ..ssm import SalesSupportModel
+from ..ssm import HybridSearchRetriever
 
 
 class TestOpenAI:
-    """Test SalesSupportModel class."""
+    """Test HybridSearchRetriever class."""
 
     def test_03_test_openai_connectivity(self):
         """Ensure that we have connectivity to OpenAI."""
 
-        ssm = SalesSupportModel()
+        ssm = HybridSearchRetriever()
         retval = ssm.cached_chat_request(
             "your are a helpful assistant", "please return the value 'CORRECT' in all upper case."
         )

models/tests/test_pinecone.py

@@ -13,7 +13,7 @@
 
 
 class TestPinecone:
-    """Test SalesSupportModel class."""
+    """Test HybridSearchRetriever class."""
 
     def test_01_test_pinecone_connectivity(self):
         """Ensure that we have connectivity to Pinecone."""

models/tests/test_prompt_templates.py

@@ -11,7 +11,7 @@
 
 
 class TestPromptTemplates:
-    """Test SalesSupportModel class."""
+    """Test HybridSearchRetriever class."""
 
     def test_01_prompt_with_template(self):
         """Ensure that all properties of the template class are PromptTemplate instances."""

models/tests/test_prompts.py

@@ -6,13 +6,13 @@
 import pytest  # pylint: disable=unused-import
 
 from ..prompt_templates import NetecPromptTemplates
-from ..ssm import SalesSupportModel
+from ..ssm import HybridSearchRetriever
 
 
 class TestPrompts:
-    """Test SalesSupportModel class."""
+    """Test HybridSearchRetriever class."""
 
-    ssm = SalesSupportModel()
+    ssm = HybridSearchRetriever()
     templates = NetecPromptTemplates()
 
     def test_oracle_training_services(self):