§
    éŽ’jñ  ã                   ó¶   — d Z ddlZddlmZmZ ddlZddlmZ ddlm	Z	 ddl
mZ ddlmZ ddlmZ dd	lmZ  G d
„ deej        ¦  «        Z G d„ de¦  «        ZdS )zLiteRT Agent Config.é    N)ÚAnyÚCallable)Útypes)Ú
connection)ÚBaseLocalAgentConfig)Úhooks)Úpolicy)Útriggersc                   ó   — e Zd ZdZdZdZdS )ÚLiteRTBackendÚcpuÚgpuÚnpuN)Ú__name__Ú
__module__Ú__qualname__ÚCPUÚGPUÚNPU© ó    úkC:\Projects\DSCons\.venv\Lib\site-packages\google\antigravity\connections\local\litert_connection_config.pyr   r      s   € € € € € Ø€#Ø€#Ø€#€#€#r   r   c            1       ó   — e Zd ZU dZ ej        dd¬¦  «        Zeed<    ej        e	j
        d¬¦  «        Ze	ed<    ej        d	d
¬¦  «        Zeed<    ej        dd¬¦  «        Zedz  ed<    ej        dd¬¦  «        Ze	dz  ed<    ej        dd¬¦  «        Ze	dz  ed<    ej        dd¬¦  «        Zeed<    ej        d	d¬¦  «        Zeed<    ej        dd¬¦  «        Zej        dz  ed<   e	j
        d	ddddd	ddddddddddddddddœdede	ez  dededz  de	ez  dz  de	ez  dz  dedededz  deej        z  dz  dej        dz  deedef                  dz  deej                 dz  deej                 dz  d ee j!                 dz  d!eej"                 dz  d"eej#                 dz  d#ee         dz  d$edz  d%edz  d&edz  d'e$eef         e%ej&                 z  ez  dz  d(ee         dz  d)ef0d*„Z'd+ed,ed-d.fd/„Z(dS )0ÚLiteRTAgentConfigzJConfiguration for local Gemma models using managed LiteRT-LM PyPI backend..z!Path to the .litertlm model file.)ÚdescriptionÚ
model_pathz!Hardware backend (cpu, gpu, npu).)Údefaultr   ÚbackendFz5Enable speculative decoding (Multi-Token Prediction).Úenable_speculative_decodingNz5Path to a writable directory for compilation caching.Ú	cache_dirz&Backend override for audio processing.Úaudio_backendz-Backend override for vision/image processing.Úvision_backendr   z<The port to bind the local server to. 0 picks a random port.Úportz2Automatically download weights. Defaults to False.Údownload_if_missingz0Maximum number of tokens for the context window.Úmax_context_tokens)r   r   r    r!   r"   r#   r$   r%   Úsystem_instructionsÚcapabilitiesÚtoolsÚpoliciesr   r
   Úmcp_serversÚ	subagentsÚ
workspacesÚconversation_idÚsave_dirÚapp_data_dirÚresponse_schemaÚskills_pathsr&   r'   r(   r)   r   r
   r*   r+   r,   r-   r.   r/   r0   r1   Úkwargsc                óˆ  — t          |t          ¦  «        r!t          |                     ¦   «         ¦  «        }t          |t          ¦  «        r!t          |                     ¦   «         ¦  «        }t          |t          ¦  «        r!t          |                     ¦   «         ¦  «        }|€t	          j        ddddd¬¦  «        }d„ t          ¦   «                              ¦   «         D ¦   «         }d|v r?|                     d¦  «        }t          |t          ¦  «        r| 
                    |¦  «         t          j        j        | fi |¤Ž d S )NT)Ú
