o i'@sdZddlmZmZmZmZmZddlmZm Z m Z m Z m Z m Z mZmZmZmZddlmZmZddlmZmZmZddlmZddlmZmZddlmZdd l m!Z!gd Z"Gd d d Z#Gd dde Z$Gddde Z%GdddZ&dS)z$Testing utilities for OpenAI agents.) AsyncIteratorCallableOptionalSequenceUnion) AgentOutputSchemaBaseHandoffModel ModelProvider ModelResponse ModelSettings ModelTracingToolTResponseInputItemUsage)TResponseOutputItemTResponseStreamEvent)ResponseFunctionToolCallResponseOutputMessageResponseOutputText)Client)StatefulMCPServerProviderStatelessMCPServerProvider)ModelActivityParameters)OpenAIAgentsPlugin)AgentEnvironmentResponseBuilders TestModelTestModelProviderc@sleZdZdZededefddZedede fddZ ed ed edefd d Z ededefd dZ dS)rzBuilders for creating model responses for testing. .. warning:: This API is experimental and may change in the future. outputreturncCst|gtddS)zCreate a ModelResponse with the given output. .. warning:: This API is experimental and may change in the future. N)rusage response_id)r r)rr#n/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/temporalio/contrib/openai_agents/testing.pymodel_response/s zResponseBuilders.model_responsetextcCstdt|gddgddddS)zCreate a ResponseOutputMessage with text content. .. warning:: This API is experimental and may change in the future. output_text)r& annotationstype assistant completedmessage)idcontentrolestatusr*)rrr&r#r#r$response_output_message<sz(ResponseBuilders.response_output_message argumentsnamec Cstt|d|ddddS)zCreate a ModelResponse with a function tool call. .. warning:: This API is experimental and may change in the future. call function_callr.r,)r4call_idr5r*r.r1)rr%r)r4r5r#r#r$ tool_callQszResponseBuilders.tool_callcCstt|S)zCreate a ModelResponse with an output message. .. warning:: This API is experimental and may change in the future. )rr%r3r2r#r#r$output_messagecszResponseBuilders.output_messageN) __name__ __module__ __qualname____doc__ staticmethodrr r%strrr3r9r:r#r#r#r$r(s rc@s<eZdZdZdZdefddZdeedfdefd d Z dS) rzTest model provider which simply returns the given module. .. warning:: This API is experimental and may change in the future. FmodelcC ||_dS)zInitialize a test model provider with a model. .. warning:: This API is experimental and may change in the future. N_model)selfrAr#r#r$__init__x zTestModelProvider.__init__ model_nameNr cCs|jS)zGet a model from the model provider. .. warning:: This API is experimental and may change in the future. rC)rErHr#r#r$ get_modelszTestModelProvider.get_model) r;r<r=r>__test__r rFrr@rIr#r#r#r$ros rc@seZdZdZdZdegefddfddZdee dfd ee e e fd e d e e d eedfd e ededefddZdee d ee e e fd e d e e d eed e ededeefddZede eddfddZdS)rzTest model for use mocking model responses. .. warning:: This API is experimental and may change in the future. Ffnr NcCrB)zInitialize a test model with a callable. .. warning:: This API is experimental and may change in the future. NrK)rErKr#r#r$rFrGzTestModel.__init__system_instructionsinputmodel_settingstools output_schemahandoffstracingc s |S)zXGet a response from the mocked model, by calling the callable passed to the constructor.rL rErMrNrOrPrQrRrSkwargsr#r#r$ get_responses zTestModel.get_responsec Kst)z6Get a streamed response from the model. Unimplemented.)NotImplementedErrorrTr#r#r$stream_responses zTestModel.stream_response responsescst|tfddS)zCreate a mock model which sequentially returns responses from a list. .. warning:: This API is experimental and may change in the future. cstS)N)nextr#ir#r$sz/TestModel.returning_responses..)iterr)rYr#r[r$returning_responsesszTestModel.returning_responses)r;r<r=r>rJrr rFrr@listrr rrrr rVrrrrXr?r_r#r#r#r$rsP    rc @seZdZdZdZ     ddeedeedeed e e e e fd e d df d d ZdddZdddZded efddZed efddZdS)raTesting environment for OpenAI agents with Temporal integration. This async context manager provides a convenient way to set up testing environments for OpenAI agents with mocked model calls and Temporal integration. .. warning:: This API is experimental and may change in the future. Example: >>> from temporalio.contrib.openai_agents.testing import AgentEnvironment, TestModelProvider, ResponseBuilders >>> from temporalio.client import Client >>> >>> # Create a mock model that returns predefined responses >>> mock_model = TestModel.returning_responses([ ... ResponseBuilders.output_message("Hello, world!"), ... ResponseBuilders.output_message("How can I help you?") ... ]) >>> >>> async with AgentEnvironment(model=mock_model) as env: ... client = env.applied_on_client(client) ... # Use client for testing workflows with mocked model calls FNr#T model_paramsmodel_providerrAmcp_server_providersregister_activitiesr cCsD||_d|_|dur||_n |durt||_||_||_d|_dS)aInitialize the AgentEnvironment. Args: model_params: Configuration parameters for Temporal activity execution of model calls. If None, default parameters will be used. model_provider: Optional model provider for custom model implementations. Only one of model_provider or model should be provided. If both are provided, model_provider will be used. model: Optional model for custom model implementations. Use TestModel for mocking model responses. Equivalent to model_provider=TestModelProvider(model). Only one of model_provider or model should be provided. If both are provided, model_provider will be used. mcp_server_providers: Sequence of MCP servers to automatically register with the worker. register_activities: Whether to register activities during worker execution. .. warning:: This API is experimental and may change in the future. N) _model_params_model_providerr_mcp_server_providers_register_activities_plugin)rErarbrArcrdr#r#r$rFs  zAgentEnvironment.__init__cs t|j|j|j|jd|_|S)z Enter the async context manager.)rarbrcrd)rrerfrgrhrirEr#r#r$ __aenter__szAgentEnvironment.__aenter__csdS)zExit the async context manager.Nr#)rEargsr#r#r$ __aexit__szAgentEnvironment.__aexit__clientcCsH|jdur td|}|dg}t||jg|d<tdi|S)abApply the agent environment's plugin to a client and return a new client instance. Args: client: The base Temporal client to apply the plugin to. Returns: A new Client instance with the OpenAI agents plugin applied. .. warning:: This API is experimental and may change in the future. Nz:AgentEnvironment must be entered before applying to clientpluginsr#)ri RuntimeErrorconfiggetr`r)rErn new_configexisting_pluginsr#r#r$applied_on_clients  z"AgentEnvironment.applied_on_clientcCs|jdur td|jS)zGet the underlying OpenAI agents plugin. .. warning:: This API is experimental and may change in the future. Nz8AgentEnvironment must be entered before accessing plugin)rirprjr#r#r$openai_agents_plugin)s z%AgentEnvironment.openai_agents_plugin)NNNr#T)r r)r N)r;r<r=r>rJrrr r rrrrboolrFrkrmrrupropertyrrvr#r#r#r$rs8  ' rN)'r>typingrrrrragentsrrr r r r r rrr agents.itemsrropenai.types.responsesrrrtemporalio.clientr%temporalio.contrib.openai_agents._mcprr2temporalio.contrib.openai_agents._model_parametersr8temporalio.contrib.openai_agents._temporal_openai_agentsr__all__rrrrr#r#r#r$s0    G8