智谱AI的API调用

最新推荐文章于 2025-08-12 14:19:08 发布

原创

最新推荐文章于 2025-08-12 14:19:08 发布 · 710 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#网络 #服务器 #运维

国内好用的通用大模型

实现的是mqtt的AI问答。简单示例，这个是在之前利用MQTT转TCP透传代码上改的，所以有点慢，但是复制粘贴就能用

# -*- coding: GB2312 -*-
import json
from zhipuai import ZhipuAI
import os
import sys
import socket
import paho.mqtt.client as mqtt
import threading

# 配置文件名
config_file = 'aicfg.txt'

# 示例API_KEY配置
example_config = {
    "API_KEY": "your_api_key_here"
}

def write_default_config():
    with open(config_file, 'w', encoding='utf-8') as f:
        json.dump(example_config, f, ensure_ascii=False, indent=4)
    print(f"配置文件 {config_file} 已创建，并写入示例API_KEY配置。")

def read_config():
    with open(config_file, 'r', encoding='utf-8') as f:
        config = json.load(f)
    return config

def get_response_from_zhipuai(question, api_key):
    try:
        client = ZhipuAI(api_key=api_key)
        response = client.chat.completions.create(
            model="GLM-4-Flash",
            messages=[
                {"role": "user", "content": question},
            ],
            stream=True