Update llmriddles/llms/chatglm.py
Browse files
llmriddles/llms/chatglm.py
CHANGED
|
@@ -6,7 +6,7 @@ def ask_chatglm(message: str, api_key: str):
|
|
| 6 |
zhipuai.api_key = api_key
|
| 7 |
|
| 8 |
response = zhipuai.model_api.invoke(
|
| 9 |
-
model="
|
| 10 |
prompt=[{
|
| 11 |
"role": "user",
|
| 12 |
"content": message
|
|
|
|
| 6 |
zhipuai.api_key = api_key
|
| 7 |
|
| 8 |
response = zhipuai.model_api.invoke(
|
| 9 |
+
model="chatglm_turbo",
|
| 10 |
prompt=[{
|
| 11 |
"role": "user",
|
| 12 |
"content": message
|