uPj"dZddlZddlmZmZddlmZmZddlZddl m Z m Z m Z m Z mZmZddlmZddlmZerddlmZd d lmZeZeZneZeZGd d eZy) a Base transformation class for Interactions API implementations. This follows the same pattern as BaseResponsesAPIConfig for the Responses API. Per OpenAPI spec (https://ai.google.dev/static/api/interactions.openapi.json): - Create: POST /{api_version}/interactions - Get: GET /{api_version}/interactions/{interaction_id} - Delete: DELETE /{api_version}/interactions/{interaction_id} N)ABCabstractmethod) TYPE_CHECKINGAny)CancelInteractionResultDeleteInteractionResultInteractionInput$InteractionsAPIOptionalRequestParamsInteractionsAPIResponse InteractionsAPIStreamingResponse)GenericLiteLLMParams) LlmProviders)LoggingBaseLLMExceptionc eZdZdZdZeedefdZe dZ ede de e fdZ edede d ed zdefd Ze d$d e d zde d zd e d zd ed zded zde f dZede d zd e d zded zded ededefdZede d zdej.dedefdZede d zdededefdZede d e d ededee eff dZedej.dedefdZede d e d ededee eff dZ edej.dede de!fdZ"ede d e d ededee eff dZ#edej.dede$fdZ%de d e&deejNzde(fd!Z) d%de d zded zd"e d zdefd#Z*y )&BaseInteractionsAPIConfiga Base configuration class for Google Interactions API implementations. Per OpenAPI spec, the Interactions API supports two types of interactions: - Model interactions (with model parameter) - Agent interactions (with agent parameter) Implementations should override the abstract methods to provide provider-specific transformations for requests and responses. cyNselfs v/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/base_llm/interactions/transformation.py__init__z"BaseInteractionsAPIConfig.__init__5s returncy)z#Return the LLM provider identifier.Nrrs rcustom_llm_providerz-BaseInteractionsAPIConfig.custom_llm_provider8rc |jjDcic]`\}}|jdsJ|jds9t|tj tj ttfs|||bc}}Scc}}w)N___abc) __dict__items startswith isinstancetypes FunctionTypeBuiltinFunctionType classmethod staticmethod)clskvs r get_configz$BaseInteractionsAPIConfig.get_config=s **, 1<<%LL(&&--   qD   sA%Bmodelcy)zN Return the list of supported parameters for the given model. Nr)rr1s rget_supported_paramsz.BaseInteractionsAPIConfig.get_supported_paramsPr rheaderslitellm_paramsNciS)zN Validate and prepare environment settings including headers. r)rr4r1r5s rvalidate_environmentz.BaseInteractionsAPIConfig.validate_environmentVs  rapi_baseagentstreamc | td|S)a Get the complete URL for the interaction request. Per OpenAPI spec: POST /{api_version}/interactions Args: api_base: Base URL for the API model: The model name (for model interactions) agent: The agent name (for agent interactions) litellm_params: LiteLLM parameters stream: Whether this is a streaming request Returns: The complete URL for the request zapi_base is required) ValueError)rr8r1r9r5r:s rget_complete_urlz*BaseInteractionsAPIConfig.get_complete_url]s0  34 4rinputoptional_paramscy)a Transform the input request into the provider's expected format. Per OpenAPI spec, the request body should be either: - CreateModelInteractionParams (with model) - CreateAgentInteractionParams (with agent) Args: model: The model name (for model interactions) agent: The agent name (for agent interactions) input: The input content (string, content object, or list) optional_params: Optional parameters for the request litellm_params: LiteLLM-specific parameters headers: Request headers Returns: The transformed request body as a dictionary Nr)rr1r9r>r?r5r4s rtransform_requestz+BaseInteractionsAPIConfig.transform_requestyr r raw_response logging_objcy)z Transform the raw HTTP response into an InteractionsAPIResponse. Per OpenAPI spec, the response is an Interaction object. Nr)rr1rBrCs rtransform_responsez,BaseInteractionsAPIConfig.transform_responser r parsed_chunkcy)z Transform a parsed streaming response chunk into an InteractionsAPIStreamingResponse. Per OpenAPI spec, streaming uses SSE with various event types. Nr)rr1rFrCs rtransform_streaming_responsez6BaseInteractionsAPIConfig.transform_streaming_responser rinteraction_idcy)z Transform the get interaction request into URL and query params. Per OpenAPI spec: GET /{api_version}/interactions/{interaction_id} Returns: Tuple of (URL, query_params) NrrrIr8r5r4s r!transform_get_interaction_requestz;BaseInteractionsAPIConfig.transform_get_interaction_requestr rcy)z9 Transform the get interaction response. NrrrBrCs r"transform_get_interaction_responsezBaseInteractionsAPIConfig.transform_delete_interaction_requestr rcy)z< Transform the delete interaction response. Nr)rrBrCrIs r%transform_delete_interaction_responsez?BaseInteractionsAPIConfig.transform_delete_interaction_responser rcy)z Transform the cancel interaction request into URL and body. Returns: Tuple of (URL, request_body) NrrKs r$transform_cancel_interaction_requestz>BaseInteractionsAPIConfig.transform_cancel_interaction_requestr rcy)z< Transform the cancel interaction response. NrrNs r%transform_cancel_interaction_responsez?BaseInteractionsAPIConfig.transform_cancel_interaction_responser r error_message status_codec$ddlm}||||)zC Get the appropriate exception class for an error. rr)rYmessager4)chat.transformationr)rrXrYr4rs rget_error_classz)BaseInteractionsAPIConfig.get_error_classs ;#!  rrcy)z Returns True if litellm should fake a stream for the given model. Override in subclasses if the provider doesn't support native streaming. Fr)rr1r:rs rshould_fake_streamz,BaseInteractionsAPIConfig.should_fake_streamsr)NNNr)+__name__ __module__ __qualname____doc__rpropertyrrrr+r0strlistr3dictr r7boolr=r r rAhttpxResponseLiteLLMLoggingObjr rEr rHtuplerLrOrQrrSrUrrWintHeadersrr]r_rrrrr)s  2\22  $ # $s)  DNbeiNinr  !&*" *TzTz  t  t  6 Tz Tz  $&  >  -      8  Tz  nn  '  !      Tz    '  *       -    sDy     nn '  !      -    sDy     nn '   !         -    sDy      nn '  !     S  s  TTYTaTaMa  fv  +/ Tz t  !4Z   rr)rcr(abcrrtypingrrrilitellm.types.interactionsrrr r r r litellm.types.routerr litellm.types.utilsr*litellm.litellm_core_utils.litellm_loggingr_LiteLLMLoggingObjr\r_BaseLLMExceptionrkrrrrrwsY  #% 6,XK*(~~r