Skip to content

Commit eb7c7aa

Browse files
committed
chore: remove refactored tests
1 parent 57b012c commit eb7c7aa

File tree

1 file changed

+0
-19
lines changed

1 file changed

+0
-19
lines changed

models/tests/test_ssm.py

Lines changed: 0 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -3,14 +3,12 @@
33
"""
44
Test integrity of base class.
55
"""
6-
import pinecone
76
import pytest # pylint: disable=unused-import
87
from langchain.chat_models import ChatOpenAI
98
from langchain.embeddings import OpenAIEmbeddings
109
from langchain.text_splitter import RecursiveCharacterTextSplitter
1110
from langchain.vectorstores.pinecone import Pinecone
1211

13-
from ..const import Credentials
1412
from ..ssm import SalesSupportModel
1513

1614

@@ -30,20 +28,3 @@ def test_02_class_aatribute_types(self):
3028
assert isinstance(ssm.pinecone_index, Pinecone)
3129
assert isinstance(ssm.text_splitter, RecursiveCharacterTextSplitter)
3230
assert isinstance(ssm.openai_embedding, OpenAIEmbeddings)
33-
34-
def test_03_test_openai_connectivity(self):
35-
"""Ensure that we have connectivity to OpenAI."""
36-
37-
ssm = SalesSupportModel()
38-
retval = ssm.cached_chat_request(
39-
"your are a helpful assistant", "please return the value 'CORRECT' in all upper case."
40-
)
41-
assert retval == "CORRECT"
42-
43-
def test_04_test_pinecone_connectivity(self):
44-
"""Ensure that we have connectivity to Pinecone."""
45-
# pylint: disable=broad-except
46-
try:
47-
pinecone.init(api_key=Credentials.PINECONE_API_KEY, environment=Credentials.PINECONE_ENVIRONMENT)
48-
except Exception as e:
49-
assert False, f"pinecone.init() failed with exception: {e}"

0 commit comments

Comments
 (0)