j <ddlmZddlZddlmZddlZGddeejZGddeejZ Gdd ej Z Gd d ej Z Gd d ej Z Gddej ZGddej ZdS)) annotationsN)Anyc:eZdZdZdZdZdZdZed d Z d S) AiProviderTypez"Canonical enterprise AI providers. antigravity openrouterlmstudiogeminivalue str | Nonereturn'AiProviderType'c|s|jSt|}|dvr|jS|dvr|jS|dvr|jS|dvr|jS|jS)zLNormalize various string forms (e.g. 'antigravity_sdk') into canonical Enum.)rantigravity_sdkgoogle_antigravity)r open_router)r lm_studiolocal)r google gemini_direct) ANTIGRAVITYstrlowerstrip OPENROUTERLMSTUDIO GEMINI_DIRECT)clsr vs /C:\Projects\DSCons\app\core\ai\domain\models.pyfrom_strzAiProviderType.from_strs #? " JJ     $ $ & & H H H? " / / /> ! 4 4 4<  7 7 7$ $N)r r r r) __name__ __module__ __qualname____doc__rrrr classmethodr!r"r rrsL,,KJHM   [   r"rc"eZdZdZdZdZdZdZdS) AiCapabilityz/Capabilities supported by models in DSCons ERP.text_generationmultimodal_visionstructured_jsoncode_executionN)r#r$r%r&TEXT_GENERATIONMULTIMODAL_VISIONSTRUCTURED_JSONCODE_EXECUTIONr(r"r r*r*!s)99'O+'O%NNNr"r*ceZdZUdZded<ded<ded<ded<ejeZd ed <d Z d ed <dZ d ed<dZ ded<dZ ded<ej dZdS) AiModelSpeczCSpecification of an AI model registered in the Enterprise Registry.r canonical_idr provider_typeprovider_model_id display_name)default_factoryzlist[AiCapability] capabilitiesiintcontext_windowimax_output_tokensNr fallback_canonical_idTbool is_active)frozen)r#r$r%r&__annotations__pydanticFieldlistr:r<r=r>r@ ConfigDict model_configr(r"r r4r4*sMM!!!!'5x~d'K'K'KLKKKK!N!!!!!!!!!(,,,,,I&8&d333LLLr"r4ceZdZUdZdZded<dZded<dZded <dZded <dZ d ed <dZ ded <dZ ded<dZ ded<dZ ded<dS) AiTextRequestz5Input contract for standard text and chat completion.rpromptNr system_promptzlist[dict[str, Any]] | Nonemessagescanonical_model_idAiProviderType | Nonepreferred_provider agent_codeg?float temperaturez int | None max_tokensdict[str, Any] | None extra_body)r#r$r%r&rKrBrLrMrNrPrQrSrTrVr(r"r rIrI:s??F $M$$$$,0H0000%)))))044444!J!!!!K!J!!!!(,J,,,,,,r"rIceZdZUdZded<dZded<dZded<dZd ed <dZd ed <dZ d ed <dZ d ed<dZ d ed<dZ ded<dZ ded<dS)AiVisionRequestz9Input contract for multimodal vision / document analysis.rrKNz bytes | None image_bytesz image/jpeg image_mimer data_uri file_pathrNrOrPrQ皙?rRrSir;rT)r#r$r%r&rBrYrZr[r\rNrPrQrSrTr(r"r rXrXHsCCKKK $K$$$$"J""""H I    %)))))044444!J!!!!KJr"rXcreZdZUdZded<dZded<dZded<dZded <dZd ed <dZ ded <d Z ded<dS)AiStructuredRequestz2Input contract for strictly typed JSON extraction.rrKNr rLrU schema_dictrNrOrPrQr]rRrS) r#r$r%r&rBrLr`rNrPrQrSr(r"r r_r_Ws<<KKK $M$$$$)-K----%)))))044444!J!!!!Kr"r_ceZdZUdZded<dZded<ded<ded <d Zd ed <d Zded<d Zded<d Z d ed<dZ ded<dS) AiResponsez)Normalized response from any AI provider.rcontentNrUstructured_data model_usedr provider_usedgrR latency_msrr; prompt_tokenscompletion_tokenscost_usdFr? is_fallback) r#r$r%r&rBrdrgrhrirjrkr(r"r rbrbcs33LLL-1O1111OOO!!!!JMHKr"rb) __future__renumtypingrrCrEnumrr* BaseModelr4rIrXr_rbr(r"r rqs"""""" S$)2&&&&&3 &&& 4 4 4 4 4($ 4 4 4 - - - - -H& - - -     h(        (,        #     r"