From c0debab0cbfd33f9758811a24fc925b8e29b6f75 Mon Sep 17 00:00:00 2001
From: xds <iam@voroninv.ru>
Date: Thu, 5 Feb 2026 22:19:57 +0300
Subject: [PATCH] feat: Add `use_profile_image` and detailed token usage fields
 to generation models.

---
 .../google_adapter.cpython-313.pyc            | Bin 7479 -> 7787 bytes
 adapters/google_adapter.py                    |  10 +++++++++-
 api/models/GenerationRequest.py               |   3 +++
 .../GenerationRequest.cpython-313.pyc         | Bin 2600 -> 2775 bytes
 .../generation_service.cpython-313.pyc        | Bin 18279 -> 18666 bytes
 api/service/generation_service.py             |   8 ++++++--
 models/Generation.py                          |   3 +++
 models/__pycache__/Generation.cpython-313.pyc | Bin 2224 -> 2399 bytes
 8 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/adapters/__pycache__/google_adapter.cpython-313.pyc b/adapters/__pycache__/google_adapter.cpython-313.pyc
index e61c572219462ef389e5d207417d6cd0581aecac..221010d0d5c3b72691cd3c4249ec4296b12e2eb1 100644
GIT binary patch
delta 584
zcmdmP_1cE-GcPX}0}x0&Y{`t1*vPk?mr0Io^Kss7EF8fsd4fPJG<hSx*yLzVPDY){
z>-i-`T|mlFL7pH`jkvlSswjw;CltyR${s2hDikcN$C4);${i{^xmzfhF<|q3p&TYg
zmdQS%nliTpGxG{cOX5rNvs3fpON$fJQ*R07mzIFUCTEGNF{*5yD9X>McT1?CC_lFV
zrZ73bG_T~ASaM=sN@hx8Noq0F0I;~G$mB(06=sn@M-=G+iCa8Kb`;;@g&GgGviKHz
zZfZ$UW^!?{2vDhl!em2nFUIqeYs3SXoESHs6+gll^*~yt!{>&KY=`d+e(~%4YCv>D
zUj4ef*+qG?>++r#<vl0x+)&WCu3&yq!Th>{-$ez#3A`WJ81$`g$ZOqDHN2r_{E?GY
zoUg<E0}q3U%7X9<!a5gtbbfr%oV-C&6BxF)Bpn^R#Tkx?Ia+(EF`W=}5&`l~sd0IW
zu%0rI@aAVd&A|>7InB=s6ge%z3}TCO`S38GR#fw0XFkKu>ccZRT51|&#^g^@T{?b@
ztUinjf<G{T89R8Oj1%Qh#)Z_(3t9OWQwxCn$t$E2Il$^b96_1Y0$HG7QvwkJAfjmU
b85s%2>d8-K90lE&7=0K&F@UKeS)eQc9bv0o

delta 356
zcmaEDv)zjCGcPX}0}%MlZpr*9wvlf;FVhUx&BuASvG9d*g|Y_=>oMgChH{4rP8Jjn
zX5`(RDxAZ_$TazqsODrPF=<An&30n^j7*wBldHum#KVCG73l$qTkN^1B}JLZ#l=EE
z-egS)FOJDf3=9nnA9yA=@JLRcB4NO2wfUID5k~nBYz+F=H{`WGvNDVFb+~`vU=UGR
z5Pm^e=K_z;k1vvwpGavk%1q{yc69I(VK^q{Xzi)YbVATk5Xd{J%;hD>dQw-yi-+|T
zD?3o+6b~y<<dh&Yh%Lh9&B=U9PR*Nz`7{fwH|OMj>1m8flhtIpblezOeHa%6e_#MJ
zcJM$MC(5CW3#pkGvhpvc76AE^@5m%_fYpIGuCl8IQb68R0ucfrB4?7E1Y_xBNjXQs
RIz~nx#!n1jsz??n2>>N`US$9P

diff --git a/adapters/google_adapter.py b/adapters/google_adapter.py
index 7298066..d40d60f 100644
--- a/adapters/google_adapter.py
+++ b/adapters/google_adapter.py
@@ -131,9 +131,17 @@ class GoogleAdapter:
             else:
                 logger.warning("No images text generated from parts")
 
+            input_tokens = 0
+            output_tokens = 0
+            if response.usage_metadata:
+                input_tokens = response.usage_metadata.prompt_token_count
+                output_tokens = response.usage_metadata.candidates_token_count
+
             metrics = {
                 "api_execution_time_seconds": api_duration,
-                "token_usage": token_usage
+                "token_usage": token_usage,
+                "input_token_usage": input_tokens,
+                "output_token_usage": output_tokens
             }
             return generated_images, metrics
 
diff --git a/api/models/GenerationRequest.py b/api/models/GenerationRequest.py
index 06f8a96..02e6436 100644
--- a/api/models/GenerationRequest.py
+++ b/api/models/GenerationRequest.py
@@ -14,6 +14,7 @@ class GenerationRequest(BaseModel):
     quality: Quality = Quality.ONEK
     prompt: str
     telegram_id: Optional[int] = None
+    use_profile_image: bool = True
     assets_list: List[str]
 
 
@@ -32,6 +33,8 @@ class GenerationResponse(BaseModel):
     execution_time_seconds: Optional[float] = None
     api_execution_time_seconds: Optional[float] = None
     token_usage: Optional[int] = None
+    input_token_usage: Optional[int] = None
+    output_token_usage: Optional[int] = None
     progress: int = 0
     created_at: datetime = datetime.now(UTC)
     updated_at: datetime = datetime.now(UTC)
diff --git a/api/models/__pycache__/GenerationRequest.cpython-313.pyc b/api/models/__pycache__/GenerationRequest.cpython-313.pyc
index ff498b0c63c3f89ea26d6f101def9d416d5f9e61..dce78439a7eecf9a676f44ab034945bc55c4fdd3 100644
GIT binary patch
delta 841
zcmZvZ&ui0Q7{{9?X_Ee!P1CftUB~ELx7c=WR$-GO$Us4-$SNojD0Rz9XqVJCsq-Q|
zC=PEc@1=@|9o3t6{{^8?(2||>=4l6e6?~s}YIiV)&&&Hf@AG`WPo8)2uj<s#NF?YZ
z-}7%T%ys_bl+G|8aFcnJuZk$D%eqjN7D!r;=>Z;2p&EpHHH4IgUkl@~|Ka>SPwqHI
zt&Qv*vt`uG-Fn0DLUqS6T&LDD9rwLYi{m~2eo!T)9{0$$Tlkkh%Or7z-D2+JE_*#O
zlN)gAfji`KBUwGz>ZQJLX*7oi><se+AG6=rI26}KU7;@cYLvR*ka{X6(^w28bVcW@
z3iX}bT*gseO{Fd~M4NmlttUw~emYy8WHU6Y(oP99-QcvO7rS9t2CBQJZEYG(+qN9z
zhl(HY*O&x;!*5J`vT3zDZq2ouhE?l0Fn-MLxDdlfd^wOLx>^>8#B?_c79_xVKmZU0
zz!p#lPyoP~s0b(#c>GSwuDfUn0v7<YfMvi6;39w}@L0><LmXVofJ=ZJ08F6N9Pk_<
z{}-?n=Lmla^s);DSi;jow)bqv6-JB20hj2Dhsx{7kSpR1VI@H8^Z2ot!f%C}v(%wN
z8hDYGX*G@Q+R56f+8s@TnM-u$D<DUwnz)&mS%oa1h;NHCCn~E%<*rP}1BE{=1!%cO
zTmG+61zrN6&JhUqeWUQ1m$wiVtKfO5tI^S^Bz;c*%s(%lSlP9=J1yfXTF0NI#j^sF
a=}TiDImU5?DfLrhA34TLI1)^9Q1%<Od$V=`

delta 684
zcmZ8dOKTHR6rMZpNlnZoGifF*8cW4YA6jUXE<_Q;S06~TQU*$B#}bmxlsgeOf(T-9
zmvS~0`~eDX+_@3lh<`v-<{!9rx{)qC=f>9RU3}-<bI$#~@7!<hYp3utpU+8n_S0vb
z&+5BEz}PG3vKQ5$80Z={0;6S~!`QO8y(#-SsLA*3bfldmVLw^xb&~!Y$uGf6c?YJI
zaaMs_$~s$ux5|y8?GDy=y{~GGpFsj%fV$<$D-4#+ak#F2vWwga%7NChi9c{xXJA`%
zz0r#duXuPhaIsrD-kty3^+dNUyg9tKsrqhOx*0|xZ}&TKv>GPcag>BV9C)u?V=nyC
zmS9V7j8u@?ALFAK;Hg#9Es|=$N4?thDW#?P6u~q?sy*n%?LMEO!YsjQg7X9lLaIdZ
zBd*f*0>K<Xoq%+Pi%(ECc?b?B$p=_AcHzL7)5X_J($~PEdFQOKP{xro-|Ix(aAR%w
z(}Hfrw~l-_E)k(HTc_+lxhB;Jd^j*q4ON<`a<?L^lfoZL<A}0ZKdc_}y+AbrdY2do
zJJR<PC!a44XBOa#^~4v;2}v>jub0%kycKUe?1fkORd}3h%<F7y&&nhWS;=OzdtN4C
Y$fjgAF}R&c7_uB=#lgLO34_r24aX{rRsaA1

diff --git a/api/service/__pycache__/generation_service.cpython-313.pyc b/api/service/__pycache__/generation_service.cpython-313.pyc
index 07a5cf1b9a32086bd958443a9a1399463c952be9..cd01f6132ab4de4834efe789a9ae4f13c9645512 100644
GIT binary patch
delta 2154
zcmbtUZ){W76~D*#*(Qnqf&G%$aU8!mj_ufvNt}P+q`@KKUnm;y&12++WoQIQp@sxd
z0aZ;$ODU?>L653Rl#Nm*I;d1Y{xE6#FttKsup-Dp6mSq)yAP8l(J_idrG41WwQ-30
zbx-y=zw<li-h19T=e~cu3;#R`wr{Q0az@XwuihNnwd{)Rfl^^3^9}lcd&GuMm7L)h
z3{OO=@z)hjtXolu59>!cs4je`sm5HxSVmH-u*L1<Cbb%O1r+e$ctpmJ10L|=SLIHq
z!>ND_e)4(nUc_O_a#|9=bdwXC8vM{u2>)qA*c}pih$mxgha$2u&7z0!Y=cq}E;5X%
zO$gO8XKEak6<$v%G;8)TJy8nE7&8*K1gj|1)<SknZ7Ur4lb{zm3+UaT#p<1pFH?)`
z?}@GbKI?XrPRYP--}k*Wix+t7N?sq!nAOeLTkXMr1?`~Ye}gF)!Bk_w>eO6=OV_nz
z?S7_*@n)kBk2ls<>?p0&WW3VowpQvpON{ywy|T-zRcY1jh9YTWNlD*5ACMJA#&7DK
zm<Sn-?q$ZT15ZY(p0!|HlmHKgys#GE4=GS+7~Tl|Kz5XN(ECbAoxqluY!8%d%|kD}
z9DfxK8}#*fGUl|-H|ob5cc;jaf<xjxC5r38r{NIPU|qy*^kzWqqVBp(S}nolMd%~g
zU4(T6UnoL9!NUY)u;8(Xf~|2Uo@{pEi5fdTh{Qk<JMwy(g+E4RW1tXC6C^j-nHEf%
zOAYWd^1NCY%|Fl_G4(aVad@4i$Fhw|_hRKUT1@5}qG-yir5Am!^`j-6k=54=^CPQo
ziA-%;W7437mKn4#oz#F^qjH5uVl=0bB-fho=g}08KFcyz$VFU+Z=#j)TZ`CgX+u9r
zzTpFf=_^V{nAz4tjAhFgbf8~!pklrQ`dZu=lOedE2N1%Cy>$!9<inn33(9dP{1EkK
zufLvYOETp&C5Gp<`TtLF+Y^-f^Q$<FkdX0kbGf~5JL}<fY&$3m=f2F`N%c0XV)}#l
zb+c>1H5O}==52+7Ji=6>9X5%3X^cNBo6Ddz<C#PP61iK6O8~#Y$(FCZ`Nd<|y$AM>
z?BBC%G&^!&-_{+Y`6YV}>=zvVsP%K%67!E`i|&0rhpjO`IY-c*d<y+&Zrk9@FFvqu
zbY$<up6z41M@Pn9rVTLD_MYiY>c_-U{CgY1Tlh-*IXI3r9ajk7>^KV3sCAz8+TFAE
zDE;m_lvzi3$`Sspdp5FqDzf^HV-1?SK8HrU-qp3n#oD`nH+W_6_KM-@wk>b3yJKm&
z8;pITpVcqdP6vAhQGYkudMR}wb-Q!jbaYTK;k&WqrNIk>x4YI)$6kDU-HauY3obha
zlRFHL7~5%>Fgz8ipZ~k4h}>Zq;G_n)DwM{za0=1*UJ)V`kt!b-YMdVw>JVQkQOF@a
zBvFV^L>}b2Q1Y<!gt4+M(%)=Le!27b&RJXil&yZ&miumb1GqArxyRwLjd^u=TyKio
zzI)=!E$u!AldsU1`bb~da4&pO0U3|<g{99*##J|DYNe~5K1Rl$YML_vcC8FDe(73y
z+DFEpBSCUpw;P8{?79nvxO82X1a`L*f5QfPtLcV4-9g5eM$p?$Uz*|qCoA<X>1J8K
zfsCmN&?kGODN8y))?b&phB=#Fl&&Uq#}C8IbVunRJX0l+d!`nI^aC9}6DS{Id?ucb
zZBqEG8#Z}OvmUd+ij^F*&3(qrDSj^rn>$SR+ROr{jO55OKBTZb2TC+=at---e$vBf
z+k7nu!&kscqn_Y%i;TKUnzMrLSTtuVU?&LP<<g_(Ir@|mZUTX$lE4X(<XpQ*a<0on
z@;(Df{yr}s?&kNc!QoDRztt%CPNQJD#aTZy<dVGuhU9rMyPuHA^Owi66BmU^7mrQo
zLW~P>PKZ51C_;RMpQh7r3LmB;uq0R2_aVS(%&lAvr*i*U>4U|ufpHsLZhFEB;8CB1
tO#@!j8L{+yZv8+bIQ}xf*J*)n;gMA#cz_pIeP(}&L+@iI<irc)`yY2u3nu^o

delta 1914
zcmb_cZETZO6uzhTZM$||_tCyvyLD~fwr{`It{bJ@IyaycHiB+(?Mk-3iIIU*!k7#V
z!9|=X$`2EXo)`jXf|-fG6tGc$=pTXL2$*GIj(rdl{9`mRA`m}fqTbu>WtiwcoAli0
zJm;Q!&ppq*?+<Ulcdvu_ZbgNWu%BW1=$@MP^X6})8t%W${&(jp2=Nj(c?%zHmeA#S
zP4hg#4PhG|PlxbF&l6P0ir8wmP+hhb(=iL4@tVPgvmq;9^fo{}-ZEOS#crjLb*{$s
zcr08cjM2P+N9-a#;&X$ilyJlsvdg>>{>y6<u!77aXHsOm=#$F*D-0poAc9S#B-?~(
z*@~0lGN(FyOz$L0RGpaD=7l_Do7prGT1u6Y>^hYE7JN~5K%#`s$)<{i`x%Dp3VTyF
z;pehJ=OlT3Hphzh>K&{ueKyZ<J)&QNA^h0yt*g$}JRIHLJXYx@%36hwu?nq71C13s
z)unTmJm7e_wTd0dstl`ib~pA1#0D`}o6xLqW)c^ak^Awd@JJvEb$B+g+ELhxtw9I=
z5wJi8D}xd+4T8JGN;O02R8e-6H>hgVtr9s~9!A^nt6)g#RynnGEaFqS!T!;9HY<o5
zLl$_3pP2P=v;)UO0cvN**hH;m>=^Z|xk;v;sBK`Vy@VVLJ<Ao<#LF-dmS8)-1<<=H
zRf;yQ<hxnEQ(05IT&;uguF@JzCed9<a#CEc#@vKO4q~dQLH`+3gJcge9EAydvPp(R
zrHu}Av899RN!CVXQpI^OJHaml&NYc#4NqK?^QJU8A6wkR??eU3*#yEOa-8P;7>hjA
z82FD;gQbG_jXxc!iykT?J+$x7lI@p>(xGmtCp(`bWX$-0HXC?HOl#UG8T>0Eve&@W
z5&u;k+@E&+qn$I&LA)5XKp*yo9c~ZFRppKB_Gs4H^?&b|s1(_;*6#TZJR1}30%0Es
z7wHlDf+M7hy2uDf>bnWS%8Pj==6c`-^Srozd#O?L_<bw^{k-wo&}utH2xD(c44R6E
zTh0PZVRYTMn2djdhV?%|8h>BkhlRwO_)FqA9K~0YQ*aE!ZRZ)j+4c%Nj>GNmT1<^M
zOuku@@66B*IWsF~=1g1AmHGm_cq`QoHCW%_reB*p#ho3eAUv$OO~@|IKFzOu*g}6^
z85lXN*-F_UcGG+8BiTcXdpqM`?WM#%4tGq%p}-xFE^7Ymo~`&%*D;tp$Ua;zZ8j)D
zxKIuQRJdSDcQWTv84T!zONPyaIhT#i115609R>`-<<vTczl_L?pRUmk#>jLX3`T@$
zo4~On!T2lHP-xIysYxf9bF~Z#PTf^Ql;dP=K@zT+GC}6dtcQYsQkY4kz08}nTMH2~
z+nTnq;kp3|A>DOjlHoZbu<)D?xHSOVe9mMX4AHsTbYzE-&TC+YRyR-e92+ZGV7_j%
zcF;!`95Cq8Ei~#m_GwvQAw-9aWFZ2q(1n<FNKY5(BxTDk>VP}?5;pn^HoDlV9qJMm
zlQ5JjTWl-gPT+i(wP4jRRcjbtvH-_cfnysla>>byT#|I$hJY2n6fr6?T}sM|n=UtK
z8Nck-GG<xkd+pMY;-{OoXyQ}s-5q$3nI*b!&)9+Yx!cB#!p%50XSjKq8;P5DusM^4
z6SzMU0(0?H<}|=bOm6Lg6U7&{I-z<3w7cP4_#P>N+Y=N&=yvGd=BZP~nQlK+P2XSZ
bBuCd!yDb3A7~S?^<p_n1cL=u!Hn7;=fx)4o

diff --git a/api/service/generation_service.py b/api/service/generation_service.py
index be67fa9..705671c 100644
--- a/api/service/generation_service.py
+++ b/api/service/generation_service.py
@@ -154,7 +154,8 @@ class GenerationService:
             char_info = await self.dao.chars.get_character(generation.linked_character_id, with_image_data=True)
             if char_info is None:
                 raise Exception(f"Character ID {generation.linked_character_id} not found")
-            media_group_bytes.append(char_info.character_image_data)
+            if generation.use_profile_image:
+                media_group_bytes.append(char_info.character_image_data)
             generation_prompt = f"""You are creating image for {char_info.character_bio}"""
 
         reference_assets = await self.dao.assets.get_assets_by_ids(generation.assets_list)
@@ -182,9 +183,12 @@ class GenerationService:
                 gemini=self.gemini
             )
                 
+            # Update metrics from API (Common for both)
             # Update metrics from API (Common for both)
             generation.api_execution_time_seconds = metrics.get("api_execution_time_seconds")
             generation.token_usage = metrics.get("token_usage")
+            generation.input_token_usage = metrics.get("input_token_usage")
+            generation.output_token_usage = metrics.get("output_token_usage")
             
         except GoogleGenerationException as e:
             generation.status = GenerationStatus.FAILED
@@ -247,7 +251,7 @@ class GenerationService:
         end_time = datetime.now()
         generation.execution_time_seconds = (end_time - start_time).total_seconds()
         
-        logger.info(f"DEBUG: Saving generation {generation.id}. Metrics: api_exec={generation.api_execution_time_seconds}, tokens={generation.token_usage}, exec={generation.execution_time_seconds}")
+        logger.info(f"DEBUG: Saving generation {generation.id}. Metrics: api_exec={generation.api_execution_time_seconds}, tokens={generation.token_usage}, in_tokens={generation.input_token_usage}, out_tokens={generation.output_token_usage}, exec={generation.execution_time_seconds}")
         
         await self.dao.generations.update_generation(generation)
         logger.info(f"Generation {generation.id} completed successfully. {len(created_assets)} assets created. Total Time: {generation.execution_time_seconds:.2f}s")
diff --git a/models/Generation.py b/models/Generation.py
index 9bd313b..2cb018b 100644
--- a/models/Generation.py
+++ b/models/Generation.py
@@ -19,6 +19,7 @@ class Generation(BaseModel):
     failed_reason: Optional[str] = None
     linked_character_id: Optional[str] = None
     telegram_id: Optional[int] = None
+    use_profile_image: bool = True
     aspect_ratio: AspectRatios
     quality: Quality
     prompt: str
@@ -29,6 +30,8 @@ class Generation(BaseModel):
     execution_time_seconds: Optional[float] = None
     api_execution_time_seconds: Optional[float] = None
     token_usage: Optional[int] = None
+    input_token_usage: Optional[int] = None
+    output_token_usage: Optional[int] = None
     
     created_at: datetime = Field(default_factory=lambda: datetime.now(UTC))
     updated_at: datetime = Field(default_factory=lambda: datetime.now(UTC))
diff --git a/models/__pycache__/Generation.cpython-313.pyc b/models/__pycache__/Generation.cpython-313.pyc
index d31ee7023e7984ab0e9d2743c3a7f4dbb4cef236..cc4f421bfad6a77050503af18be7db9af131c96e 100644
GIT binary patch
delta 577
zcmYk1O-lkn9EY8CT~}+ZH8tNWsWh~AvXm$~7J-NDP&6O~DofVJxI!1{l9z%==2C~i
zTi5yk-TDB67uid6>(DK_H~(!3_wej*p83t&v;6Lrz9q?RV_);*N~JGSgG0CUhHFkL
zA{oNN*r~XZY!=Hbxy|HdX~0YZ-LXe6u@4Ils5li5XoVww?D%&>rW+;!>|qVrYK*c*
z&|2lST&T~DST4+*$Ljek7wL2UWa00~XrBx~5yuiXWi)J4f;mLuc#It$`uAbWkDESX
zZX6{O6>cU*KkWtIo*~z3rQ$i!%4)S#R8Nm8CA!MDqK2%Vo!9lEuAP+5iuD@vhF7cW
z;OHy=h$iWw<2tp%E<myXA0P&Z1J(dJhA9YnDhXkfpebSh0PbM$$C_3p128uTNCO1G
zEMN{W&tN#pRqaS83m_H&OMqp73$PBb12*WJFwJp@ehCG4JkjC(cY2pk(7f|B-b<!o
xe9^YI8eKl!OJzEI;9hKxJV;$WL!X=vXp?S<NwiH>F)RDgF1ic-*x+xz=@(UQc&PvY

delta 433
zcmcaFv_X*fGcPX}0}y<_)siW`k++tKF=BEzQ)_(;e++vthdNLuMgYj-1hd$HEG`f$
zn5jrGhBbzb6(r9d%ng<mg7bNxYJ`#0@PftoibP`gfI9f0Dnx-Q#Gs}KK*hv?Vv#UG
zL8zbvP>>I*PzWRzBdNp?EG)$kEMm!6Bo!kKG*lEWE(R5s0gH=+&5$UPO=r@SoGi!U
zKe>@*;$%zKNsNk<&#-oB7y*qdG6oSMAVL;I$bkq`5Mc%+lt2vg$#rZ}jB=A_vPHOX
zg9La$gcgWk0}(nPLKj5n0f}3zX*v0cB}Mum0Rs@h2_h^(1T%=Rp6tc0#&nZmauK`#
m<TLC!lfyV>GTKc3z@g6QI$46#Oi+l?hjD`BR|XIbb`$^z0ZdT<