From eb7c7aa6e66035795968bf872f5fcacd4100e74f Mon Sep 17 00:00:00 2001 From: lpm0073 Date: Fri, 1 Dec 2023 10:38:39 -0600 Subject: [PATCH] chore: remove refactored tests --- models/tests/test_ssm.py | 19 ------------------- 1 file changed, 19 deletions(-) diff --git a/models/tests/test_ssm.py b/models/tests/test_ssm.py index 9bb5cff..94ef163 100644 --- a/models/tests/test_ssm.py +++ b/models/tests/test_ssm.py @@ -3,14 +3,12 @@ """ 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 @@ -30,20 +28,3 @@ 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}"