You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
90 lines
2.4 KiB
90 lines
2.4 KiB
package migu
|
|
|
|
import (
|
|
"yunyan/comm"
|
|
"yunyan/modules"
|
|
"fmt"
|
|
"net/http"
|
|
"time"
|
|
|
|
"yunyan/lego/base"
|
|
"yunyan/lego/core"
|
|
"yunyan/lego/sys/log"
|
|
|
|
httpSwagger "github.com/swaggo/http-swagger"
|
|
|
|
_ "yunyan/swagger/migu" // migu 专属 swagger 文档（instanceName=migu）
|
|
)
|
|
|
|
/*
|
|
模块名：移动灵犀聊天服务（migu）
|
|
说明：独立部署，对外暴露普通聊天接口 POST /chat（支持流式 SSE 与非流式 JSON），
|
|
内部按咪咕「智能体能力统一对外输出API」格式调用咪咕灵犀（sys/migu），
|
|
并把咪咕响应（文本/思考/音乐等结构化卡片/追问词）整理后返回。不使用 gateway 的路由转发机制。
|
|
*/
|
|
|
|
func NewModule() core.IModule {
|
|
return new(Migu)
|
|
}
|
|
|
|
type Migu struct {
|
|
modules.ModuleBase
|
|
service base.ISingleService
|
|
options *Options
|
|
}
|
|
|
|
func (this *Migu) GetType() core.M_Modules {
|
|
return comm.ModuleMigu
|
|
}
|
|
|
|
func (this *Migu) NewOptions() (options core.IModuleOptions) {
|
|
return new(Options)
|
|
}
|
|
|
|
func (this *Migu) Init(service core.IService, module core.IModule, options core.IModuleOptions) (err error) {
|
|
this.service = service.(base.ISingleService)
|
|
this.options = options.(*Options)
|
|
if err = this.ModuleBase.Init(service, module, options); err != nil {
|
|
return
|
|
}
|
|
return
|
|
}
|
|
|
|
func (this *Migu) Start() (err error) {
|
|
if err = this.ModuleBase.Start(); err != nil {
|
|
return
|
|
}
|
|
go this.run()
|
|
return
|
|
}
|
|
|
|
func (this *Migu) OnInstallComp() {
|
|
this.ModuleBase.OnInstallComp()
|
|
}
|
|
|
|
// run 启动独立的 HTTP 服务入口；启动失败直接 panic 退出进程
|
|
func (this *Migu) run() {
|
|
mux := http.NewServeMux()
|
|
mux.HandleFunc("/chat", this.handleChat)
|
|
// migu 专属 swagger UI：访问 /swagger/index.html（instanceName=migu）
|
|
mux.HandleFunc("/swagger/", httpSwagger.Handler(httpSwagger.InstanceName("migu")))
|
|
mux.HandleFunc("/health", func(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusOK)
|
|
_, _ = w.Write([]byte(`{"status":"ok"}`))
|
|
})
|
|
|
|
handler := withCORS(withRecover(mux))
|
|
httpServer := &http.Server{
|
|
Addr: fmt.Sprintf(":%d", this.options.Port),
|
|
Handler: handler,
|
|
ReadTimeout: 5 * time.Minute,
|
|
WriteTimeout: 30 * time.Minute, // 流式回复可能较长
|
|
IdleTimeout: 5 * time.Minute,
|
|
}
|
|
|
|
// TLS 由外层 nginx 处理，本服务只监听明文 HTTP
|
|
log.Infof("migu openai-compatible server listening on :%d", this.options.Port)
|
|
if err := httpServer.ListenAndServe(); err != nil {
|
|
log.Panicf("migu server error: %v", err)
|
|
}
|
|
}
|
|
|