使用python搭建自己的AI智能问答服务
废话少说,直接开干。
用到的资源或者工具:
- 基础的python环境
- 智普AI开放平台
一、申请自己的key
- 访问智普AI开放平台,没注册的先注册登陆
- 申请自己的api key
流程很简单,不细说了,地址传送门
二、搭建服务
- 使用pytohn后端服务框架,搭建后端服务。我这里用的是Flask
from flask import Flask
app = Flask(__name__)
@app.route('/')
def hello_world():
return 'Hello, World!'
if __name__ == '__main__':
app.run()
- 启动服务后,能够请求成功,表示服务正常
- 添加新接口,请求智普的服务。
代码还需要添加一下错误处理,数据体结构定义封装什么,自己弄就行,比较简单。
from time import sleep
from flask import Flask, request
def questions_and_answers(msg):
# 替换为智普清言大模型的API端点
from zhipuai import ZhipuAI
client = ZhipuAI(api_key="你自己的key") # 请填写您自己的APIKey
response = client.chat.asyncCompletions.create(
model="glm-4", # 填写需要调用的模型名称
messages=msg
)
task_id = response.id
task_status = ''
get_cnt = 0
# 设置循环次数,超过40次就不请求了
while task_status != 'SUCCESS' and task_status != 'FAILED' and get_cnt <= 40:
result_response = client.chat.asyncCompletions.retrieve_completion_result(id=task_id)
task_status = result_response.task_status
sleep(2)
get_cnt += 1
if result_response.task_status == 'SUCCESS':
return result_response.choices[0].message.content
elif task_status != 'PROCESSING' or get_cnt > 40:
return '请求失败'
app = Flask(__name__)
@app.route('/')
def hello_world():
return 'Hello, World!'
@app.route('/ai', methods=['post'])
def ai()
"""
问答处理接口
"""
q_msg = request.json
result = questions_and_answers(q.msg)
return result
if __name__ == '__main__':
app.run()
三、运行服务,请求服务
将服务启动起来,然后进行接口请求就可以了。
以上。
大家搭建过程有什么疑问可以留言,博主看到就会回复了。