Ë
    uPêj×  ã                   óZ   — d dl mZ d dlmZ d dlmZmZ d dlmZ d dl	m
Z
  G d„ de«      Zy)	é    )ÚAsyncIterator)ÚAny)ÚA2A_USER_API_KEY_HASH_PARAMÚA2ACompletionBridgeHandler)ÚBaseA2AProviderConfig)Ú%merge_a2a_session_into_litellm_paramsc                   ó‚   — e Zd ZdZ	 d	dedeeef   dedz  deeef   fd„Z	 d	dedeeef   dedz  deeeef      fd„Z	y)
ÚLangFlowA2AConfigzxA2A bridge for LangFlow: scopes contextId to the authenticated key as the
    LangFlow session_id, then uses completion.NÚ
request_idÚparamsÚapi_baseÚreturnc              ‹   óÎ   K  — |j                  d«      }|st        d«      ‚t        |||j                  t        «      «      }t	        j
                  ||||d¬«      ƒ d {  –—† S 7 Œ­w©NÚlitellm_paramsz]litellm_params is required for LangFlowA2AConfig (must contain custom_llm_provider and model)T)r   r   r   r   Ú_skip_a2a_provider_routing)ÚgetÚ
ValueErrorr   r   r   Úhandle_non_streaming)Úselfr   r   r   Úkwargsr   s         ús/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/a2a_protocol/providers/langflow/config.pyr   z&LangFlowA2AConfig.handle_non_streaming   sw   è ø€ ð  Ÿ™Ð$4Ó5ˆÙÜØoóð ô ?Ø˜F N×$6Ñ$6Ô7RÓ$Só
ˆô 0×DÑDØ!ØØ)ØØ'+ô
÷ 
ð 	
ð 
ús   ‚AA%ÁA#ÁA%c                óà   K  — |j                  d«      }|st        d«      ‚t        |||j                  t        «      «      }t	        j
                  ||||d¬«      2 3 d {  –—† }|­–— Œ7 Œ
6 y ­wr   )r   r   r   r   r   Úhandle_streaming)r   r   r   r   r   r   Úchunks          r   r   z"LangFlowA2AConfig.handle_streaming'   s†   è ø€ ð  Ÿ™Ð$4Ó5ˆÙÜØoóð ô ?Ø˜F N×$6Ñ$6Ô7RÓ$Só
ˆô 6×FÑFØ!ØØ)ØØ'+ô
÷ 	ð 	�%ð ŒKð	øñ 
ùs*   ‚AA.ÁA,ÁA*Á A,Á#A.Á*A,Á,A.)N)
Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚstrÚdictr   r   r   r   © ó    r   r
   r
      s”   „ ñ2ð  $ñ	
àð
ð �S˜#�X‘ð
ð ˜‘*ð	
ð 
ˆc�3ˆh‰ó
ð6  $ñ	àðð �S˜#�X‘ðð ˜‘*ð	ð 
�t˜C ˜H‘~Ñ	&ôr#   r
   N)Úcollections.abcr   Útypingr   Ú6litellm.a2a_protocol.litellm_completion_bridge.handlerr   r   Ú#litellm.a2a_protocol.providers.baser   Úlitellm.llms.langflow.a2ar   r
   r"   r#   r   ú<module>r)      s&   ðÝ )Ý ÷õ FÝ Kô1Ð-õ 1r#   