+
    5mj  c                  X   ^ RI Ht ^ RIt^ RIHu Ht ^ RIt^ RI	H
t
 ^ RIt^ RIHtHt R R ltR R ltR	 R
 ltR R ltR R ltR R ltR R lt]P,                  P/                  RRR]P0                  ! R. /4      ]P0                  ! RRR/./4      .4      R R l4       tR R ltR# )    )annotationsN)Path)FALLBACK_EFFORTSModelCatalogc               $    V ^8  d   QhRRRRRR/# )   tmp_pathr   modelsz
list[dict]return )formats   "c/home/user/overdeck-private-tests/workstation-ai-broad2/modules/systray/tests/test_model_catalog.py__annotate__r      s!      D *      c                j    V R ,          pVP                  \        P                  ! RV/4      RR7       V# )models_cache.jsonr	   utf-8encoding)
write_textjsondumps)r   r	   paths   && r   write_catalogr      s0    ))DOODJJ&12WOEKr   c                    V ^8  d   QhRRRR/# r   r   r   r
   Noner   )r   s   "r   r   r      s     : :4 :D :r   c                x   \        V R RRRRRR/RR/./.4      p\        V4      pVP                  pRqC! V4      qU'       g   RRR	\        P                  ! 4       9   g   \
        P                  ! V4      '       d   \
        P                  ! V4      MR	R
\
        P                  ! V4      R\
        P                  ! V4      R\
        P                  ! V4      /,          p\        \
        P                  ! V4      4      hR;p;rEVP                  q3! 4       pRqtV8H  qU'       g   \
        P                  ! RV3RWG34      RR	\        P                  ! 4       9   g   \
        P                  ! V4      '       d   \
        P                  ! V4      MR	R
\
        P                  ! V4      R\
        P                  ! V4      R\
        P                  ! V4      /,          pRRV/,          p	\        \
        P                  ! V	4      4      hR;p;p;rWVP                  pRqC! V4      pRqV
8H  q'       g   \
        P                  ! RV3RWZ34      RR	\        P                  ! 4       9   g   \
        P                  ! V4      '       d   \
        P                  ! V4      MR	R
\
        P                  ! V4      R\
        P                  ! V4      R\
        P                  ! V4      R\
        P                  ! V
4      /,          p	RRV	/,          p\        \
        P                  ! V4      4      hR;p;p;p;rVP                  pRqC! V4      pRqV
8H  q'       g   \
        P                  ! RV3RWZ34      RR	\        P                  ! 4       9   g   \
        P                  ! V4      '       d   \
        P                  ! V4      MR	R
\
        P                  ! V4      R\
        P                  ! V4      R\
        P                  ! V4      R\
        P                  ! V
4      /,          p	RRV	/,          p\        \
        P                  ! V4      4      hR;p;p;p;rR# )sluggpt-testdefault_reasoning_levelmediumsupported_reasoning_levelseffortlowzGassert %(py6)s
{%(py6)s = %(py2)s
{%(py2)s = %(py0)s.known
}(%(py4)s)
}py0catalogpy2py4py6Npy7zassert %(py9)spy9zassert %(py11)spy11)r   ==)zE%(py4)s
{%(py4)s = %(py2)s
{%(py2)s = %(py0)s.models
}()
} == %(py7)s)r$   r!   )zW%(py6)s
{%(py6)s = %(py2)s
{%(py2)s = %(py0)s.supported_efforts
}(%(py4)s)
} == %(py9)s)zT%(py6)s
{%(py6)s = %(py2)s
{%(py2)s = %(py0)s.default_effort
}(%(py4)s)
} == %(py9)s)r   r   known@py_builtinslocals
@pytest_ar_should_repr_global_name	_safereprAssertionError_format_explanationr	   _call_reprcomparesupported_effortsdefault_effort)r   r   r&   @py_assert1@py_assert3@py_assert5@py_format7@py_assert6@py_format8@py_format10@py_assert8@py_assert7@py_format12s   &            r   $test_reads_slugs_efforts_and_defaultrD      s~    
)8,%/@8XBV.W	
	D 4 G==$$=$$$$$$$$$7$$$$7$$$=$$$$$$$$$$$$$>>,>,},},,,,,,,,,,,7,,,,7,,,>,,,,,,},,,,,,,,$$EZE$Z0E4EE4EEEEEE0EEEEEE7EEEE7EEE$EEEZEEE0EEE4EEEEEEEEE!!9*9!*-9999999-999999799997999!999*999-999999999999r   c                    V ^8  d   QhRRRR/# r   r   )r   s   "r   r   r   #   s     C CT Cd Cr   c                Z   \        V R RRRR/./.4      p\        V4      q"P                  pRqC! V4      pRqeV8H  qw'       Eg_   \        P                  ! RV3RWV34      RR\
        P                  ! 4       9   g!   \        P                  ! \        4      '       d   \        P                  ! \        4      MRRR\
        P                  ! 4       9   g   \        P                  ! V4      '       d   \        P                  ! V4      MRR	\        P                  ! V4      R
