F()jN LddlZddlZddlZddlZddlmZmZmZddlmZddl m Z m Z m Z m Z ddlmZmZddlmZddlmZddlmZddlZddlZddlZddlZdd lmZej<d Zd Z dZ!e r^ ejDejFejHejJejLejNd d Z!ejQdnejQdGdde)Z,Gdde,Z-hdZ.ddiZ/e.jae/jcZ2dZ3dZ4dddddd Z5d!e6fd"Z7d#Z8d$Z9d%Z:d&Z;d'Zfd,Z?d-e6d.e@d/efd0ZAd8d-e6d!e6d1ee6d2e>fd3ZBd4ZCd9d5e>d+e6fd6ZDd7ZEy#e)$rZ*ejWde*YdZ*[*dZ*[*wwxYw):N)datetime timedeltatimezone) relativedelta)fetch_tenant_limitsget_ticket_countget_vector_db_sizeget_total_token_usage)get_ai_response_message_countget_conversation_thread_count)client)settings)QUOTA_MAPPING_PROMPT)Optionalquotascttddxsdjj}|dvryt ttddS)NENV>devtestlocal developmentFQUOTA_REDIS_CACHE_ENABLEDT)getattrrstriplowerbool)env_names 2/var/www/html/strategist-ai/app/services/quotas.py_quota_redis_enabledr sH%,299;AACH:: "=tD EET)hostportusernamepasswordssldecode_responsessocket_timeoutz1Redis client for quotas initialized successfully.z#Failed to initialize Redis client: z0Redis quota cache disabled for this environment.c(eZdZdZdeffd ZxZS)QuotaExceededErrorz?Exception raised when a tenant has exceeded its resource quota.messagecF||_t| |jyN)r,super__init__)selfr, __class__s rr0zQuotaExceededError.__init__4s  &r!)__name__ __module__ __qualname____doc__strr0 __classcell__)r2s@rr+r+2sI'''r!r+ceZdZdZy)QuotaValidationErrorz@Exception raised when quota limits or usage cannot be validated.N)r3r4r5r6r!rr:r:8sJr!r:> ai_tokens total_ticketsai_conversationsstorage_vector_capacity ai_responsesuSWe’re unable to validate your service limits right now. Please try again shortly.v2)tokentokens character characters)ticketticketssupporthelpdeskzcustomer queryzcustomer queries)storagevector knowledgekbdatabasefilefilesindexing) conversation conversationsthreadthreadsz chat sessionz chat sessions)response responseszassistant messagezassistant messageszai replyz ai repliesr<r=r?r>r@ metric_typec|dk(s|dk(ry|ttt|trttt|trU|j j dd}|jdk(ry|jstt|} t|}|dkrtt|S#ttf$rttwxYw)NrAz-1,r unlimited) r:QUOTA_VALIDATION_MESSAGE isinstancerr7rreplacerisdigitint TypeError ValueError)rZ raw_limitcleanedlimits r _coerce_limitrhRsB)t+"#;<<)T""#;<<)S!//#++C4 ==?k ) &'?@ @ =I rz"#;<< L z "="#;<<=s " CC"c dt|tstttj |j }|r0tjdt|ttit|}tDcic]}|t||j| c}Scc}w)Nz%Quota mapping missing required keys: )r_dictr:r^CORE_REQUIRED_LIMIT_KEYS differencekeysloggererrorsortedOPTIONAL_LIMIT_DEFAULTSREQUIRED_LIMIT_KEYSrhget)mappingmissingkeys r_validate_mapped_limitsrwns gt $"#;<<&11',,.AG ;;s# #$ r!c tDcic]}|dc} |D]\}t|t|}|tj D]*\}}t fd|Dst || |<\^t fdtDr5tj D]\}} j|| |< StDcgc]} j||}}tjd|ycc}wcc}w)Nc3&K|]}|v ywr.r;).0keyword feature_texts r z,_map_limits_from_features..sCw7l*Csc3DK|]}j|duywr.rs)rrvrts rrz,_map_limits_from_features..s LC7;;s 4 ' Ls z;Could not deterministically map all quota limits. Missing: ) rrrrFEATURE_KEYWORDSitemsanyrhallrkrqrsrnwarning) aivora_featuresrvrrerZkeywordsdefaultrurrts @@r_map_limits_from_featuresrs$78SsDy8G" ,W5 *73   %5%;%;%=  !KC(CC'4[)'L $   L3K LL399; 'LC{{3'&  '6Ss'++c:J:RsSGS NNPQXPYZ[ +9&Ts DD %D c|rt|tsiStd|Dd}|r|jdsiS|d}t j |d}t j|jj}dtd|}trL tj|}|r3tjdtt j|S t%|}|r_tj'd trB tj)|d t j |tj'd |S|Stj'd t j |d} t+j,| } t.j0j2j5t6j8d| dgddi} tt j| j:dj<j>} tj'dt j | trB tj)|d t j | tj'd| S| S#t $r#}tj#d |Yd}~d}~wwxYw#t $r#}tj#d |Yd}~|Sd}~wwxYw#t $r#}tj#d|Yd}~| Sd}~wwxYw#t $r.}tjAd|dtCtDd}~wwxYw)z Intelligently maps Aivora features to internal quota categories using an LLM. Caches the results in Redis for 10 minutes. c3JK|]}|jddk(s|yw) productNameAivoraNr)rps rrz$get_mapped_limits..s Xaquu]7Kx7WXs##NfeaturesT sort_keysz quota_map::z#Found quota mapping in Redis cache.z Error reading from Redis cache: zLSuccessfully extracted quota limits deterministically from product features.iXz?Stored deterministic quota mapping in Redis cache (10 min TTL).z:Error writing deterministic quota mapping to Redis cache: z3Quota mapping cache miss. Triggering LLM mapping...)indent) features_listuser)rolecontenttype json_object)modelmessagesresponse_formatrz-Successfully extracted quota limits via LLM: z5Stored new quota mapping in Redis cache (10 min TTL).zError writing to Redis cache: zFailed to map quotas via LLM: exc_info)#r_listnextrsjsondumpshashlibmd5encode hexdigestQUOTA_CACHE_VERSION redis_clientrndebugrwloads Exceptionrrinfosetexrformat openai_clientchat completionscreater LLM_MODELchoicesr,rror:r^) product_features aivora_datar features_json features_hash cache_key cached_valedeterministic_mappingfeatures_list_strpromptrWrts rget_mapped_limitsrs :.>#E X#3XZ^_K kooj9 !*-OJJ$?MKK 4 4 67AACM01=/BI C%)))4J BC.tzz*/EFF 6oF bc  a""9c4::>S3TU ]^%$$$ KKEF= JJqA%,,;LM %%1188$$%&9:#]39  *$**X5E5Ea5H5P5P5X5X*YZ CDJJwDWCXYZ  E""9c4::g3FG STwK C NN=aSA B B C a![\][^_``$$ a0 E!?sCDD E = 5aS9D I"#;<<=szA J ?J:*CL?K)L L J7J22J7: K&K!!K&) L2L LLL M!)M  Mbaselinec`|r;|j/|jtjj d}t j }|j|jz dz|jz|jz }|t|z}||kDr|t|dz zS|S)zPCalculates the start of the current monthly cycle relative to the baseline date.Ntzinfo )months) r astimezonerutcr`rutcnowyearmonthr)rnow months_diff candidates rcalculate_anniversary_startrsHOO/&&x||4<FFdFS  5HOO$5*844!)-I // CKK +;t L S S UV``bM/5#'' 2 JJ|,E--eO.DEHyy1HSXS\S\]lSmX33EIIo4NOswN"N=( @ Kghi!%Mmn6T ? {Jbcpbqqrst(!%&.%7%7%9%2BO]%<%<%>UY"%--/    DJJy,AB KKA)M N$+H!)!3!3!5!.>K!8!8!:QU!mmo I   Y 9(= >+844 5 ? {"QCPQ 'C*8445sGJ K +K>K K user_idrequested_amountc $tjd|d|d|dt|\}}|s(tjd|dt t t |}|s(tjd|dt t |j|}t||}|dk(rtjd |d y t|||}d } d } |dk(r"t||} | td |z|kDrd} n|dk(r"t||} | td |z|kDrd} n|dk(r*t|} |dzdz} | } | td |z| kDrxd} nu|dk(r"t||} | td |z|kDrQd} nN|dk(r"t!||} | td |z|kDr*d} n'tjd|t t | rTtj%d|d|d| d|d t'|||dd d!d"d#d$}|j|d%}t)|tjd&|d|d| d'|d(|d y #t$rt"$r4} tjd|d|d| dt t d } ~ wwxYw))zK Checks the current usage against the monthly (or overall) limits. z Checking z quota for tenant: z (User: )z#No quota features found for tenant z6. Blocking request because limits cannot be validated.z&No quota mapping generated for tenant rAz Quota for z is unlimited.NFrr<) start_dateTr=r?ir>r@z Unknown quota metric requested: z$Could not calculate quota usage for z in tenant rrzQuota exceeded for z (Tenant: z , Usage: z , Limit: uKWe’re unable to process your request right now. Please try again shortly.umWe couldn’t create your support request at the moment. Please contact our team through the Contact Us page.usThat information isn’t available at the moment. Please get in touch with our support team for further assistance.zJThis service is currently unavailable. Please try again in a little while.uOYou’ve hit your AI response quota. Upgrade or wait for the reset to continue.rYuLYou’ve hit your resource quota. Upgrade or wait for the reset to continue.zQuota check passed for /z, Start Date: )rnrrror:r^rrsrhrrr rrr r r rrsend_quota_notificationr+)rrZrrrrlimitsrgr quota_hitusage usage_bytes limit_bytesrmsg_maperr_msgs r check_quotar Js KK)K=(;I;hwiWXYZ':)&D#m  :9+E{|}"#;<</ 0F  =i[H~ A"#;<< JJ{ #E +u -E { z+n=>&i1A=QJI E= + %)) KEs1.//%7T O +$Y:FEs1.//%7T 5 5,Y7K$,-KES$455 CQUY . .1) SEs1.//%7T N *1) SEs1.//%7T LL;K=I J&'?@ @,[MI;iX]W^^ghmgnnopq ;@ gM(] lm  ++k+yz )) KK)+j 9UZT[[\]b\ccqr|q}}~A-  = ;K= T]S^^`ab`cdos t"#;<<=sC2I J/J  JcKt}d|dd}tjd4d{} |jtj ||d{}t jd|jdddd{y7b79#t$r"}t jd|Yd}~;d}~wwxYw77#1d{7swYyxYww) NzBearer zapplication/json) Authorizationz Content-Typeg$@)timeout)rheaderszQuota notification API result: z#Failed to send quota notification: ) generate_system_tokenhttpx AsyncClientpostrNOTIFICATION_API_URLrnr status_coderro)payloadrCrr rrs r _do_notifyrs ! #E#5'**G   .DD& Dkk("?"?gW^k__A KK9!--I JDDD_ D LL>qcB C C D DDDDsy-CBCC%BB&B C C CB CB<7C<CCCC C CCexpires_in_secctjstjdyt t j }dtj |||zd}tj|tjd}|S)zR Generates a system JWT for authenticating with the notification service. z5SYSTEM_JWT_KEY not configured. Notification may fail.rsystem)rserviceiatexpHS256) algorithm) rSYSTEM_JWT_KEYrnrrbtimeSYSTEM_SERVICE_NAMEjwtr)rrrrCs rrrsj  " "NO diik C//^# G JJw 7 77 KE Lr!c|stjdydddddd}|j|d\}}|d |d d |gd }tjd |d| t j }|j r|jt|yt jt|y#t$r"}tjd|Yd}~yd}~wwxYw)Nz5No user_id provided for notification alert. Skipping.)u Alert 🚨 Aivora AI Token LimituKAI token quota approaching limit — chatbot responses will be paused soon.)u$Alert 🚨 Aivora Ticket Count Limitu;Ticket count limit reached — new submissions are on hold.)uAlert 🚨 Aivora DB Storageu@Vector DB Storage at capacity — new training uploads disabled.)u*Alert 🚨 Aivora Conversation Count LimituBConversation limit has exhausted — Live chat temporarily paused.)u#Alert 🚨 Aivora AI Response Limitz8Aivora AI response limit hit. Upgrade or wait for reset.rY)zQuota Limit hitz"A resource limit has been reached.z

z

T)r{r} isSelectedAlluserIdsz"Triggering quota notification for z to user z'Error orchestrating notification task: ) rnrrsrasyncioget_event_loop is_running create_taskrrunrro) rrZrr r{descrlooprs rrrs  LMIA$HQ{ G++k+deKE4TF$'9 G KK4[M7)TUD%%' ??    Z0 1 KK 7+ , D >qcBCCDs>B<B<< C'C""C')Nr)i)Floggingrrredisrrrdateutil.relativedeltarapp.services.databaserrr r app.services.firestore_servicer r app.core.llm_clientr rapp.core.configrapp.core.promptsrrr)r%r#typingr getLoggerrnr rRedis REDIS_HOST REDIS_PORT REDIS_USERREDIS_PASSWORD REDIS_SSLrrrror+r:rkrqunionrmrrr^rrr7rhrwrrrrrrbrrrr rrrr;r!rrAs$  220 h7$1     8 $F  @"u{{$$$$((,,""!   GH KKBC''  - B/445L5Q5Q5STp@gpor s8 V (0B=H(&"s"s" E5CE54E5PXE5NQA3QASQA8C=QAcfQAfD #$ De @ :1#>??@s