Python Tkinter 集成 DocsGPT 开发 AI 代码助手

GPT API 客户端 (gpt_client.py)

"""GPT API 客户端模块
处理与 GPT API 的通信
"""
import requests
import json
from typing import Optional, Dict, Any
import threading
from database import db

class GPTClient:
    def __init__(self, api_key: str = None, api_url: str = None):
        self.api_key = api_key or db.get_setting('api_key') or ""
        self.api_url = api_url or db.get_setting('api_url') or "https://api.openai.com/v1/chat/completions"
        self.session = requests.Session()
        self.session.headers.update({
            "Authorization": f"Bearer {self.api_key}",
            "Content-Type": "application/json"
        })

    def set_api_key(self, api_key: str):
        """设置 API 密钥"""
        self.api_key = api_key
        self.session.headers.update({"Authorization": f"Bearer {api_key}"})
        db.update_setting('api_key', api_key)

    def set_api_url(self, api_url: str):
        """设置 API URL"""
        self.api_url = api_url
        db.update_setting('api_url', api_url)

    def ask_question(self, question: str, code_context: str = "", model: str = "gpt-3.5-turbo", 
                     max_tokens: int = 1000, temperature: float = 0.7, callback=None) -> Optional[str]:
        """
        向 GPT 提问
        Args:
            question: 问题文本
            code_context: 相关代码上下文
            model: 使用的模型
            max_tokens: 最大 token 数
            temperature: 温度参数
            callback: 回调函数，用于异步更新 UI
        Returns:
            GPT 的回答
        """
        if not self.api_key:
            return "错误：请先设置 API 密钥"

        # 构建消息
        messages = []
        if code_context:
            messages.append({
                "role": "system",
                "content": f"以下是相关代码上下文：\n```python\n{code_context}\n```\n请基于此代码上下文回答问题。"
            })
        messages.append({
            "role": "user",
            "content": question
        })

        # 构建请求数据
        data = {
            "model": model,
            "messages": messages,
            "max_tokens": max_tokens,
            "temperature": temperature,
            "stream": False
        }

        try:
            if callback:
                # 异步调用
                thread = threading.Thread(target=self._async_ask, args=(data, callback))
                thread.daemon = True
                thread.start()
                return "正在处理..."
            else:
                # 同步调用
                return self._sync_ask(data)
        except Exception as e:
            return f"请求错误：{str(e)}"

    def _async_ask(self, data: Dict, callback):
        """异步调用 GPT"""
        try:
            response = self.session.post(self.api_url, json=data, timeout=30)
            if response.status_code == 200:
                result = response.json()
                answer = result['choices'][0]['message']['content']
                callback(answer)
            else:
                error_msg = f"API 错误：{response.status_code}\n{response.text}"
                callback(error_msg)
        except requests.exceptions.Timeout:
            callback("错误：请求超时，请检查网络连接")
        except requests.exceptions.ConnectionError:
            callback("错误：无法连接到 API 服务器")
        except Exception as e:
            callback(f"未知错误：{str(e)}")

    def _sync_ask(self, data: Dict) -> str:
        """同步调用 GPT"""
        try:
            response = self.session.post(self.api_url, json=data, timeout=30)
            if response.status_code == 200:
                result = response.json()
                return result['choices'][0]['message']['content']
            else:
                return f"API 错误：{response.status_code}\n{response.text}"
        except requests.exceptions.Timeout:
            return "错误：请求超时，请检查网络连接"
        except requests.exceptions.ConnectionError:
            return "错误：无法连接到 API 服务器"
        except Exception as e:
            return f"未知错误：{str(e)}"

    def get_models(self) -> list:
        """获取可用的模型列表（需要相应权限）"""
        if not self.api_key:
            return []
        try:
            response = self.session.get("https://api.openai.com/v1/models", timeout=10)
            if response.status_code == 200:
                models = response.json()['data']
                return [model['id'] for model in models if 'gpt' in model['id'].lower()]
            return []
        except:
            return ["gpt-3.5-turbo", "gpt-4", "gpt-4-turbo"]

# 全局 GPT 客户端实例
gpt_client = GPTClient()

使用说明

功能特点

• 代码编辑器：支持语法高亮、文件操作（新建、打开、保存）、查找替换、撤销/重做。
• GPT 助手：集成 OpenAI GPT API，支持代码上下文、历史记录管理、搜索功能。
• 界面优化：左右分栏设计、海蓝色主题按钮、深浅色主题切换、响应式布局。
• 数据库：SQLite 存储历史记录、持久化设置、快速搜索查询。

该程序提供了一个完整的 Python 开发环境，集成了 GPT 智能助手，非常适合学习和开发使用。