830f6f9f90
Refactor toolkit implementation examples following the removal of two Spotify tools in [PR#196](https://github.com/ArcadeAI/arcade-ai/pull/196), due to Spotify API deprecation announcement.
80 lines
2.5 KiB
Python
80 lines
2.5 KiB
Python
"""
|
|
Example script demonstrating how to call multiple tools (sequentially) using an LLM with authentication.
|
|
|
|
For this example, we are using the prebuilt Spotify toolkit to search for a song, start playing it, and
|
|
get info about the currently playing song.
|
|
|
|
Steps:
|
|
1. Search for the song
|
|
2. Start playing the song
|
|
3. Get info about the currently playing song
|
|
4. Inform the user which song is now playing
|
|
"""
|
|
|
|
import os
|
|
|
|
from openai import OpenAI
|
|
|
|
|
|
def call_tool(client: OpenAI, user_id: str, tool: str, message: dict, history: list[dict]) -> str:
|
|
"""Make a tool call with a specific tool and message."""
|
|
response = client.chat.completions.create(
|
|
messages=[
|
|
*history,
|
|
message,
|
|
],
|
|
model="gpt-4o",
|
|
user=user_id,
|
|
tools=[tool],
|
|
tool_choice="generate",
|
|
)
|
|
return response
|
|
|
|
|
|
def call_tools_with_llm(
|
|
client: OpenAI, user_id: str, song_name: str, artist_name: str
|
|
) -> list[dict]:
|
|
"""Use an LLM to execute the sequence of tools to search for a song and start playback."""
|
|
tools = [
|
|
"Spotify.Search",
|
|
"Spotify.StartTracksPlaybackById",
|
|
"Spotify.GetCurrentlyPlaying",
|
|
]
|
|
|
|
messages = [
|
|
{"role": "user", "content": f"Search for '{song_name}' by {artist_name}' on Spotify."},
|
|
{"role": "user", "content": "Start playing the song. Just one tool call."},
|
|
{"role": "user", "content": "Get the currently playing song."},
|
|
]
|
|
|
|
history = []
|
|
for i in range(len(messages)):
|
|
response = call_tool(client, user_id, tools[i], messages[i], history)
|
|
print("\n\n", response.choices[0].message.content)
|
|
if (
|
|
response.choices[0].tool_authorizations
|
|
and response.choices[0].tool_authorizations[0].get("status") == "pending"
|
|
):
|
|
input("\nPress Enter once you have authorized...")
|
|
response = call_tool(client, user_id, tools[i], messages[i], history)
|
|
history.append(messages[i])
|
|
history.append({"role": "assistant", "content": response.choices[0].message.content})
|
|
|
|
return history
|
|
|
|
|
|
if __name__ == "__main__":
|
|
arcade_api_key = os.environ.get("ARCADE_API_KEY")
|
|
cloud_host = "https://api.arcade-ai.com/v1"
|
|
|
|
openai_client = OpenAI(
|
|
api_key=arcade_api_key,
|
|
base_url=cloud_host,
|
|
)
|
|
|
|
user_id = "you@example.com"
|
|
song_name = input("Enter the song name: ")
|
|
artist_name = input("Enter the artist name: ")
|
|
|
|
history = call_tools_with_llm(openai_client, user_id, song_name, artist_name)
|
|
print("\n\n", history)
|