P1i\_ddlmZddlmZddlmZddlmZddl m Z ddl m Z m Z mZddlmZdd lmZGd d ej&ZGd d ej&ZGddej&ZGddej&ZGddej&ZGddej&ZGddej&ZGddej&Zy))models)Qsettings) gettext_lazy)ValidationError)validate_templateget_template_statisticslog_template_change)timezone)get_user_modelc beZdZdZgdZddgZgdZejde dZ ejd e d  Z eje d  Z ejdee dZejdede de dZejdede dZej"ej&ej(d d de dZej"ej&ej(d d de dZej.d e dZej.d e d Zej4d d e d!e d"#Zej.d d e d$%ZGd&d'Zd(Zfd)Zfd*Z d+Z!d,Z"xZ#S)-PromptTemplatez Model for storing and managing AI prompt templates. Templates can have different statuses (draft, active, archived). )baziBaZinumberz Number PowerliuyaoLiuYao)zh-hanszChinese (Simplified))enEnglish))draftDraft)activeActive)archivedArchiveddName) max_length verbose_nameT Descriptionblankr%Contentr% Divination Typer$choicesr%rLanguagez!Language for this prompt template)r$r.defaultr% help_textrStatus)r$r.r0r%created_templates Created By on_deletenullr( related_namer%modified_templatesLast Modified By Created At auto_now_addr% Updated Atauto_nowr%zValidation Warningsz!Warnings from template validationr7r(r%r1zLast Validated)r7r(r%cxeZdZdgZedZedZgdZejddge dd gZ y ) PromptTemplate.Meta -updated_atzPrompt TemplatezPrompt Templates)divination_typenamelanguagerErGrstatus#unique_active_template_per_language)fields conditionrFN) __name__ __module__ __qualname__ordering_r%verbose_name_pluralunique_togetherrUniqueConstraintr constraints+/home/cursorai/projects/iching/ai/models.pyMetarCZsO!?*+  23A #F # #):68,:  rWrYcT|jd|jd|jdS)N (, ))rFrErGselfs rX__str__zPromptTemplate.__str__gs*))Bt334Bt}}oQGGrWc,t|t|j|j\}}|r|nd|_|j dk(rC|s@|Dcgc] }t|}}tdtddj|ziyycc}w)z,Validate the template content before saving.NrcontentzTTemplate must be valid before it can be activated. Please fix the following issues: r\) supercleanr rbrEvalidation_warningsrIstrrrQjoin)r_is_validwarningswarningwarning_messages __class__s rXrdzPromptTemplate.cleanjs  /t||T=Q=QR(088T  ;;( "85"rWrceZdZdZej deddZejedZ ejded Z ej d e jed  Z Gd dZdZy)PromptPlaceholderz Model for documenting placeholders used in prompt templates. This helps administrators understand what variables are available for templates. r"zPlaceholder NameT)r$r%uniquer&r*z Example Valuer'r+r,r-c2eZdZdgZedZedZy)PromptPlaceholder.MetarFzPrompt PlaceholderzPrompt PlaceholdersNrMrNrOrPrQr%rRrVrWrXrYrs 8-.  56rWrYc@d|jd|jddS)Nz{{z}} - )rFrr^s rXr`zPromptPlaceholder.__str__s'dii[(8(8"(='>??rWN)rMrNrOrrrrQrFrr example_valuerrrErYr`rVrWrXrrs 6  s;M9NW[ \D"&""-0@AK$F$$4a>PQM&f&&66()O 77 @rWrc eZdZdZgdZddgZgdZgdZejdede d e d  Z ejd ee d e dZ ejde de dZ ejde de dZej de dZej de dZej&ej*ej,ddde dZej&ej*ej,ddde dZGd d!Zd"Zfd#Zed$Zed%ZxZS)&AIProviderConfigz Model for storing AI provider and model configurations for different divination types. Allows admins to dynamically configure which AI provider and model to use as default. )rrr)bazi_conversationzBaZi Conversation)liuyao_conversationzLiuYao ConversationgroqGroqopenaiOpenAI))llama-3.3-70b-versatilez-Llama 3.3 70B (Best quality & context - 128k))z-meta-llama/llama-4-maverick-17b-128e-instructz*Llama 4 Maverick 17B (Good balance - 128k))zllama-3.1-8b-instantzLlama 3.1 8B (Faster - 128k)))gpt-4oz%GPT-4o (Latest version, best quality))z gpt-4-turboz$GPT-4 Turbo (Fast with good quality))z gpt-3.5-turboz/GPT-3.5 Turbo (Fastest, good for simpler tasks))zgpt-4zGPT-4 (Original GPT-4))zgpt-4.1zGPT-4.1)zgpt-5zGPT-5Tr,z4The type of divination this configuration applies to)r$r.rr%r1r+ AI Providerz/The AI provider to use for this divination typer$r.r%r1r"Modelz4The specific model to use with the selected provider)r$r%r1 Is Active.Whether this configuration is currently activer0r%r1r;r<r>r?created_ai_configsr4r5modified_ai_configsr:c2eZdZdgZedZedZy)AIProviderConfig.MetarEzAI Provider ConfigurationzAI Provider ConfigurationsNrrVrWrXrYrs!%&45  <=rWrYcb|jd|jd|jS)N: z - )get_divination_type_displayget_provider_displaymodelr^s rXr`zAIProviderConfig.__str__s62245R8Q8Q8S7TTWX\XbXbWcddrWct||jdk(r_|jDcgc]}|d }}|j|vr4t dt djdj|iy|jdk(r_|jDcgc]}|d }}|j|vr4t dt djdj|iyycc}wcc}w) zAValidate that the model is appropriate for the selected provider.rrrz@Selected model is not valid for Groq provider. Valid options: {}r\rzBSelected model is not valid for OpenAI provider. Valid options: {}N) rcrdproviderGROQ_MODEL_CHOICESrrrQformatrgOPENAI_MODEL_CHOICES)r_choice valid_modelsrls rXrdzAIProviderConfig.cleans   ==F "484K4KL4K&F1I4KLLzz-%Qabii ,/'. ]]h &484M4MN4M&F1I4MLNzz-%Qcdkk ,/'.'MOs C- C2c  |jj|d}|j|jdS#|j$r;ddlm}t|dd}|dk(rt|dd }n t|d d }||dcYSwxYw) z Get the active configuration for a divination type. Returns fallback from settings if no config exists. T)rE is_active)rrrrDEFAULT_LLM_PROVIDERr GROQ_MODELr OPENAI_MODELr)rrgetrr DoesNotExist django.confrgetattr)clsrEconfigrrrs rX get_configzAIProviderConfig.get_config.s  [[___PT_UF"OO   ,x)?HH6!,8QR.(C%  s58ABBcJ|dk(r |jS|dk(r |jSgS)z-Get available models for a specific provider.rr)rr)rrs rXget_available_modelsz%AIProviderConfig.get_available_modelsKs1 v )) )  !++ + rW) rMrNrOrrPROVIDER_CHOICESrrrrrQrErr BooleanFieldrrrrrrrrrxrvrYr`rd classmethodrrrrs@rXrrs    'f&&'()JK O v }%EF H F  wZJK E $##{^DEI &%%4a oVJ%%%t!L/RJ"""  // )|_ J$&##  // *)* K>> e*8rWrc eZdZdZej dededZejdededZ ejded  Z ejded  Z ejejej ddd edZejejej dddedZGddZdZfdZfdZedZxZS)ConversationConfigz Model for storing conversation configuration settings. Allows admins to dynamically configure conversation limits and settings. Only one active configuration should exist at a time. zMax Messages Per Conversationz\Maximum number of messages allowed per conversation (only messages with status="sent" count)rTrrr;r<r>r?created_conversation_configsr4r5modified_conversation_configsr:c4eZdZddgZedZedZy)ConversationConfig.Metaz -is_activerDzConversation ConfigurationzConversation ConfigurationsNrrVrWrXrYr{s# -056  =>rWrYcJd|jd|jrddSddS)NzMax Messages: r[rInactiver]) max_messagesrr^s rXr`zConversationConfig.__str__s5 1 12"X4``abbU_4``abbrWclt||jdkrtdt diy)z'Validate that max_messages is positive.rrz#Max messages must be greater than 0N)rcrdrrrQ)r_rls rXrdzConversationConfig.cleans<      !!"G H#  "rWc|jrJtjjdj |j j dt| |i|y)z@When a config is set to active, deactivate other active configs.TrrnFN) rrrrrsrtrorurcry)r_rzr{rls rXryzConversationConfig.savesS >>  & & - - - = E E E Q X Xch X i  d%f%rWc |jjd}|jS#|j$rddlm}t |ddcYSwxYw)z Get the maximum messages per conversation from active configuration. Returns fallback from settings if no config exists. TrrrCONVERSATION_MAX_MESSAGESr)rrrrrrrr)rrrs rXget_max_messagesz#ConversationConfig.get_max_messagessR  F[[__t_4F&& & F ,8%@"E E Fs'*"AA)rMrNrOrr IntegerFieldrQrrrrrrrrrrrxrvrYr`rdryrrrrs@rXrrUs" '6&&67rsL $##{^DEI &%%4a oVJ%%%t!L/RJ"""  // 3|_ J$&##  // 4)* K?? c&  F FrWrceZdZddgZej dZejZej deZ ej dZ ejdd Z ej dd Z ejeejdd Zej"d Zej"dZej(edZGddZdZy)AdminAIAnalysisrr)r$r)r$r.r"T)r(r7pending)r$r0)r6r7r()r=)r@)r0r(c2eZdZdgZedZedZy)AdminAIAnalysis.Meta)can_send_admin_llm_analysisz#Can send admin LLM analysis requestz AI AnalysisN)rMrNrO permissionsrQr%rRrVrWrXrYrs# R '  .rWrYc|jS)N)titler^s rXr`zAdminAIAnalysis.__str__s zzrWN)rMrNrOrrrrrpromptrrresponserIrrrrxrrrrdictmetarYr`rVrWrXrrs F   ,E V   Fv27GHH F   ,Evd6H V  Y ?F""">#3vUYaefJ%%%48J%%%t4J 6  D 5D//rWrceZdZdZddgZej deededZejeded  Z ejd ed  Z GddZ dZdZedZy)ConversationSubjectz Pivot table for linking conversations to different divination types. Supports BaZi (Person), LiuYao, and future divination types. rrr+z Content Typez1Type of divination record this subject representsrz Object IDzCID of the related object (Person ID for BaZi, LiuYao ID for LiuYao))r%r1Tr;r<cfeZdZdgZedZedZddgZejddggZ y)ConversationSubject.Meta -created_atzConversation SubjectzConversation Subjects content_type object_idrKN) rMrNrOrPrQr%rRrSrIndexindexesrVrWrXrYrsD!?/0  78);7 FLL = > rWrYc@|jd|jS)Nz #)get_content_type_displayrr^s rXr`zConversationSubject.__str__s"//12"T^^4DEErWcH|jdk(r-ddlm} |jj |j S|jdk(r-ddlm} |jj |j Sy#|j $rYywxYw#|j $rYywxYw)z-Get the actual object this subject refers to.rr)PersonrnNr)r) r bazi.modelsrrrrrr liuyao.modelsr)r_rrs rX get_objectzConversationSubject.get_objects    & * ~~))T^^)<<  ( * , ~~))T^^)<<&&   &&  s#%A:%B:B  B B! B!cF|jj||\}}|S)zMGet or create a ConversationSubject for the given content type and object ID.)rr)rr get_or_create)rrrsubjectrqs rXget_or_create_subjectz)ConversationSubject.get_or_create_subjects/;;44%5 rWN)rMrNrOrCONTENT_TYPE_CHOICESrrrQrrrrrrYr`rrrrVrWrXrrs  $6##$~&GH L $##{^YZI&%%4a oVJ  F rWrc eZdZdZej ejejde dZ ej e ejddde de dZ ej d ejddd e d e d Z ejd de de dZej de dZej de dZej&dde de dZej dde de dZej,dde de dZGddZdZd Zd!Zd"Zy#)$ Conversationz Model representing a conversation thread between a user and AI about a divination record. Each conversation is tied to a ConversationSubject (which links to BaZi, LiuYao, etc.) and user. conversationsUserr6r8r%TSubjectz0The divination record this conversation is about)r6r7r(r8r%r1z bazi.Personconversations_legacyzPerson (Legacy)zMLegacy field - use subject instead. The BaZi chart this conversation is aboutrTitlez#Optional title for the conversation)r$r(r%r1r;r<r>r?zContext Summaryz3AI-generated summary of older conversation messages)r(r7r%r1zContext Summary Updated Atz)When the context summary was last updatedrAzLast Summarized Message IDz.ID of the last message included in the summaryceZdZdgZedZedZejddgejddgejddggZ y) Conversation.MetarDr ConversationsuserrrpersonN rMrNrOrPrQr%rRrrrrVrWrXrYr2s^!?( 0 FLL 7 8 FLLM : ; FLL= 9 : rWrYc|jr|jj}|rk|jjdk(rt|dr |jnd}nw|jjdk(rt|dr |j nd}nCd}n@d}n=|j r/t|j dr|j jnd}nd}|jxsd|}|d|jdS) NrrFUnknownrquestionzConversation about r[r]) rrrhasattrrFrrrr)r_ subject_objrFrs rXr`zConversation.__str__<s <<,,113K<<,,6/6{F/K;++QZD\\..(:3:; 3S;//YbD$D  [['.t{{F'C4;;##DD : 3D6:499+Q''rWcf|jr|jjS|jS)z1Get the actual object this conversation is about.)rrrr^s rXget_subject_objectzConversation.get_subject_objectPs& <<<<**, ,{{rWcT|jjdjS)z*Get the last message in this conversation.r)messagesorder_byfirstr^s rXget_last_messagezConversation.get_last_messageWs }}%%m4::<|  m FLL > ? rWrYc|j}t|jdkDr|jdddzn |j}|d|S)N2z...r)get_role_displaylenrb)r_ role_displaycontent_previews rXr`zMessage.__str__sP,,. 7:4<<7H27M$,,s+e3SWS_S_r/!233rWN)rMrNrOr ROLE_CHOICESrrrrr$rQr5rrolerrbrrrI error_messagerrrrrrrrYr`rVrWrXr)r)es "L N %6$$..~& L 6  vYHI  D fAiL9G%%%4a oVJV   x[qr F%F$$ '78 M v 11 }%78 H F   wZ45  E 6  z];<  D  4rWr)N) django.dbrdjango.db.modelsrrrdjango.utils.translationrrQdjango.core.exceptionsrutils.template_validationr r r django.utilsr django.contrib.authrrrrrrrrrr)rVrWrXrFs 62ff!.Z"V\\Z"z@ @.Rv||RjHFHFVfll69&,,9xk6<<k\Q4fllQ4rW