make format
This commit is contained in:
parent
865c0a3ba0
commit
ce66d908a9
3 changed files with 8 additions and 4 deletions
|
|
@ -138,7 +138,7 @@ def _type_to_str(t: type[Any]) -> str:
|
|||
# It's a simple type like `str`, `int`, etc.
|
||||
return t.__name__
|
||||
elif args:
|
||||
args_str = ', '.join(_type_to_str(arg) for arg in args)
|
||||
args_str = ", ".join(_type_to_str(arg) for arg in args)
|
||||
return f"{origin.__name__}[{args_str}]"
|
||||
else:
|
||||
return str(t)
|
||||
|
|
|
|||
|
|
@ -11,6 +11,7 @@ class ModelSettings:
|
|||
This class holds optional model configuration parameters (e.g. temperature,
|
||||
top_p, penalties, truncation, etc.).
|
||||
"""
|
||||
|
||||
temperature: float | None = None
|
||||
top_p: float | None = None
|
||||
frequency_penalty: float | None = None
|
||||
|
|
|
|||
|
|
@ -49,13 +49,16 @@ def test_resp_set_default_openai_client():
|
|||
|
||||
|
||||
def test_set_default_openai_api():
|
||||
assert isinstance(OpenAIProvider().get_model("gpt-4"), OpenAIResponsesModel), \
|
||||
assert isinstance(OpenAIProvider().get_model("gpt-4"), OpenAIResponsesModel), (
|
||||
"Default should be responses"
|
||||
)
|
||||
|
||||
set_default_openai_api("chat_completions")
|
||||
assert isinstance(OpenAIProvider().get_model("gpt-4"), OpenAIChatCompletionsModel), \
|
||||
assert isinstance(OpenAIProvider().get_model("gpt-4"), OpenAIChatCompletionsModel), (
|
||||
"Should be chat completions model"
|
||||
)
|
||||
|
||||
set_default_openai_api("responses")
|
||||
assert isinstance(OpenAIProvider().get_model("gpt-4"), OpenAIResponsesModel), \
|
||||
assert isinstance(OpenAIProvider().get_model("gpt-4"), OpenAIResponsesModel), (
|
||||
"Should be responses model"
|
||||
)
|
||||
|
|
|
|||
Loading…
Reference in a new issue