99 lines
2.4 KiB
Go
99 lines
2.4 KiB
Go
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
|
||
}
|