122 lines
11 KiB
Plaintext
122 lines
11 KiB
Plaintext
|
Ë
|
|||
|
|
ôŸØi*" ã óð — d Z ddlZddlZddlZddlmZmZmZmZm Z ddl
|
|||
|
|
m Z m Z ddl
|
|||
|
|
mZmZ ddlmZmZ ej& e« Ze G d„ d« « Z G d „ d
|
|||
|
|
« Z G d „ d « Z dd
|
|||
|
|
ee defd„Zy)uL
|
|||
|
|
DeepSeek API 客户端�装
|
|||
|
|
支æŒ�å�Œæ¥/异æ¥è°ƒç”¨ï¼Œä¸ŽCAMEL AI兼容
|
|||
|
|
é N)ÚDictÚListÚOptionalÚAnyÚUnion)Ú dataclassÚfield)ÚOpenAIÚ AsyncOpenAI)ÚChatCompletionÚChatCompletionMessagec óª — e Zd ZU dZeed< dZeed< dZeed< dZe ed< d Z
|
|||
|
|
e ed
|
|||
|
|
< d Z e ed < d
|
|||
|
|
Z
|
|||
|
|
e ed< d
|
|||
|
|
Ze ed< dZe ed< dZeeeef ed< y)ÚDeepSeekConfigu DeepSeek APIé…�ç½®Úapi_keyzhttps://api.deepseek.comÚbase_urlz
|
|||
|
|
deepseek-chatÚ
|
|||
|
|
model_nameg333333Ó?Ú temperaturei Ú
|
|||
|
|
max_tokensgÍÌÌÌÌÌì?Útop_pg Úfrequency_penaltyÚpresence_penaltyg N@ÚtimeoutNÚ
|
|||
|
|
extra_headers)Ú__name__Ú
|
|||
|
|
__module__Ú __qualname__Ú__doc__ÚstrÚ__annotations__r r r Úfloatr Úintr r r r r r r © ó úPC:\Users\24019\Desktop\backman-camel\text2sql_agent_camel\llm\deepseek_client.pyr r ss … á Ø
|
|||
|
|
ƒLØ.€HˆcÓ.Ø%€J�Ó%Ø €K�Ó Ø€J�ÓØ€Eˆ5ÓØ "Ð�uÓ"Ø !Ð�eÓ!Ø€GˆUÓØ.2€M�8˜D c ™NÑ+Ô2r# r c ó¾ — e Zd ZdZdefd„Zdeeeef de fd„Z
|
|||
|
|
deeeef deee f fd„Z dded ed
|
|||
|
|
edefd „Z
|
|||
|
|
d ed edeee f fd
|
|||
|
|
„Zdededeee f fd„Zy)ÚDeepSeekClientub
|
|||
|
|
DeepSeek API 客户端（å�Œæ¥ï¼‰
|
|||
|
|
|
|||
|
|
使用 OpenAI 兼容接�调用 DeepSeek 模型
|
|||
|
|
Úconfigc óÒ — || _ t |j |j |j ¬« | _ t j d|j › d|j › �« y )N©r r r u$ [OK] DeepSeekClientåˆ�始化: model=z , base_url=) r' r
|
|||
|
|
r r r ÚclientÚloggerÚinfor ©Úselfr' s r$ Ú__init__zDeepSeekClient.__init__' s] € Ø ˆŒ ô Ø—N‘NØ—_‘_Ø—N‘Nô
|
|||
|
|
ˆŒ ô � ‰ Ø2°6×3DÑ3DÐ2Eð FØ Ÿ™Ð(ð
|
|||
|
|
*õ
|
|||
|
|
r# ÚmessagesÚreturnc óÊ — | j j ||j d| j j « |j d| j j « |j d| j j
|
|||
|
|
« |j d| j j « |j d| j j « |j d| j j « dœ}| j j r| j j |d< | j j j j di |¤Ž}|j d j }|j }t" j% d
|
|||
|
|
|j& › d |j( › d |j* › �« |S # t, $ r }t" j/ d
|
|||
|
|
|› �« ‚ d}~ww xY w)u
|
|||
|
|
���天请求
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
messages: 消æ�¯åˆ—è¡¨ï¼Œæ ¼å¼�为 [{"role": "user", "content": "..."}, ...]
|
|||
|
|
**kwargs: 覆盖默认�数的�外�数
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
ChatCompletionMessage对象（包�.role和.content属性）
|
|||
|
|
r r r r r r )Úmodelr0 r r r r r r r r u$ DeepSeek调用完æˆ�: prompt_tokens=z, completion_tokens=z, total_tokens=u DeepSeek API调用失败: Nr" )r' r Úgetr r r r r r r r* ÚchatÚ completionsÚcreateÚchoicesÚmessageÚusager+ ÚdebugÚ
|
|||
|
|
prompt_tokensÚcompletion_tokensÚ total_tokensÚ ExceptionÚerror)r. r0 ÚkwargsÚparamsÚresponser9 r: Úes r$ r5 zDeepSeekClient.chat6 s• € ð" —[‘[×+Ñ+Ø Ø!Ÿ:™: m°T·[±[×5LÑ5LÓMØ Ÿ*™* \°4·;±;×3IÑ3IÓJØ—Z‘Z ¨¯©×):Ñ):Ó;Ø!'§¡Ø# T§[¡[×%BÑ%Bó"ð !'§
|
|||
|
|
¡
|
|||
|
|
Ø" D§K¡K×$@Ñ$@ó!ð —z‘z )¨T¯[©[×-@Ñ-@ÓAñ
|
|||
|
|
|
|||
|
|
ˆð �;‰;× $Ò $Ø&*§k¡k×&?Ñ&?ˆF�?Ñ #ð Ø'J t§{¡{×'7Ñ'7×'CÑ'C×'JÑ'JÑ'TÈVÑ'TˆHØ ×&Ñ& qÑ)×1Ñ1ˆGð —N‘NˆEÜ �L‰Lð!Ø!&×!4Ñ!4Ð 5ð 6%Ø%*×%<Ñ%<Ð$=ð > Ø %× 2Ñ 2Ð 3ð5ô
|
|||
|
|
ð ˆNøäò Ü �L‰LÐ5°a°SÐ9Ô :Ø ûð ús Ä(BF; Æ; G"ÇG Ç G"c óö — | j |fi |¤Ž}|j j « } 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|icY d}~S d}~ww xY w)uÇ
|
|||
|
|
å�‘é€�è�Šå¤©è¯·æ±‚并期望JSONæ ¼å¼�返回
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
messages: 消�列表
|
|||
|
|
**kwargs: �外�数
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
解æž�å�Žçš„JSONå—å…¸
|
|||
|
|
z```jsoné ú```é u' JSON解æž�失败，返回原始内容: Ú raw_contentN) r5 ÚcontentÚstripÚfindÚjsonÚloadsÚJSONDecodeErrorr+ Úwarning)r. r0 rA r9 rJ ÚstartÚendrD s r$ Úchat_with_jsonz DeepSeekClient.chat_with_jsonk sô € ð �$—)‘)˜HÑ/¨Ñ/ˆØ—/‘/×'Ñ'Ó)ˆð ,à˜GÑ#à Ÿ ™ YÓ/°!Ñ3�Ø —l‘l 5¨%Ó0�Ø! %¨Ð,×2Ñ2Ó4‘ؘ'Ñ!à Ÿ ™ UÓ+¨aÑ/�Ø —l‘l 5¨%Ó0�Ø! %¨Ð,×2Ñ2Ó4�ä—:‘:˜gÓ&Ð &øÜ×#Ñ#ò ,Ü �N‰NÐ DÀQÀCÐHÔ IØ! 7Ð+Õ +ûð ,ús ¯BB? Â?C8ÃC3Ã-C8Ã3C8ÚpromptÚschemaÚdialectc ó — ddl m}m} d|dœd|j |||¬« dœg} | j |fi |¤Ž}|j
|
|||
|
|
j
|
|||
|
|
« } d| v r;| j d« dz }
|
|||
|
|
| j d |
|
|||
|
|
« } | |
|
|||
|
|
| j
|
|||
|
|
« } | S d | v r9| j d « d
|
|||
|
|
z }
|
|||
|
|
| j d |
|
|||
|
|
« } | |
|
|||
|
|
| j
|
|||
|
|
« } | S ) u
|
|||
|
|
便�方法：生�SQL
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
prompt: 用户问题
|
|||
|
|
schema: Schema�述
|
|||
|
|
dialect: SQL方言（默认 T-SQL / SQL Server）
|
|||
|
|
**kwargs: �外�数
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
SQLå—符串
|
|||
|
|
r )ÚSQL_GENERATOR_SYSTEMÚSQL_GENERATOR_USERÚsystem©ÚrolerJ Úuser)rU ÚquestionrV z```sqlé rG rH )Úconfig.promptsrX rY Úformatr5 rJ rK rL ) r. rT rU rV rA rX rY r0 rC rJ rQ rR s r$ Ú generate_sqlzDeepSeekClient.generate_sql� sù € ÷& Lð Ð*>Ñ ?à Ø-×4Ñ4Ø!Ø #Ø #ð 5ó ñ
|
|||
|
|
ð
|
|||
|
|
|
|||
|
|
ˆð �4—9‘9˜XÑ0¨Ñ0ˆØ×"Ñ"×(Ñ(Ó*ˆð �wÑ Ø—L‘L Ó*¨QÑ.ˆEØ—,‘,˜u eÓ,ˆCØ ˜e CÐ(×.Ñ.Ó0ˆGð ˆð
|
|||
|
|
�gÑ
|
|||
|
|
à—L‘L Ó'¨!Ñ+ˆEØ—,‘,˜u eÓ,ˆCØ ˜e CÐ(×.Ñ.Ó0ˆGàˆr# Úsqlc ón — ddl m}m} d|dœd|j ||¬« dœg} | j |fi |¤ŽS )uÉ
|
|||
|
|
便�方法：验�SQL
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
sql: SQLè¯å�¥
|
|||
|
|
schema: Schema�述
|
|||
|
|
**kwargs: �外�数
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
验è¯�结果å—å…¸
|
|||
|
|
r )ÚVALIDATOR_SYSTEMÚVALIDATOR_USERrZ r[ r] )rc rU )r` re rf ra rS )r. rc rU rA re rf r0 s r$ Ú validate_sqlzDeepSeekClient.validate_sqlÀ sO € ÷" Dð Ð*:Ñ ;à Ø)×0Ñ0°SÀÐ0ÓHñ
|
|||
|
|
ð
|
|||
|
|
ˆð #ˆt×"Ñ" 8Ñ6¨vÑ6Ð6r# r^ Ú
|
|||
|
|
table_listc ón — ddl m}m} d|dœd|j ||¬« dœg} | j |fi |¤ŽS )uÿ
|
|||
|
|
便�方法：选择相关表
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
question: 用户问题
|
|||
|
|
table_list: å�¯ç”¨è¡¨åˆ—表å—符串
|
|||
|
|
**kwargs: �外�数
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
包å�«relevant_tableså’Œreasoningçš„å—å…¸
|
|||
|
|
r )ÚSCHEMA_LINKER_SYSTEMÚSCHEMA_LINKER_USERrZ r[ r] )r^ rh )r` rj rk ra rS )r. r^ rh rA rj rk r0 s r$ Ú
|
|||
|
|
select_tablesz DeepSeekClient.select_tablesÝ sU € ÷" Lð Ð*>Ñ ?à Ø-×4Ñ4Ø %Ø )ð 5ó ñ
|
|||
|
|
ð
|
|||
|
|
ˆð #ˆt×"Ñ" 8Ñ6¨vÑ6Ð6r# N)Útsql)r r r r r r/ r r r r
|
|||
|
|
r5 r rS rb rg rl r" r# r$ r& r& sè „ ñð
|
|||
|
|
|
|||
|
|
˜~ó
|
|||
|
|
|
|||
|
|
ð 3à�t˜C ˜H‘~Ñ&ð3ð
|
|||
|
|
ó 3ðj",à�t˜C ˜H‘~Ñ&ð",ð
|
|||
|
|
ˆc�3ˆh‰ ó ",ðP ñ /àð/ð ð/ð ð /ð
|
|||
|
|
|
|||
|
|
ó
|