itcwc / go-zhipu

go智谱清言扩展

go-zhipu

• 新版内容已更新完毕，参数详情请看官方文档，欢迎提交bug~
• 首次写第三方扩展，有代码问题还请见谅。

---

前言

• go-智谱清言（go-zhipu）是基于 go 语言开发的智谱清言 API 接口包，主要用于智谱清言模型的调用，支持通用模型、图像大模型、超拟人大模型、向量模型、Batch API、模型微调、知识管理等功能。
• 详情见官方文档：智谱清言官方文档

---

支持模型

• 通用模型
  • GLM-4
    • sse调用
    • 异步调用
    • 任务结果查询
  • GLM-4V
    • sse调用
  • GLM-3-Turbo
    • sse调用
    • 异步调用
    • 任务结果查询
• 图像大模型
• 超拟人大模型
  • CharacterGLM
    • sse调用
    • 异步调用
    • 任务结果查询
  • Emohaa
    • sse调用
    • 异步调用
    • 任务结果查询
• 向量模型
  • 创建向量模型同步请求
• Batch API
  • 创建 Batch
  • 取消 Batch
  • 列出 Batch
  • 下载 Batch 结果
• 模型微调
  • 创建微调任务
  • 查询微调任务事件
  • 查询微调任务
  • 查询个人微调任务
  • 删除微调任务
  • 取消微调任务
  • 删除微调模型
• 知识管理
  • 知识库管理
    • 创建知识库
    • 编辑知识库
    • 检索知识库列表
    • 删除知识库
    • 知识库使用量查询
  • 文件管理
    • 文件上传
    • 编辑知识库文件
    • 查询文件列表
    • 删除知识库文件
    • 查询知识库文件详情

---

安装使用

• 安装

  go get -u github.com/itcwc/go-zhipu

• 使用

  import "github.com/itcwc/go-zhipu"

---

示例

package examples

import (
"fmt"

zhipu "github.com/itcwc/go-zhipu/model_api"
)

func Example() {
  expireAtTime := int64(1719803252) // token 过期时间
  mssage := zhipu.PostParams{
    Model: "glm-3-turbo",
    Messages: []zhipu.Message{
      {
        Role:    "user",    // 消息的角色信息 详见文档
        Content: "content", // 消息内容
      },
    },
  }
  
  apiKey := "your api key"

  postResponse, err := zhipu.BeCommonModel(expireAtTime, mssage, apiKey)
  if err != nil {
    fmt.Println(err)
    return
  }

  fmt.Println(postResponse)
}

---

其他模型调用

• 通用模型

  // sse调用
  go-zhipu.model_api.BeCommonModel(expireAtTime int64, postParams PostParams, apiKey string)
  // 异步调用
  go-zhipu.model_api.ModelAsynchronousCall(expireAtTime int64, postParams PostParams, apiKey string)
  // 任务结果查询
  go-zhipu.model_api.ModelTaskResultQuery(expireAtTime int64, id int, apiKey string)

• 图像大模型

  go-zhipu.model_api.ImageLargeModel(expireAtTime int64, prompt string, apiKey string, model string, userId string)

• 超拟人大模型

  // 同步调用
  go-zhipu.model_api.SuperhumanoidModel(expireAtTime int64, postParams PostSuperhumanoidParams, apiKey string)
  // 异步调用
  go-zhipu.model_api.SHMAsyncCall(expireAtTime int64, postParams PostSuperhumanoidParams, apiKey string)

• 向量模型

  go-zhipu.model_api.VectorModel(expireAtTime int64, input string, apiKey string, model string)

• Batch API

  // 创建 Batch
  go-zhipu.model_api.BatchAPICreate(expireAtTime int64, postParams PostBatchParams, apiKey string)
  // 检索 Batch
  go-zhipu.model_api.BatchSearch(expireAtTime int64, batchId int, apiKey string)
  // 取消 Batch
  go-zhipu.model_api.BatchCancel(expireAtTime int64, batchId int, apiKey string)
  // 列出 Batch
  go-zhipu.model_api.BatchList(expireAtTime int64, after string, limit int, apiKey string)
  // 下载 Batch 结果
  go-zhipu.model_api.BatchDownload(expireAtTime int64, fileId int, apiKey string)

• 模型微调

  // 创建微调任务
  go-zhipu.model_api.CreateModelFineTuning(expireAtTime int64, trainingFile string, apiKey string, model string)
  // 查询微调任务事件
  go-zhipu.model_api.QueryModelFineTuningEvent(expireAtTime int64, jobId int, after string, limit int, apiKey string)
  // 查询微调任务
  go-zhipu.model_api.QueryModelFineTuning(expireAtTime int64, jobId int, after string, limit int, apiKey string)
  // 查询个人微调任务
  go-zhipu.model_api.QueryPersonalModelFineTuning(expireAtTime int64, after string, limit int, apiKey string)
  // 删除微调任务
  go-zhipu.model_api.DeleteModelFineTuning(expireAtTime int64, jobId int, apiKey string)
  // 取消微调任务
  go-zhipu.model_api.CancelModelFineTuning(expireAtTime int64, jobId int, apiKey string)
  // 删除微调模型
  go-zhipu.model_api.DeleteModelFineTuningModel(expireAtTime int64, fineTunedModel string, apiKey string)

• 知识管理

  • 知识库管理

    // 创建知识库
    go-zhipu.model_api.Knowledge(expireAtTime int64, postParams PostKnowledgeParams, apiKey string, model string)
    // 编辑知识库
    go-zhipu.model_api.EditKnowledge(expireAtTime int64, postParams PostKnowledgeItemParams, apiKey string)
    // 检索知识库列表
    go-zhipu.model_api.QueryKnowledgeList(expireAtTime int64, page int, size int, apiKey string)
    // 删除知识库
    go-zhipu.model_api.DeleteKnowledge(expireAtTime int64, knowledgeId string, apiKey string)
    // 知识库使用量查询
    go-zhipu.model_api.KnowledgeUsage(expireAtTime int64, apiKey string)

  • 文件管理

    // 文件管理
    go-zhipu.model_api.FileManagement(expireAtTime int64, purpose string, apiKey string, model string, file *FileHeader)
    // 编辑知识库文件
    go-zhipu.model_api.EditKnowledgeFile(expireAtTime int64, postParams KnowledgeFileParams, apiKey string)
    // 查询文件列表
    go-zhipu.model_api.QueryFileList(expireAtTime int64, postParams QueryFileListParams, apiKey string)
    // 删除知识库文件
    go-zhipu.model_api.DeleteKnowledgeFile(expireAtTime int64, id string, apiKey string)
    // 查询知识库文件详情
    go-zhipu.model_api.QueryKnowledgeFileDetail(expireAtTime int64, id string, apiKey string)

联系我

• 邮箱：[email protected]