{ "cells": [ { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "# Load env variables and create client\n", "from dotenv import load_dotenv\n", "from anthropic import Anthropic\n", "\n", "load_dotenv()\n", "\n", "client = Anthropic()\n", "model = \"claude-3-7-sonnet-latest\"" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "# Helper functions\n", "def add_user_message(messages, text):\n", " user_message = {\"role\": \"user\", \"content\": text}\n", " messages.append(user_message)\n", "\n", "\n", "def add_assistant_message(messages, text):\n", " assistant_message = {\"role\": \"assistant\", \"content\": text}\n", " messages.append(assistant_message)\n", "\n", "\n", "def chat(messages, system=None, temperature=1.0, stop_sequences=[]):\n", " params = {\n", " \"model\": model,\n", " \"max_tokens\": 1000,\n", " \"messages\": messages,\n", " \"temperature\": temperature,\n", " \"stop_sequences\": stop_sequences,\n", " }\n", "\n", " if system:\n", " params[\"system\"] = system\n", "\n", " message = client.messages.create(**params)\n", " return message.content[0].text" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "# Tools and Schemas\n", "\n", "from datetime import datetime, timedelta\n", "\n", "\n", "def add_duration_to_datetime(\n", " datetime_str, duration=0, unit=\"days\", input_format=\"%Y-%m-%d\"\n", "):\n", " date = datetime.strptime(datetime_str, input_format)\n", "\n", " if unit == \"seconds\":\n", " new_date = date + timedelta(seconds=duration)\n", " elif unit == \"minutes\":\n", " new_date = date + timedelta(minutes=duration)\n", " elif unit == \"hours\":\n", " new_date = date + timedelta(hours=duration)\n", " elif unit == \"days\":\n", " new_date = date + timedelta(days=duration)\n", " elif unit == \"weeks\":\n", " new_date = date + timedelta(weeks=duration)\n", " elif unit == \"months\":\n", " month = date.month + duration\n", " year = date.year + month // 12\n", " month = month % 12\n", " if month == 0:\n", " month = 12\n", " year -= 1\n", " day = min(\n", " date.day,\n", " [\n", " 31,\n", " 29\n", " if year % 4 == 0 and (year % 100 != 0 or year % 400 == 0)\n", " else 28,\n", " 31,\n", " 30,\n", " 31,\n", " 30,\n", " 31,\n", " 31,\n", " 30,\n", " 31,\n", " 30,\n", " 31,\n", " ][month - 1],\n", " )\n", " new_date = date.replace(year=year, month=month, day=day)\n", " elif unit == \"years\":\n", " new_date = date.replace(year=date.year + duration)\n", " else:\n", " raise ValueError(f\"Unsupported time unit: {unit}\")\n", "\n", " return new_date.strftime(\"%A, %B %d, %Y %I:%M:%S %p\")\n", "\n", "\n", "def set_reminder(content, timestamp):\n", " print(\n", " f\"----\\nSetting the following reminder for {timestamp}:\\n{content}\\n----\"\n", " )\n", "\n", "\n", "add_duration_to_datetime_schema = {\n", " \"name\": \"add_duration_to_datetime\",\n", " \"description\": \"Adds a specified duration to a datetime string and returns the resulting datetime in a detailed format. This tool converts an input datetime string to a Python datetime object, adds the specified duration in the requested unit, and returns a formatted string of the resulting datetime. It handles various time units including seconds, minutes, hours, days, weeks, months, and years, with special handling for month and year calculations to account for varying month lengths and leap years. The output is always returned in a detailed format that includes the day of the week, month name, day, year, and time with AM/PM indicator (e.g., 'Thursday, April 03, 2025 10:30:00 AM').\",\n", " \"input_schema\": {\n", " \"type\": \"object\",\n", " \"properties\": {\n", " \"datetime_str\": {\n", " \"type\": \"string\",\n", " \"description\": \"The input datetime string to which the duration will be added. This should be formatted according to the input_format parameter.\",\n", " },\n", " \"duration\": {\n", " \"type\": \"number\",\n", " \"description\": \"The amount of time to add to the datetime. Can be positive (for future dates) or negative (for past dates). Defaults to 0.\",\n", " },\n", " \"unit\": {\n", " \"type\": \"string\",\n", " \"description\": \"The unit of time for the duration. Must be one of: 'seconds', 'minutes', 'hours', 'days', 'weeks', 'months', or 'years'. Defaults to 'days'.\",\n", " },\n", " \"input_format\": {\n", " \"type\": \"string\",\n", " \"description\": \"The format string for parsing the input datetime_str, using Python's strptime format codes. For example, '%Y-%m-%d' for ISO format dates like '2025-04-03'. Defaults to '%Y-%m-%d'.\",\n", " },\n", " },\n", " \"required\": [\"datetime_str\"],\n", " },\n", "}\n", "\n", "set_reminder_schema = {\n", " \"name\": \"set_reminder\",\n", " \"description\": \"Creates a timed reminder that will notify the user at the specified time with the provided content. This tool schedules a notification to be delivered to the user at the exact timestamp provided. It should be used when a user wants to be reminded about something specific at a future point in time. The reminder system will store the content and timestamp, then trigger a notification through the user's preferred notification channels (mobile alerts, email, etc.) when the specified time arrives. Reminders are persisted even if the application is closed or the device is restarted. Users can rely on this function for important time-sensitive notifications such as meetings, tasks, medication schedules, or any other time-bound activities.\",\n", " \"input_schema\": {\n", " \"type\": \"object\",\n", " \"properties\": {\n", " \"content\": {\n", " \"type\": \"string\",\n", " \"description\": \"The message text that will be displayed in the reminder notification. This should contain the specific information the user wants to be reminded about, such as 'Take medication', 'Join video call with team', or 'Pay utility bills'.\",\n", " },\n", " \"timestamp\": {\n", " \"type\": \"string\",\n", " \"description\": \"The exact date and time when the reminder should be triggered, formatted as an ISO 8601 timestamp (YYYY-MM-DDTHH:MM:SS) or a Unix timestamp. The system handles all timezone processing internally, ensuring reminders are triggered at the correct time regardless of where the user is located. Users can simply specify the desired time without worrying about timezone configurations.\",\n", " },\n", " },\n", " \"required\": [\"content\", \"timestamp\"],\n", " },\n", "}\n", "\n", "batch_tool_schema = {\n", " \"name\": \"batch_tool\",\n", " \"description\": \"Invoke multiple other tool calls simultaneously\",\n", " \"input_schema\": {\n", " \"type\": \"object\",\n", " \"properties\": {\n", " \"invocations\": {\n", " \"type\": \"array\",\n", " \"description\": \"The tool calls to invoke\",\n", " \"items\": {\n", " \"type\": \"object\",\n", " \"properties\": {\n", " \"name\": {\n", " \"type\": \"string\",\n", " \"description\": \"The name of the tool to invoke\",\n", " },\n", " \"arguments\": {\n", " \"type\": \"string\",\n", " \"description\": \"The arguments to the tool, encoded as a JSON string\",\n", " },\n", " },\n", " \"required\": [\"name\", \"arguments\"],\n", " },\n", " }\n", " },\n", " \"required\": [\"invocations\"],\n", " },\n", "}\n", "\n", "pass" ] } ], "metadata": { "kernelspec": { "display_name": ".venv", "language": "python", "name": "python3" }, "language_info": { "codemirror_mode": { "name": "ipython", "version": 3 }, "file_extension": ".py", "mimetype": "text/x-python", "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.12.9" } }, "nbformat": 4, "nbformat_minor": 2 }