mirror of
https://github.com/RYDE-WORK/Langchain-Chatchat.git
synced 2026-01-20 13:59:32 +08:00
d0846f88cc
- 重构 api.py：
- 按模块划分为不同的 router
- 添加 openai 兼容的转发接口，项目默认使用该接口以实现模型负载均衡
- 添加 /tools 接口，可以获取/调用编写的 agent tools
- 移除所有 EmbeddingFuncAdapter，统一改用 get_Embeddings
- 待办：
- /chat/chat 接口改为 openai 兼容
- 添加 /chat/kb_chat 接口，openai 兼容
- 改变 ntlk/knowledge_base/logs 等数据目录位置
31 lines
676 B
Python
31 lines
676 B
Python
import sys
|
|
from pathlib import Path
|
|
sys.path.append(str(Path(__file__).parent.parent.parent))
|
|
|
|
from pprint import pprint
|
|
import requests
|
|
|
|
from server.utils import api_address
|
|
|
|
|
|
api_base_url = f"{api_address()}/tools"
|
|
|
|
def test_tool_list():
|
|
resp = requests.get(api_base_url)
|
|
assert resp.status_code == 200
|
|
data = resp.json()["data"]
|
|
pprint(data)
|
|
assert "calculate" in data
|
|
|
|
|
|
def test_tool_call():
|
|
data = {
|
|
"name": "calculate",
|
|
"kwargs": {"a":1,"b":2,"operator":"+"},
|
|
}
|
|
resp = requests.post(f"{api_base_url}/call", json=data)
|
|
assert resp.status_code == 200
|
|
data = resp.json()["data"]
|
|
pprint(data)
|
|
assert data == 3
|