Skip to content

Commit 1232766

Browse files
committed
feat: Update package version to 0.7.7
fix: Improve error handling for unsupported models in NOVITA and AsyncNOVITA classes
1 parent c883316 commit 1232766

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

setup.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@
5656

5757
setup(
5858
name="python-tgpt",
59-
version="0.7.6",
59+
version="0.7.7",
6060
license="MIT",
6161
author="Smartwa",
6262
maintainer="Smartwa",

src/pytgpt/novita/main.py

+6-6
Original file line numberDiff line numberDiff line change
@@ -36,10 +36,10 @@ class NOVITA(OPENAI):
3636
"""Novita AI provider"""
3737

3838
def __init__(self, *args, **kwargs):
39-
kwargs.setdefault("model", model)
40-
if not model in available_models:
39+
model_choice = kwargs.setdefault("model", model)
40+
if not model_choice in available_models:
4141
raise UnsupportedModelError(
42-
f"Model '{model}' is not yet supported. Choose from {available_models}"
42+
f"Model '{model_choice}' is not yet supported. Choose from {available_models}"
4343
)
4444
super().__init__(*args, **kwargs)
4545
self.chat_endpoint = "https://api.novita.ai/v3/openai/chat/completions"
@@ -49,10 +49,10 @@ class AsyncNOVITA(AsyncOPENAI):
4949
"""Async Novita AI provider"""
5050

5151
def __init__(self, *args, **kwargs):
52-
kwargs.setdefault("model", model)
53-
if not model in available_models:
52+
model_choice = kwargs.setdefault("model", model)
53+
if not model_choice in available_models:
5454
raise UnsupportedModelError(
55-
f"Model '{model}' is not yet supported choose from {available_models}"
55+
f"Model '{model_choice}' is not yet supported. Choose from {available_models}"
5656
)
5757
super().__init__(*args, **kwargs)
5858
self.chat_endpoint = "https://api.novita.ai/v3/openai/chat/completions"

0 commit comments

Comments
 (0)