mirror of
https://github.com/VectifyAI/PageIndex.git
synced 2026-04-24 23:56:21 +02:00
26 lines
776 B
Python
26 lines
776 B
Python
from pageindex.events import QueryEvent
|
|
from pageindex.backend.protocol import AgentTools
|
|
|
|
|
|
def test_query_event():
|
|
event = QueryEvent(type="answer_delta", data="hello")
|
|
assert event.type == "answer_delta"
|
|
assert event.data == "hello"
|
|
|
|
|
|
def test_query_event_types():
|
|
for t in ["reasoning", "tool_call", "tool_result", "answer_delta", "answer_done"]:
|
|
event = QueryEvent(type=t, data="test")
|
|
assert event.type == t
|
|
|
|
|
|
def test_agent_tools_default_empty():
|
|
tools = AgentTools()
|
|
assert tools.function_tools == []
|
|
assert tools.mcp_servers == []
|
|
|
|
|
|
def test_agent_tools_with_values():
|
|
tools = AgentTools(function_tools=["tool1"], mcp_servers=["server1"])
|
|
assert len(tools.function_tools) == 1
|
|
assert len(tools.mcp_servers) == 1
|