AskaEth
673ff752c5
- 新增 backend/pkg/plugins/ 共享模块：SDK 接口、PluginManager、ToolRegistry（含环形缓冲区调用日志） - 13 个通用插件从 plugin-manager 迁移至共享模块（import 路径统一） - ai-core 切换至共享 ToolRegistry，进程内执行（零网络开销），包装 6 个专属工具 - plugin-manager 迁移至共享模块，保留管理 REST API - 新增 DevTools 插件管理面板（侧边栏 → 🔌 插件管理） - 移除 tool-engine 服务（从 go.work、DevTools 配置、编译系统） - 工具调用记录 API 从 Tool-Engine 迁至 AI-Core（/api/v1/tools/calls） - ai-core ContextStore 启动时从 PostgreSQL 恢复会话历史 - 清理所有过时引用和备份文件 Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
101 lines
2.9 KiB
Go
101 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/yourname/cyrene-ai/pkg/plugins/calculator"
|
|
"github.com/yourname/cyrene-ai/pkg/plugins/crypto"
|
|
"github.com/yourname/cyrene-ai/pkg/plugins/datetime"
|
|
fileplugin "github.com/yourname/cyrene-ai/pkg/plugins/file"
|
|
httpplugin "github.com/yourname/cyrene-ai/pkg/plugins/http"
|
|
iotcontrol "github.com/yourname/cyrene-ai/pkg/plugins/iot_control"
|
|
iotquery "github.com/yourname/cyrene-ai/pkg/plugins/iot_query"
|
|
jsonplugin "github.com/yourname/cyrene-ai/pkg/plugins/json"
|
|
"github.com/yourname/cyrene-ai/pkg/plugins/manager"
|
|
"github.com/yourname/cyrene-ai/pkg/plugins/markdown"
|
|
"github.com/yourname/cyrene-ai/pkg/plugins/random"
|
|
"github.com/yourname/cyrene-ai/pkg/plugins/sdk"
|
|
"github.com/yourname/cyrene-ai/pkg/plugins/text"
|
|
webfetch "github.com/yourname/cyrene-ai/pkg/plugins/web_fetch"
|
|
websearch "github.com/yourname/cyrene-ai/pkg/plugins/web_search"
|
|
|
|
"github.com/yourname/cyrene-ai/plugin-manager/internal/config"
|
|
"github.com/yourname/cyrene-ai/plugin-manager/internal/handler"
|
|
)
|
|
|
|
func main() {
|
|
cfg := config.Load()
|
|
|
|
var iotAPI iotquery.IoTClient
|
|
if cfg.IoTSvcURL != "" {
|
|
iotAPI = newIoTClient(cfg.IoTSvcURL)
|
|
}
|
|
|
|
registry := manager.NewToolRegistry()
|
|
host := newHostAPI(registry)
|
|
mgr := manager.NewPluginManager(registry, host)
|
|
|
|
builtins := []sdk.Plugin{
|
|
&calculator.CalculatorPlugin{},
|
|
&datetime.DatetimePlugin{},
|
|
&text.TextPlugin{},
|
|
&crypto.CryptoPlugin{},
|
|
&random.RandomPlugin{},
|
|
&markdown.MarkdownPlugin{},
|
|
&jsonplugin.JSONPlugin{},
|
|
fileplugin.NewFilePlugin(cfg.DataDir),
|
|
httpplugin.NewHTTPPlugin(),
|
|
websearch.NewWebSearchPlugin(),
|
|
webfetch.NewWebFetchPlugin(),
|
|
iotquery.NewIoTQueryPlugin(iotAPI),
|
|
}
|
|
for _, p := range builtins {
|
|
if err := mgr.Install(p); err != nil {
|
|
println("WARN: install plugin failed:", err.Error())
|
|
}
|
|
}
|
|
if iotAPI != nil {
|
|
ctrlPlugin := iotcontrol.NewIoTControlPlugin(newIoTControllerAdapter(iotAPI, cfg.IoTSvcURL))
|
|
if err := mgr.Install(ctrlPlugin); err != nil {
|
|
println("WARN: install plugin failed:", err.Error())
|
|
}
|
|
}
|
|
|
|
ctx := context.Background()
|
|
errs := mgr.EnableAll(ctx)
|
|
for _, e := range errs {
|
|
println("WARN: enable plugin failed:", e.Error())
|
|
}
|
|
println("Plugin Manager: all built-in plugins enabled")
|
|
|
|
mux := http.NewServeMux()
|
|
ph := handler.NewPluginHandler(mgr)
|
|
ph.RegisterRoutes(mux)
|
|
|
|
println("Plugin Manager listening on port", cfg.Port)
|
|
srv := &http.Server{Addr: ":" + cfg.Port, Handler: mux}
|
|
|
|
go func() {
|
|
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
println("FATAL:", err.Error())
|
|
os.Exit(1)
|
|
}
|
|
}()
|
|
|
|
quit := make(chan os.Signal, 1)
|
|
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
|
|
<-quit
|
|
println("Shutting down Plugin Manager...")
|
|
|
|
shutdownCtx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
mgr.Shutdown(shutdownCtx)
|
|
srv.Shutdown(shutdownCtx)
|
|
println("Plugin Manager stopped")
|
|
}
|