当前位置：首页 > article >正文

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

article 2025/2/18 20:10:33

使用 Go 调用 SiliconFlow 图像生成 API

概述

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

环境准备

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

依赖管理

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

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

代码实现

`main.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)
    }
}