mirror of
https://github.com/RYDE-WORK/Langchain-Chatchat.git
synced 2026-01-19 13:23:16 +08:00
03e55e11c4
* move get_default_llm_model from webui to ApiRequest 增加API接口及其测试用例： - /server/get_prompt_template: 获取服务器配置的 prompt 模板 - 增加知识库多线程访问测试用例 支持lite模式：无需安装torch等重依赖，通过在线API实现LLM对话和搜索引擎对话 * fix bug in server.api --------- Co-authored-by: imClumsyPanda <littlepanda0716@gmail.com>
48 lines
1.0 KiB
Python
48 lines
1.0 KiB
Python
import sys
|
|
from pathlib import Path
|
|
root_path = Path(__file__).parent.parent.parent
|
|
sys.path.append(str(root_path))
|
|
|
|
from webui_pages.utils import ApiRequest
|
|
|
|
import pytest
|
|
from pprint import pprint
|
|
from typing import List
|
|
|
|
|
|
api = ApiRequest()
|
|
|
|
|
|
def test_get_default_llm():
|
|
llm = api.get_default_llm_model()
|
|
|
|
print(llm)
|
|
assert isinstance(llm, tuple)
|
|
assert isinstance(llm[0], str) and isinstance(llm[1], bool)
|
|
|
|
|
|
def test_server_configs():
|
|
configs = api.get_server_configs()
|
|
pprint(configs, depth=2)
|
|
|
|
assert isinstance(configs, dict)
|
|
assert len(configs) > 0
|
|
|
|
|
|
def test_list_search_engines():
|
|
engines = api.list_search_engines()
|
|
pprint(engines)
|
|
|
|
assert isinstance(engines, list)
|
|
assert len(engines) > 0
|
|
|
|
|
|
@pytest.mark.parametrize("type", ["llm_chat", "agent_chat"])
|
|
def test_get_prompt_template(type):
|
|
print(f"prompt template for: {type}")
|
|
template = api.get_prompt_template(type=type)
|
|
|
|
print(template)
|
|
assert isinstance(template, str)
|
|
assert len(template) > 0
|