From dee344dbea593b567aeb654c2ee9e0e453e0b589 Mon Sep 17 00:00:00 2001
From: blade <8019068@qq.com>
Date: Wed, 22 Oct 2025 18:05:38 +0800
Subject: [PATCH] make code stronger

---
 .../truth_social_retriever.cpython-312.pyc    | Bin 14759 -> 15205 bytes
 core/media/truth_social_retriever.py          |  19 +++++++++++-------
 2 files changed, 12 insertions(+), 7 deletions(-)

diff --git a/core/media/__pycache__/truth_social_retriever.cpython-312.pyc b/core/media/__pycache__/truth_social_retriever.cpython-312.pyc
index d25757a8c304a80a9526e0a32c3ced693724c4c7..a48cd429a9c3341f9249e2665eb17ed7ab9a5cee 100644
GIT binary patch
delta 1450
zcmYk5Yiv|S6vt=o?%ntMdAAR`?{`_Yz;-K?rb=O<wDJ<9DWR2H9tF0zTPz{>R!oc-
zQEYP(EFnUT5{lK(W+8zPl@F3M6w_TwY_on42~hCU*75=Upw3+a+4=CF-#K$~=FFLy
zi|I%2X(rWb1tNnTeZKFH__$`jntDjX=k789bp!6mOshW=;UVbMH`zK^8)sylD}`tW
zA(ucL>UP1P-3UMGb=VD4`kmMV)dnl}LX)9d>l?*Fc7!k*pg!FV9~jnQKU5jZ@oAVd
z)Z&1^#+$5&H}#wPa34A<+JX9TAed<}j*5V>I^em*u1M`BDN-zvOr5ol!gd(9`EeWM
zZ7cA`jHT=Wc0ZFKrhKRl`bY3xYhbpvZb+VWF3mB^W@EJi*5sI_Ag<WRJY*J~SDsO3
zy<PX1y?@UiqS~-L`utfEk`-?|;_J24ZEa<9guWftG$VQvSqM*BG|fSJGDtT^GIJH%
zDLkHOs4|Q3jm%rFIL0YB<z6S|aJLk0E-Tx(8V%r`IfU9Ngm{V<$q^@FDH|!C7ZVZ{
zi6$g2l7y3_LcMabTpXO_Wy@Ilz-n^pN@(<3;jqUkJ+&9{Qcenbof%6Y=dr_-$AzUu
z`mRzT#)y@H!<(QuIehQ6VL4pzIvk2URPoA`oct{MlH!$|l2vgEfsd(4O-hLRtZ>k;
zm9gsLhCh)F!{GCUs$baemK3LaRh;ISa6A=BhB#Fj>b41ISMh32$*EZM(n0YWk}XKF
zTDa<4fpswNtH*kX_%|u<i4ljlAQ$Q_gOmPhY=G<ja^t@&;<f$SlBZyVIe(R|oir!a
zC`gAO6nFxWl4k*11J|(zG{KZKPp5lR`wzqM-~-7NiA}G9eIYxX3tbbfYQfMS_Evf8
zuk~c>TeI#r2U=!Lj;yc#S~6STlJ&IaOsxYg@O5~HeJ0eF3$;y$+J`pGhT_?}&Dpvw
z+0fQkZctQTDrZRk8*3E_1vR0NCt~!3kx4}9i71_j!Bq91ltz@fvG@drMukzv8V>kL
z<@=yvi52cdKEX!#B~sO8Vs)dWlfvL-$x2Srt$AUU>RH5^$%?Mj_kzhaig+oj<7J$b
zlkK634K=)sbEA}#m-m`L@2|ub_+d#1TjA-F2a<~<;^_q3i871w*nY%B%ts%8^!VJs
z<5MG#PoH@7&G?fm7hbamRPbLkY|bmwy`8;B(z|x|uw8kvps7pXuZiKyu&d^{e29GV
z<ETJ6;YrP}@9UTutt+Q>4X5vFgZE7pUu+oMFwBh9Oq&+}Ze|9z4B3WXpEmnO@Vp3{
zLvtmr(&d62>74}?vN_JH&ZvgBkGwO@L}!@gIcE7-@@D5WlNhA$`yv<lulV7fu}xz~
zXKL5xYS(6CszlbkE~|6QGPVI4@=N2ECQFkI-LlEoh())YOjCpCRs+38obfL^Nnuas
zN^Bdp_mBpBh!&_aI=m1!(Wd$EeoRM)-&jDxT-XM|cvUb=+eS8DZokyNfCv`aF{T4T
z2$>2J7tM?r7Z8~W>oKI_-^P9>IrEakok_Mcok|iTbyX0RAa@066(mmNAKA)7J?xQ#
WU8_>R!RER(+72<c{DlN92K^5Y1DJ{c

delta 1210
zcmYjQZA@EL7(VB=rytyYK<_R6fTrKQEv1EyuL6UGz&0EVk!=hjKSsP88!BqqM3>ub
z8WU%nhUJZ6#;Dtj$|Q5sCI+I(G&6C57%#F+q9Gb0G0`7_><4P1o>Sby`SG0Ry!V{<
z<9Tvdo_>GSGG{gyAiAD^YvZV??jy@FvrZ)lO(3vV6a$!d;NIMV`4tU*6$XoTl3oQ3
z2jP5J@CZ|wy+bG~D8dku1DCOD^Mr;&aH&E7&en{B&|-7o5cJy;*6_U-7>#r1`r$R(
z12_Ud+Dh;__}f;8CB}tMWKC8)C63`S^o*t-jp0}{Hz-VLAmQ+V$>Gl(bc}Jh0KRcW
zaUX2B8gW;y+P#cTKRetP>fzKR{?-wLhSF~LT^(|H-{xQE)yU!7&Y?foj&Xh1nA*5I
z43|s${K-9pn<bu(FqsWl_SBNu+G5J*OqLEGne&m3aIVC2ki*w<qh7lf&*WYTrm=3n
z_P#btp`;*l3YIn7>D;oGGC~;?ghL(E2n}#l8ZN6bB1I$X^3rH2)8`qX%#dd~MVE;m
zH!>mFQgru3lC_VvA(Zh@9~>|q4(Lw|p{!ofLs6j}8(}T%gWtnJDg<Lcc?z@`$vm_~
z+5@L~*`yfXKyPv<&9nzz;bpT@u;pxos}UF0z`GGYPmXY`eQK+n*$V$e3iXzJ<#+Ox
zlTiV7NQt`5*nusb^7lKno@BF{XdR@O+$d{Sc*P_m$}IoqC<a5c0T)4Qv<chbx#&au
zr&^@oeJF^IJ7FoBzyk2G5@AI~S?ek57V|}@k9oaT+G6Gr2DB2&$P5M;v@kdXM`IOm
zDK>-6@NiiMZkH|VzNQ~Wcf)6KpIwTbHLAAYydkt0kr)lk8-nmdd?soSs8aLvX|?%5
zHQYTfc8~8}6#c5yG(Dm=?N!5F^I{hbHFd&7c?6cq`z#ka7vpKQzDKR!r^ffgfy9+k
z=HeG*S88`7_f?~@T_U%%&h`Mg6(H>rxRSqeWn7L_yojMXCDPe%RC*;rE}UX>(3LeP
zdc|;r%PV2pj3E?4nLySEDZdws@LMX5?cmt9jNgNQw>iOG<);4ERF%WCRaN*yxLS3_
z7^DB<8MMkdp|<)?sogWtdA{?aV=}NHCcYC>^J40H@^k-!*mia&Ow^={TSKi9x+xjk
zY?_;u&Nj2=OEYP+<yLB5=5R^wsdO*)jnH0RK<k{Fl&_~nBCeGWVUoJox{lbs+5^AT
zdt*uBn(VnYaAjZ}QM!5vV-jOm=(ehJ5$APb9no#I7AhM)sJ%f`ey@9Sa6}#)&9MBL
rj~TG3Gt90`7lSl~zYN<Se?mU-*x}~P9Vl$vW$o4C;y)0>{H6Z@CHp<G

diff --git a/core/media/truth_social_retriever.py b/core/media/truth_social_retriever.py
index cb8affa..f4e7b67 100644
--- a/core/media/truth_social_retriever.py
+++ b/core/media/truth_social_retriever.py
@@ -267,14 +267,19 @@ class TruthSocialRetriever:
                 search_options={"forced_search": True},  # 强制联网搜索
                 result_format="message",
             )
-            response_contents = (
-                response.get("output", {})
-                .get("choices", [])[0]
-                .get("message", {})
-                .get("content", "")
-            )
+
             # 获取response的token
-            token = response.get("usage", {}).get("total_tokens", 0)
+            if response.status_code == 200:
+                response_contents = (
+                    response.get("output", {})
+                    .get("choices", [])[0]
+                    .get("message", {})
+                    .get("content", "")
+                )
+                token = response.get("usage", {}).get("total_tokens", 0)
+            else:
+                response_contents = f"{response.code} {response.message}"
+                token = 0
             return response_contents, token
         except Exception as e:
             logger.error(f"分析推文失败: {e}")