Skip to content

Commit

Permalink
feat: 增加LIKE知识库联网搜索和切换模型能力,新增相关配置项 (#421)
Browse files Browse the repository at this point in the history
  • Loading branch information
destoryD authored Feb 7, 2025
1 parent d4dd447 commit 9c48613
Show file tree
Hide file tree
Showing 2 changed files with 19 additions and 1 deletion.
13 changes: 12 additions & 1 deletion api/answer.py
Original file line number Diff line number Diff line change
Expand Up @@ -237,7 +237,7 @@ class TikuLike(Tiku):
def __init__(self) -> None:
super().__init__()
self.name = 'Like知识库'
self.ver = '1.0.6'
self.ver = '1.0.6' #对应官网API版本
self.query_api = 'https://api.datam.site/search'
self.balance_api = 'https://api.datam.site/balance'
self.homepage = 'https://www.datam.site'
Expand Down Expand Up @@ -307,8 +307,19 @@ def load_token(self):
token = self._conf['tokens'].split(',')[-1] if ',' in self._conf['tokens'] else self._conf['tokens']
self._token = token

def load_config(self):
var_params = {"likeapi_search":self._search,"likeapi_model":self._model}
config_params = {"likeapi_search":False, "likeapi_model":None}

for k,v in config_params.items():
if k in self._conf:
var_params[k] = self._conf[k]
else:
var_params[k] = v

def _init_tiku(self):
self.load_token()
self.load_config()
self.update_times()

class TikuAdapter(Tiku):
Expand Down
7 changes: 7 additions & 0 deletions config_template.ini
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,14 @@ provider=TikuYanxi
; 对于那些需要解锁的章节,你必须要提交章节检测才能继续下一章节的学习,自行决定是否开启
submit=false
; 用于言溪题库的TOKEN,同样使用英文逗号隔开多个,会按顺序去使用
; 或用于LIKE知识库的TOKEN,在使用LIKE知识库时仅会调用最后一个TOKEN,请注意!
tokens=
; 下面是用于LIKE知识库模型的专属配置,其他题库无需关注下列选项
; likeapi_search=true表示启用模型的联网搜索功能,false则不启用联网搜索
; likeapi_model=deepseekv3表示使用deepseekv3模型,其他模型请自行查询
; 支持模型列表:https://www.datam.site/doc/api_doc.html#%E6%A8%A1%E5%9E%8B%E6%94%AF%E6%8C%81%E5%88%97%E8%A1%A8
likeapi_search=false
likeapi_model=deepseekv3
; 用于TikuAdapter题库的url
url=
; 用于AI大模型答题的API Endpoint和Key
Expand Down

0 comments on commit 9c48613

Please sign in to comment.