简介:首先注册,申请获取使用资格(申请流程网上一堆),主要步骤如:
1、注册申请 讯飞星火大模型-AI大语言模型-星火大模型-科大讯飞
2、复制代码,更改自己控制台中APPID 、key等
3、设计logoText to ASCII Art Generator (TAAG) (patorjk.com)
4、使用Python命令打包成.exe
5、发送桌面快捷方式
其中,亮点对对话内容持久化、自己设计logo,多一句,在安装python后建议全局配置清华等镜像源,没有配置的现在配置也可以,在pip相关轮子时如果不成功,可以看看是否配置国内镜像源,如果还不行,可以采用去官网下载,本地安装即可(注意版本问题,如果配置代码有误,面向浏览器)。
配置代码
# 阿里云
pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/
# 中科大
pip config set global.index-url https://pypi.mirrors.ustc.edu.cn/simple/
# 清华
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple/
# 豆瓣
pip config set global.index-url https://pypi.douban.com/simple/
1、结果展示
1.1 对话展示
1.2 持久化展示
2、实现流程
2.1 代码部分
2.1.1 代码及来源说明
备注:该代码为二创原创,基于官网
复制代码前在pycharm中新建名为xfyun的文件夹(自定义未尝不可),如图中所示
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time : 2023/10/26 0026 18:14
# @Author : 程序员ck君
# @File : 讯飞星火.py
# @Software: PyCharm
import _thread as thread
import base64
import datetime
import hashlib
import hmac
import json
import time
from urllib.parse import urlparse
import ssl
from datetime import datetime
from time import mktime
from urllib.parse import urlencode
from wsgiref.handlers import format_date_time
import websocket
answer = ""
class Ws_Param(object):
# 初始化
def __init__(self, APPID, APIKey, APISecret, Spark_url):
self.APPID = APPID
self.APIKey = APIKey
self.APISecret = APISecret
self.host = urlparse(Spark_url).netloc
self.path = urlparse(Spark_url).path
self.Spark_url = Spark_url
# 生成url
def create_url(self):
# 生成RFC1123格式的时间戳
now = datetime.now()
date = format_date_time(mktime(now.timetuple()))
# 拼接字符串
signature_origin = "host: " + self.host + "\n"
signature_origin += "date: " + date + "\n"
signature_origin += "GET " + self.path + " HTTP/1.1"
# 进行hmac-sha256进行加密
signature_sha = hmac.new(self.APISecret.encode('utf-8'), signature_origin.encode('utf-8'),
digestmod=hashlib.sha256).digest()
signature_sha_base64 = base64.b64encode(signature_sha).decode(encoding='utf-8')
authorization_origin = f'api_key="{self.APIKey}", algorithm="hmac-sha256", headers="host date request-line", signature="{signature_sha_base64}"'
authorization = base64.b64encode(authorization_origin.encode('utf-8')).decode(encoding='utf-8')
# 将请求的鉴权参数组合为字典
v = {
"authorization": authorization,
"date": date,
"host": self.host
}
# 拼接鉴权参数,生成url
url = self.Spark_url + '?' + urlencode(v)
# 此处打印出建立连接时候的url,参考本demo的时候可取消上方打印的注释,比对相同参数时生成的url与自己代码生成的url是否一致
return url
# 收到websocket错误的处理
def on_error(ws, error):
print("错误信息 error:", error)
# 收到websocket关闭的处理
def on_close(ws, one, two):
print(" ")
# 收到websocket连接建立的处理
def on_open(ws):
thread.start_new_thread(run, (ws,))
def run(ws, *args):
data = json.dumps(gen_params(appid=ws.appid, domain=ws.domain, question=ws.question))
ws.send(data)
# 收到websocket消息的处理
def on_message(ws, message):
# print(message)
data = json.loads(message)
code = data['header']['code']
if code != 0:
print(f'请求错误: {code}, {data}')
ws.close()
else:
choices = data["payload"]["choices"]
status = choices["status"]
content = choices["text"][0]["content"]
print(content, end="")
global answer
answer += content
# print(1)
if status == 2:
ws.close()
def gen_params(appid, domain, question):
"""
通过appid和用户的提问来生成请参数
"""
data = {
"header": {
"app_id": appid,
"uid": "1234"
},
"parameter": {
"chat": {
"domain": domain,
"random_threshold": 0.5,
"max_tokens": 2048,
"auditing": "default"
}
},
"payload": {
"message": {
"text": question
}
}
}
return data
def main(appid, api_key, api_secret, Spark_url, domain, question):
# print("星火:")
wsParam = Ws_Param(appid, api_key, api_secret, Spark_url)
# websocket.enableTrace(False)
wsUrl = wsParam.create_url()
ws = websocket.WebSocketApp(wsUrl, on_message=on_message, on_error=on_error, on_close=on_close, on_open=on_open)
ws.appid = appid
ws.question = question
ws.domain = domain
ws.run_forever(sslopt={"cert_reqs": ssl.CERT_NONE})
"""
填写自己对应的
"""
appid = "" # 填写控制台中获取的 APPID 信息
api_secret = "" # 填写控制台中获取的 APISecret 信息
api_key = "" # 填写控制台中获取的 APIKey 信息
# 用于配置大模型版本,默认“general/generalv2”
# domain = "general" # v1.5版本
domain = "generalv2" # v2.0版本
# 云端环境的服务地址
# Spark_url = "ws://spark-api.xf-yun.com/v1.1/chat" # v1.5环境的地址
Spark_url = "ws://spark-api.xf-yun.com/v2.1/chat" # v2.0环境的地址
text = []
# length = 0
def getText(role, content):
jsoncon = {}
jsoncon["role"] = role
jsoncon["content"] = content
text.append(jsoncon)
return text
def getlength(text):
length = 0
for content in text:
temp = content["content"]
leng = len(temp)
length += leng
return length
def checklen(text):
while (getlength(text) > 8000):
del text[0]
return text
if __name__ == '__main__':
dateValue = time.strftime('%Y-%m-%d_%H-%M-%S', time.localtime()) # 获取时间, 用于命名文件
text.clear()
file_object = open("./mylog.log") # logo
try:
all_the_text = file_object.read()
finally:
file_object.close()
print(all_the_text)
print("------------------使用须知------------------\n1) 退出 -- exit")
with open(f'{dateValue}.txt', 'w', encoding="utf8") as file_xfyun: # 数据持久化
try:
while 1:
Input = input("\n我: ")
if Input == "exit":
file_xfyun.close() # 退出关闭流
exit()
question = checklen(getText("user", Input))
answer = ""
print("讯飞星火1.5v[(o)_(o)]: ", end="")
main(appid, api_key, api_secret, Spark_url, domain, question)
getText("assistant", answer)
file_xfyun.write("-----------------------------------------------------------------\n" +
Input + "\n" + answer + "\n") # 数据写入
except Exception as e:
print(e)
print("\n---------------- 程序中止 !!! -----------------------") # 非正常退出
finally:
file_xfyun.close() # 关闭文件流
print("关闭流!!!")
2.1.2 更改对应的Key
2.2 设计logo
复制设计logo的logo Text to ASCII Art Generator (TAAG) (patorjk.com) 名为mylog.log的文件放于同目录下。注意:后面打包运行exe后可能因为该文件报错。后面说解决方法
2.3 打包
在使用命令打包之前,应该运行一下是否都配置成功,然后再打包
pip install pyinstaller
注意要到讯飞星火.py这个目录下使用打包命令,PyInstaller将自动分析脚本的依赖项并生成一个dist
文件夹,其中包含可执行文件。可以在dist
文件夹中找到可执行文件。
pyinstaller 讯飞星火.py
打包结果预览