mirror of
https://github.com/VectifyAI/PageIndex.git
synced 2026-04-24 23:56:21 +02:00
14 lines
476 B
Python
14 lines
476 B
Python
from pageindex.agent import AgentRunner, SYSTEM_PROMPT
|
|
from pageindex.backend.protocol import AgentTools
|
|
|
|
|
|
def test_agent_runner_init():
|
|
tools = AgentTools(function_tools=["mock_tool"])
|
|
runner = AgentRunner(tools=tools, model="gpt-4o")
|
|
assert runner._model == "gpt-4o"
|
|
|
|
|
|
def test_system_prompt_has_tool_instructions():
|
|
assert "list_documents" in SYSTEM_PROMPT
|
|
assert "get_document_structure" in SYSTEM_PROMPT
|
|
assert "get_page_content" in SYSTEM_PROMPT
|