33 lines
692 B
Python
33 lines
692 B
Python
|
import sys
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
sys.path.append(str(Path(__file__).parent.parent.parent))
|
||
|
|
|
||
|
|
from pprint import pprint
|
||
|
|
|
||
|
|
import requests
|
||
|
|
|
||
|
|
from chatchat.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
|