
    2Rj4                       d dl mZ d dlZd dlZd dlZd dlZd dlmZmZm	Z	m
Z
mZmZmZmZmZmZ d dlmZ d dlmZmZ d dlmZmZ d dlZddlmZmZmZmZmZ  ed	      Z ed
e	e df         Z! ede
e"e f         Z# edee          Z$ ededef         Z%erddl&m'Z'm(Z( d>dZ)dd	 	 	 	 	 	 	 d?dZ*d@dZ+	 	 	 	 	 	 	 	 	 	 	 	 dAdZ,dBdZ-dCdZ.dDdZ/dEdZ0dFdZ1dGdZ2dHdZ3dIdZ4dJdZ5dKd Z6d!d"d#dLd$Z7dMd%Z8dNd&Z9 ed'      Z: ed(      Z;edOd)       Z<edPd*       Z<edQd+       Z<dRd,Z<dSd-Z=dTd.Z>dUd/Z?dVd0Z@dWd1ZAdXd2ZBdYd3ZCdZd4ZDd[d5ZEd\d6ZFd]d7ZGd8d9d^d:ZHd_d;ZId`d<ZJdad=ZKy)b    )annotationsN)
TYPE_CHECKINGAnyTupleMappingTypeVarCallableIterableSequencecastoverload)Path)datedatetime)	TypeGuardget_args   )OmitNotGiven	FileTypesArrayFormatHeadersLike_T_TupleT.)bound	_MappingT
_SequenceT	CallableT)AzureOpenAIAsyncAzureOpenAIc                @    | D cg c]  }|D ]  }|  c}}S c c}}w N )tsublistitems      \/var/www/html/meridian/meridian-ai/venv/lib/python3.12/site-packages/openai/_utils/_utils.pyflattenr(   %   s"     !6Wg6dD6D666s   brackets)array_formatc          
     V    g }|D ]!  }|j                  t        | |dd|             # |S )a  Recursively extract files from the given dictionary based on specified paths.

    A path may look like this ['foo', 'files', '<array>', 'data'].

    ``array_format`` controls how ``<array>`` segments contribute to the emitted
    field name. Supported values: ``"brackets"`` (``foo[]``), ``"repeat"`` and
    ``"comma"`` (``foo``), ``"indices"`` (``foo[0]``, ``foo[1]``).

    Note: this mutates the given dictionary.
    r   Nindexflattened_keyr*   )extend_extract_items)querypathsr*   filespaths        r'   extract_filesr5   )   s9    $ *,E j^E4q[ghijL    c           
         | dk(  ry| dk(  rd| dS | dk(  s| dk(  ryt        d	|  d
dj                  t        t                           )Nr)   z[]indices[]repeatcomma zUnknown array_format value: z, choose from , )NotImplementedErrorjoinr   r   )r*   array_indexs     r'   _array_suffixrB   A   se    z!y ;-q!!x<7#: 

