基于Python本地接入讯飞星火语言模型

简介:首先注册,申请获取使用资格(申请流程网上一堆),主要步骤如:

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 打包

在使用命令打包之前,应该运行一下是否都配置成功,然后再打包

安装Pyinstaller

pip install pyinstaller 

 注意要到讯飞星火.py这个目录下使用打包命令,PyInstaller将自动分析脚本的依赖项并生成一个dist文件夹,其中包含可执行文件。可以在dist文件夹中找到可执行文件。

pyinstaller 讯飞星火.py

打包结果预览

  • 4
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值