081865733a
## PR Description This PR adds 7 examples. * `call_a_tool_directly_with_auth.py` - Simple example that uses Arcade client to execute a tool that lists Gmail emails * `call_a_tool_directly.py` - Simple example that uses Arcade client to execute a tool that adds two numbers together * `call_a_tool_with_llm.py` - Simple example that uses the LLM api to star the arcade-ai repository * `get_auth_token.py` - Simple example that gets a Google auth token and then calls the Google API * `call_multiple_tools_directly_with_auth.py` - A more involved example that directly calls multiple spotify tools sequentially * `call_multiple_tools_with_llm.py` - A more involved example that uses an llm to call multiple spotify tools sequentially * `simple_chatbot.py` - Simple chatbot that uses arcade tools and has history --------- Co-authored-by: Nate Barbettini <nathanaelb@gmail.com>
91 lines
3.1 KiB
Python
91 lines
3.1 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 start playing similar songs to the currently playing song.
|
|
|
|
Steps:
|
|
1. Get the currently playing song
|
|
2. Get audio features for the currently playing song
|
|
3. Get song recommendations that are similar to the currently playing song
|
|
4. Start playing the recommended songs
|
|
5. Inform the user which recommended 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, user_country_code: str) -> list[dict]:
|
|
"""Use an LLM to execute the sequence of tools to get recommendations and start playback."""
|
|
tools = [
|
|
"Spotify.GetCurrentlyPlaying",
|
|
"Spotify.GetTracksAudioFeatures",
|
|
"Spotify.GetRecommendations",
|
|
"Spotify.StartTracksPlaybackById",
|
|
"Spotify.GetCurrentlyPlaying",
|
|
]
|
|
|
|
messages = [
|
|
{"role": "user", "content": "Get the currently playing song."},
|
|
{"role": "user", "content": "Retrieve its audio features."},
|
|
{
|
|
"role": "user",
|
|
"content": "Get song recommendations similar to it. "
|
|
"Do not include the currently playing song in the recommendations "
|
|
"or any remixed versions of the song. "
|
|
"Also only include tracks that are available in the user's country. "
|
|
f"The current user resides in {user_country_code}.",
|
|
},
|
|
{"role": "user", "content": "Start playing the recommended songs. 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"
|
|
user_country_code = "US"
|
|
|
|
while True:
|
|
history = call_tools_with_llm(openai_client, user_id, user_country_code)
|
|
|
|
print("\nPress Enter to get more recommendations...")
|
|
input()
|