liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Reformat codes

Browse Source
This commit is contained in:
Changhua
2024-02-24 11:16:46 +08:00
parent a74fb6b8bf
commit 388edcd062
1 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
base/func_zhipu.py
View File

@@ -1,16 +1,17 @@
from zhipuai import ZhipuAI from zhipuai import ZhipuAI
class ZhiPu(): class ZhiPu():
def __init__(self, conf: dict) -> None: def __init__(self, conf: dict) -> None:
self.api_key = conf.get("api_key") self.api_key = conf.get("api_key")
self.model = conf.get("model", "glm-4") # 默认使用 glm-4 模型 self.model = conf.get("model", "glm-4") # 默认使用 glm-4 模型
self.client = ZhipuAI(api_key=self.api_key) self.client = ZhipuAI(api_key=self.api_key)
self.converstion_list = {} self.converstion_list = {}
@staticmethod @staticmethod
def value_check(conf: dict) -> bool: def value_check(conf: dict) -> bool:
if conf and conf.get("api_key") : if conf and conf.get("api_key"):
return True return True
return False return False
def __repr__(self): def __repr__(self):
@@ -33,6 +34,7 @@ class ZhiPu():
content = {"role": role, "content": str(msg)} content = {"role": role, "content": str(msg)}
self.converstion_list[wxid].append(content) self.converstion_list[wxid].append(content)
if __name__ == "__main__": if __name__ == "__main__":
from configuration import Config from configuration import Config
config = Config().ZHIPU config = Config().ZHIPU