Go 语言调用 SiliconFlow 的 Deepseek AI Janus-Pro-7B 模型进行图像生成

老大白菜2025-02-16 23:38

使用 Go 调用 SiliconFlow 图像生成 API

概述

本文档介绍如何使用 Go 语言调用 SiliconFlow 的 Deepseek AI Janus-Pro-7B 模型进行图像生成。

环境准备

Go 1.20 或更高版本
网络连接
SiliconFlow API 令牌

依赖管理

在项目根目录创建 go.mod 文件：

bash 复制代码

go mod init image-generator
go get github.com/go-resty/resty/v2

代码实现

`main.go`

go 复制代码

package main

import (
    "fmt"
    "log"

    "github.com/go-resty/resty/v2"
)

type ImageGenerationRequest struct {
    Model  string `json:"model"`
    Prompt string `json:"prompt"`
    Seed   int64  `json:"seed"`
}

func generateImage(apiToken string, prompt string) error {
    client := resty.New()

    req := &ImageGenerationRequest{
        Model:  "deepseek-ai/Janus-Pro-7B",
        Prompt: prompt,
        Seed:   4999999999,
    }

    resp, err := client.R().
        SetHeader("Authorization", "Bearer "+apiToken).
        SetHeader("Content-Type", "application/json").
        SetBody(req).
        Post("https://api.siliconflow.cn/v1/images/generations")

    if err != nil {
        return fmt.Errorf("请求发送失败: %v", err)
    }

    if resp.IsError() {
        return fmt.Errorf("API错误: %s", resp.String())
    }

    fmt.Println("图像生成响应:", resp.String())
    return nil
}

func main() {
    apiToken := "<你的SiliconFlow令牌>"
    prompt := "海边的岛屿，海鸥，月光洒在海面上，灯塔，背景有船只，海面上有飞鱼"

    err := generateImage(apiToken, prompt)
    if err != nil {
        log.Fatalf("图像生成失败: %v", err)
    }
}

运行项目

bash 复制代码

go mod tidy
go run main.go

注意事项

替换 <你的SiliconFlow令牌> 为实际的 API 令牌
建议使用环境变量或配置文件管理敏感信息
处理 API 响应和错误

错误处理

检查网络连接
验证 API 令牌
处理 API 返回的错误消息

最佳实践

使用错误处理和日志记录
考虑实现重试机制
保护 API 令牌安全

扩展功能建议

支持更多图像生成参数
实现图像保存功能
添加更复杂的错误处理逻辑

参考资源

SiliconFlow API 文档
Go Resty 库：https://github.com/go-resty/resty

快来体验 siliconflow 图像生成服务：https://cloud.siliconflow.cn/i/vnCCfVaQ