110 lines
15 KiB
Plaintext
110 lines
15 KiB
Plaintext
|
Ë
|
|||
|
|
(YßiU/ ã óð — d Z ddlmZ ddlZddlZddlZddlmZ ddlm Z m
|
|||
|
|
Z
|
|||
|
|
m Z m Z ddl
|
|||
|
|
mZmZ ddlmZmZ ej& e« Ze G d„ d « « Z G d
|
|||
|
|
„ d « Z G d „ d
|
|||
|
|
« Z d dd„Zy)u‘
|
|||
|
|
OpenAI API 客户端å°�装（å�Œæ¥/异æ¥ï¼‰
|
|||
|
|
|
|||
|
|
�供与 :mod:`llm.deepseek_client` �形�的接�，便于在需�时切� LLM 路由。
|
|||
|
|
é )Ú annotationsN)Ú dataclass)ÚAnyÚDictÚListÚOptional)Ú AsyncOpenAIÚOpenAI)ÚChatCompletionÚChatCompletionMessagec óš — e Zd ZU dZded< dZded< dZded< dZd ed
|
|||
|
|
< d Zd ed
|
|||
|
|
< dZ d ed< dZ
|
|||
|
|
d ed< dZ d ed< dZ d ed< dZ
|
|||
|
|
ded< y)Ú OpenAIConfigu4 OpenAI API é…�置（支æŒ� OpenAI 兼容网关）。ÚstrÚapi_keyzhttps://api.openai.com/v1Úbase_urlz gpt-4o-miniÚ
|
|||
|
|
model_nameg333333Ó?ÚfloatÚ temperaturei ÚintÚ
|
|||
|
|
max_tokensgÍÌÌÌÌÌì?Útop_pç Úfrequency_penaltyÚpresence_penaltyg N@ÚtimeoutNzOptional[Dict[str, str]]Ú
|
|||
|
|
extra_headers)Ú__name__Ú
|
|||
|
|
__module__Ú __qualname__Ú__doc__Ú__annotations__r r r r r r r r r © ó ú backend/llm/openai_client.pyr r sg … á>à
|
|||
|
|
ƒLØ/€HˆcÓ/Ø#€J�Ó#Ø €K�Ó Ø€J�ÓØ€Eˆ5ÓØ "Ð�uÓ"Ø !Ð�eÓ!Ø€GˆUÓØ.2€MÐ+Ô2r# r c óº — e Zd ZdZdd„Zdd„Z dd„Z d dd„Zdd„Z dd„Z dd„Z
|
|||
|
|
dd „Z d
|
|||
|
|
d œ dd „Z d
|
|||
|
|
dœ dd„Z
|
|||
|
|
y)Ú OpenAIClientuï
|
|||
|
|
OpenAI API 客户端（å�Œæ¥ï¼‰ã€‚
|
|||
|
|
|
|||
|
|
说明：本项目其他模�仅�求具备 ``chat`` / ``chat_with_json`` 方法；
|
|||
|
|
这里�外�供若干便�方法，与 DeepSeekClient 对�，便于�用�一套 prompt。
|
|||
|
|
c óÊ — || _ t |j |j |j ¬« | _ t j d|j |j « y )N©r r r u1 [OK] OpenAIClientåˆ�始化: model=%s, base_url=%s) Úconfigr
|
|||
|
|
r r r ÚclientÚloggerÚinfor ©Úselfr) s r$ Ú__init__zOpenAIClient.__init__0 sK € Ø ˆŒ Ü Ø—N‘NØ—_‘_Ø—N‘Nô
|
|||
|
|
ˆŒ ô
|
|||
|
|
� ‰ Ø ?Ø × Ñ Ø �O‰Oõ
|
|||
|
|
r# c
|
|||
|
|
ó — |j d| j j « }|j dd « }| j j ||j d| j j « |j d| j j
|
|||
|
|
« |j d| j j « |j d| j j « |j d| j j « dœ}|�||d< n||d< | j j r| j j |d < | j j j j di |¤Ž}|j d
|
|||
|
|
j }|j }|�6t" j% d |j& |j( |j* « |S # t, $ r¬} t/ | « }
|
|||
|
|
d |
|
|||
|
|
v r�d|
|
|||
|
|
v r}d|
|
|||
|
|
v ryd|vru|j1 dd « ||d< | j j j j di |¤Ž}|j d
|
|||
|
|
j cY d } ~ S # t, $ r Y nw xY wt" j3 d
|
|||
|
|
| « ‚ d } ~ ww xY w)Nr Úmax_completion_tokensr r r r r )ÚmodelÚmessagesr r r r r r r uK OpenAI调用完æˆ�: prompt_tokens=%s, completion_tokens=%s, total_tokens=%szUnsupported parameteru OpenAI API调用失败: %sr" )Úgetr) r r r r r r r r r* ÚchatÚ completionsÚcreateÚchoicesÚmessageÚusager+ ÚdebugÚ
|
|||
|
|
prompt_tokensÚcompletion_tokensÚ total_tokensÚ Exceptionr ÚpopÚerror) r. r3 Úkwargsr r1 ÚparamsÚresponser9 r: ÚeÚmsgs r$ r5 zOpenAIClient.chat= s@ € Ø—Z‘Z ¨d¯k©k×.DÑ.DÓEˆ
|
|||
|
|
Ø &§
|
|||
|
|
¡
|
|||
|
|
Ð+BÀDÓ IÐ ð —[‘[×+Ñ+Ø Ø!Ÿ:™: m°T·[±[×5LÑ5LÓMØ—Z‘Z ¨¯©×):Ñ):Ó;Ø!'§¡Ø# T§[¡[×%BÑ%Bó"ð !'§
|
|||
|
|
¡
|
|||
|
|
Ø" D§K¡K×$@Ñ$@ó!ð —z‘z )¨T¯[©[×-@Ñ-@ÓAñ "
|
|||
|
|
ˆð" !Ð ,Ø.CˆFÐ*Ò +à#-ˆF�<Ñ à �;‰;× $Ò $Ø&*§k¡k×&?Ñ&?ˆF�?Ñ #ð Ø'J t§{¡{×'7Ñ'7×'CÑ'C×'JÑ'JÑ'TÈVÑ'TˆHØ ×&Ñ& qÑ)×1Ñ1ˆGØ —N‘NˆEØÐ Ü— ‘ ØaØ×'Ñ'Ø×+Ñ+Ø×&Ñ&ô ð ˆNøÜò ä�a“&ˆCà'¨3Ñ.Ø CÑ'Ø+¨sÑ2Ø+°6Ñ9à—
|
|||
|
|
‘
|
|||
|
|
˜<¨Ô.Ø2<�Ð.Ñ/ðØ B˜tŸ{™{× /Ñ /× ;Ñ ;× BÑ BÑ LÀVÑ L�HØ#×+Ñ+¨AÑ.×6Ñ6Õ6øÜ ò Ùðúô
|
|||
|
|
�L‰LÐ5°qÔ 9Ø ûð% ús> ÅBG Ç J Ç 2JÈAI!ÉJ É! I-É*JÉ,I-É-JÊJ c óü — | j |fi |¤Ž}|j xs dj « } d|v r:|j d« dz }|j d|« }||| j « }n=d|v r9|j d« dz }|j d|« }||| j « }t j
|
|||
|
|
|« S # t j $ r%}t j d|« d|dœcY d }~S d }~ww xY w) NÚ z```jsoné ú```é u) JSON解æž�失败，返回原始内容: %sT)Ú_json_decode_failedÚ raw_content) r5 ÚcontentÚstripÚfindÚjsonÚloadsÚJSONDecodeErrorr+ Úwarning)r. r3 rB r9 rN ÚstartÚendrE s r$ Úchat_with_jsonzOpenAIClient.chat_with_jsonz s÷ € ð �$—)‘)˜HÑ/¨Ñ/ˆØ—?‘?Ò( b×/Ñ/Ó1ˆð IؘGÑ#Ø Ÿ ™ YÓ/°!Ñ3�Ø —l‘l 5¨%Ó0�Ø! %¨Ð,×2Ñ2Ó4‘ؘ'Ñ!Ø Ÿ ™ UÓ+¨aÑ/�Ø —l‘l 5¨%Ó0�Ø! %¨Ð,×2Ñ2Ó4�Ü—:‘:˜gÓ&Ð &øÜ×#Ñ#ò IÜ �N‰NÐFÈÔ JØ+/ÀÑHÕ Hûð Iús ³BC ÃC;ÃC6Ã0C;Ã6C;c óð — ddl m}m} d|dœd|j |||¬« dœg}|j dd« |j d d
|
|||
|
|
« | j
|
|||
|
|
|fi |¤Ž}|j xs d j « } d | v r;| j d « d
|
|||
|
|
z }
|
|||
|
|
| j d|
|
|||
|
|
« } | |
|
|||
|
|
| j « } | S d| v r9| j d« dz }
|
|||
|
|
| j d|
|
|||
|
|
« } | |
|
|||
|
|
| j « } | S )Nr )ÚSQL_GENERATOR_SYSTEMÚSQL_GENERATOR_USERÚsystem©ÚrolerN Úuser)ÚschemaÚquestionÚdialectr r r ç ð?rH z```sqlé rJ rK ) Úconfig.promptsrY rZ ÚformatÚ
|
|||
|
|
setdefaultr5 rN rO rP ) r. Úpromptr_ ra rB rY rZ r3 rD rN rU rV s r$ Ú generate_sqlzOpenAIClient.generate_sql� s € ÷ Lð Ð*>Ñ ?à Ø-×4Ñ4Ø!¨F¸Gð 5ó ñ
|
|||
|
|
ð
|
|||
|
|
ˆð ×ј-¨Ô-Ø×ј' 3Ô'Ø �4—9‘9˜XÑ0¨Ñ0ˆØ×#Ñ#Ò) r×0Ñ0Ó2ˆà �wÑ Ø—L‘L Ó*¨QÑ.ˆEØ—,‘,˜u eÓ,ˆCØ ˜e CÐ(×.Ñ.Ó0ˆGð ˆð �gÑ
|
|||
|
|
Ø—L‘L Ó'¨!Ñ+ˆEØ—,‘,˜u eÓ,ˆCØ ˜e CÐ(×.Ñ.Ó0ˆGàˆr# c ó¶ — ddl m}m} d|dœd|j ||¬« dœg}|j dd« |j d d
|
|||
|
|
« | j
|
|||
|
|
|fi |¤ŽS ) Nr )ÚVALIDATOR_SYSTEMÚVALIDATOR_USERr[ r\ r^ )Úsqlr_ r r r rb )rd rj rk re rf rW )r. rl r_ rB rj rk r3 s r$ Ú validate_sqlzOpenAIClient.validate_sql® si € ßCð Ð*:Ñ ;ب×(=Ñ(=À#ÈfÐ(=Ó(UÑ Vð
|
|||
|
|
ˆð
|
|||
|
|
×ј-¨Ô-Ø×ј' 3Ô'Ø"ˆt×"Ñ" 8Ñ6¨vÑ6Ð6r# c ó¶ — ddl m}m} d|dœd|j ||¬« dœg}|j dd« |j d d
|
|||
|
|
« | j
|
|||
|
|
|fi |¤ŽS ) Nr )ÚSCHEMA_LINKER_SYSTEMÚSCHEMA_LINKER_USERr[ r\ r^ )r` Ú
|
|||
|
|
table_listr r r rb )rd ro rp re rf rW )r. r` rq rB ro rp r3 s r$ Ú
|
|||
|
|
select_tableszOpenAIClient.select_tablesº st € ÷ Lð Ð*>Ñ ?à Ø-×4Ñ4Ø %°*ð 5ó ñ
|
|||
|
|
ð
|
|||
|
|
ˆð ×ј-¨Ô-Ø×ј' 3Ô'Ø"ˆt×"Ñ" 8Ñ6¨vÑ6Ð6r# c óF — ddl m}m} |xs dj « }|syd|dœd|j |¬« dœg}| j |dd d
|
|||
|
|
¬ « }|j xs dj « }|r!|j « d j « nd}|j d « S )
|
|||
|
|
Nr )Ú#CANONICALIZE_NL_FOR_TEXT2SQL_SYSTEMÚ!CANONICALIZE_NL_FOR_TEXT2SQL_USERrH r[ r\ r^ )r` r rb é )r r r1 u 「ã€�"'“â€�)rd rt ru rO re r5 rN Ú
|
|||
|
|
splitlines) r. r` rt ru Úqr3 rF ÚtextÚlines r$ Ú"normalize_nl_question_for_text2sqlz/OpenAIClient.normalize_nl_question_for_text2sqlÍ s§ € ÷
|
|||
|
|
ð
|
|||
|
|
Š^˜× "Ñ "Ó $ˆÙØà Ð*MÑ NØÐ(I×(PÑ(PÐZ[Ð(PÓ(\Ñ ]ð
|
|||
|
|
ˆð �i‰i˜¨c¸ÐTWˆiÓXˆØ— ‘ Ò!˜r×(Ñ(Ó*ˆÙ/3ˆt�‰Ó Ñ#×)Ñ)Ô+¸ˆØ�z‰zÐ+Ó,Ð,r# c ó$ — | j |« S ©N)r{ )r. r` s r$ Útranslate_nl_question_to_zhz(OpenAIClient.translate_nl_question_to_zhß s € Ø×6Ñ6°xÓ@Ð@r# i@ )Úmax_schema_charsc óÌ — ddl m}m} |xs dd | }d|dœd|j |xs d||¬« dœg} | j | fd d
|
|||
|
|
d œ|¤Ž}
|
|||
|
|
|
|
|||
|
|
j
|
|||
|
|
xs dj
|
|||
|
|
« S ) Nr )Ú EMPTY_RESULT_FEEDBACK_SYSTEMÚEMPTY_RESULT_FEEDBACK_USERrH r[ r\ r^ õ (æ— ))r` rl r_ gš™™™™™Ù?rv )r r1 )rd r� r‚ re r5 rN rO ) r. r` rl r_ r rB r� r‚ Ú schema_snipr3 rF s r$ Úempty_result_user_feedbackz'OpenAIClient.empty_result_user_feedbackâ sŠ € ÷ \à ’| Ð%6Ð&6Ð7ˆ à Ð*FÑ Gà Ø5×<Ñ<Ø %Ò 0¨ØØ&ð =ó ñ
|
|||
|
|
ð
|
|||
|
|
|
|||
|
|
ˆð ˆd�i‰i˜ÐW¨cÈÑWÐPVÑWˆØ—‘Ò" ×)Ñ)Ó+Ð,r# i )Ú
|
|||
|
|
max_sql_charsc ó¼ — ddl m}m} |xs dj « xs d}|xs dj « }t |« |kD r|d |dz
|
|||
|
|
j « dz }d|dœd |j
|
|||
|
|
||¬
|
|||
|
|
« dœg} |j d d « |j d
|
|||
|
|
d« |j dd« | j | fi |¤Ž}
|
|||
|
|
|
|
|||
|
|
j xs dj « S )Nr )Ú!SQL_PROBE_SUCCESS_DELIVERY_SYSTEMÚ SQL_PROBE_SUCCESS_DELIVERY_USERrH rƒ é u
|
|||
|
|
-- …(已截æ–)r[ r\ r^ )r` rl r gš™™™™™É?r rb r1 i@ )
|
|||
|
|
rd rˆ r‰ rO ÚlenÚrstripre rf r5 rN ) r. r` rl r† rB rˆ r‰ rx Úsr3 rF s r$ Ú"sql_probe_success_delivery_messagez/OpenAIClient.sql_probe_success_delivery_messageü sí € ÷
|
|||
|
|
ð
|
|||
|
|
Š^˜× "Ñ "Ó $Ò /¨ˆØ
|
|||
|
|
ŠY�B× Ñ Ó ˆÜ ˆq‹6�MÒ !ØÐ&�M BÑ&Ð'×.Ñ.Ó0Ð3HÑHˆAð Ð*KÑ Là Ø:×AÑAÈ1ÐRSÐAÓTñ
|
|||
|
|
ð
|
|||
|
|
ˆð ×ј-¨Ô-Ø×ј' 3Ô'Ø×ÑÐ1°3Ô7؈d�i‰i˜Ñ+ FÑ+ˆØ— ‘ Ò!˜r×(Ñ(Ó*Ð*r# N©r) r ©r3 úList[Dict[str, str]]Úreturnr )r3 r‘ r’ úDict[str, Any])Útsql)rg r r_ r ra r r’ r )rl r r_ r r’ r“ )r` r rq r r’ r“ )r` r r’ r ) r` r rl r r_ r r r rB r r’ r )
|
|||
|
|
r` r rl r r† r rB r r’ r )r r r r r/ r5 rW rh rm rr r{ r~ r… rŽ r" r# r$ r& r& ( sÿ „ ñó
|
|||
|
|
ó;ðzIØ,ðIà óIð, 8>ð Øð Ø#&ð Ø14ð à ó ó>
|
|||
|
|
7ð7Øð7Ø),ð7à ó7ó&-ó$Að !%ñ
|
|||
|
|
-àð-ð ð-ð ð -ð ð
|
|||
|
|
-ð ð-ð
|
|||
|
|
|