2a96f4c6f9
后端： 1.阶段 6 agent / memory 服务化收口 - 新增 cmd/agent 独立进程入口，承载 agent zrpc server、agent outbox relay / consumer 和运行时依赖初始化 - 补齐 services/agent/rpc 的 Chat stream 与 conversation meta/list/timeline、schedule-preview、context-stats、schedule-state unary RPC - 新增 gateway/client/agent 与 shared/contracts/agent，将 /api/v1/agent chat 和非 chat 门面切到 agent zrpc - 收缩 gateway 本地 AgentService 装配，双 RPC 开关开启时不再初始化本地 agent 编排、LLM、RAG 和 memory reader fallback - 将 backend/memory 物理迁入 services/memory，私有实现收入 internal，保留 module/model/observe 作为 memory 服务门面 - 调整 memory outbox、memory reader 和 agent 记忆渲染链路的 import 与服务边界，cmd/memory 独占 memory worker / consumer - 关闭 gateway 侧 agent outbox worker 所有权，agent relay / consumer 由 cmd/agent 独占，gateway 仅保留 HTTP/SSE 门面与迁移期开关回退 - 更新阶段 6 文档，记录 agent / memory 当前切流点、smoke 结果，以及 backend/client 与 gateway/shared 的目录收口口径
61 lines
1.5 KiB
Go
61 lines
1.5 KiB
Go
package rpc
|
||
|
||
import (
|
||
"errors"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/LoveLosita/smartflow/backend/services/agent/rpc/pb"
|
||
agentsv "github.com/LoveLosita/smartflow/backend/services/agent/sv"
|
||
"github.com/zeromicro/go-zero/core/service"
|
||
"github.com/zeromicro/go-zero/zrpc"
|
||
"google.golang.org/grpc"
|
||
)
|
||
|
||
const (
|
||
defaultListenOn = "0.0.0.0:9089"
|
||
defaultTimeout = 0
|
||
)
|
||
|
||
type ServerOptions struct {
|
||
ListenOn string
|
||
Timeout time.Duration
|
||
Service *agentsv.AgentService
|
||
}
|
||
|
||
// NewServer 创建 agent zrpc 服务端。
|
||
//
|
||
// 职责边界：
|
||
// 1. 只负责 zrpc server 配置与 gRPC handler 注册；
|
||
// 2. 不创建数据库、Redis、LLM 或业务服务，它们由 cmd/agent 管理；
|
||
// 3. Chat 是长连接 server-stream，默认不设置 RPC timeout，避免截断 SSE 转发。
|
||
func NewServer(opts ServerOptions) (*zrpc.RpcServer, string, error) {
|
||
if opts.Service == nil {
|
||
return nil, "", errors.New("agent service dependency not initialized")
|
||
}
|
||
|
||
listenOn := strings.TrimSpace(opts.ListenOn)
|
||
if listenOn == "" {
|
||
listenOn = defaultListenOn
|
||
}
|
||
timeout := opts.Timeout
|
||
if timeout < 0 {
|
||
timeout = defaultTimeout
|
||
}
|
||
|
||
server, err := zrpc.NewServer(zrpc.RpcServerConf{
|
||
ServiceConf: service.ServiceConf{
|
||
Name: "agent.rpc",
|
||
Mode: service.DevMode,
|
||
},
|
||
ListenOn: listenOn,
|
||
Timeout: int64(timeout / time.Millisecond),
|
||
}, func(grpcServer *grpc.Server) {
|
||
pb.RegisterAgentServer(grpcServer, NewHandler(opts.Service))
|
||
})
|
||
if err != nil {
|
||
return nil, "", err
|
||
}
|
||
return server, listenOn, nil
|
||
}
|