Files
ai-chat-system/services/__pycache__/ai_service.cpython-310.pyc
T

42 lines
3.2 KiB
Plaintext
Raw Normal View History

o
êÈÙiSã@sRdZddlZddlmZmZmZddlZddlZe e ¡Z
GdddƒZ e ƒZ dS)u
AIæœåŠ¡ - 调用大模型API
éN)ÚListÚDictÚAsyncGeneratorc@s^eZdZddZdededefddZdeed efd
d Zdeed e ed ffd
dZ
d S)Ú AIServicecCsd|_d|_d|_d|_dS)T)Úapi_baseÚapi_keyÚmodelÚuse_mock)Úself©r úI/home/xian/.openclaw/workspace-coder/works/ai-chat/services/ai_service.pyÚ__init__
s
zAIService.__init__rrr cCs@||_||_||_|o | |_t d|d|d|j¡dS)u æ›´æ–°é…ç½®uAIé…置已更新: api_base=ú, model=z , use_mock=N)rrr r
ÚloggerÚinfo)r rrr r r r
Ú
update_configs
 "zAIService.update_configÚmessagesÚreturnc
ÃsX|jrt d¡|r|ddnd}d|dS|jd}d|jd d
œ}|j|d d d
œ}t d|d|j¡zAtjdd4IdHš)}|j|||dIdH}|  ¡| 
¡}|ddddWdƒIdHWS1IdHsvwYWdSt } z"t  d| ¡|r•|ddnd}dt
| ƒd|WYd} ~ Sd} ~ ww)u,
调用AI模型进行对è¯
u使用Mock模å¼å›žå¤éÿÿÿÿÚcontentõ你好õ*这是一个测试回å¤ã€‚您说的是:uN
请é…置有效的AIæœåŠ¡åœ°å€å’Œæ¨¡åž‹ï¼Œæ‰èƒ½èŽ·å¾—çœŸæ­£çš„AI回å¤ã€‚ú/chat/completionsúBearer úapplication/json©Ú
Authorizationz Content-Typeçffffffæ?éÐ)r rÚ temperatureÚ
max_tokensu调用AI API: rçN@©ÚtimeoutN©ÚheadersÚjsonÚchoicesrÚmessageuAI API调用失败: u#AIæœåŠ¡æš‚æ—¶ä¸å¯ç”¨ï¼ˆé”™è¯¯ï¼šu)。您说的是:)r
rrrrr ÚhttpxÚ AsyncClientÚpostÚraise_for_statusr'Ú ExceptionÚerrorÚstr)
r rÚlast_msgÚurlr&ÚpayloadÚclientÚresponseÚdataÚer r r
Úchats8
  
þü üzAIService.chatNc Cs|jr|r |ddnd}d|}|D]}|VqdS|jd}d|jdd œ}|j|d
d d d
œ}tjdd4IdHš}}|jd|||d4IdHšY} |  ¡2zH3dHW}
|
 d¡rš|
dd} | dkrin3z&t  
| ¡} d| vrŽt | dƒdkrŽ| dd  di¡}
d|
vrŽ|
dVWqRt j
y™YqRwqR6WdƒIdHn 1IdHs¬wYWdƒIdHdS1IdHsÂwYdS)u&
æµå¼è°ƒç”¨AI模型