mirror of
https://github.com/7836246/cursor2api.git
synced 2026-05-22 20:41:03 +08:00
c68aa3bfb6
- internal/tools/types.go: 定义工具相关类型 - internal/tools/executor.go: 工具执行器（bash/文件操作） - internal/tools/parser.go: 解析 AI 输出中的工具调用 - internal/handler/anthropic.go: 支持 tools 参数和 tool_use 响应 - internal/handler/tools.go: 工具执行和列表接口 - cmd/server/main.go: 注册工具路由 支持的工具： - bash: 执行命令 - read_file: 读取文件 - write_file: 写入文件 - list_dir: 列出目录 - edit: 查找替换编辑
69 lines
1.7 KiB
Go
69 lines
1.7 KiB
Go
// Cursor2API - 将 Cursor API 转换为 OpenAI/Anthropic 兼容格式
|
|
//
|
|
// 本项目通过浏览器自动化技术调用 Cursor 的 AI 接口，
|
|
// 并将其转换为标准的 OpenAI 和 Anthropic API 格式，
|
|
// 使得各种 AI 客户端可以直接使用 Cursor 的服务。
|
|
package main
|
|
|
|
import (
|
|
"log"
|
|
|
|
"cursor2api/internal/browser"
|
|
"cursor2api/internal/config"
|
|
"cursor2api/internal/handler"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func main() {
|
|
// 加载配置
|
|
cfg := config.Get()
|
|
|
|
// 初始化浏览器服务
|
|
log.Println("[启动] 正在初始化浏览器服务...")
|
|
browser.GetService()
|
|
|
|
// 创建 Gin 引擎
|
|
r := gin.Default()
|
|
|
|
// ==================== 路由配置 ====================
|
|
|
|
// OpenAI 兼容接口
|
|
r.GET("/v1/models", handler.ListModels)
|
|
r.POST("/v1/chat/completions", handler.ChatCompletions)
|
|
|
|
// Anthropic Messages API 兼容接口
|
|
r.POST("/v1/messages", handler.Messages)
|
|
r.POST("/messages", handler.Messages)
|
|
r.POST("/v1/messages/count_tokens", handler.CountTokens)
|
|
r.POST("/messages/count_tokens", handler.CountTokens)
|
|
|
|
// 工具相关接口
|
|
r.GET("/tools", handler.ListTools)
|
|
r.POST("/tools/execute", handler.ExecuteTool)
|
|
|
|
// 健康检查
|
|
r.GET("/health", func(c *gin.Context) {
|
|
c.JSON(200, gin.H{"status": "ok"})
|
|
})
|
|
|
|
// 浏览器状态
|
|
r.GET("/browser/status", func(c *gin.Context) {
|
|
svc := browser.GetService()
|
|
hasToken := svc.GetXIsHuman() != ""
|
|
c.JSON(200, gin.H{"hasToken": hasToken})
|
|
})
|
|
|
|
// 静态文件
|
|
r.Static("/static", "./static")
|
|
r.GET("/", func(c *gin.Context) {
|
|
c.File("./static/index.html")
|
|
})
|
|
|
|
// 启动服务
|
|
log.Printf("[启动] 服务运行在端口 %s", cfg.Port)
|
|
if err := r.Run(":" + cfg.Port); err != nil {
|
|
log.Fatalf("[错误] 启动失败: %v", err)
|
|
}
|
|
}
|