jyps/getEs/tools.go

package main

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

// ZpAI 智普AI
func ZpAI(apiKey, model, content string) (rest map[string]interface{}) {
	expireAtTime := int64(1719803252) // token 过期时间
	sys := "请根据我给出的公司名称，依据其单位性质、业务范围和单位职能，准确给出最符合的一个国标行业分类标签，分别给出大类、中类和小类。输出结果以JSON格式返回，格式如下:{\"label1\":\"大类\",\"label2\":\"中类\",\"label3\":\"小类\"}。我只要最匹配 的一个标签，不要返回多个字段;如果无法识别出类别，直接给我空字符串。按照以上要求输出，不要联想，不要无中生有，不要生成解释。单位名称是："
	text := fmt.Sprintf(sys, content)

	mssage := zhipu.PostParams{
		Model: model,
		Messages: []zhipu.Message{
			{
				Role:    "user", // 消息的角色信息 详见文档
				Content: text,
			},
		},
	}

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

	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)
					}
				}
			}
		}
	}
	return
}

func checkString(s string) bool {
	for _, char := range s {
		if ('a' <= char && char <= 'z') || ('A' <= char && char <= 'Z') || strings.ContainsAny(string(char), "12345678910") {
			return true
		}
	}
	return false
}

// ZpAI1 智普问答，附带业务范围
func ZpAI1(apiKey, model, name string, businessScope string) (rest map[string]interface{}) {
	expireAtTime := int64(1719803252) // token 过期时间
	sys := "请根据我给出的公司名称，依据其单位性质、单位职能和业务范围，准确给出最符合的一个国标行业分类标签，分别给出大类、中类和小类。输出结果以JSON格式返回，格式如下:{\"label1\":\"大类\",\"label2\":\"中类\",\"label3\":\"小类\"}。我只要最匹配 的一个标签，不要返回多个字段;如果无法识别出类别，直接给我空字符串。按照以上要求输出，不要联想，不要无中生有，不要生成解释。单位名称是："
	text := fmt.Sprintf(sys, name)

	if businessScope != "" {
		text = text + "；业务范围是：" + businessScope
	}
	mssage := zhipu.PostParams{
		Model: model,
		Messages: []zhipu.Message{
			{
				Role:    "user", // 消息的角色信息 详见文档
				Content: text,
			},
		},
	}

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

	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)
					}
				}
			}
		}
	}
	return
}