feat: 新增执行记录实体与文件上传能力

workflow/controller/flow/flow_execution_controller.go

@@ -4,32 +4,19 @@ import (
 	flowDto "ai-agent/workflow/model/dto/flow"
 	flowService "ai-agent/workflow/service/flow"
 	"context"
-
-	"gitea.redpowerfuture.com/red-future/common/beans"
 )
 
 type flowExecution struct{}
 
 var FlowExecution = new(flowExecution)
 
-func (c *flowExecution) Execute(ctx context.Context, req *flowDto.ExecuteReq) (res *flowDto.ExecuteRes, err error) {
-	return flowService.FlowExecutionService.Execute(ctx, req)
-}
-
-func (c *flowExecution) ComposeCallBack(ctx context.Context, req *flowDto.ComposeCallbackReq) (res *beans.ResponseEmpty, err error) {
-	err = flowService.FlowExecutionService.ComposeCallback(ctx, req)
-	return
-}
-
-func (c *flowExecution) ModelCallback(ctx context.Context, req *flowDto.ModelCallbackReq) (res *beans.ResponseEmpty, err error) {
-	err = flowService.FlowExecutionService.ModelCallback(ctx, req)
-	return
-}
-
-func (c *flowExecution) VideoCallback(ctx context.Context, req *flowDto.VideoCallbackReq) (res *beans.ResponseEmpty, err error) {
-	err = flowService.FlowExecutionService.VideoCallback(ctx, req)
-	return
-}
+//func (c *flowExecution) Execute(ctx context.Context, req *flowDto.ExecuteReq) (res *flowDto.ExecuteRes, err error) {
+//	return flowService.FlowExecutionService.Execute(ctx, req)
+//}
+//
+//func (c *flowExecution) ReExecute(ctx context.Context, req *flowDto.ReExecuteReq) (res *flowDto.ExecuteRes, err error) {
+//	return flowService.FlowExecutionService.ReExecute(ctx, req)
+//}
 
 func (c *flowExecution) Get(ctx context.Context, req *flowDto.GetFlowExecutionReq) (res *flowDto.VOFlowExecution, err error) {
 	return flowService.FlowExecutionService.Get(ctx, req)

workflow/controller/session/session_controller.go

@@ -1,18 +1,34 @@
 package session
 
 import (
+	flowDto "ai-agent/workflow/model/dto/flow"
+	sessionDto "ai-agent/workflow/model/dto/session"
+	"ai-agent/workflow/service/flow"
+	sessionService "ai-agent/workflow/service/session"
 	"context"
 
-	sessionDto "ai-agent/workflow/model/dto/session"
-	sessionService "ai-agent/workflow/service/session"
-
 	"gitea.redpowerfuture.com/red-future/common/beans"
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/os/glog"
 )
 
 type session struct{}
 
 var Session = new(session)
 
+func (c *session) WsExecute(ctx context.Context, req *sessionDto.WebSocketConnectReq) (res *beans.ResponseEmpty, err error) {
+	r := g.RequestFromCtx(ctx)
+	// 首次连接仅升级 WebSocket，不区分普通对话/工作流；由后续消息 type 路由到对应处理器
+	err = flow.WsConnect(ctx, r, req)
+	if err != nil {
+		glog.Errorf(ctx, "ws connect failed: %v", err)
+		return nil, err
+	}
+	// 标记响应缓冲区，使 MiddlewareHandlerResponse 跳过
+	r.Response.Writeln("")
+	return
+}
+
 func (c *session) Get(ctx context.Context, req *sessionDto.GetSessionInfoReq) (res *sessionDto.GetSessionInfoRes, err error) {
 	return sessionService.SessionService.Get(ctx, req)
 }
@@ -27,3 +43,14 @@ func (c *session) Delete(ctx context.Context, req *sessionDto.DeleteSessionReq)
 	}
 	return &beans.ResponseEmpty{}, nil
 }
+
+func (c *session) DeleteRecord(ctx context.Context, req *sessionDto.DeleteSessionRecordReq) (res *beans.ResponseEmpty, err error) {
+	if err = sessionService.SessionService.DeleteRecord(ctx, req); err != nil {
+		return
+	}
+	return &beans.ResponseEmpty{}, nil
+}
+
+func (c *session) ResultList(ctx context.Context, req *sessionDto.ListWorkflowResultReq) (res *flowDto.ListFlowExecutionTreeRes, err error) {
+	return sessionService.SessionService.ResultList(ctx, req)
+}

workflow/controller/tool/tool_controller.go

@@ -0,0 +1,15 @@
+package tool
+
+import (
+	toolDto "ai-agent/workflow/model/dto/tool"
+	toolService "ai-agent/workflow/service/tool"
+	"context"
+)
+
+type tool struct{}
+
+var Tool = new(tool)
+
+func (c *tool) List(ctx context.Context, req *toolDto.ToolListReq) (res *toolDto.ToolListRes, err error) {
+	return toolService.ToolService.List(ctx, req)
+}