补充requirements.txt,优化代码

This commit is contained in:
yudong
2024-02-20 12:38:33 +08:00
parent 8c564fcb60
commit b5ee4bf449
3 changed files with 6 additions and 9 deletions

View File

@@ -1,19 +1,15 @@
from pyexpat import model
from zhipuai import ZhipuAI
class ZhiPu():
def __init__(self, conf: dict) -> None:
api_key = conf.get("api_key")
model = conf.get("model", "glm-4") # 默认使用 glm-4 模型
self.api_key = api_key
self.model = model
self.client = ZhipuAI(api_key=api_key)
self.api_key = conf.get("api_key")
self.model = conf.get("model", "glm-4") # 默认使用 glm-4 模型
self.client = ZhipuAI(api_key=self.api_key)
self.converstion_list = {}
@staticmethod
def value_check(conf: dict) -> bool:
if conf:
if conf.get("api_key") :
if conf and conf.get("api_key") :
return True
return False