From ac98fc10a547d503b6916d5dc289c93ed2d6f53f Mon Sep 17 00:00:00 2001 From: Jneeee Date: Sat, 3 May 2025 23:48:37 +0800 Subject: [PATCH] [fix] use openai model provider as default --- src/agents/run.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/src/agents/run.py b/src/agents/run.py index 849da7bf..f21f40d6 100644 --- a/src/agents/run.py +++ b/src/agents/run.py @@ -27,14 +27,19 @@ ModelBehaviorError, OutputGuardrailTripwireTriggered, ) -from .guardrail import InputGuardrail, InputGuardrailResult, OutputGuardrail, OutputGuardrailResult +from .guardrail import ( + InputGuardrail, + InputGuardrailResult, + OutputGuardrail, + OutputGuardrailResult, +) from .handoffs import Handoff, HandoffInputFilter, handoff from .items import ItemHelpers, ModelResponse, RunItem, TResponseInputItem from .lifecycle import RunHooks from .logger import logger from .model_settings import ModelSettings from .models.interface import Model, ModelProvider -from .models.multi_provider import MultiProvider +from .models.openai_provider import OpenAIProvider from .result import RunResult, RunResultStreaming from .run_context import RunContextWrapper, TContext from .stream_events import AgentUpdatedStreamEvent, RawResponsesStreamEvent @@ -56,7 +61,7 @@ class RunConfig: agent. The model_provider passed in below must be able to resolve this model name. """ - model_provider: ModelProvider = field(default_factory=MultiProvider) + model_provider: ModelProvider = field(default_factory=OpenAIProvider) """The model provider to use when looking up string model names. Defaults to OpenAI.""" model_settings: ModelSettings | None = None