mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
Create/Get Agent API - fixes and example improvements (#3246)
This commit is contained in:
committed by
GitHub
Unverified
parent
975884f32d
commit
b773830e4b
@@ -660,6 +660,7 @@ class OpenAIAssistantsClient(
|
||||
"json_schema": {
|
||||
"name": response_format.__name__,
|
||||
"schema": response_format.model_json_schema(),
|
||||
"strict": True,
|
||||
},
|
||||
}
|
||||
|
||||
|
||||
@@ -784,6 +784,27 @@ def test_prepare_options_with_mapping_tool(mock_async_openai: MagicMock) -> None
|
||||
assert run_options["tool_choice"] == "auto"
|
||||
|
||||
|
||||
def test_prepare_options_with_pydantic_response_format(mock_async_openai: MagicMock) -> None:
|
||||
"""Test _prepare_options sets strict=True for Pydantic response_format."""
|
||||
from pydantic import BaseModel, ConfigDict
|
||||
|
||||
class TestResponse(BaseModel):
|
||||
name: str
|
||||
value: int
|
||||
model_config = ConfigDict(extra="forbid")
|
||||
|
||||
chat_client = create_test_openai_assistants_client(mock_async_openai)
|
||||
messages = [ChatMessage(role=Role.USER, text="Test")]
|
||||
options = {"response_format": TestResponse}
|
||||
|
||||
run_options, _ = chat_client._prepare_options(messages, options) # type: ignore
|
||||
|
||||
assert "response_format" in run_options
|
||||
assert run_options["response_format"]["type"] == "json_schema"
|
||||
assert run_options["response_format"]["json_schema"]["name"] == "TestResponse"
|
||||
assert run_options["response_format"]["json_schema"]["strict"] is True
|
||||
|
||||
|
||||
def test_prepare_options_with_system_message(mock_async_openai: MagicMock) -> None:
|
||||
"""Test _prepare_options with system message converted to instructions."""
|
||||
chat_client = create_test_openai_assistants_client(mock_async_openai)
|
||||
|
||||
Reference in New Issue
Block a user