tests/test_graph_tool.py

import asyncio
import sys

if sys.platform == "win32":
    asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())

from app.tools.graph_tool import graph_query_tool


async def main():
    result = await graph_query_tool(query_type="most_connected_entities", limit=5)
    print(result)

asyncio.run(main())