jne ddlZddlZddlZddlZddlmZmZmZddlmZddl m Z m Z m Z m Z mZddlmZddlmZddlmZddlZddlZddlZddlZddlmZej8d Zd ZdZer^ ej@ejBejDejFejHejJd d ZejMdnejMdGdde'Z*Gdde*Z+hdZ,ddiZ-e,j]e-j_Z0dZ1dZ2e1e,e2dhiZ3e1e-e2iiZ4e2hZ5dZ6dZ7dddd d!d"d#Z8d$e9fd%Z:e1fd&Z;d'Ze1fd*Z?d+efd,Z@d-e9d.eAfd/ZBd0e9d1eCd2efd3ZDdd4e1fd0e9d$e9d5ee9d6eAd7e9f d8ZEd9ZFd=d:eAd.e9fd;ZGd<ZHy#e'$rZ(ejSde(YdZ([(dZ([(wwxYw)>N)datetime timedeltatimezone) relativedelta)fetch_tenant_limitsget_catalogue_product_countget_ticket_countget_vector_db_sizeget_total_token_usage)client)settings)QUOTA_MAPPING_PROMPT)Optionalquotascttddxsdjj}|dvryt ttddS)NENV>devtestlocal developmentFQUOTA_REDIS_CACHE_ENABLEDT)getattrr striplowerbool)env_names 8/var/www/html/strategist-ai/app/services/infra/quotas.py_quota_redis_enabledrsH%,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 rr/zQuotaExceededError.__init__9s  &r )__name__ __module__ __qualname____doc__strr/ __classcell__)r1s@rr*r*7sI'''r r*ceZdZdZy)QuotaValidationErrorz@Exception raised when quota limits or usage cannot be validated.N)r2r3r4r5r rr9r9=sJr r9> ai_tokens total_ticketsai_conversationsstorage_vector_capacity ai_responsesAivora BrandForge product_countuSWe’re unable to validate your service limits right now. Please try again shortly.v2)z product countrCproducts cataloguecatalogskuskus)tokentokens character characters)ticketticketssupporthelpdeskzcustomer queryzcustomer queries)storagevector knowledgekbdatabasefilefilesindexing) conversation conversationsthreadthreadsz chat sessionz chat sessions)response responseszassistant messagezassistant messageszai replyz ai replies)rCr;r<r>r=r? metric_typec|dk(s|dk(ry|ttt|trttt|trU|j j dd}|jdk(ry|jstt|} t|}|dkrtt|S#ttf$rttwxYw)Nr@z-1,r unlimited) r9QUOTA_VALIDATION_MESSAGE isinstancerr6rreplacerisdigitint TypeError ValueError)r` raw_limitcleanedlimits r _coerce_limitrnqsB)t+"#;<<)T""#;<<)S!//#++C4 ==?k ) &'?@ @ =I rz"#;<< L z "="#;<<=s " CC"c t|tstttj |t }tj |t}|j|j}|r3tjd|dt|tti||}|j|j}|Dcic]}|t||j | c}Scc}w)NzQuota mapping for z missing required keys: )redictr9rdREQUIRED_BY_PRODUCTgetCORE_REQUIRED_LIMIT_KEYSOPTIONAL_BY_PRODUCTOPTIONAL_LIMIT_DEFAULTS differencekeysloggererrorsortedunionrn)mapping product_namerequiredoptionalmissingrwkeys r_validate_mapped_limitsrs gt $"#;<<"&&|5MNH"&&|5LMH!!',,.1G ),7Ow(* +"#;<<%%W%G >>(--/ *DAE F#CsGKK$45 5 FF Fs#C7ct|tsyg}dD]0}|j|}|s|jt |2dj |j S)Nr)name featureNametitlelabel descriptionr resourceName )rerprrappendr6joinr)feature text_partsrvalues r_extract_feature_textrs` gt $J^* C     c%j )* 88J  % % ''r cbt|tsydD]}||vs|j|cSy)N) cappingValue capping_valuermrquotacap)rerprr)rrs r_extract_capping_valuers9 gt $R$ '>;;s# #$ r c tj|t}tj|t}|j |j Dcic]}|dc} |D]a}t| t|}|tjD]/\}}| vr t fd|Ds t|| |<act fd|Dr1|jD]\}} j|| |< S|Dcgc]} j||} }tjd|d| ycc}wcc}w)Nc3&K|]}|v ywr-r:).0keyword feature_texts r z,_map_limits_from_features..sCw7l*Csc3DK|]}j|duywr-rr)rrr|s rrz,_map_limits_from_features..s 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.}tjCd|dtEtFd}~wwxYw)a Maps one product's plan features to internal quota categories, using an LLM where the keywords cannot settle it. Caches in Redis for 10 minutes. The product is a parameter because each one measures something different: Aivora counts tokens and conversations, BrandForge counts products. Reading Aivora's numbers for a BrandForge tenant is how a customer who never bought Aivora ends up unable to build anything. c3JK|]}|jdk(r|yw) productNameNr)rpr}s rrz$get_mapped_limits..s+BqEE-0L@Bs #NfeaturesT sort_keysz quota_map::z#Found quota mapping in Redis cache.z Error reading from Redis cache: No z% limits found in the plan; unmetered.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)$relistnextrrjsondumpshashlibmd5encode hexdigestQUOTA_CACHE_VERSION redis_clientrxdebugrloads ExceptionrrUNMETERED_IS_ALLOWEDinfosetexrformat openai_clientchat completionscreater LLM_MODELchoicesr+rryr9rd)rr} product_dataproduct_plan_features features_json features_hash cache_key cached_valedeterministic_mappingfeatures_list_strpromptr^r|s ` rget_mapped_limitsrs? :.>#E B$4BCGIL |// ; (4JJ4EMKK 4 4 67AACM01<.-QI C%)))4J BC.tzz*/E/;==66K6BD$9M)M  c,'LMN  bc  a""9c4::>S3TU ]^%$$$ KKEF= JJ'?K(CM6?L5M7M9 K%K  K%( L1LL M L>8M>MM M=)M88M=baselinec`|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 astimezonerutcrfrutcnowyearmonthr)rnow months_diff candidates rcalculate_anniversary_startr%sHOO/&&x||4<sw--9%00>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 ruser_idrequested_amountr}c tjd|d|d|d|d t|\}}|sP|tvr tjd|d|d|d y tj d |d t t t||}|sP|tvr tjd |d|d|dy tj d|d t t |j|}|$|tvrtj|d|dy t||}|dk(rtjd|dy t|||} d} d} |dk(r"t|| } | td|z|kDrd} n|dk(r"t|| } | td|z|kDrd} n|dk(r*t|} |dzdz} | } | td|z| kDrd} n|dk(r%dd lm}||| } | td|z|kDr}d} nz|d!k(r%dd"lm}||| } | td|z|kDrSd} nP|d#k(r$t'|} td|}|r2| |z|kDr*d} n'tj d$|t t | rYtj+d)|d*|d+| d,|d t-|||d-d.d/d0d1d2|d3d4}|j|d5}t/|tjd6|d*|d+| d7|d8| d y #t $rt($r4}tj d%|d&|d'|d(t t d }~wwxYw)9a Checks the current usage against the monthly (or overall) limits. product_name selects whose numbers apply. Catalogue work passes BrandForge, which caps products and meters nothing else; the chat passes Aivora, which meters tokens, conversations and tickets. z Checking z quota for tenant: z (User: z , Product: )zNo plan features for z; z is unmetered under rNz#No quota features found for tenant z6. Blocking request because limits cannot be validated.rz limits on z 's plan; z is unmetered.z&No quota mapping generated for tenant z does not meter z ; allowing.r@z Quota for z is unlimited.Frr;) start_dateTr<r>ir=)get_conversation_thread_countr?)get_ai_response_message_countrCz 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.zYour plan allows zc products and this would take you past that. Remove some products or upgrade your plan to add more.)r;r<r>r=r?rCuLYou’ve hit your resource quota. Upgrade or wait for the reset to continue.zQuota check passed for /z, Start Date: )rxrrrryr9rdrrrrnrr r rr r #app.services.integrations.firestorerrrrrsend_quota_notificationr*)rr`r r r}rrlimitsrmr quota_hitusage usage_bytes limit_bytesrraddingrmsg_maperr_msgs r check_quotars KK)K=(;I;G!+l^1>?':)&D#m  / / KK/ {"[MJ++7.; <  :9+E{|}"#;<< / >F  / / KK#l^;yk&-~7 8  =i[H~ A"#;<< JJ{ #E })==  |n$4[MMN +u -E { z+n=>&i1A=QJI E1= + %)) KEs1.//%7T O +$Y:FEs1.//%7T 5 5,Y7K$,-KES$455 CQUY . . 2) SEs1.//%7T N * 2) SEs1.//%7T O +0 :E ,-F%&.50d) LL;K=I J&'?@ @,[MI;iX]W^^ghmgnnopq ;@ gM(] lm 18>?  ++k+yz )) KK)+j 9UZT[[\]b\ccqr|q}}~A9  = ;K= T]S^^`ab`cdos t"#;<<=sD!K77L= /L88L=cKt}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 AsyncClientpostr NOTIFICATION_API_URLrxr status_coderry)payloadrJr r rrs r _do_notifyr)s ! #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) r SYSTEM_JWT_KEYrxrrhtimeSYSTEM_SERVICE_NAMEjwtr)r*rr'rJs rr!r!sj  " "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.)r;r<r>r=r?)zQuota Limit hitz"A resource limit has been reached.z