\        P                  ! V4      R\        P                  ! V4      R\        P                  ! V4      R\        P                  ! V4      /,          pRRV/,          p	\        \        P                  ! V	4      4      hR;p;p;p;p;rvR# )r   r   r"   r#   highr%   r   py1r   py3py5r*   r+   py12zassert %(py14)spy14Nr-   )zs%(py9)s
{%(py9)s = %(py5)s
{%(py5)s = %(py3)s
{%(py3)s = %(py0)s(%(py1)s)
}.default_effort
}(%(py7)s)
} == %(py12)s)r   r   r9   r2   r7   r0   r1   r3   r4   r5   r6   )
r   r   @py_assert2@py_assert4r>   rA   @py_assert11@py_assert10@py_format13@py_format15s
   &         r   6test_default_falls_back_to_first_supported_when_absentrS   #   s   
*:h=O<P	QRD B,,BZB,Z8BFBFBBBBB8BBBBBB<BBBB<BBBBBBBBBBBBBBBBB,BBBZBBB8BBBFBBBBBBBBBr   c                    V ^8  d   QhRRRR/# r   r   )r   s   "r   r   r   +   s     
> 
>$ 
>4 
>r   c                @   \        V R RRRR/RR/./R RRRR/RR/./.4      p\        V4      pVP                  pRpRqS! WE4      pRqvV8H  q'       Eg	   \        P                  ! RV3RWg34      R	R
\
        P                  ! 4       9   g   \        P                  ! V4      '       d   \        P                  ! V4      MR
R\        P                  ! V4      R\        P                  ! V4      R\        P                  ! V4      R\        P                  ! V4      R\        P                  ! V4      /,          p	RRV	/,          p
\        \        P                  ! V
4      4      hR;p;p;p;p;rVP                  pRpRqS! WE4      pRqvV8H  q'       Eg	   \        P                  ! RV3RWg34      R	R
\
        P                  ! 4       9   g   \        P                  ! V4      '       d   \        P                  ! V4      MR
R\        P                  ! V4      R\        P                  ! V4      R\        P                  ! V4      R\        P                  ! V4      R\        P                  ! V4      /,          p	RRV	/,          p
\        \        P                  ! V
4      4      hR;p;p;p;p;rR# )r   z
with-ultrar"   r#   maxultrazno-ultrarG   pror%   r&   r'   r(   r)   py8r,   zassert %(py13)spy13Nr-   )z^%(py8)s
{%(py8)s = %(py2)s
{%(py2)s = %(py0)s.resolve_effort
}(%(py4)s, %(py6)s)
} == %(py11)sr   r   resolve_effortr2   r7   r0   r1   r3   r4   r5   r6   )r   r   r&   r:   r;   r<   rB   rP   @py_assert9rC   @py_format14s   &          r   +test_pro_resolves_to_highest_supported_tierr_   +   s   \#?8UBSV^`gUhAijZ!=6@RU]_dTe?fg	
D 4 G!!A,AA!,6A'A'AAAAA6AAAAAA7AAAA7AAA!AAA,AAAAAA6AAA'AAAAAAAAA!!=*=e=!*4=======4======7====7===!===*===e===4============r   c                    V ^8  d   QhRRRR/# r   r   )r   s   "r   r   r   8   s     L L4 LD Lr   c                   \        V R RRRR/./.4      p\        V4      q"P                  pRpRqS! WE4      pRqvV8H  q'       Egu   \        P                  ! RV3RWg34      RR\
        P                  ! 4       9   g!   \        P                  ! \        4      '       d   \        P                  ! \        4      MRRR	\
        P                  ! 4       9   g   \        P                  ! V4      '       d   \        P                  ! V4      MR	R
