加入豆包AI :doubao-1-5-lite-32k-250115

This commit is contained in:
liuwei
2025-02-28 16:33:27 +08:00
parent 24502ed18c
commit 06718629e7
7 changed files with 165 additions and 7 deletions

View File

@@ -9,6 +9,7 @@ from threading import Thread
from datetime import datetime, timedelta
import random
from base.func_doubao import Doubao
from base.func_epic import is_friday, get_free
from base.func_zhipu import ZhiPu
@@ -74,6 +75,8 @@ class Robot(Job):
self.chat = ZhiPu(self.config.ZhiPu)
elif chat_type == ChatType.CLAUDE.value and Claude.value_check(self.config.CLAUDE):
self.chat = Claude(self.config.CLAUDE)
elif chat_type == ChatType.DOUBAO.value and Claude.value_check(self.config.DOUBAO):
self.chat = Doubao(self.config.DOUBAO)
else:
self.LOG.warning("未配置模型")
self.chat = None
@@ -92,6 +95,8 @@ class Robot(Job):
self.chat = ZhiPu(self.config.ZhiPu)
elif Claude.value_check(self.config.CLAUDE):
self.chat = Claude(self.config.CLAUDE)
elif Doubao.value_check(self.config.DOUBAO):
self.chat = Doubao(self.config.DOUBAO)
else:
self.LOG.warning("未配置模型")
self.chat = None