diff --git a/test/test_settings_embedder_api.py b/test/test_settings_embedder_api.py index 784decb..05b3dce 100644 --- a/test/test_settings_embedder_api.py +++ b/test/test_settings_embedder_api.py @@ -23,7 +23,7 @@ class TestSettingsEmbedderApi(unittest.TestCase): """SettingsEmbedderApi unit test stubs""" def setUp(self): - self.api = cheshire_cat_api.api.settings_embedder_api.SettingsEmbedderApi() # noqa: E501 + self.api = cheshire_cat_api.api.embedder_api.EmbedderApi() # noqa: E501 def tearDown(self): pass diff --git a/test/test_settings_general_api.py b/test/test_settings_general_api.py index e07e220..d71b512 100644 --- a/test/test_settings_general_api.py +++ b/test/test_settings_general_api.py @@ -23,7 +23,7 @@ class TestSettingsGeneralApi(unittest.TestCase): """SettingsGeneralApi unit test stubs""" def setUp(self): - self.api = cheshire_cat_api.api.settings_general_api.SettingsGeneralApi() # noqa: E501 + self.api = cheshire_cat_api.api.settings_api.SettingsApi() # noqa: E501 def tearDown(self): pass diff --git a/test/test_settings_large_language_model_api.py b/test/test_settings_large_language_model_api.py index dc171c3..5e18f67 100644 --- a/test/test_settings_large_language_model_api.py +++ b/test/test_settings_large_language_model_api.py @@ -23,7 +23,7 @@ class TestSettingsLargeLanguageModelApi(unittest.TestCase): """SettingsLargeLanguageModelApi unit test stubs""" def setUp(self): - self.api = cheshire_cat_api.api.settings_large_language_model_api.SettingsLargeLanguageModelApi() # noqa: E501 + self.api = cheshire_cat_api.api.large_language_model_api.LargeLanguageModelApi() # noqa: E501 def tearDown(self): pass diff --git a/test/test_settings_prompt_api.py b/test/test_settings_prompt_api.py index 2527038..510e346 100644 --- a/test/test_settings_prompt_api.py +++ b/test/test_settings_prompt_api.py @@ -23,7 +23,7 @@ class TestSettingsPromptApi(unittest.TestCase): """SettingsPromptApi unit test stubs""" def setUp(self): - self.api = cheshire_cat_api.api.settings_prompt_api.SettingsPromptApi() # noqa: E501 + self.api = cheshire_cat_api.api.prompt_api.PromptApi() # noqa: E501 def tearDown(self): pass