.Ui ddlmZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZmZmZdd lmZerdd lmZGd d ej,ZGddej0ZGddej0Zy)) annotationsN) AsyncIterator) TYPE_CHECKING)ResponsePromptParam)AgentOutputSchemaBase)Handoff) ModelResponseTResponseInputItemTResponseStreamEvent)Tool) ModelSettingsc.eZdZdZ dZ dZ ddZddZy) ModelTracingrrc(|tjk(SN)rDISABLEDselfs s/home/jay/workspace/tools/ai-image-gen/jaaz-app/server/venv/lib/python3.12/site-packages/agents/models/interface.py is_disabledzModelTracing.is_disableds|,,,,c(|tjk(Sr)rENABLEDrs r include_datazModelTracing.include_data s|++++rN)returnbool)__name__ __module__ __qualname__rrENABLED_WITHOUT_DATArrrrrrs%H'G7B-,rrceZdZdZej ddZej ddZy)Modelz&The base interface for calling an LLM.c Kyw)a Get a response from the model. Args: system_instructions: The system instructions to use. input: The input items to the model, in OpenAI Responses format. model_settings: The model settings to use. tools: The tools available to the model. output_schema: The output schema to use. handoffs: The handoffs available to the model. tracing: Tracing configuration. previous_response_id: the ID of the previous response. Generally not used by the model, except for the OpenAI Responses API. conversation_id: The ID of the stored conversation, if any. prompt: The prompt config to use for the model. Returns: The full model response. Nr# rsystem_instructionsinputmodel_settingstools output_schemahandoffstracingprevious_response_idconversation_idprompts r get_responsezModel.get_response's B scy)aMStream a response from the model. Args: system_instructions: The system instructions to use. input: The input items to the model, in OpenAI Responses format. model_settings: The model settings to use. tools: The tools available to the model. output_schema: The output schema to use. handoffs: The handoffs available to the model. tracing: Tracing configuration. previous_response_id: the ID of the previous response. Generally not used by the model, except for the OpenAI Responses API. conversation_id: The ID of the stored conversation, if any. prompt: The prompt config to use for the model. Returns: An iterator of response stream events, in OpenAI Responses format. Nr#r's rstream_responsezModel.stream_responseJsB rN)r( str | Noner)str | list[TResponseInputItem]r*rr+ list[Tool]r,AgentOutputSchemaBase | Noner- list[Handoff]r.rr/r5r0r5r1ResponsePromptParam | Nonerr )r(r5r)r6r*rr+r7r,r8r-r9r.rr/r5r0r5r1r:rz#AsyncIterator[TResponseStreamEvent])rr r!__doc__abcabstractmethodr2r4r#rrr%r%$s0  '  .  &    4       )  $  +       D   '  .  &    4       )  $  +   -    rr%c6eZdZdZej ddZy) ModelProviderzoThe base interface for a model provider. Model provider is responsible for looking up Models by name. cy)zGet a model by name. Args: model_name: The name of the model to get. Returns: The model. Nr#)r model_names r get_modelzModelProvider.get_modeltsrN)rAr5rr%)rr r!r;r<r=rBr#rrr?r?ns!    rr?) __future__rr<enumcollections.abcrtypingr,openai.types.responses.response_prompt_paramr agent_outputrr-r itemsr r r toolr r*rEnumrABCr%r?r#rrrMs_" ) L0KK.,499,"G CGGG T CGG r