Skip to content

Commit 915d674

Browse files
authored
fix: do not hardcode glm model (#547)
Signed-off-by: Frost Ming <[email protected]>
1 parent 011637e commit 915d674

File tree

2 files changed

+1
-6
lines changed

2 files changed

+1
-6
lines changed

xiaogpt/bot/glm_bot.py

+1-5
Original file line numberDiff line numberDiff line change
@@ -11,13 +11,11 @@
1111

1212
class GLMBot(ChatHistoryMixin, BaseBot):
1313
name = "Chat GLM"
14-
default_options = {"model": "chatglm_turbo"}
14+
default_options = {"model": "glm-4"} # Change glm model here
1515

1616
def __init__(self, glm_key: str) -> None:
1717
from zhipuai import ZhipuAI
1818

19-
self.model = "glm-4" # Change glm model here
20-
2119
self.history = []
2220
self.client = ZhipuAI(api_key=glm_key)
2321

@@ -28,7 +26,6 @@ def from_config(cls, config):
2826
def ask(self, query, **options):
2927
ms = self.get_messages()
3028
kwargs = {**self.default_options, **options}
31-
kwargs["model"] = self.model
3229
ms.append({"role": "user", "content": f"{query}"})
3330
kwargs["messages"] = ms
3431
try:
@@ -45,7 +42,6 @@ def ask(self, query, **options):
4542
async def ask_stream(self, query: str, **options: Any):
4643
ms = self.get_messages()
4744
kwargs = {**self.default_options, **options}
48-
kwargs["model"] = self.model
4945
ms.append({"role": "user", "content": f"{query}"})
5046
kwargs["messages"] = ms
5147
kwargs["stream"] = True

xiaogpt/bot/qwen_bot.py

-1
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@ class QwenBot(ChatHistoryMixin, BaseBot):
1515

1616
def __init__(self, qwen_key: str) -> None:
1717
import dashscope
18-
from dashscope.api_entities.dashscope_response import Role
1918

2019
self.history = []
2120
dashscope.api_key = qwen_key

0 commit comments

Comments
 (0)