From 8ed85b05230f88dff093e3ff69fd2025f445cf6c Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Sat, 9 Sep 2023 16:10:41 -0700
Subject: [PATCH] rename apimanager to budget manager

---
 litellm/__init__.py                          |   8 +++++---
 litellm/__pycache__/__init__.cpython-311.pyc | Bin 8075 -> 8070 bytes
 litellm/__pycache__/main.cpython-311.pyc     | Bin 30921 -> 31623 bytes
 litellm/__pycache__/utils.cpython-311.pyc    | Bin 92489 -> 90253 bytes
 litellm/{apimanager.py => budget_manager.py} |   7 +++++--
 litellm/tests/test_api_manager.py            |  18 ++++++------------
 6 files changed, 16 insertions(+), 17 deletions(-)
 rename litellm/{apimanager.py => budget_manager.py} (86%)

diff --git a/litellm/__init__.py b/litellm/__init__.py
index 23dff09d5eb..704b36e3bbc 100644
--- a/litellm/__init__.py
+++ b/litellm/__init__.py
@@ -36,9 +36,11 @@ caching_with_models = False  # if you want the caching key to be model + prompt
 cache: Optional[Cache] = None # cache object
 model_alias_map: Dict[str, str] = {}
 
-####### APIManager ###################
-from .apimanager import APIManager
-apiManager = APIManager()
+####### BudgetManager ###################
+from .budget_manager import BudgetManager
+budget_manager = BudgetManager()
+
+#############################################
 
 def get_model_cost_map():
     url = "https://raw.githubusercontent.com/BerriAI/litellm/main/model_prices_and_context_window.json"
diff --git a/litellm/__pycache__/__init__.cpython-311.pyc b/litellm/__pycache__/__init__.cpython-311.pyc
index 1d6568e1e56bd228840101147e3acf0da8f1d138..f23105fe1b09a23d396ed6d1cdf7611985c50ccc 100644
GIT binary patch
delta 523
zcmZ9`J4?e*6bJC!wA30&L9i9kw!R;ISl>p|Hnpjbs-ugSYEkYXDy@U)Qjm5K2L(~0
zo0Efr;!v`=xH~!e37j09eSoKzg5o9of5|x~hkL9YYVwUNO9GzK#}~UV%MW5#FY_G;
zG$0BPxW>j^rOOWuZVEI64v-*hzKLz5-*nj;y%C&@{F@wL?YLu*5i@S{6b*rOToJ0E
zo#+JGg-)W~sEYQWQM4D0u|5_sL>gy77D6?whq0bO`_U;hf+o=cR7VHV6euuJz^lR}
zPnBo1P^^?<{3F3L&Qs$##WTq>#Z#{&oyMg@xOC=sDX0*I1SBB^F^EI;e>Qw6(OEWP
z5IV<N*k~@DPt!CJ@O>;dzl7GX@!Udwk($f)SaY?JsWfn4g{$ON^I2+^?QZMIzC~)}
z<Q_9%*HRoQk%X;%=LR8B^R+fZR8yCpiEcI2-AV+kM|Ue}aIN#{Y<_Kf&48U9Ki=Q^
k^c;!g!a8iCqBrm?^!<Ha>V2o)Gyl22<a&2n-_0ksA8KET7ytkO

