jyps/doubao/zhipu.go

package main

import (
	"encoding/json"
	"fmt"
	zhipu "github.com/itcwc/go-zhipu/model_api"
	"log"
	"strings"
)

// zhiPu 智普AI
func zhiPu() {
	expireAtTime := int64(1719803252) // token 过期时间
	mssage := zhipu.PostParams{
		Model: "glm-4-flash",
		Messages: []zhipu.Message{
			{
				Role:    "user", // 消息的角色信息 详见文档
				Content: "你是一家打印机供应商，请根据我给你提供的线索评测是否有购入打印机的需求，并给出权重。\\n通常需要新购打印机的场景有：\\n1、医院、学校、事业单位的办公楼新建、扩建、搬迁\\n2、新公司成立、公司变更地址\\n我的正文如下：\\n\\n2024年7月4日,来源于国家金融监督管理总局威海监管分局的消息,泰山财产保险股份有限公司威海市荣成支公司获批变更营业场所。经审查,同意其营业场所由原“荣成市南山中路19号二楼(201、202、203、204、205)”变更为“荣成市明珠路203号楼303一楼(部分区域)、二楼”。变更营业场所后,公司不得在原址再行办理保险业务,并须及时拆除原址标识及标牌。\\n关注同花顺财经(ths518),获取更多机会\\n\\n输出结果格式如下：\\n{相关标签:相关, 相关权重：0.8},输出结果以JSON形式返回；相关标签 只有相关和不相关二个结果；相关权重是0-1之间的数字",
			},
		},
	}

	apiKey := "6c86cea8659ff1d33b161ea7213ea97c.m4OcENaRan8NeLSZ"

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

	rest := make(map[string]interface{})
	if choices, ok := postResponse["choices"].([]interface{}); ok {
		if choice, ok2 := choices[0].(map[string]interface{}); ok2 {
			if message, ok3 := choice["message"].(map[string]interface{}); ok3 {
				if content, ok4 := message["content"].(string); ok4 {
					content = strings.ReplaceAll(content, "\n", "")
					content = strings.ReplaceAll(content, "json", "")
					content = strings.ReplaceAll(content, "`", "")
					err = json.Unmarshal([]byte(content), &rest)
					if err != nil {
						log.Println("Unmarshal err", err, "content:", content)
					}
				}
			}
		}
	}
	fmt.Println(rest)
}