i#dZddlZddlmZmZddlmZmZmZm Z m Z m Z m Z ddl Z ddlmZmZmZmZmZmZddlmZddlmZerddlmZd d lmZeZeZneZeZGd d eZ y) 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_CHECKINGAnyDictListOptionalTupleUnion)CancelInteractionResultDeleteInteractionResultInteractionInput$InteractionsAPIOptionalRequestParamsInteractionsAPIResponse InteractionsAPIStreamingResponse)GenericLiteLLMParams) LlmProviders)LoggingBaseLLMExceptionceZdZdZdZeedefdZe dZ ede de e fdZ edede d eedefd Ze d$d ee dee d ee d eedeede f dZedee d ee deeded ededefdZedee dej2dedefdZedee dededefdZede d e d ededee eff dZ edej2dedefdZ!ede d e d ededee eff dZ"edej2dede de#fdZ$ede d e d ededee eff dZ%edej2dede&fdZ'de d e(de)eejTfde+fd!Z, d%dee deed"ee defd#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 /Users/manta/Documents/Projects/TheRoad-I1/.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_provider8s r!c |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)rr5s rget_supported_paramsz.BaseInteractionsAPIConfig.get_supported_paramsQs r!headerslitellm_paramsciS)zN Validate and prepare environment settings including headers. r)rr8r5r9s rvalidate_environmentz.BaseInteractionsAPIConfig.validate_environmentXs  r!Napi_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)rr<r5r=r9r>s rget_complete_urlz*BaseInteractionsAPIConfig.get_complete_urlas0  34 4r!inputoptional_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)rr5r=rBrCr9r8s rtransform_requestz+BaseInteractionsAPIConfig.transform_request}s8 r! raw_response logging_objcy)z Transform the raw HTTP response into an InteractionsAPIResponse. Per OpenAPI spec, the response is an Interaction object. Nr)rr5rFrGs rtransform_responsez,BaseInteractionsAPIConfig.transform_response r! parsed_chunkcy)z Transform a parsed streaming response chunk into an InteractionsAPIStreamingResponse. Per OpenAPI spec, streaming uses SSE with various event types. Nr)rr5rKrGs rtransform_streaming_responsez6BaseInteractionsAPIConfig.transform_streaming_responserJr!interaction_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) NrrrNr<r9r8s r!transform_get_interaction_requestz;BaseInteractionsAPIConfig.transform_get_interaction_request r!cy)z9 Transform the get interaction response. NrrrFrGs r"transform_get_interaction_responsezBaseInteractionsAPIConfig.transform_delete_interaction_requestrRr!cy)z< Transform the delete interaction response. Nr)rrFrGrNs r%transform_delete_interaction_responsez?BaseInteractionsAPIConfig.transform_delete_interaction_responses r!cy)z Transform the cancel interaction request into URL and body. Returns: Tuple of (URL, request_body) NrrPs r$transform_cancel_interaction_requestz>BaseInteractionsAPIConfig.transform_cancel_interaction_requests r!cy)z< Transform the cancel interaction response. NrrTs r%transform_cancel_interaction_responsez?BaseInteractionsAPIConfig.transform_cancel_interaction_responserVr! error_message status_codec$ddlm}||||)zC Get the appropriate exception class for an error. rr)r`messager8)chat.transformationr)rr_r`r8rs rget_error_classz)BaseInteractionsAPIConfig.get_error_classs ;#!  r!r$cy)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)rr5r>r$s rshould_fake_streamz,BaseInteractionsAPIConfig.should_fake_stream+sr!)NNNr).__name__ __module__ __qualname____doc__r propertyrrr$r/r4strrr7dictr rr;boolrArrrrEhttpxResponseLiteLLMLoggingObjrrIrrMr rQrUrXr rZr\r r^intr Headersrrdrfrr!rrr)s   \    $ # $s)  $'9ABV9W  $)-!% 3-}}  !    6 } } ()  >  -      :  }  nn  '  !      }    '  *    "   -    sDy    " nn '  !      -    sDy    "  nn  '    !          -    sDy      nn '  !        /2  =B4CV=W    $.2 }  &c]   r!r)!rjr,abcrrtypingrrrrr r r rolitellm.types.interactionsr r rrrrlitellm.types.routerrlitellm.types.utilsr*litellm.litellm_core_utils.litellm_loggingr_LiteLLMLoggingObjrcr_BaseLLMExceptionrqrrr!rr|s_  #III 6,XK*(MMr!