admin-valentin/openai-agents-python
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

make format

Browse source
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
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/agents/agent_output.py
View file

@ -138,7 +138,7 @@ def _type_to_str(t: type[Any]) -> str:
# It's a simple type like `str`, `int`, etc. # It's a simple type like `str`, `int`, etc.
return t.__name__ return t.__name__
elif args: 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}]" return f"{origin.__name__}[{args_str}]"
else: else:
return str(t) return str(t)

1
src/agents/model_settings.py
View file

@ -11,6 +11,7 @@ class ModelSettings:
This class holds optional model configuration parameters (e.g. temperature, This class holds optional model configuration parameters (e.g. temperature,
top_p, penalties, truncation, etc.). top_p, penalties, truncation, etc.).
""" """
temperature: float | None = None temperature: float | None = None
top_p: float | None = None top_p: float | None = None
frequency_penalty: float | None = None frequency_penalty: float | None = None

9
tests/test_config.py
View file

@ -49,13 +49,16 @@ def test_resp_set_default_openai_client():
def test_set_default_openai_api(): 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" "Default should be responses"
)
set_default_openai_api("chat_completions") 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" "Should be chat completions model"
)
set_default_openai_api("responses") 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" "Should be responses model"
)