z

T)rr isSelectedAlluserIdsz"Triggering quota notification for z to user z'Error orchestrating notification task: ) rxrrrrasyncioget_event_loop is_running create_taskr)runrry) rr`r rrdescr'looprs rrr*s  LMIA$HQ{ G++k+deKE4TF$'9 G KK4[M7)TUD%%' ??    Z0 1 KK 7+ , D >qcBCCDs>B<B<< C'C""C')i)Iloggingrrredisrrrdateutil.relativedeltarapp.services.infra.databaserrr r r app.core.llm_clientr rapp.core.configr app.core.promptsrr"r9r5r3typingr getLoggerrxrrRedis REDIS_HOST REDIS_PORT REDIS_USERREDIS_PASSWORD REDIS_SSLrrrryr*r9rsrur{rwREQUIRED_LIMIT_KEYSAIVORA BRANDFORGErqrtrrdrrr6rnrrrrrrrhrrr rr)r!rr:r rrRs  2208$1     8 $F  @"u{{$$$$((,,""!   GH KKBC''  - B/445L5Q5Q5ST    $!  ##|p 0?gpor s839G" (>D:6<T=l(&"s"s" E5CE54E5PXE5NLP()vAA3AASAA8C=AA"%AA9<AAFD #$ Dq @ :1#>??@s6AFF=F88F=