feat: 新增创作作品管理模块及相关配置

workflow/service/einograph/model.go

@@ -0,0 +1,98 @@
+package einograph
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"time"
+
+	"github.com/cloudwego/eino-ext/components/model/qwen"
+	"github.com/cloudwego/eino/components/model"
+	"github.com/gogf/gf/v2/util/gconv"
+)
+
+// NewChatModel component initialization function of node 'ChatModel1' in graph 'test'
+func NewChatModel(ctx context.Context) (cm model.ChatModel, err error) {
+	// TODO Modify component configuration here.
+	config := &qwen.ChatModelConfig{
+		APIKey:      "sk-4a8b82770bf74bc490eb3e4c5a8e2be9",
+		Model:       "qwen-turbo",
+		BaseURL:     "https://dashscope.aliyuncs.com/compatible-mode/v1", // 必须加！
+		MaxTokens:   gconv.PtrInt(2000),
+		Temperature: gconv.PtrFloat32(float32(0.7))}
+	cm, err = qwen.NewChatModel(ctx, config)
+	if err != nil {
+		return nil, err
+	}
+	return cm, nil
+}
+
+func GenerateRealImage(prompt string) (string, error) {
+	apiKey := "sk-4a8b82770bf74bc490eb3e4c5a8e2be9"
+	url := "https://dashscope.aliyuncs.com/api/v1/services/aigc/multimodal-generation/generation"
+
+	body := map[string]any{
+		"model": "qwen-image",
+		"input": map[string]any{
+			"messages": []map[string]any{
+				{
+					"role": "user",
+					"content": []map[string]string{
+						{"type": "text", "text": prompt},
+					},
+				},
+			},
+		},
+		"parameters": map[string]any{
+			"size":      "1024*1364",
+			"n":         1,
+			"watermark": false,
+		},
+	}
+
+	payload, _ := json.Marshal(body)
+	req, _ := http.NewRequest("POST", url, bytes.NewBuffer(payload))
+	req.Header.Set("Authorization", "Bearer "+apiKey)
+	req.Header.Set("Content-Type", "application/json")
+
+	client := &http.Client{Timeout: 30 * time.Second}
+	resp, err := client.Do(req)
+	if err != nil {
+		return "", err
+	}
+	defer resp.Body.Close()
+
+	data, _ := io.ReadAll(resp.Body)
+
+	// ✅ 修复：plus 模型返回的字段是 image 不是 image_url
+	var result struct {
+		Output struct {
+			Choices []struct {
+				Message struct {
+					Content []struct {
+						Image string `json:"image"`
+					} `json:"content"`
+				} `json:"message"`
+			} `json:"choices"`
+		} `json:"output"`
+		Code string `json:"code"`
+	}
+
+	err = gconv.Struct(data, &result)
+
+	if len(result.Output.Choices) == 0 || result.Code != "" {
+		return "", fmt.Errorf("生成失败: %s", string(data))
+	}
+
+	// ✅ 修复：直接取 base64，不再下载
+	imgBase64 := result.Output.Choices[0].Message.Content[0].Image
+	if imgBase64 == "" {
+		return "", fmt.Errorf("图片为空")
+	}
+
+	// 解码 base64
+	return imgBase64, nil
+}