// Engine 核心接口 - 保持不变
public interface AIServiceProvider {
    String generateContent(AIChatRequest request, String apiKey, AIResponseListener listener);
    ValidationResult validateConfiguration(String apiKey);
    List<String> getAvailableModels(String apiKey);
}

// 标准 Provider 实现 - 保持不变
public abstract class AICompatibleProvider implements AIServiceProvider {
    // 统一的 HTTP 请求处理
    // 统一的错误处理
    // 统一的重试机制
}

public class CodefreeProvider extends AICompatibleProvider {
    
    public CodefreeProvider(Project project, AIProviderConfig config, 
                           AIModelParameters modelParameters, 
                           AIRuntimeSettings runtimeSettings) {
        super(project, config, modelParameters, runtimeSettings);
        // 关键：将 baseUrl 指向本地 Agent
        this.config.baseUrl = "http://127.0.0.1:10011/v1";
    }
    
    // 无需重写任何方法！
    // AICompatibleProvider 已经处理了所有 HTTP 请求
}

// Agent 内部实现（独立项目）
public class OpenAiApiServer {
    private HttpServer server;
    
    public void start() {
        server = HttpServer.create(new InetSocketAddress(10011), 0);
        
        // 提供 OpenAI 兼容接口
        server.createContext("/v1/chat/completions", this::handleChatCompletions);
        server.createContext("/v1/models", this::handleModels);
        
        server.start();
    }
    
    private void handleChatCompletions(HttpExchange exchange) {
        // 1. 解析 OpenAI 格式请求
        OpenAIRequest request = parseRequest(exchange);
        
        // 2. 转换为服务商协议（WebSocket）
        WebSocketMessage wsMessage = convertToWebSocket(request);
        
        // 3. 调用服务商 API
        String response = callProviderViaWebSocket(wsMessage);
        
        // 4. 转换为 OpenAI 格式响应
        OpenAIResponse openAIResponse = convertToOpenAI(response);
        
        // 5. 返回标准响应
        sendResponse(exchange, openAIResponse);
    }
}

// AIProviderType.java - 添加一行
CODEFREE(
    "codefree",
    "Codefree",
    "http://127.0.0.1:10011/v1",  // 指向本地 Agent
    "codefree-chat",
    false,  // Agent 内部处理认证
    false,  // URL 不可编辑（固定为本地）
    List.of("codefree-chat")
)

// CodefreeProvider.java - 约 20 行代码
public class CodefreeProvider extends AICompatibleProvider {
    public CodefreeProvider(Project project, AIProviderConfig config,
                           AIModelParameters modelParameters,
                           AIRuntimeSettings runtimeSettings) {
        super(project, config, modelParameters, runtimeSettings);
    }
    // 无需重写任何方法，直接使用父类的 HTTP 处理逻辑
}

// AIServiceFactory.java - 添加一个 case
return switch (providerType) {
    // ... 其他 case
    case CODEFREE -> new CodefreeProvider(project, config, modelParameters, runtimeSettings);
};

// 使用 JDK HttpServer（零依赖）
HttpServer server = HttpServer.create(
    new InetSocketAddress("127.0.0.1", 10011), 0
);

server.createContext("/v1/chat/completions", exchange -> {
    // 1. 解析请求
    String body = new String(exchange.getRequestBody().readAllBytes(), UTF_8);
    JSONObject request = JSON.parseObject(body);
    
    // 2. 协议转换
    WebSocketMessage wsMsg = convertToWebSocket(request);
    
    // 3. 调用 WebSocket 服务
    String response = webSocketClient.send(wsMsg);
    
    // 4. 格式转换
    JSONObject openAIResponse = convertToOpenAIFormat(response);
    
    // 5. 返回响应
    byte[] respBytes = openAIResponse.toJSONString().getBytes(UTF_8);
    exchange.getResponseHeaders().add("Content-Type", "application/json");
    exchange.sendResponseHeaders(200, respBytes.length);
    exchange.getResponseBody().write(respBytes);
    exchange.close();
});

server.start();

// CodefreeAgentManager.java
public class CodefreeAgentManager {
    
    // 1. 下载 Agent JAR
    public Path downloadJar(String version, ProgressIndicator indicator);
    
    // 2. 启动 Agent 进程
    public long startAgent(CodefreeAgentSettings settings);
    
    // 3. 停止 Agent 进程
    public void stopAgent();
    
    // 4. 检查运行状态
    public boolean isRunning();
}

// CodefreeAgentLifecycleListener.java
public class CodefreeAgentLifecycleListener 
    implements ApplicationInitializedListener, Disposable {
    
    @Override
    public void componentsInitialized() {
        // IDE 启动时，如果配置了自动启动，则启动 Agent
        if (settings.autoStart) {
            manager.startAgent(settings);
        }
    }
    
    @Override
    public void dispose() {
        // IDE 关闭时，自动停止 Agent
        if (manager.isRunning()) {
            manager.stopAgent();
        }
    }
}

// Agent 侧：优雅降级
try {
    return callProviderViaWebSocket(request);
} catch (WebSocketException e) {
    // 返回标准错误格式
    return createOpenAIErrorResponse(e);
}

// Engine 侧：统一处理
// AICompatibleProvider 已处理所有 HTTP 错误
// 无需 Provider 实现类额外处理

// 未来可支持：根据请求类型自动选择协议
if (request.requiresStreaming()) {
    return useWebSocketProvider();
} else {
    return useHttpProvider();
}

// 未来可支持：多个 Agent 实例负载均衡
List<AgentInstance> agents = discoverAgents();
AgentInstance selected = loadBalancer.select(agents);