file_readsÚfile_writesÚcommand_executionr+   Úmcpc                 ó*   — i | ]\  }}|d k    ¯|®||“ŒS )Úselfr   )Ú.0ÚkÚvs      r   ú
<dictcomp>z.LiteRTAgentConfig.__init__.<locals>.<dictcomp>{   s.   € ð ð ð Ù��A¨Q°&ª[¨[¸Q¸]ˆˆ1¸]¸]¸]r   r2   )Ú
isinstanceÚstrr   Úlowerr   ÚCapabilitiesConfigÚlocalsÚitemsÚpopÚdictÚupdateÚpydanticÚ	BaseModelÚ__init__)r9   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r   r
   r*   r+   r,   r-   r.   r/   r0   r1   r2   Ú	init_dataÚkwargs_dicts                              r   rI   zLiteRTAgentConfig.__init__M   s?  € õ< �'�3ÑÔð /Ý˜gŸmšm™oœoÑ.Ô.€gÝ�-¥Ñ%Ô%ð ;Ý# M×$7Ò$7Ñ$9Ô$9Ñ:Ô:€mÝ�.¥#Ñ&Ô&ð =Ý$ ^×%9Ò%9Ñ%;Ô%;Ñ<Ô<€nàÐÝÔ-ØØØ ØØðñ ô €lðð Ý™œŸšÑ)Ô)ðñ ô €Ið �9ÐÐØ—M’M (Ñ+Ô+€kÝ	�K¥Ñ	&Ô	&ð &Ø×Ò˜Ñ%Ô%Ð%ÝÔÔ Ð2Ð2¨	Ð2Ð2Ð2Ð2Ð2r   Útool_runnerÚhook_runnerÚreturnzconnection.ConnectionStrategyc                óÂ  — 	 ddl m} n)# t          t          f$ r}t	          d¦  «        |‚d }~ww xY w|                     | j        | j        | j        | j	        | j
        | j        | j        | j        | j        |||                      ¦   «         | j        | j        |                      ¦   «         | j        | j        | j        | j        | j        | j        | j        | j        ¬¦  «        S )Nr   )Úlitert_connectionzÊLiteRT backend is not available. To run local LiteRT/Gemma models, please install the optional LiteRT dependencies (e.g., pip install litert-lm or include the LiteRT backend in your build dependencies).)r   r   r   r    r!   r"   r#   r$   r%   rL   rM   r&   Úcapabilities_configr-   r.   r,   r/   r1   r*   r+   ÚenvÚdebug_configÚretry_config)Ú$google.antigravity.connections.localrP   ÚImportErrorÚModuleNotFoundErrorÚRuntimeErrorÚLiteRTConnectionStrategyr   r   r   r    r!   r"   r#   r$   r%   Ú_get_system_instructionsr'   r-   Ú_get_or_create_save_dirr,   r/   r1   r*   r+   rR   rS   rT   )r9   rL   rM   rP   Úes        r   Úcreate_strategyz!LiteRTAgentConfig.create_strategy„   s  € ðØHÐHÐHÐHÐHÐHÐHøÝÕ,Ð-ð ð ð ÝðQñô ð ð	øøøøðøøøð ×5Ò5Ø”?Ø”Ø$(Ô$DØ”.ØÔ(ØÔ*ØŒYØ Ô4ØÔ2ØØØ ×9Ò9Ñ;Ô;Ø Ô-ØÔ,Ø×-Ò-Ñ/Ô/Ø”?ØÔ&ØÔ&ØÔ$Ø”.ØŒHØÔ&ØÔ&ð/ 6ñ ô ð s   ‚	 ‰/š*ª/))r   r   r   Ú__doc__rG   ÚFieldr   r?   Ú__annotations__r   r   r   r   Úboolr    r!   r"   r#   Úintr$   r%   ÚPositiveIntr   ÚSystemInstructionsrA   Úlistr   r   r	   ÚPolicyÚ	hooks_modÚHookÚtriggers_modÚTriggerÚMcpServerConfigÚSubagentConfigrE   ÚtyperH   rI   r]   r   r   r   r   r   $   s?  € € € € € € ØRÐRà"�H”NØ	Ð:ðñ ô €*ˆcð ð ñ ð *˜8œ>ØÔØ5ðñ ô €'ˆ=ð ð ñ ð '5 h¤nØØIð'ñ 'ô 'Ð˜tð ð ñ ð )˜(œ.ØØIðñ ô €)ˆS�4‰Zð ð ñ ð )7¨¬ØØ:ð)ñ )ô )€-� Ñ%ð ð ñ ð *8¨¬ØØAð*ñ *ô *€.�- $Ñ&ð ð ñ ð ˆhŒnØà
Hðñ ô €$ˆð ð ñ ð -˜hœnØØFðñ ô Ð�tð ð ñ ð 5C°H´NØØDð5ñ 5ô 5Ð�hÔ*¨TÑ1ð ð ñ ð &3Ô%6Ø*/Ø"Ø26Ø37ØØ"'Ø'+ØCGØ6:Ø/3Ø-1Ø+/Ø48Ø8<Ø59Ø%)Ø$(Ø!Ø!%ð Ø'+ð753ð 53ð 53ð ð53ð ˜sÑ"ð	53ð
 $(ð53ð �t‘ð53ð # SÑ(¨4Ñ/ð53ð $ cÑ)¨DÑ0ð53ð ð53ð  ð53ð  ™*ð53ð  Ô!9Ñ9¸DÑ@ð53ð Ô,¨tÑ3ð53ð �(˜3 ˜8Ô$Ô%¨Ñ,ð53ð �V”]Ô# dÑ*ð53ð  �)”.Ô! DÑ(ð!53ð" �\Ô)Ô*¨TÑ1ð#53ð$ ˜Ô-Ô.°Ñ5ð%53ð& �eÔ*Ô+¨dÑ2ð'53ð( �s”)˜dÑ"ð)53ð* ˜T‘zð+53ð, �d‘
ð-53ð. ˜$‘Jð/53ð2 ˆs�CˆxŒ.˜4 Ô 2Ô3Ñ
3°cÑ
9¸DÑ
@ð353ð6 ˜”I Ñ$ð753ð8 ð953ð 53ð 53ð 53ðn)ð ð)ð ð	)ð
 'ð)ð )ð )ð )ð )ð )r   r   )r^   ÚenumÚtypingr   r   rG   Úgoogle.antigravityr   Úgoogle.antigravity.connectionsr   Ú<google.antigravity.connections.local.local_connection_configr   Úgoogle.antigravity.hooksr   rg   r	   Úgoogle.antigravity.triggersr
   ri   r?   ÚEnumr   r   r   r   r   ú<module>rv      s  ðð Ð à €€€Ø  Ð  Ð  Ð  Ð  Ð  Ð  Ð  à €€€à $Ð $Ð $Ð $Ð $Ð $Ø 5Ð 5Ð 5Ð 5Ð 5Ð 5Ø ]Ð ]Ð ]Ð ]Ð ]Ð ]Ø 7Ð 7Ð 7Ð 7Ð 7Ð 7Ø +Ð +Ð +Ð +Ð +Ð +Ø @Ð @Ð @Ð @Ð @Ð @ðð ð ð ð �C˜œñ ô ð ðIð Ið Ið Ið IÐ,ñ Iô Ið Ið Ið Ir   