o @bojX@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<dZded<dZded <dS) LlmReplyrtextN dict | Nonerawrint prompt_tokenscompletion_tokens)r r r rrrrrrrrrs   rc(eZdZdZddd fd d ZZS) LlmFatalErrorz:Error de LLM que no debe reintentarse (cuota, auth, etc.).codemessagerrreturnNonecst|||_dSN)super__init__rselfrr __class__rrr"s  zLlmFatalError.__init__rrrrrrr r r __doc__r" __classcell__rrr%rrrcr) LlmQuotaExceededz%Cuota / rate limit de OpenAI agotada.rate_limit_exceededrrrrrrcstj||ddS)Nr)r!r"r#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)rr0r1r2rr4r5rrrrr.&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]r8ConversationContext | NonerrcdSr r)r$r9r8rrrreply:szLanguageModel.reply)r8 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$r9r8r>rrrextract_call_resultCs z!LanguageModel.extract_call_resultrcr@)z6Precarga el modelo (p. ej. Ollama). No-op por defecto.Nrr$rrrwarmupQszLanguageModel.warmupcr<r rrBrrrcloseUszLanguageModel.close)r9r:r8r;rr)r9r:r8r;r>r?rr)rr) r r r r)rrr=rArCrDrrrrr62s   r6N)r) __future__rabcrr dataclassesrrrr RuntimeErrorrr,r.r6rrrrs