Skip to content

Commit

Permalink
feat: Add bounded params to the toolbox SDK with temporary dummy data.
Browse files Browse the repository at this point in the history
This is to make the insert_ticket tool work so that the app can actually book a flight.

Note that we are skipping the step of validating a ticket. That would be added later once toolbox supports more features, like custom functions.
  • Loading branch information
anubhav756 committed Nov 15, 2024
1 parent 80261b6 commit a074269
Showing 1 changed file with 10 additions and 1 deletion.
11 changes: 10 additions & 1 deletion llm_demo/orchestrator/langchain_tools/tools.py
Original file line number Diff line number Diff line change
Expand Up @@ -273,7 +273,16 @@ async def list_tickets():

# Tools for agent
async def initialize_tools(client: aiohttp.ClientSession):
toolbox_client = ToolboxClient(BASE_URL, client)
toolbox_client = ToolboxClient(
BASE_URL,
client,
)
# TODO: Remove the hardcoded values and replace with actual values on successful login.
toolbox_client.set_bounded_params({
"user_id": "12345678900",
"user_name": "Someone",
"user_email": "something@somewhere.com",
})
return await toolbox_client.load_toolset()


Expand Down

0 comments on commit a074269

Please sign in to comment.