国内接口 四行代码搭建一个属于自己的AI聊天机器人_国内聊天ai代码

一、背景

AI语言模型在最近很火，今天介绍一个中文的语言模型api：SKchat-v3聊天

二、单轮对话实现

api文档：

具体的代码：

导入库

import requests
'
运行
运行

定义函数

def chat(text):
	r = requests.get('https://4147093qp2.imdo.co/openapi/fastchat?inp='+text)
	return r.json()['data']['text']
'
运行
运行

就短短4行，但是有些时候会有一些问题：

>>> chat('你叫什么名字啊？')
'我叫秦老师，你好？\n  秦老师：嗯嗯，你也好啊。\n  我的'

由于是基于SKchat-v3小说续写 开发的，所以有时候返回的内容会过多，这时候我们就要按‘\n’截断：

def chat(text):
	r = requests.get('https://4147093qp2.imdo.co/openapi/fastchat?inp='+text)
	str_ = r.json()['data']['text']
	if '\n' in str_:
		str_=str_.split('\n')[0]
	return str_
'
运行
运行

再把代码精简一下：

def chat(text):
	r = requests.get('https://4147093qp2.imdo.co/openapi/fastchat?inp='+text)
	return r.json()['data']['text'].split('\n')[0]
'
运行
运行

这就是完整的代码，回复时间大概在9秒左右。不支持大量并发或频繁调用。

三、多轮对话实现

内容差不多，就是url换了。

def chat2(text,usern):
	"""
    text: 聊天内容
    usern:你的用户名（暂时不需要注册，直接起一个就行）
	"""
	r = requests.get(f'https://4147093qp2.imdo.co/openapi/user-chat?inp={text}&user={usern}')
	return r.json()['data']['result'].split('\n')[0]
'
运行
运行

在线体验： 

输入昵称 (imdo.co)https://4147093qp2.imdo.co/getname.html

API文档：

        小思框架-api接口文档​​​​​​https://github.com/Ericsjq/xiaothinkNLP-API/blob/main/xiaothink-api%E6%8E%A5%E5%8F%A3%E6%96%87%E6%A1%A3.md​

没做完控制台，接口不需要API Token。 

注：由于是基于小说续写模型做的接口，所以不能写代码等任务，但是闲聊效果还是比较好的。需要自己定制AI角色形象的私信我，可以根据小说片段对人物的描写生成角色，模仿人物的语气、风格。接口默认的是小思（开朗、活泼......）。