mirror of
https://github.com/dataforcanada/d4c-service-geo-assistant.git
synced 2026-06-13 14:31:01 +02:00
feat: add tests (#5)
This commit is contained in:
@@ -0,0 +1,14 @@
|
||||
from langchain_core.tools.base import ToolCall
|
||||
from geo_assistant.tools.overture import get_place
|
||||
|
||||
|
||||
def test_get_place():
|
||||
command = get_place.invoke(
|
||||
ToolCall(
|
||||
name="get_place",
|
||||
type="tool_call",
|
||||
id="test_id",
|
||||
args={"place_name": "Neighboourhood Cafe Lisbon"},
|
||||
)
|
||||
)
|
||||
assert "place" in command.update
|
||||
Reference in New Issue
Block a user