14 lines
490 B
Python
14 lines
490 B
Python
|
from fastapi.testclient import TestClient
|
||
|
|
from app.main import create_app
|
||
|
|
|
||
|
|
def test_tools_ollama_schema():
|
||
|
|
app = create_app()
|
||
|
|
with TestClient(app) as c:
|
||
|
|
r = c.get("/tools/ollama")
|
||
|
|
assert r.status_code == 200
|
||
|
|
data = r.json()
|
||
|
|
assert "tools" in data and isinstance(data["tools"], list)
|
||
|
|
names = [t["function"]["name"] for t in data["tools"] if t.get("function")]
|
||
|
|
assert "mindnet_query" in names
|
||
|
|
assert "mindnet_subgraph" in names
|