hong-red/sql-optimization-agent
0
1o
2 d��i � 3 @ s� d dl Z d dlZd dlZdZdZdZddd�Zd Zd4d � Ze d krXe5d
� zddd�gZ ee �Z e6de � d�� e� W dS e
yW Z ze7de� �� W Y dZ[dS dZ[ww dS )� Nz3sk-OysQAuYdnQbrL1ObbEZ234lFVeqFDalGcFz11SdyFnhcp36dz+https://api.moonshot.cn/v1/chat/completionsz kimi-k2.5� �� c C s~ ddt � �d�}t| ||d�}ddd�}tjt||d|d�}|jd kr3|�� }|d8 d d d
�� S td|j� d|j � ���)u% 通过 requests 直接调用 Kimi APIzapplication/jsonzBearer )z Content-TypeZ
Authorization)Zmodel�messages� temperature�9max_tokensN)ZhttpZhttpsF)�headers�jsonZverify�proxies�� Zchoicesr �message�contentu API 请求失败 (状态码 z): )10� KIMI_API_KEY�MODEL�requestsZpost�BASE_URLZ status_coder �strip� Exception�text)r r r r Zpayloadr Zresponse�result� r �f:\myjob\sql_agent.py�ask_kimi s ��1112r u 13你是一个资深的数据库架构师 + SQL 优化专家，精通 MySQL/PostgreSQL。14 15你的任务是：161. 根据用户描述的业务需求，设计合理的数据库表结构（表名、字段类型、主键、外键、常见索引建议）。172. 输出格式必须严格用 Markdown，包括：18 - ## 表结构设计19 - 表名: xxx20 - id: BIGINT PRIMARY KEY AUTO_INCREMENT21 - ...22 - 索引建议: INDEX idx_user_id (user_id)23 - ## 建表 SQL（MySQL 方言）24 253. 当用户问 SQL 问题时：26 - 先理解需求27 - 写出原始 SQL28 - 分析潜在性能问题（全表扫描、JOIN顺序、索引缺失等）29 - 给出优化后的 SQL + 解释为什么更快30 31保持专业、严谨、一步一步思考。32当前数据库类型：MySQL 8.033c 34 C s� dt d�g} td� ztd�}W n35 ty Y d S w |�� �� dv r&d S | �d|d�� td� zt| �}td |� td36� | �d |d�� W n tyc } ztd |� �� W Y d }~d S d }~ww q )
N�system��roler u/ 智能体已启动！输入 '退出' 结束。37Tu 你: )u 退出�exit�q�useru
思考中...38u39 智能体:z<------------------------------------------------------------� assistantu 发生错误: ) �
SYSTEM_PROMPT�print�input�EOFErrorr �lower�appendr r )r Z40user_input�reply�er r r � run_agent; s. �41���r' �__main__u 正在测试 API 连接...r u$ 你好，请用一句话介绍自己r u API 测试成功！Kimi 回复: �42u API 测试失败: )r r )r r �osr
r r r r r' �__name__r Ztest_msgZ43test_replyr r& r r r r �<module> s* 44 ���