\        P                  ! V4      R\        P                  ! V4      R\        P                  ! V4      R\        P                  ! V4      R\        P                  ! V4      R\        P                  ! V4      /,          p	RRV	/,          p
\        \        P                  ! V
4      4      hR;p;p;p;p;p;rR# )r   r   r"   r#   r!   medr%   r   rH   r   rI   rJ   r*   r+   r,   rL   zassert %(py16)spy16Nr-   )z~%(py11)s
{%(py11)s = %(py5)s
{%(py5)s = %(py3)s
{%(py3)s = %(py0)s(%(py1)s)
}.resolve_effort
}(%(py7)s, %(py9)s)
} == %(py14)sr[   )r   r   rM   rN   r>   rA   rP   @py_assert13@py_assert12rR   @py_format17s   &          r   test_med_alias_canonicalizesrg   8   s   FJ(DRZG[F\]^D K,,KZKK,Z?K8K8KKKKK?KKKKKK<KKKK<KKKKKKKKKKKKKKKKK,KKKZKKKKKK?KKK8KKKKKKKKKr   c                    V ^8  d   QhRRRR/# r   r   )r   s   "r   r   r   ?   s     ? ?d ?t ?r   c                    \        V R RRRR/./.4      p\        P                  ! \        RR7      ;_uu_ 4        \	        V4      P                  RR4       RRR4       R#   + '       g   i     R# ; i)	r   r   r"   r#   r$   zsupported: lowmatchxhighN)r   pytestraises
ValueErrorr   r\   )r   r   s   & r   'test_unsupported_effort_names_what_fitsrp   ?   s_    FJ(DRWGXFYZ[D 
z)9	:	:T))*g> 
;	:	:	:s   AA0	c                    V ^8  d   QhRRRR/# r   r   )r   s   "r   r   r   G   s     0 0$ 04 0r   c                   \        V R RRRR/./.4      p\        V4      pVP                  pRqC! V4      qU'       * qf'       g   RRR\        P                  ! 4       9   g   \
        P                  ! V4      '       d   \
        P                  ! V4      MRR	\
        P                  ! V4      R
\
        P                  ! V4      R\
        P                  ! V4      /,          p\        \
        P                  ! V4      4      hR;p;p;rV\        P                  ! \        RR7      ;_uu_ 4        VP                  RR4       RRR4       R#   + '       g   i     R# ; i)r   r   r"   r#   r$   absentzKassert not %(py6)s
{%(py6)s = %(py2)s
{%(py2)s = %(py0)s.known
}(%(py4)s)
}r%   r&   r'   r(   r)   Nzunknown model 'absent'rj   )r   r   r/   r0   r1   r2   r3   r4   r5   r6   rm   rn   ro   r\   )r   r   r&   r:   r;   r<   rB   r?   s   &       r   test_unknown_model_rejectedrt   G   s    FJ(DRWGXFYZ[D 4 G}}&X&}X&&&&&&&&&&&w&&&&w&&&}&&&X&&&&&&&&&&&	z)A	B	Bx/ 
C	B	B	Bs   %EE	content znot jsonr	   r   xc               $    V ^8  d   QhRRRRRR/# )r   r   r   ru   strr
   r   r   )r   s   "r   r   r   T   s&     J J$ J J Jr   c                   V R ,          pVP                  VRR7       \        V4      q3P                  qD! 4       p\        \        4      p\        V4      quV8H  q'       Ega   \        P                  ! RV3RWW34      RR\        P                  ! 4       9   g!   \        P                  ! \        4      '       d   \        P                  ! \        4      MRRR\        P                  ! 4       9   g   \        P                  ! V4      '       d   \        P                  ! V4      MRR\        P                  ! V4      R\        P                  ! V4      R	\        P                  ! V4      R
