mirror of
https://github.com/dataforcanada/d4c-service-geo-assistant.git
synced 2026-06-13 22:41:01 +02:00
Wei Ji
2d5af2adb9
* Set asyncio_mode="auto" in pytest.ini config Xref https://pytest-asyncio.readthedocs.io/en/latest/reference/configuration.html#asyncio-mode * Add test-python GitHub Actions CI workflow * Add `--color=yes` flag to force GitHub Actions logs to have color Xref https://github.com/pytest-dev/pytest/issues/7443 * Refactor contents of test_api.py slightly Using some ruff rules * Delete test_api.py * Make test_get_place async * Make get_place async, and call .ainvoke method on it
16 lines
387 B
Python
16 lines
387 B
Python
from langchain_core.tools.base import ToolCall
|
|
|
|
from geo_assistant.tools.overture import get_place
|
|
|
|
|
|
async def test_get_place():
|
|
command = await get_place.ainvoke(
|
|
ToolCall(
|
|
name="get_place",
|
|
type="tool_call",
|
|
id="test_id",
|
|
args={"place_name": "Neighbourhood Cafe Lisbon"},
|
|
)
|
|
)
|
|
assert "place" in command.update
|