make format

This commit is contained in:
Rohan Mehta 2025-03-11 14:54:10 -07:00
parent 865c0a3ba0
commit ce66d908a9
3 changed files with 8 additions and 4 deletions

View file

@ -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)

View file

@ -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

View file

@ -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"
)