create_app_logic.go 7.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. package fastgpt
  2. import (
  3. "bytes"
  4. "context"
  5. "fmt"
  6. "github.com/suyuan32/simple-admin-common/msg/errormsg"
  7. "net/http"
  8. "strconv"
  9. "wechat-api/internal/svc"
  10. "wechat-api/internal/types"
  11. "github.com/zeromicro/go-zero/core/logx"
  12. )
  13. type CreateAppLogic struct {
  14. logx.Logger
  15. ctx context.Context
  16. svcCtx *svc.ServiceContext
  17. }
  18. func NewCreateAppLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CreateAppLogic {
  19. return &CreateAppLogic{
  20. Logger: logx.WithContext(ctx),
  21. ctx: ctx,
  22. svcCtx: svcCtx}
  23. }
  24. func (l *CreateAppLogic) CreateApp(req *types.CreateAppsReq) (resp *types.BaseMsgResp, err error) {
  25. organizationId := l.ctx.Value("organizationId").(uint64)
  26. organizationIdStr := strconv.FormatUint(organizationId, 10)
  27. token, err := GetToken(organizationIdStr)
  28. if err != nil {
  29. return nil, fmt.Errorf("invalid token")
  30. }
  31. AppCreate(req.Type, req.Name, token)
  32. return &types.BaseMsgResp{Msg: errormsg.Success}, nil
  33. }
  34. func AppCreate(apps_type string, name string, token string) {
  35. url := "https://fastgpt.gkscrm.com/api/core/app/create"
  36. jsonData := ""
  37. if apps_type == "simple" {
  38. jsonData = fmt.Sprintf(`{"parentId":"","avatar":"core/app/type/simpleFill","name":"%s","type":"simple","modules":[{"nodeId":"userGuide","name":"common:core.module.template.system_config","intro":"common:core.module.template.config_params","avatar":"core/workflow/template/systemConfig","flowNodeType":"userGuide","position":{"x":531.2422736065552,"y":-486.7611729549753},"version":"481","inputs":[{"key":"welcomeText","renderTypeList":["hidden"],"valueType":"string","label":"core.app.Welcome Text","value":""},{"key":"variables","renderTypeList":["hidden"],"valueType":"any","label":"core.app.Chat Variable","value":[]},{"key":"questionGuide","valueType":"object","renderTypeList":["hidden"],"label":"core.app.Question Guide","value":{"open":false}},{"key":"tts","renderTypeList":["hidden"],"valueType":"any","label":"","value":{"type":"web"}},{"key":"whisper","renderTypeList":["hidden"],"valueType":"any","label":"","value":{"open":false,"autoSend":false,"autoTTSResponse":false}},{"key":"scheduleTrigger","renderTypeList":["hidden"],"valueType":"any","label":"","value":null}],"outputs":[]},{"nodeId":"448745","name":"common:core.module.template.work_start","intro":"","avatar":"core/workflow/template/workflowStart","flowNodeType":"workflowStart","position":{"x":558.4082376415505,"y":123.72387429194112},"version":"481","inputs":[{"key":"userChatInput","renderTypeList":["reference","textarea"],"valueType":"string","label":"common:core.module.input.label.user question","required":true,"toolDescription":"common:core.module.input.label.user question"}],"outputs":[{"id":"userChatInput","key":"userChatInput","label":"core.module.input.label.user question","valueType":"string","type":"static"}]},{"nodeId":"loOvhld2ZTKa","name":"common:core.module.template.ai_chat","intro":"common:core.module.template.ai_chat_intro","avatar":"core/workflow/template/aiChat","flowNodeType":"chatNode","showStatus":true,"position":{"x":1097.7317280958762,"y":-244.16014496351386},"version":"481","inputs":[{"key":"model","renderTypeList":["settingLLMModel","reference"],"label":"core.module.input.label.aiModel","valueType":"string","value":"gpt-4o-mini"},{"key":"temperature","renderTypeList":["hidden"],"label":"","value":0,"valueType":"number","min":0,"max":10,"step":1},{"key":"maxToken","renderTypeList":["hidden"],"label":"","value":2000,"valueType":"number","min":100,"max":4000,"step":50},{"key":"isResponseAnswerText","renderTypeList":["hidden"],"label":"","value":true,"valueType":"boolean"},{"key":"quoteTemplate","renderTypeList":["hidden"],"label":"","valueType":"string"},{"key":"quotePrompt","renderTypeList":["hidden"],"label":"","valueType":"string"},{"key":"systemPrompt","renderTypeList":["textarea","reference"],"max":3000,"valueType":"string","label":"core.ai.Prompt","description":"core.app.tip.systemPromptTip","placeholder":"core.app.tip.chatNodeSystemPromptTip","value":""},{"key":"history","renderTypeList":["numberInput","reference"],"valueType":"chatHistory","label":"core.module.input.label.chat history","required":true,"min":0,"max":30,"value":6},{"key":"userChatInput","renderTypeList":["reference","textarea"],"valueType":"string","label":"common:core.module.input.label.user question","required":true,"toolDescription":"common:core.module.input.label.user question","value":["448745","userChatInput"]},{"key":"quoteQA","renderTypeList":["settingDatasetQuotePrompt"],"label":"","debugLabel":"common:core.module.Dataset quote.label","description":"","valueType":"datasetQuote"}],"outputs":[{"id":"history","key":"history","label":"core.module.output.label.New context","description":"core.module.output.description.New context","valueType":"chatHistory","type":"static"},{"id":"answerText","key":"answerText","label":"core.module.output.label.Ai response content","description":"core.module.output.description.Ai response content","valueType":"string","type":"static"}]}],"edges":[{"source":"448745","target":"loOvhld2ZTKa","sourceHandle":"448745-source-right","targetHandle":"loOvhld2ZTKa-target-left"}]}`, name)
  39. } else {
  40. jsonData = fmt.Sprintf(`{"parentId":"","avatar":"core/app/type/workflowFill","name":"%s","type":"advanced","modules":[{"nodeId":"userGuide","name":"common:core.module.template.system_config","intro":"common:core.module.template.system_config_info","avatar":"core/workflow/template/systemConfig","flowNodeType":"userGuide","position":{"x":262.2732338817093,"y":-476.00241136598146},"version":"481","inputs":[{"key":"welcomeText","renderTypeList":["hidden"],"valueType":"string","label":"core.app.Welcome Text","value":""},{"key":"variables","renderTypeList":["hidden"],"valueType":"any","label":"core.app.Chat Variable","value":[]},{"key":"questionGuide","valueType":"any","renderTypeList":["hidden"],"label":"core.app.Question Guide","value":{"open":false}},{"key":"tts","renderTypeList":["hidden"],"valueType":"any","label":"","value":{"type":"web"}},{"key":"whisper","renderTypeList":["hidden"],"valueType":"any","label":"","value":{"open":false,"autoSend":false,"autoTTSResponse":false}},{"key":"scheduleTrigger","renderTypeList":["hidden"],"valueType":"any","label":"","value":null}],"outputs":[]},{"nodeId":"448745","name":"common:core.module.template.work_start","intro":"","avatar":"core/workflow/template/workflowStart","flowNodeType":"workflowStart","position":{"x":632.368838596004,"y":-347.7446492944009},"version":"481","inputs":[{"key":"userChatInput","renderTypeList":["reference","textarea"],"valueType":"string","label":"common:core.module.input.label.user question","required":true,"toolDescription":"common:core.module.input.label.user question"}],"outputs":[{"id":"userChatInput","key":"userChatInput","label":"common:core.module.input.label.user question","type":"static","valueType":"string"}]}],"edges":[]}`, name)
  41. }
  42. requestBody := []byte(jsonData)
  43. // 创建一个新的请求
  44. req1, err := http.NewRequest("POST", url, bytes.NewBuffer(requestBody))
  45. if err != nil {
  46. fmt.Printf("Error creating request: %v\n", err)
  47. return
  48. }
  49. // 设置请求头
  50. req1.Header.Set("Content-Type", "application/json")
  51. req1.Header.Set("Cookie", fmt.Sprintf("fastgpt_token=%s", token))
  52. // 使用 http.Client 发送请求
  53. client := &http.Client{}
  54. resp1, err := client.Do(req1)
  55. logx.Info("AppCreate resp1:", resp1)
  56. logx.Info("AppCreate err:", err)
  57. if err != nil {
  58. fmt.Printf("Error sending request: %v\n", err)
  59. return
  60. }
  61. defer resp1.Body.Close()
  62. return
  63. }