Langchain-Chatchat

1
import base64
2
import datetime
3
import hashlib
4
import hmac
5
from urllib.parse import urlparse
6
from datetime import datetime
7
from time import mktime
8
from urllib.parse import urlencode
9
from wsgiref.handlers import format_date_time
10

11

12
class Ws_Param(object):
13
    # 初始化
14
    def __init__(self, APPID, APIKey, APISecret, Spark_url):
15
        self.APPID = APPID
16
        self.APIKey = APIKey
17
        self.APISecret = APISecret
18
        self.host = urlparse(Spark_url).netloc
19
        self.path = urlparse(Spark_url).path
20
        self.Spark_url = Spark_url
21

22
    # 生成url
23
    def create_url(self):
24
        # 生成RFC1123格式的时间戳
25
        now = datetime.now()
26
        date = format_date_time(mktime(now.timetuple()))
27

28
        # 拼接字符串
29
        signature_origin = "host: " + self.host + "\n"
30
        signature_origin += "date: " + date + "\n"
31
        signature_origin += "GET " + self.path + " HTTP/1.1"
32

33
        # 进行hmac-sha256进行加密
34
        signature_sha = hmac.new(self.APISecret.encode('utf-8'), signature_origin.encode('utf-8'),
35
                                 digestmod=hashlib.sha256).digest()
36

37
        signature_sha_base64 = base64.b64encode(signature_sha).decode(encoding='utf-8')
38

39
        authorization_origin = f'api_key="{self.APIKey}", algorithm="hmac-sha256", headers="host date request-line", signature="{signature_sha_base64}"'
40

41
        authorization = base64.b64encode(authorization_origin.encode('utf-8')).decode(encoding='utf-8')
42

43
        # 将请求的鉴权参数组合为字典
44
        v = {
45
            "authorization": authorization,
46
            "date": date,
47
            "host": self.host
48
        }
49
        # 拼接鉴权参数，生成url
50
        url = self.Spark_url + '?' + urlencode(v)
51
        # 此处打印出建立连接时候的url,参考本demo的时候可取消上方打印的注释，比对相同参数时生成的url与自己代码生成的url是否一致
52
        return url
53

54

55
def gen_params(appid, domain, question, temperature, max_token):
56
    """
57
    通过appid和用户的提问来生成请参数
58
    """
59
    data = {
60
        "header": {
61
            "app_id": appid,
62
            "uid": "1234"
63
        },
64
        "parameter": {
65
            "chat": {
66
                "domain": domain,
67
                "random_threshold": 0.5,
68
                "max_tokens": max_token,
69
                "auditing": "default",
70
                "temperature": temperature,
71
            }
72
        },
73
        "payload": {
74
            "message": {
75
                "text": question
76
            }
77
        }
78
    }
79
    return data
80