&|nN499XVaMbCcBde r6   c                  	 ||   }|dz  }t        |       rK	 t        d ||d  D              r| j                  |      }n| |   }||}n	|d| dz  }t        |||||      S t	        |       rM|d	k7  rg S t        t        |       D cg c]&  \  }}t        |||||nd
t        ||      z   |      ( c}}      S g S # t         $ r t        |       sg cY S ddlm} |J t	        |       rYg }t        |       D ]E  \  }}	t        ||      }
|r||
z   n|
} ||	|       |j                  |t        t        |	      f       G |cY S  || |       |t        t        |       fgcY S w xY w# t        $ r g cY S w xY wc c}}w )Nr   )assert_is_file_content)key   c              3  &   K   | ]	  }|d k(    yw)<array>Nr#   ).0ps     r'   	<genexpr>z!_extract_items.<locals>.<genexpr>u   s     8a1	>8s   r9   r:   r,   rH   r=   )
IndexErroris_given_filesrD   is_list	enumeraterB   appendr   r   is_dictallpopKeyErrorr0   r(   )objr4   r-   r.   r*   rE   rD   r3   rA   entrysuffixemitted_keyr&   s                r'   r0   r0   O   s   75k0 
QJEs|	 84<88wws|3x  MqQZ'M'%
 	
 
)I *33 &K *7*C}]ikvOww!-
 	
  II  7}I 	4 (((3<13E&/n D"U&|[A:G}v5V&u+>k4	5+ABC	D
 Ls6Y 4566+7@  	 I		*s5   B< ,E% +E6
<E"A,E"E"!E"%E32E3c                J    t        | t               xr t        | t               S r"   )
isinstancer   r   rV   s    r'   rM   rM      s!    #x((FC1F-FFr6   c                "    t        | t              S r"   r[   tupler\   s    r'   is_tupler`          c5!!r6   c                "    t        | t              S r"   r^   r\   s    r'   
is_tuple_trc      ra   r6   c                "    t        | t              S r"   r[   r   r\   s    r'   is_sequencerf          c8$$r6   c                "    t        | t              S r"   re   r\   s    r'   is_sequence_tri      rg   r6   c                "    t        | t              S r"   r[   r   r\   s    r'   
is_mappingrl          c7##r6   c                "    t        | t              S r"   rk   r\   s    r'   is_mapping_tro      rm   r6   c                "    t        | t              S r"   )r[   dictr\   s    r'   rR   rR          c4  r6   c                "    t        | t              S r"   )r[   listr\   s    r'   rO   rO      rr   r6   c                "    t        | t              S r"   )r[   r
   r\   s    r'   is_iterablerv      rg   r6   r>   or)delimfinalc                   t        |       }|dk(  ry|dk(  r| d   S |dk(  r| d    d| d| d    S |j                  | d d       d| d| d    z   S )Nr   r=   rF   r    )lenr@   )seqrx   ry   sizes       r'   
human_joinr      sv    s8Dqyqy1vqya&5'3q6(++::c#2hAeWAc"gY"777r6   c                    d|  dS )zOAdd single quotation marks around the given string. Does *not* do any escaping.'r#   )strings    r'   quoter      s    vha=r6   c                      d fd}|S )a4  Decorator to enforce a given set of arguments or variants of arguments are passed to the decorated function.

    Useful for enforcing runtime validation of overloaded functions.

    Example usage:
    ```py
    @overload
    def foo(*, a: str) -> str: ...


    @overload
    def foo(*, b: bool) -> str: ...


    # This enforces the same constraints that a static type checker would
    # i.e. that either a or b must be passed to the function
    @required_args(["a"], ["b"])
    def foo(*, a: str | None = None, b: bool | None = None) -> str: ...
    ```
    c                $    t        j                         j                  }|j                         D cg c]+  \  }}|j                  |j
                  |j                  hv r|- c}}t        j                         d fd       }|S c c}}w )Nc                 l   t               t        |       D ]  \  }}	 j                  |           |j                         D ]  }j                  |        D ]  }t        fd|D              }|s n t              dkD  rTt        D cg c],  }dt        |D cg c]  }t        |       c}d      z   d	z   . c}}      }d
| d}	t	        |	      t              dkD  sJ t        t        d         z
        }
t        |
      dkD  r0dt        |
D cg c]  }t        |       c}       }	t	        |	      dt        |
d          }	t	        |	       | i |S # t        $ r2 t	        j
                   dt               dt        |        d      d w xY wc c}w c c}}w c c}w )Nz	() takes z argument(s) but z were givenc              3  &   K   | ]  }|v  
 y wr"   r#   )rI   paramgiven_paramss     r'   rK   z@required_args.<locals>.inner.<locals>.wrapper.<locals>.<genexpr>  s     Ju4Js   rF   (and)ry   )z,Missing required arguments; Expected either z arguments to be givenr   zMissing required arguments: zMissing required argument: )setrP   addrL   	TypeError__name__r}   keysrS   r   r   rt   )argskwargsi_rE   variantmatchesarg
variationsmsgmissingr   func
positionalvariantss              @r'   wrapperz-required_args.<locals>.inner.<locals>.wrapper  s   %(UL!$  1  $$Z]3  {{} &  %& $ %J'JK%
 x=1$!+muvbiz*I#5:*IQVWWZ]]v"J ITjkC  n$ x=1,,, #3x{#3l#BCG7|a' <Z_fHgX[sHg=h<ij  n$ !<E'!*<M;NOn$(((9 "  #==/3z?2CCTUXY]U^T__jk   +Jv Ihs)   E(F+
!F&3F+
F1
(;F#&F+
)r   objectr   r   returnr   )	inspect	signature
parametersitemskindPOSITIONAL_ONLYPOSITIONAL_OR_KEYWORD	functoolswraps)r   paramsnamer   r   r   r   s   `    @r'   innerzrequired_args.<locals>.inner   s    ""4(33  &||~
ezz%%++ 

 
	!	) 
!	)F ]
s   0B)r   r   r   r   r#   )r   r   s   ` r'   required_argsr      s    ,0d Lr6   _K_Vc                     y r"   r#   r\   s    r'   strip_not_givenr   5  s    (+r6   c                     y r"   r#   r\   s    r'   r   r   9  s    FIr6   c                     y r"   r#   r\   s    r'   r   r   =  s    ,/r6   c                    | yt        |       s| S | j                         D ci c]  \  }}t        |t              r|| c}}S c c}}w )zHRemove all top-level keys where their values are instances of `NotGiven`N)rl   r   r[   r   )rV   rE   values      r'   r   r   A  sA    
{c?
),X:3Juh<WCJXXXs
   AAc                    t        | d      S )N
   )base)intvals    r'   coerce_integerr   L  s    sr6   c                    t        |       S r"   )floatr   s    r'   coerce_floatr   P  s    :r6   c                (    | dk(  xs | dk(  xs | dk(  S )Ntrue1onr#   r   s    r'   coerce_booleanr   T  s    &=5C3J5#+5r6   c                    | y t        |       S r"   )r   r   s    r'   maybe_coerce_integerr   X      
{#r6   c                    | y t        |       S r"   )r   r   s    r'   maybe_coerce_floatr   ^  s    
{r6   c                    | y t        |       S r"   )r   r   s    r'   maybe_coerce_booleanr   d  r   r6   c                D    | j                  |      r| t        |      d S | S )zXRemove a prefix from a string.

    Backport of `str.removeprefix` for Python < 3.9
    N)
startswithr}   )r   prefixs     r'   removeprefixr   j  s(    
  c&km$$Mr6   c                F    | j                  |      r| dt        |        S | S )zXRemove a suffix from a string.

    Backport of `str.removesuffix` for Python < 3.9
    N)endswithr}   )r   rX   s     r'   removesuffixr   t  s(    
 vnV%%Mr6   c                z    t        |       j                         }t        j                  j	                  |       }||fS r"   )r   
