prompts-core

controller/compose_session.go

@@ -0,0 +1,20 @@
+package controller
+
+import (
+	"context"
+
+	"prompts-core/model/dto"
+	"prompts-core/service"
+
+	"gitea.com/red-future/common/beans"
+)
+
+type session struct{}
+
+// Prompt 提示词配置控制器
+var Session = new(session)
+
+// SessionCallback 会话回调
+func (c *session) SessionCallback(ctx context.Context, req *dto.SessionCallbackReq) (res *beans.ResponseEmpty, err error) {
+	return service.Session.SessionCallback(ctx, req)
+}

controller/prompt_controller.go

@@ -0,0 +1,85 @@
+package controller
+
+import (
+	"context"
+
+	"prompts-core/model/dto"
+	"prompts-core/service"
+
+	"gitea.com/red-future/common/beans"
+)
+
+type prompt struct{}
+
+// Prompt 提示词配置控制器
+var Prompt = new(prompt)
+
+// ComposeMessages 调用 model-gateway 异步任务并同步等待结果
+func (c *prompt) ComposeMessages(ctx context.Context, req *dto.ComposeMessagesReq) (res *dto.ComposeMessagesRes, err error) {
+	return service.Prompt.ComposeMessages(ctx, req)
+}
+
+// ComposeMessagesCallback model-gateway 提示词回调
+func (c *prompt) Callback(ctx context.Context, req *dto.CallbackReq) (res *beans.ResponseEmpty, err error) {
+	err = service.Prompt.Callback(ctx, req)
+	return
+}
+
+// GetComposeTask 查询拼接任务结果
+func (c *prompt) GetComposeTask(ctx context.Context, req *dto.GetComposeTaskReq) (res *dto.GetComposeTaskRes, err error) {
+	return service.Prompt.GetComposeTask(ctx, req.TaskId)
+}
+
+// CreatePrompt 添加配置（默认启用）
+func (c *prompt) CreatePrompt(ctx context.Context, req *dto.CreatePromptReq) (res *dto.CreatePromptRes, err error) {
+	return service.Prompt.Create(ctx, req)
+}
+
+// UpdatePrompt 更新配置
+func (c *prompt) UpdatePrompt(ctx context.Context, req *dto.UpdatePromptReq) (res *beans.ResponseEmpty, err error) {
+	err = service.Prompt.Update(ctx, req)
+	return
+}
+
+// DeletePrompt 删除配置
+func (c *prompt) DeletePrompt(ctx context.Context, req *dto.DeletePromptReq) (res *beans.ResponseEmpty, err error) {
+	err = service.Prompt.Delete(ctx, req.ID)
+	return
+}
+
+// GetPrompt 获取配置详情
+func (c *prompt) GetPrompt(ctx context.Context, req *dto.GetPromptReq) (res *dto.GetPromptRes, err error) {
+	m, err := service.Prompt.Get(ctx, req.ID)
+	if err != nil {
+		return nil, err
+	}
+	return &dto.GetPromptRes{Prompt: m}, nil
+}
+
+// ListPrompt 配置列表
+func (c *prompt) ListPrompt(ctx context.Context, req *dto.ListPromptReq) (res *dto.ListPromptRes, err error) {
+	pageNum, pageSize := 1, 10
+	if req != nil && req.Page != nil {
+		if req.Page.PageNum > 0 {
+			pageNum = int(req.Page.PageNum)
+		}
+		if req.Page.PageSize > 0 {
+			pageSize = int(req.Page.PageSize)
+		}
+	}
+	var modelTypeID *int
+	modelType := ""
+	if req != nil {
+		modelTypeID = req.ModelTypeId
+		modelType = req.ModelType
+	}
+
+	list, total, err := service.Prompt.List(ctx, pageNum, pageSize, modelTypeID, modelType)
+	if err != nil {
+		return nil, err
+	}
+	return &dto.ListPromptRes{
+		List:  list,
+		Total: total,
+	}, nil
+}