【大模型API调用初尝试一】智谱AI && 通义千问 |
您所在的位置：网站首页 › 智谱无线电找不到考场 › 【大模型API调用初尝试一】智谱AI && 通义千问 |
【大模型API调用初尝试一】智谱AI && 通义千问
|
大模型API调用初尝试一
调用大模型API能干什么智谱AI大模型API调用的过程获取API_KEYGLM_4同步调用GLM_4异步调用文生图大模型API调用
阿里云通义千问API调用过程单轮会话多轮会话
调用大模型API能干什么
大模型的参数非常庞大，功能非常强大，但是训练成本高昂，因此个人或者小企业自己去训练一个大模型是不可能的。我们可以通过直接调用大模型的API，将大模型集成到自己的应用中。 大模型的API就是一个接口，类似MaaS，用户通过调用API访问大模型，获得大模型针对用户prompt（问题）的输出，一般输出是json格式的，然后我们利用这个输出进行后续的操作。 但是大模型是一个已经训练好的模型，类似一个封装好的盒子，其能够运用的知识是有限的，比如chatgpt的知识截至2021年9月，让它提供实时的天气预报是不可行的，因此现在的大模型API中还增加了外部API/函数的调用功能，大模型能够根据用户的prompt确定要去调用的外部API/function–>调用外部API/function并获得结果–>大模型整合结果再输出，如下图所示： 我们可以参考某一个大模型对应的官方文档，学习如何调用其API。 智谱AI大模型API调用的过程智谱AI是清华大学计算机系技术成果转化而来的公司，公司产品包括：中英双语千亿级超大规模预训练模型GLM-130B，并基于此推出对话模型ChatGLM；高效率代码模型CodeGeeX；多模态理解模型CogVLM和文生图模型CogView等。 获取API_KEY调用API首先需要获取API，每个用户需要自己注册申请对应的API。进入智谱AI官网，注册后申请API，复制API_KEY以便后续调用： 同步调用就是创建一个调用任务后，一直运行这个任务直到收到大模型的响应输出；对应的是异步调用，创建调用任务后获得这个任务的request-id，之后可以去干其他事情，后续通过这个任务ID查看模型的输出。 from zhipuai import ZhipuAI # 先安装ZhipuAI的包 pip install ZhipuAI client = ZhipuAI(api_key="xxxx") # 填写您自己的APIKey response = client.chat.completions.create( model="glm-4", # 填写需要调用的模型名称 messages=[ # messages是json格式的数据，大模型逐条响应 {"role": "user", "content": "作为一名营销专家，请为我的产品创作一个吸引人的slogan"}, {"role": "assistant", "content": "当然，为了创作一个吸引人的slogan，请告诉我一些关于您产品的信息"}, {"role": "user", "content": "智谱AI开放平台"}, {"role": "assistant", "content": "智启未来，谱绘无限一智谱AI，让创新触手可及!"}, {"role": "user", "content": "创造一个更精准、吸引人的slogan"} ], ) # 直接输出response，查看响应的具体内容 print(response)输出的完整json数据： 上面例子传入大模型的message列表里面，有user的信息，也有assistant的信息，大模型实际响应的只有user对应的content，但是assistant的内容可以为大模型提供一些上下文或者提示。 message只传入用户的输入，并多次问答的例子如下： from zhipuai import ZhipuAI client = ZhipuAI(api_key="xxxxxx") # 填写您自己的APIKey # 循环提问/对话 while True: # 接收用户输入作为问题 prompt = input("\nuser:") response = client.chat.completions.create( model="glm-4", # 填写需要调用的模型名称 messages=[ {"role": "user", "content": prompt} ], ) answer = response.choices[0].message.content print("\nZhipuAI:",answer) # 只输出大模型响应的message.context结果： 参考官方文档，异步调用获取结果包括两步：①获取模型的响应ID，②根据ID查询响应结果； import time from zhipuai import ZhipuAI client = ZhipuAI(api_key="xxxxxx") # 请填写您自己的APIKey response = client.chat.asyncCompletions.create( model="glm-4", # 填写需要调用的模型名称 messages=[ { "role": "user", "content": "请你作为童话故事大王，写一篇短篇童话故事，故事的主题是要永远保持一颗善良的心，要能够激发儿童的学习兴趣和想象力，同时也能够帮助儿童更好地理解和接受故事中所蕴含的道理和价值观。" } ], ) # 获取响应ID task_id = response.id task_status = '' get_cnt = 0 while task_status != 'SUCCESS' and task_status != 'FAILED' and get_cnt |
现在打开API调用参考文档：通用大模型和图像大模型，直接复制其代码，看看输出是什么样的！
如果只关注最终输出的message的content，可以只取response.choices[0].message.content。
上述例子虽然user能够无限次的与大模型进行对话（交互），但由于message中只有用户单次的问题，没有保存上下文的信息，因此如果前后问题有衔接，这样的方式是不行的。应该使用多轮对话，将每轮对话的问答保存在message中，传给大模型为其提供对话的上下文。【本文地址】