read_bytesosr4   basename)r4   contents	file_names      r'   file_from_pathr   ~  s3    Dz$$&H  &Ix  r6   c                   |j                         }t        |       r@| j                         D ]-  \  }}|j                         |k(  st        |t              s+|c S  t        j                  dd |j                               }|||j                         |fD ]  }| j                  |      }|s|c S  t        d| d      )Nz([^\w])(\w)c                d    | j                  d      | j                  d      j                         z   S )NrF   r   )groupupper)pats    r'   <lambda>z%get_required_header.<locals>.<lambda>  s%    #))A,STI[I[I]:] r6   zCould not find z header)lowerro   r   r[   strresub
capitalizer   get
ValueError)headersheaderlower_headerkvintercaps_headernormalized_headerr   s           r'   get_required_headerr     s    <<>LGMMO 	DAqwwyL(Z3-?	
 vvn.]_e_p_p_rs$lFLLNDTU -.L
 vhg6
77r6   c                 J    	 t        j                         S # t        $ r Y yw xY w)Nfalse)sniffiocurrent_async_library	Exceptionr#   r6   r'   get_async_libraryr     s(    ,,.. s    	""   maxsizec                N    t        j                  |       }t        t        |      S )zmA version of functools.lru_cache that retains the type signature
    for the wrapped function arguments.
    r   )r   	lru_cacher   r   )r   r   s     r'   r   r     s&     !!G Wr6   c                h   t        |       r6| j                         D ci c]  \  }}t        |      t        |       c}}S t        |       r4t	        | t
        t        t        f      s| D cg c]  }t        |       c}S t	        | t        t        f      r| j                         S | S c c}}w c c}w )zvTranslates a mapping / sequence recursively in the same fashion
    as `pydantic` v2's `model_dump(mode="json")`.
    )rl   r   	json_saferv   r[   r   bytes	bytearrayr   r   	isoformat)datarE   r   r&   s       r'   r   r     s     $CG::<PZS%	#	% 00PP4D3y2I!J,01D	$11$4()~~K Q 2s   B),B/c                &    ddl m} t        | |      S )Nr   )r   )	lib.azurer   r[   )clientr   s     r'   is_azure_clientr    s    'fk**r6   c                &    ddl m} t        | |      S )Nr   )r    )r  r    r[   )r  r    s     r'   is_async_azure_clientr    s    ,f.//r6   )r$   zIterable[Iterable[_T]]r   zlist[_T])r1   zMapping[str, object]r2   zSequence[Sequence[str]]r*   r   r   list[tuple[str, FileTypes]])r*   r   rA   r   r   r   )rV   r   r4   Sequence[str]r-   r   r.   
