Go語言調(diào)用DeepSeek?API實(shí)現(xiàn)流式輸出和對話

更新時(shí)間：2025年02月07日 09:24:33 作者：老大白菜

DeepSeek是一個強(qiáng)大的AI模型服務(wù)平臺,本文將詳細(xì)介紹如何使用Go語言調(diào)用DeepSeek?API實(shí)現(xiàn)流式輸出和對話功能,感興趣的小伙伴可以了解一下

引言

DeepSeek 是一個強(qiáng)大的 AI 模型服務(wù)平臺，本文將詳細(xì)介紹如何使用 Go 語言調(diào)用 DeepSeek API，實(shí)現(xiàn)流式輸出和對話功能。

Deepseek的api因?yàn)楸还粢巡荒苡?，本文?DeepSeek：https://cloud.siliconflow.cn/i/vnCCfVaQ 為例子進(jìn)行講解。

1. 環(huán)境準(zhǔn)備

首先，我們需要準(zhǔn)備以下內(nèi)容：

Go 語言環(huán)境

DeepSeek API 訪問權(quán)限

開發(fā)工具（如 VS Code）

2. 基礎(chǔ)代碼實(shí)現(xiàn)

2.1 創(chuàng)建項(xiàng)目結(jié)構(gòu)

mkdir deepseek-go
cd deepseek-go
go mod init deepseek-go

2.2 核心代碼實(shí)現(xiàn)

package main

import (
    "bufio"
    "encoding/json"
    "fmt"
    "net/http"
    "os"
    "strings"
    "time"
)

// 定義響應(yīng)結(jié)構(gòu)
type ChatResponse struct {
    Choices []struct {
        Delta struct {
            Content string `json:"content"`
        } `json:"delta"`
    } `json:"choices"`
}

func main() {
    // 創(chuàng)建輸出文件
    file, err := os.OpenFile("conversation.txt", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
    if err != nil {
        fmt.Printf("Error opening file: %v\n", err)
        return
    }
    defer file.Close()

    // API 配置
    url := "https://api.siliconflow.cn/v1/chat/completions"
    
    for {
        // 獲取用戶輸入
        fmt.Print("\n請輸入您的問題 (輸入 q 退出): ")
        reader := bufio.NewReader(os.Stdin)
        question, _ := reader.ReadString('\n')
        question = strings.TrimSpace(question)
        
        if question == "q" {
            break
        }

        // 記錄對話時(shí)間
        timestamp := time.Now().Format("2006-01-02 15:04:05")
        file.WriteString(fmt.Sprintf("\n[%s] Question:\n%s\n\n", timestamp, question))

        // 構(gòu)建請求體
        payload := fmt.Sprintf(`{
            "model": "deepseek-ai/DeepSeek-V3",
            "messages": [
                {
                    "role": "user",
                    "content": "%s"
                }
            ],
            "stream": true,
            "max_tokens": 2048,
            "temperature": 0.7
        }`, question)

        // 發(fā)送請求
        req, _ := http.NewRequest("POST", url, strings.NewReader(payload))
        req.Header.Add("Content-Type", "application/json")
        req.Header.Add("Authorization", "Bearer YOUR_API_KEY")  // 替換為你的 API Key

        // 獲取響應(yīng)
        res, _ := http.DefaultClient.Do(req)
        defer res.Body.Close()

        // 處理流式響應(yīng)
        scanner := bufio.NewReader(res.Body)
        for {
            line, err := scanner.ReadString('\n')
            if err != nil {
                break
            }

            line = strings.TrimSpace(line)
            if line == "" || line == "data: [DONE]" {
                continue
            }

            if strings.HasPrefix(line, "data: ") {
                line = strings.TrimPrefix(line, "data: ")
            }

            var response ChatResponse
            if err := json.Unmarshal([]byte(line), &response); err != nil {
                continue
            }

            if len(response.Choices) > 0 {
                content := response.Choices[0].Delta.Content
                if content != "" {
                    fmt.Print(content)
                    file.WriteString(content)
                }
            }
        }
    }
}

3. 主要特性說明

3.1 流式輸出

DeepSeek API 支持流式輸出（Stream），通過設(shè)置 "stream": true，我們可以實(shí)現(xiàn)實(shí)時(shí)顯示 AI 回復(fù)的效果。這帶來了更好的用戶體驗(yàn)：

即時(shí)看到響應(yīng)內(nèi)容
減少等待時(shí)間
更自然的對話體驗(yàn)

3.2 參數(shù)配置

{
    "model": "deepseek-ai/DeepSeek-V3",
    "messages": [...],
    "stream": true,
    "max_tokens": 2048,
    "temperature": 0.7,
    "top_p": 0.7,
    "top_k": 50,
    "frequency_penalty": 0.5
}

參數(shù)說明：