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

4
base/func_zhipu.py
View File

@@ -1,5 +1,6 @@
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")
@@ -9,7 +10,7 @@ class ZhiPu():
@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
@@ -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