str | Noner*   r   r   r  )rV   z_T | NotGiven | Omitr   zTypeGuard[_T])rV   r   r   zTypeGuard[tuple[object, ...]])rV   z_TupleT | objectr   zTypeGuard[_TupleT])rV   r   r   zTypeGuard[Sequence[object]])rV   z_SequenceT | objectr   zTypeGuard[_SequenceT])rV   r   r   zTypeGuard[Mapping[str, object]])rV   z_MappingT | objectr   zTypeGuard[_MappingT])rV   r   r   zTypeGuard[dict[object, object]])rV   r   r   zTypeGuard[list[object]])rV   r   r   zTypeGuard[Iterable[object]])r~   r	  rx   r   ry   r   r   r   )r   r   r   r   )r   r	  r    Callable[[CallableT], CallableT])rV   Noner   r  )rV   zMapping[_K, _V | NotGiven]r   zdict[_K, _V])rV   r   r   r   )rV   zobject | Noner   r   )r   r   r   r   )r   r   r   r   )r   r   r   bool)r   r
  r   
int | None)r   r
  r   zfloat | None)r   r
  r   zbool | None)r   r   r   r   r   r   )r   r   rX   r   r   r   )r4   r   r   r   )r   r   r   r   r   r   )r   r   )r   r  r   r  )r  r   r   r   )r  r   r   zTypeGuard[AzureOpenAI])r  r   r   zTypeGuard[AsyncAzureOpenAI])L
__future__r   r   r   r   r   typingr   r   r   r   r   r	   r
   r   r   r   pathlibr   r   r   typing_extensionsr   r   r   _typesr   r   r   r   r   r   r   r   r   r   r   r   r  r   r    r(   r5   rB   r0   rM   r`   rc   rf   ri   rl   ro   rR   rO   rv   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r#   r6   r'   <module>r     s   " 	 	      # 1  H HT]
)5#5
6KwsF{';<	\&)9:
KxS'9:	97 !+  
 #  !0N	N
N 	N
 N N !NbG""%%$$!!%
 48d 8
HV T]T] 
 + 
 + 
 I 
 I 
 / 
 /Y6!8& (+  +0r6   