chore: add more unit tests

Author: lpm0073

models/ssm.py

@@ -10,6 +10,7 @@
 import os
 from typing import List  # ClassVar
 
+# pinecone integration
 import pinecone
 from langchain.cache import InMemoryCache
 
@@ -76,7 +77,8 @@ def cached_chat_request(self, system_message: str, human_message: str) -> System
             HumanMessage(content=human_message),
         ]
         # pylint: disable=not-callable
-        return self.chat(messages)
+        retval = self.chat(messages).content
+        return retval
 
     def prompt_with_template(self, prompt: PromptTemplate, concept: str, model: str = DEFAULT_MODEL_NAME) -> str:
         """Prompt with template."""

models/tests/test_ssm.py

@@ -4,12 +4,14 @@
 """
 Test integrity of base class.
 """
+import pinecone
 import pytest  # pylint: disable=unused-import
 from langchain.chat_models import ChatOpenAI
 from langchain.embeddings import OpenAIEmbeddings
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.vectorstores.pinecone import Pinecone
 
+from ..const import Credentials
 from ..ssm import SalesSupportModel
 
 
@@ -29,3 +31,20 @@ def test_02_class_aatribute_types(self):
         assert isinstance(ssm.pinecone_index, Pinecone)
         assert isinstance(ssm.text_splitter, RecursiveCharacterTextSplitter)
         assert isinstance(ssm.openai_embedding, OpenAIEmbeddings)
+
+    def test_03_test_openai_connectivity(self):
+        """Ensure that we have connectivity to OpenAI."""
+
+        ssm = SalesSupportModel()
+        retval = ssm.cached_chat_request(
+            "your are a helpful assistant", "please return the value 'CORRECT' in all upper case."
+        )
+        assert retval == "CORRECT"
+
+    def test_04_test_pinecone_connectivity(self):
+        """Ensure that we have connectivity to Pinecone."""
+        # pylint: disable=broad-except
+        try:
+            pinecone.init(api_key=Credentials.PINECONE_API_KEY, environment=Credentials.PINECONE_ENVIRONMENT)
+        except Exception as e:
+            assert False, f"pinecone.init() failed with exception: {e}"