R\        P                  ! 4       9   g!   \        P                  ! \
        4      '       d   \        P                  ! \
        4      MRRR\        P                  ! 4       9   g!   \        P                  ! \        4      '       d   \        P                  ! \        4      MRRR\        P                  ! 4       9   g!   \        P                  ! \        4      '       d   \        P                  ! \        4      MRR\        P                  ! V4      R\        P                  ! V4      /
,          p	RRV	/,          p
\        \        P                  ! V
4      4      hR;p;p;p;p;rgR# )r   r   r   r%   r   rH   r   rI   rJ   r*   r+   tuplepy10sortedr,   r   rZ   py15zassert %(py17)spy17Nr-   )z%(py7)s
{%(py7)s = %(py5)s
{%(py5)s = %(py3)s
{%(py3)s = %(py0)s(%(py1)s)
}.models
}()
} == %(py15)s
{%(py15)s = %(py9)s(%(py13)s
{%(py13)s = %(py10)s(%(py11)s)
})
})r   r   r	   r}   r   r{   r2   r7   r0   r1   r3   r4   r5   r6   )r   ru   r   rM   rN   r>   re   @py_assert14rA   @py_format16@py_format18s   &&         r   #test_unusable_catalog_uses_fallbackr   Q   ss    ))DOOGgO.I$$I$&I7G0HI%0H*II*IIIIII&IIIIII<IIII<IIIIIIIIIIIIIIIII$III&IIIIII%IIII%IIIIIIIIIIIIIIIIII7GIIII7GIII0HIII*IIIIIIIIIIr   c                    V ^8  d   QhRRRR/# r   r   )r   s   "r   r   r   Z   s     ^ ^ ^$ ^r   c                   R qV,          p\        V4      q3P                  qD! 4       p\        \        4      p\	        V4      quV8H  q'       Egw   \
        P                  ! RV3RWW34      RR\        P                  ! 4       9   g!   \
        P                  ! \         4      '       d   \
        P                  ! \         4      MRRR\        P                  ! 4       9   g   \
        P                  ! V 4      '       d   \
        P                  ! V 4      MRR\
        P                  ! V4      R\
        P                  ! V4      R\
        P                  ! V4      R\
        P                  ! V4      R	R
\        P                  ! 4       9   g!   \
        P                  ! \        4      '       d   \
        P                  ! \        4      MR
RR\        P                  ! 4       9   g!   \
        P                  ! \        4      '       d   \
        P                  ! \        4      MRRR\        P                  ! 4       9   g!   \
        P                  ! \        4      '       d   \
        P                  ! \        4      MRR\
        P                  ! V4      R\
        P                  ! V4      /,          p	RRV	/,          p
\        \
        P                  ! V
4      4      hR;p;p;p;p;p;p;rgR# )zabsent.jsonr%   r   rH   r   rI   r)   rY   r|   rK   r{   rZ   r}   rL   r   rc   py18zassert %(py20)spy20Nr-   )z%(py10)s
{%(py10)s = %(py8)s
{%(py8)s = %(py6)s
{%(py6)s = %(py0)s((%(py1)s / %(py3)s))
}.models
}()
} == %(py18)s
{%(py18)s = %(py12)s(%(py16)s
{%(py16)s = %(py13)s(%(py14)s)
})
})r   r	   r}   r   r{   r2   r7   r0   r1   r3   r4   r5   r6   )r   rM   rN   r<   rB   r]   @py_assert15@py_assert17rO   @py_format19@py_format21s   &          r   "test_missing_catalog_uses_fallbackr   Z   so   #0]=0]<01]88]8:]FK[D\]eD\>]]>]]]]]]:]]]]]]<]]]]<]]]]]]]]]]]]]]=]]]1]]]8]]]:]]]]]]e]]]]e]]]]]]]F]]]]F]]]]]]]K[]]]]K[]]]D\]]]>]]]]]]]]]]r   )
__future__r   builtinsr0   _pytest.assertion.rewrite	assertionrewriter2   r   pathlibr   rm   model_catalogr   r   r   rD   rS   r_   rg   rp   rt   markparametrizer   r   r   r   r   r   <module>r      s    "      8:$C
>L?0 J

Hb> :DJJTZ\_S`RaGb<cdJJ^r   