delta 562
zcmZ9HJxClu6vubgL*j-_mmH!oe#Cruew=!q-sigBorx!ch1FyISe=*CuoJ}6MuiYe
z6q^OBKp;f~tz^5DmR6P_RoXy`RKda`{~Lt_hk3u><3Gb==AE<W*hjW)8CaiwTuF6m
ze>J;%WZPSV5~jh4FE}`<pXx=uYX<dkBU{`b{xUnnKzOeqzLmty<gJ^7vJ*3gq@ZC`
z3N?aCqef9<HA8lN7L<V>!&$f+&cRRMJp2?cz|Y{Kd@g%xCOI-8d*N~9eaKz7e+$h7
zngO^3C*d+Y2v<16*-bp2bDDY0f@V?E)pRt+H77Jnnib7*oivG=LzwyE-%QSPfzvLR
zxWYwt_-23{{%FxlnW_nzlC5$isH!Seg@HaDOwZLl8iez)LP~k$-Q@E+PX`OyW5J?Y
zq9x^%ua;?9t<Z{+Boz}!=dsr!X1TLdQLeFUibS-1XH|%7__N6qnQ+p&5N;H|du)kU
z(b@fW(Q5hYYs!D)*LY*&5q|Q$9?V<^(Vqt&OmP_PG?V$T>%!>#e}A*%yY$}M_qMIy
KvFNhppy4k{qn5A$

diff --git a/litellm/__pycache__/main.cpython-311.pyc b/litellm/__pycache__/main.cpython-311.pyc
index b35771c9646c97caacaefb81ae68c277f0920b11..2c5b0d3a41a9b43b0c6d71100a930c50d7dd1c59 100644
GIT binary patch
delta 2349
zcmZuye@q+K9e?kRvjH2Qjln+vV`F2mF}4#R0Vi349Sotgp-G!%b(IAtXGp+4!`YBQ
z&Z2*GTe3wf?d!B?LkrWktZEW1(e;nykBOp8TPiIH&fJzTO^q}uQYW=LvMgD*blrOg
z5n<ZX-S_kPzW4q0-hH3$gBAGQ$58jHva(Wu$6p^m;VS2@=w5=;YOXwKOL<}*2spsO
zq&McpXiPSxd@-LgSCM2=&9UZ`Kju#bVu4gJ7KDHT3GeVG5{nG3@;n6K0)AIVY%`8w
zN?KCZm^D=ut4g)RT2j@qYQ?W)xb>;sd4TT0?=ybRk?^BYs&R85vDNx?5pcf5M!ore
zbbq1+-Jr(S{p&W`{~v$pJM{0NT9r;F<Ec1$j!8&287&1P(z3b*!WOBlWSFA<;6{_0
zU2saerkREqLS^$5GbI8MqG{HIjQYROdLE)L^wn;IXyNKcC{lAqgSm5!Gn`8V6JQ1k
zL!49c8&1OY0MhmHR!HmkljsX$b!7?HL@YB5=Tkf=0dP*_H@qyJu9#o-@{+XDQPWLG
zYk5v;wfG>d;crTDtC5h1bfT&e(xv<*^s{OobV@g?KYb=0%G*xDDfB|^A-aTblrGiw
zlRbTDpNF)X|EV-zcN)UGQm>1Jw2FUGdbK`E(G1@#{k*|U(M7yn`i;j5X_|i%-Su9_
z7(nNmF0e~@vqP(!#hcuL!a5U5`DUr%uOrxOr`>aQ^rosZsNsgs<I}mIoDexBu6ET)
zpv2CX05_5gi=AEIkUbX>!(s%DnXPrnXIh|jollhkkPDx2h+(u4_=cWKiC_u`6ze(T
z4vgrWBL~Ciw!;S7(78Gr3;7@78u_&kq9+w<x%66g#PMifvk@LdZ&=%nqZ`{dw>x_U
zEm%UX-J<Ja=sdnM7nEy5>d)0@C4#GCIEFO-KFFi#rW%y-8|hbbT{w{$&G}W{kz7=a
za$v0h73L#r+1c#1rxEapjG^0p9~4l!(T$D-%mFcqTg^KJ#S*bdWJDX6nPs@_EOl{w
z<5A|i#qPOp;$yBrodde!Tu%uQd$<V{t~0|ESelA*d&E8Xu#T5Ko9<hvJs333uNQ!`
zSwzppNi-R(gD0h*1b+*wmR0G|1Rp;>r60_08yd~}TLWAC*>T}WTAJMKfN+QO<CdK?
zd??-BVxwRTJ#L?clj!Y^S16$x!2v6>x8NoYb;6GkA2OS|1@eiO87`5`#;4S)*Dr{^
z9SWeUp#|8CrnX&z1IV+z2Dfi?`vhH7jqTADrUuRL?1#M)yX$)}aEZhy8A|X8A)ZYO
z%dC*djHdZa;&3J_9Os9YnWM+X`4QnWY$wR**PTJwi$3Xm4mE}iRlRsJvuFn3HSmuZ
z1lsb`9cMaT>yoKM(*E#uTD!zL=Z@WB1B+~6Nis#BKqk+6WWD6<aqD0n?KyC4S_Q23
zf+cj-^M3Tk*1TnZ!Lt8U?-E=4G3&g8y*u7HkY`&9Z0qIh9d`R7yZxTov8Z?4bJowW
zOST3XK*I}gRnA10DqM31-U$`lTQ0{7?vAVGf;;@ax8Qz0U-A4*|4hHE0ui_mp4xjh
zRM5KT;svb-{d-?cS@5zu-_VYy+w)o%O7C?VF8Al_y9)JPc}KM1h%Rbv=%c+}>BfNs
zHQh&nU-TNo7(VFeQt+0xac>*A)e6bvt+vwscKX)Nw*D6Sb_;{~?e+$;uvps-2!3Rb
zvIPGWjFb}m^R`F@!Fz1CmEe7E(O>}F_bbuv2MmL|==-}OezNqbKe}a_kiWSi#RONH
zl>#1sB7p%9iWHXzj6pEb4@_NJV)4jTG~oe{8kMC-9>YY4eiVvsCQDCzVJE@==)#o*
zzp9TYg)M~JqBUe<p~tX$lS29^T&jy+=+l;K!hN-v)G~NN+WN`>glhDoSR13kKdOV$
zVr-!Zo|WD@Y=-cc=-q)qSc44l_n-m&F5Ur6=&N`qQ->|%QrD5IYPf>FJl>=EJ83QQ
z{}LXe{$V%MfDPW2CWjBw4E|e!&!rDjINU#_&v*f1mx0lCrWMaWL9?R*{2Z}ke<T;=
zlQE~Kh6J$@QA>oK2nQC++Mz`DaDqRY=5R|3r%^!oT9{NUkYWUqP#~#<Q^bP=7a9oL
zPJ|bWe--aYrMdBBVy6(rtM~z#66vDAQhLiG*zllh*&jeX8XQkiJzDhPcx1Xq3rl6N
zE|)YqnO0O4V2!d`p~XO7A(tq$6jYdHjY74c+$@(VR0p&MnN_GB6l-LILd!v!K{hJX
z1PmtGtWXOu*T`0d)_`e4rEF7pEx?)W3iSY;Y3AseL$X&<n?QNhS=ZY(*{7(@fHlu#
y&ZK0&qP_@$$|&!Hit@}xR@4+lJp@>+(10p_-6R()CNEpmaFYzygj9LrJ@{YPz<4<T

delta 2094
zcmZ`(4Qv}%9e@8jpY7O=9e>21NnFQG<HR4ao0KGNM{7G@RkdZTjBch`n&zA=ZhT44
zPST2tGO@u{wHsL<(g0PWm8MRkE6}}38<VOLjZq3xn^KrcDx`p*QYEHd0tApM%6pfV
z5(Fpv|NH&k|MT8Ey_f$6S60CKR%N9L(MMRmC%$<4qIDn4ArZzSxrm5F$9Pk*IoAw`
za_6{EjOL=nST0tK=i<dgE&+%``ADIK<+De0odbl<)BpODOVYgRac|L=^A-I$e=(Iy
z6$80|R?kh>x8}BLH7W|N4|C2TybLFVn7An)!;@TNDxUwG@1YZk(fmdo`!5~Or|{3X
zBaiY!8{_{+KKVL-8SsALP+b!&`A75R!HMzlq2kb>oG(pINTvK>sVwJ*isVOvg@Z2g
zQ^Qt(Hu4W+f#d2n<L90W;+M>uOfzFhsecn*gMFtg9`cEK5MYKhRDQ(43_j}!oT~bL
zyG?3VP+9lyw?PHZLq3g4X2mOp4s&d!B#8A2nnv>=kBVU&a($5;vp2GWXC03NUnzYL
zuQ&rCn@H=iOfjmVfmR5wSD5EyB!)@I<>Z7J1u0Fq-{B)qxqlA4S$dTOy;0yz(i(Z(
z=VatC8TK~<Um<;pUkyZowRDqffg7|X9o~;G)qaJt&bk5KC_PEOUDwZMxfA>dc)fI)
zoDUrb;K;VH4R}F1MtV1QbG%M^ntWr6hvRvvkGvi60nbSv;!m1Brnn!UXg*&FbllPP
zape(wEs(@h(XV2Nc?`SvA?hpOg?KP{_(22~3$tFOYO;C527U?OkGDI=H-2}s>GBWo
z1#c&$@$bSeYx&~CT}G(zEbedi;!BBEDBzVu3!j_K;9Yi43$HjPK`hQ@m8^&!oPwV6
z;)C|K^81gfh!P%aiNXZ_a$^IZIdcG?t*(w|**1@YQlaP+L8%odPYdFa)7(4qIeMn&
zwR2?^X1kQGGXpfkg)cPN4BZt-=@v_PDCU7OPA5HlkJ5v8MMkP5$A*?u_<FL%dFjy*
zw4GOpyfclRsSq3`+fqLP46ROsequ5&4UG|J>n4Ehq^<4qJgkyZdo2gg;MY4((}PTG
zf7ZC9U;oVU)kAJv)qcqPYdWGlOY6%!s97@RC2?|MOe*88I}(Dtldj=|I~Kr)>vvv&
zLA<!L214YsozuLop7~zf-EZ!dpJMs@n7(`PG2%~W!To)zmMTW`QeGY^PssA0Sjb&`
zBGaPlWq}Ks9<0w+`S(&$`hy#x7tmjGfGm#V9Va_p=u){plFWY0!!o(wa}TN(ZIPR{
z`ZLe}(w4YxOQ`yH_cT~x5pUW5yw`|)tv~Ge;OQUtT=(w%m3QwedzPIwi<TO4>X|$@
z*BeD|?s29mT<PelWO!BI_(TF-je{l<rv4OvwWF;+#$SsGlwV72VU9&_kDlSD!EPJF
zTP+zA!{2VtxEL<mdVCCTH|a(a==OTG+X?%~F8=neOq{vy#Jji6F>*JYF*00<X@jgH
zoh+bLoi@O#V3*zes=Ld=BG$sX=?GeD)Ld&3`*bJ2*4dq6u6xmRJ;Q%m)71>`Z%(%`
zTu8ez4I0UA&n7j}8#Sf-Ec`<6HitRgXW$q53<6zv_rO5_9UdBJ6O8n>?jqkASkS@O
z$bmy1fK#|UJOTlHYxq5|;(4(HocM~kTd1RvPY`SV0|Th|@_4U#l^rCzxaB`%U$H?5
z(||Wfwz!`cB9z`D-<_o8{z%?GA_GlnF0~6ux_$?blw|k}|E6>UUn~dX0al@gnObJ*
zm<dv|WMOx4UK*Yd$D~pDI9m_lzn35LvzcW!j?7xhteN}@b0=~CRIM{ZHT|JdwH_gr
z2Y{RS^{FO2Kjk1FPW=M#8#5<^^eX36^#6<rZ$4V&WHZJ`Gjp;TOe%V0Z<?(tuSq?!
zIaPy3ji}0{R%q0OTprb|Q44Z-)Jlz7k;SgsG+Kp>X4S4y2dcEIPK~;e-K}~w>P4O!
z)u+)KG-t0?Yc*bnXy>3tBgpEWA3iytHfgdMIsD%Wf3sGNYBGjwp83+rq8iualYsP2
o^$Vbh|9FuV1IKZHLDa19h%>(AR*hOn!DE1=iXIqp<Jf!fFG%er>i_@%

diff --git a/litellm/__pycache__/utils.cpython-311.pyc b/litellm/__pycache__/utils.cpython-311.pyc
index c8a4b9f9f99884e5be4b0e1b891a2e43e8ae7065..dee9000e5be26681575c8e4f48a4f4eb3967b1b7 100644
GIT binary patch
delta 395
zcmX?kiM972E8lWnUM>b8c=i2H%B`sr`6L*ZY*d$HRu4&0ZDEXxO;KxMh>A-IP0;}I
z;yD>$FeNNS3n-E>xqw+>asx99Q%d;cF6Kzah{;cw)fpov^RQU!N2M61dZ#$27NocU
zY1dTmG>#Ov7S^c3l*knK7RJP)6c3=`#VMXZvLq#Xav6(-Qc56@Q=Q3>&XA&=qMgQ)
z5(KoRCM6gsTARWc%%B<5{D5Wq0~W?)+wJST7{4+}RWLCy6e%(Si7z1H1&F8wiB(PC
z>&vLcS`B1x+Wy9uF^ZM3V!C@U<1@zU>3SiI{y<(`2%`#HJw)+}5XN-YHX!@Vc9BrV
z{j6*qAaS?JADyJ9e~Dt0X6py?zj16Ajb?0OVyv6KGM3Shv3~mPSVl$8TkMIM`57f8
z1=|_p7*{YcHcnraz?jPovbqRlWKsWgokT`iuBAXW(0vTWN45tiGFG!QDop>G!59Dl
D8Dn&*

delta 2380
zcmah~du&rx7(eGe`eH46uXbD4vTmbW%-j?A3d&kk9s`*ZHiS}lu3MpN=efl>c5A0B
z$YUT<lgm>lK_StY8BOM2Dq;Si@sE}?(ljBWF~ol~HEJ@AfB2o24Q3S2J@<FM^PO|P
z*FE2Nk6#tPye^t98;u48?eeX^NZt9@O=kS@hc#Bdrk_+z;elPGdI}BfCIe(O%shIu
z_>i5X7C3vNqr5pf$qO8LCVG_b$Gf6G@HKdMG)L&D_mBo@Bk7h#i3>)zwDCC|SwAHX
zJWqC$$EL8ClE*=~Or8J?kv-8DgohjTh<M3UAbEoH0b3zY1BQKQ2$4Ec_nbhs!`e$^
z2XG?<_af)B2V=rqOu&O(eSa1YheD(6_4WSXpw~Yfltb+e4b4r>DUILn9q~5)S-5?8
z+~$*Qp<&4;?;E#0>)$V}9pst&cbXPPo<r!@uK_k9jMQk)Hzlh7No=x&c?F$A1Yh97
zf`S!}aOb(Os7DGP{iUX)Mo`4}`FG*id+=skGdhR~n&6*CDt|}_YZT31KKjsRYxJ$%
zCUYZdLia3c6)n94!nA5bH!hBrp2`)%Iz^|(SHi4kbFL|@SM)BlYg5>u7-qE55Bpc;
z5Y4`f@w73N7k&Mi!iZvc7lQ%tW?Og#(JCumK-5BXL__poaE=1J2NpLVC5IR~VMbPv
z93{txhp-QCM_@3<XnOPXR3!=(E^S5*M9ad)P{~~mF)GG$=mM9~8~y6`%F?j$CFArv
z1hH`pkD(X&el&)sKWIgIHI|2qS`Tt}-Nq=8SBV<Xy*ybM<Fbb^>W5E&A;L2`%whAv
zdm@H`?nRYLi;?Avi;A75o&&~u@2IbSND4^Q8<LzHT@MaWGla>B>CZ6GN8p>$3yiwW
zmVRSYC8IVo%E>4f0}t#wP~@gGNNq$K3=&^p$hM((OHaEk;xwkTUSE*i!IZJv?~?+d
zUP+dH!2tbfQA$eZ3rW=D4UE%n5S0?;QNJ%l%bEI&Atpw>%v5ayNQr~~pe#8}w1P=k
zGCQS}??}GQ_&)-qA_A4<(Ujo#1*DW=uN(|`28Z_t_N6qLL8fk|5bvcblYSIHHUZd{
zHCoTKT85;MCzweZ4-*;;$dX46QOP?(4}p}Ab#V9y5I@2*iGJ6hmBopOsWNV=N|>r9
zA4?jGVg^UHPj=tRD^BK{(;}avNh5$sS6WvrT9XA9wewWx41dLOvNK*#pD3tLqXK=$
z{b9B_X<c>Rf5X~z-P#nlwj``AX_Rk-Q;`h*2b?6yf|4}K%j-yUxkVkxD(B^$7kAE<
z#H-p8Rc)sYNqgN5d&6~m!>l82Uz4z}Iju=n)uoaCfnJ=&)9z%2<E@Fa6KBHb!?WD=
zipE$)<4wovOB=52k2_itj!4T4PF6lRn>Vv|X6>zqI+9H-;B#v=_}p4O)05VrNAWuR
zdF{M$rf0S-?&wO^HGiW0+!|Z6HP-rMtae+>aknp|3z+JK-_5A3DvfXvIJgKm&E+v`
z+pHG=dfeQeFt^9d?GbqI?F-P-JRD-9oq5iqiq;s!i(eMyUs%w8`~S$_|GPl^mr4D{
zZf3*3si=jU^V-PKkt4zBU`$v3Uwz^5(EsgP*tf7COK^MNE_Z9urxurmziQUFG{RLY
z&*)W;wsn^Y*Ge%=u9X>Foc3CEes`Dl+QTB_bcw*3=NLV&%LfVb7K6*7op<nzuEC66
z&FC+*W$sG#%}!hrX$NhYe+2v#!dOB7sOVhwCvmsd9xx&l#?u9@NMV02z9(C3GS#IT
zWI6ei{!Wn!(rgW4IV-ao4_TVc&D?ArdP2bnOG&6dlx5;KEG2tjV>Ts|EAu8FPnOu=
z)`;x>+y@RzNnP$q_i1OM&=J=;4(kr<;HH@Ew^kHPZjl8L`Krdk4T6rPrU!~gbL-dP
z?>WmcSfO@ShB8%?)gn60c*oUsZd@y#0Q%dx=iPWf#K+XHH{jd&g!<7&yai_bChX9h
z%5Hve6CM)J0R6|@nN9dJQFD&**GIowSE`P0!=;*wFt60lo!y3C;qrMp#+F}XFb?3H
zRLlEt13sm0>BDxBP5^IiyboXD@Ehu?9eAXO6|PLED6@z%P4uFAb_Xsu%mEE%Bl+jK
it2^+7wyqkj?Zaul0$bB)Ik>QZ3)5%`^3|?Ac<bLzCygrr

diff --git a/litellm/apimanager.py b/litellm/budget_manager.py
similarity index 86%
rename from litellm/apimanager.py
rename to litellm/budget_manager.py
index e7d1f1eaf7e..fa771315650 100644
--- a/litellm/apimanager.py
+++ b/litellm/budget_manager.py
@@ -1,6 +1,6 @@
 import litellm 
 from litellm.utils import ModelResponse
-class APIManager:
+class BudgetManager:
     def __init__(self):
         self.user_dict = {}
 
@@ -22,4 +22,7 @@ class APIManager:
     def update_cost(self, completion_obj: ModelResponse, user: str):
         cost = litellm.completion_cost(completion_response=completion_obj)
         self.user_dict[user]["current_cost"] = cost + self.user_dict[user].get("current_cost", 0)
-        return self.user_dict[user]["current_cost"]
\ No newline at end of file
+        return self.user_dict[user]["current_cost"]
+    
+    def get_current_cost(self, user):
+        return self.user_dict[user].get("current_cost", 0)
\ No newline at end of file
diff --git a/litellm/tests/test_api_manager.py b/litellm/tests/test_api_manager.py
index d8d0077de91..18d507a7fa1 100644
--- a/litellm/tests/test_api_manager.py
+++ b/litellm/tests/test_api_manager.py
@@ -8,16 +8,13 @@ sys.path.insert(
     0, os.path.abspath("../..")
 )  # Adds the parent directory to the system path
 import litellm 
-from litellm import apiManager, completion 
-
-litellm.success_callback = ["api_manager"]
-
+from litellm import budget_manager, completion 
 
 ## Scenario 1: User budget enough to make call
 def test_user_budget_enough():
     user = "1234"
     # create a budget for a user
-    apiManager.create_budget(total_budget=10, user=user)
+    budget_manager.create_budget(total_budget=10, user=user)
 
     # check if a given call can be made
     data = {
@@ -26,7 +23,7 @@ def test_user_budget_enough():
     }
     model = data["model"]
     messages = data["messages"]
-    if apiManager.projected_cost(**data, user=user) <= apiManager.get_total_budget(user):
+    if budget_manager.get_current_cost(user=user) <= budget_manager.get_total_budget(user):
         response = completion(**data)
     else:
         response = "Sorry - no budget!"
@@ -37,7 +34,7 @@ def test_user_budget_enough():
 def test_user_budget_not_enough():
     user = "12345"
     # create a budget for a user
-    apiManager.create_budget(total_budget=0, user=user)
+    budget_manager.create_budget(total_budget=0, user=user)
 
     # check if a given call can be made
     data = {
@@ -46,12 +43,9 @@ def test_user_budget_not_enough():
     }
     model = data["model"]
     messages = data["messages"]
-    projectedCost = apiManager.projected_cost(**data, user=user)
-    print(f"projectedCost: {projectedCost}")
-    totalBudget = apiManager.get_total_budget(user)
-    print(f"totalBudget: {totalBudget}")
-    if projectedCost <= totalBudget:
+    if budget_manager.get_current_cost(user=user) < budget_manager.get_total_budget(user=user):
         response = completion(**data)
+        budget_manager.update_cost(completion_obj=response, user=user)
     else:
         response = "Sorry - no budget!"