yj dZddlmZddlmZmZmZddlmZm Z m Z m Z m Z ddl mZddlmZGddeeZGd d eZGd d eZGd deZGddeZdedzdedzfdZGddeZGddeZGddeZGddeZGddeZGdd eZGd!d"eZGd#d$eZGd%d&eZ Gd'd(e Z!dS))zConfiguration schemas for hierarchical settings resolution. Covers workspace, session, and message-level configuration as well as the fully-resolved variants used at runtime. )Enum)AnySelfcast) BaseModel ConfigDictFieldfield_validatormodel_validator)settings)estimate_tokensceZdZdZdZdS) DreamTypez&Types of dreams that can be triggered.omniN)__name__ __module__ __qualname____doc__OMNIA/DATA/AppData/hermes/projects/honcho/src/schemas/configuration.pyrrs00 DDDrrceZdZUeddZedzed<eddZedzed<e de dedzdedzfd Z dS) ReasoningConfigurationNz*Whether to enable reasoning functionality.default descriptionenabledzOptional custom instructions for the reasoning system on this workspace/session/message. Rejected if they exceed the deriver custom-instruction token cap.custom_instructionsvaluereturnc t|SN$_validate_custom_instructions_budgetclsr s rvalidate_custom_instructionsz3ReasoningConfiguration.validate_custom_instructions 4E:::r) rrrr rbool__annotations__rstrr classmethodr(rrrrrs 5@GTD[',eq'''t _*++;t;d ;;;[,+;;;rrcdeZdZUeddZedzed<eddZedzed<dS)PeerCardConfigurationNzGWhether to use peer card related to this peer during reasoning process.rusez/Whether to generate peer card based on content.create)rrrr r0r*r+r1rrrr/r/&szu]C %EFD4Krr/ceZdZUeddZedzed<edddZedzed<edd d Z edzed <e d de fdZ dS)SummaryConfigurationNz(Whether to enable summary functionality.rr z}Number of messages per short summary. Must be positive, greater than or equal to 10, and less than messages_per_long_summary.)rgermessages_per_short_summaryzNumber of messages per long summary. Must be positive, greater than or equal to 20, and greater than messages_per_short_summary.messages_per_long_summaryaftermoder!cT|j}|j}||||krtd|S)z@Validate that short summary threshold <= long summary threshold.NzFmessages_per_short_summary must be less than messages_per_long_summary)r6r8 ValueError)selfshortlongs rvalidate_summary_thresholdsz0SummaryConfiguration.validate_summary_thresholdsAsC/-  !1etmmX  r) rrrr rr*r+r6intr8r rrArrrr3r31s 5>GTD[.3U T...d  -2E W---sTz _'""" T   #"   rr3c:eZdZUeddZedzed<dS)DreamConfigurationNzWhether to enable dream functionality. If reasoning is disabled, dreams will also be disabled and this setting will be ignored.rr)rrrr rr*r+rrrrDrDOsJ 5VGTD[rrDrNr!c|dS|s|Stjj}|dkrt dt ||krt d|d|S)Nrz7custom_instructions are not enabled for this deploymentzDcustom_instructions exceeds DERIVER.MAX_CUSTOM_INSTRUCTIONS_TOKENS (z tokens))stripr DERIVERMAX_CUSTOM_INSTRUCTIONS_TOKENSr=r )r max_tokenss rr%r%Vs"t  $ $ & &#""!@JQRSSS*++j88 gS] g g g    rceZdZUdZedZeddZedze d<eddZ e dze d <edd Z e dze d <edd Zedze d <dS)WorkspaceConfigurationz The set of options that can be in a workspace DB-level configuration dictionary. All fields are optional. Session-level configuration overrides workspace-level configuration, which overrides global configuration. allow)extraN*Configuration for reasoning functionality.r reasoningzConfiguration for peer card functionality. If reasoning is disabled, peer cards will also be disabled and these settings will be ignored. peer_cardz(Configuration for summary functionality.summaryzConfiguration for dream functionality. If reasoning is disabled, dreams will also be disabled and these settings will be ignored.dream)rrrrr model_configr rOrr+rPr/rQr3rRrDrrrrKrKjs  :G,,,L/4u@000I%,/4e`///I$t+,15>,,,G !D ((-uX(((E  $rrKceZdZdZdS)SessionConfigurationz The set of options that can be in a session DB-level configuration dictionary. All fields are optional. Session-level configuration overrides workspace-level configuration, which overrides global configuration. N)rrrrrrrrUrUs  DrrUc>eZdZUdZeddZedzed<dS)MessageConfigurationz The set of options that can be in a message DB-level configuration dictionary. All fields are optional. Message-level configuration overrides all other configurations. NrNrrO)rrrrr rOrr+rrrrWrWsU 05u@000I%,rrWc|eZdZUeed<dZedzed<edededzdedzfdZ dS)ResolvedReasoningConfigurationrNrr r!c t|Sr#r$r&s rr(z;ResolvedReasoningConfiguration.validate_custom_instructionsr)r) rrrr*r+rr,r r-r(rrrrYrYs~ MMM&*t***_*++;t;d ;;;[,+;;;rrYc$eZdZUeed<eed<dS)ResolvedPeerCardConfigurationr0r1Nrrrr*r+rrrr\r\s" III LLLLLrr\c.eZdZUeed<eed<eed<dS)ResolvedSummaryConfigurationrr6r8N)rrrr*r+rBrrrr_r_s3 MMM ####""""""rr_ceZdZUeed<dS)ResolvedDreamConfigurationrNr]rrrraras MMMMMrraceZdZUdZeed<eed<eed<eed<e de de d e fd Z d S) ResolvedConfigurationz The final resolved configuration for a given message. Hierarchy: message > session > workspace > global configuration rOrPrQrRbeforer:datar!ct|ts|Sttttf|}d|vrd|vr|d|d<|S)z>Handle v3.0.0 migration: 'deriver' was renamed to 'reasoning'.deriverrO) isinstancedictrr,rpop)r'reconfigs rmigrate_deriver_to_reasoningz2ResolvedConfiguration.migrate_deriver_to_reasoningsb$%% Kd38nd++   ;f#<#<"(**Y"7"7F;  rN) rrrrrYr+r\r_rar r-rrlrrrrcrcs .---,,,, )))) %%%%_(###     [$#   rrcc:eZdZUeddZedzed<dS) PeerConfigNzGWhether Honcho will use reasoning to form a representation of this peerr observe_me)rrrr ror*r+rrrrnrnsG#e]Jt rrnc:eZdZUeddZedzed<dS)SessionPeerConfigNziWhether this peer should form a session-level theory-of-mind representation of other peers in the sessionrobserve_others)rrrr rrr*r+rrrrqrqsG"'%###ND4Krrq)"renumrtypingrrrpydanticrrr r r src.configr src.utils.tokensr r,rrr/r3rDr%rKrUrWrYr\r_rarcrnrqrrrrxs> """"""""""SSSSSSSSSSSSSS,,,,,,T ; ; ; ; ;Y ; ; ; I9<t4Z(Y6     1        9   ;;;;;Y;;;I #####9### I4 r