o zSoj@sdZddlmZddlmZmZddlmZmZeGdddZ eGdddZ Gd d d e Z Gd d d e Z eGd ddZGdddeZdS)z>Contrato del modelo de lenguaje (independiente del proveedor).) annotations)ABCabstractmethod) dataclassfieldc@seZdZUded<ded<dS) ChatMessagestrrolecontentN)__name__ __module__ __qualname____annotations__rr7/var/www/html/bot-asterisk-ari/backend/llm/protocols.pyr s  rc@s"eZdZUded<dZded<dS)LlmReplyrtextN dict | Noneraw)r r r rrrrrrrs rc(eZdZdZddd fd d ZZS) LlmFatalErrorz:Error de LLM que no debe reintentarse (cuota, auth, etc.).codemessagerrreturnNonecst|||_dSN)super__init__rselfrr __class__rrrs  zLlmFatalError.__init__rrrrrrr r r __doc__r __classcell__rrr"rrrcr) LlmQuotaExceededz%Cuota / rate limit de OpenAI agotada.rate_limit_exceededrrrrrrcstj||ddS)Nr)rrr r"rrr szLlmQuotaExceeded.__init__r$r%rrr"rr)r(r)c@sPeZdZUdZded<dZded<dZded<dZded <ee d Z d ed <dS) ConversationContextz&Contexto de la llamada para el prompt.rcall_idNz str | None document_id caller_numberrpoliza)default_factorydictextra) r r r r&rr-r.r/rr1r2rrrrr+$s    r+c@sTeZdZUdZded<edddd d ZddddddZdddZdddZ dS) LanguageModeluuGenera la respuesta conversacional del bot. Implementación actual: ``llm.openai_llm.OpenAiLanguageModel``. r provider_nameN)contextmessageslist[ChatMessage]r5ConversationContext | NonerrcdSrr)r!r6r5rrrreply8szLanguageModel.reply)r5 transcriptr;list[str] | NonercdS)zExtrae resultado estructurado al fin de la llamada (json_schema). Default: no soportado. OpenAI Responses lo implementa con ``text.format.type=json_schema``. Nr)r!r6r5r;rrrextract_call_resultAs z!LanguageModel.extract_call_resultrcr=)z6Precarga el modelo (p. ej. Ollama). No-op por defecto.Nrr!rrrwarmupOszLanguageModel.warmupcr9rrr?rrrcloseSszLanguageModel.close)r6r7r5r8rr)r6r7r5r8r;r<rr)rr) r r r r&rrr:r>r@rArrrrr30s   r3N)r& __future__rabcrr dataclassesrrrr RuntimeErrorrr)r+r3rrrrs