Skip to content

Commit 891f357

Browse files
authored
feat: add service toolkit to demo agent (#498)
1 parent 2a2c613 commit 891f357

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

examples/agriculture-demo.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
from langchain_core.runnables import Runnable
2020
from rai.agents.conversational_agent import State, create_conversational_agent
2121
from rai.communication.ros2.connectors import ROS2ARIConnector
22-
from rai.tools.ros2.topics import ROS2TopicsToolkit
22+
from rai.tools.ros2 import ROS2ServicesToolkit, ROS2TopicsToolkit
2323
from rai.tools.time import WaitForSecondsTool
2424
from rai.utils.model_initialization import get_llm_model
2525
from rclpy.callback_groups import ReentrantCallbackGroup
@@ -109,6 +109,7 @@ def main():
109109
system_prompt=SYSTEM_PROMPT,
110110
tools=[
111111
*ROS2TopicsToolkit(connector=connector).get_tools(),
112+
*ROS2ServicesToolkit(connector=connector).get_tools(),
112113
WaitForSecondsTool(),
113114
],
114115
)

0 commit comments

Comments
 (0)