From ccac970df329e2c4d20d62fd33bbff5032a2396e Mon Sep 17 00:00:00 2001 From: abdulmeral Date: Sun, 13 Oct 2024 20:47:51 +0300 Subject: [PATCH 1/2] week2 task1 --- week2/task1.ipynb | 366 ++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 366 insertions(+) create mode 100644 week2/task1.ipynb diff --git a/week2/task1.ipynb b/week2/task1.ipynb new file mode 100644 index 0000000..f278505 --- /dev/null +++ b/week2/task1.ipynb @@ -0,0 +1,366 @@ +{ + "cells": [ + { + "cell_type": "markdown", + "id": "ddfa9ae6-69fe-444a-b994-8c4c5970a7ec", + "metadata": {}, + "source": [ + "# Project - Airline AI Assistant with Booking Tool\n", + "\n", + "We'll now bring together what we've learned to make an AI Customer Support assistant for an Airline" + ] + }, + { + "cell_type": "code", + "execution_count": 1, + "id": "8b50bbe2-c0b1-49c3-9a5c-1ba7efa2bcb4", + "metadata": {}, + "outputs": [], + "source": [ + "# imports\n", + "\n", + "import os\n", + "import json\n", + "from dotenv import load_dotenv\n", + "from openai import OpenAI\n", + "import gradio as gr\n", + "import pandas as pd" + ] + }, + { + "cell_type": "code", + "execution_count": 2, + "id": "747e8786-9da8-4342-b6c9-f5f69c2e22ae", + "metadata": {}, + "outputs": [], + "source": [ + "# Initialization\n", + "\n", + "load_dotenv()\n", + "os.environ['OPENAI_API_KEY'] = os.getenv('OPENAI_API_KEY', 'your-key-if-not-using-env')\n", + "MODEL = \"gpt-4o-mini\"\n", + "openai = OpenAI()" + ] + }, + { + "cell_type": "code", + "execution_count": 3, + "id": "0a521d84-d07c-49ab-a0df-d6451499ed97", + "metadata": {}, + "outputs": [], + "source": [ + "system_message = \"You are a helpful assistant for an Airline called FlightAI. \"\n", + "system_message += \"Give short, courteous answers, no more than 1 sentence. \"\n", + "system_message += \"Always be accurate. If you don't know the answer, say so.\"" + ] + }, + { + "cell_type": "markdown", + "id": "36bedabf-a0a7-4985-ad8e-07ed6a55a3a4", + "metadata": {}, + "source": [ + "## Tools\n", + "\n", + "* Price and Booking" + ] + }, + { + "cell_type": "code", + "execution_count": 4, + "id": "0696acb1-0b05-4dc2-80d5-771be04f1fb2", + "metadata": {}, + "outputs": [], + "source": [ + "# Let's start by making a useful function\n", + "\n", + "ticket_prices = {\"london\": \"$799\", \"paris\": \"$899\", \"tokyo\": \"$1400\", \"berlin\": \"$499\"}\n", + "\n", + "def get_ticket_price(destination_city):\n", + " print(f\"Tool get_ticket_price called for {destination_city}\")\n", + " city = destination_city.lower()\n", + " return ticket_prices.get(city, \"Unknown\")" + ] + }, + { + "cell_type": "code", + "execution_count": 5, + "id": "4afceded-7178-4c05-8fa6-9f2085e6a344", + "metadata": {}, + "outputs": [], + "source": [ + "# There's a particular dictionary structure that's required to describe our function:\n", + "\n", + "price_function = {\n", + " \"name\": \"get_ticket_price\",\n", + " \"description\": \"Get the price of a return ticket to the destination city. Call this whenever you need to know the ticket price, for example when a customer asks 'How much is a ticket to this city'\",\n", + " \"parameters\": {\n", + " \"type\": \"object\",\n", + " \"properties\": {\n", + " \"destination_city\": {\n", + " \"type\": \"string\",\n", + " \"description\": \"The city that the customer wants to travel to\",\n", + " },\n", + " },\n", + " \"required\": [\"destination_city\"],\n", + " \"additionalProperties\": False\n", + " }\n", + "}" + ] + }, + { + "cell_type": "code", + "execution_count": 6, + "id": "6996ab93", + "metadata": {}, + "outputs": [], + "source": [ + "# \"required\": [\"destination_city\", \"num_tickets\", \"ticket_class\", \"mail_address\"]\n", + "\n", + "booking_function = {\n", + " \"name\": \"book_ticket\",\n", + " \"description\": \"Book a ticket to a destination city.\",\n", + " \"parameters\": {\n", + " \"type\": \"object\",\n", + " \"properties\": {\n", + " \"destination_city\": {\n", + " \"type\": \"string\",\n", + " \"description\": \"The city the customer wants to travel to\",\n", + " },\n", + " \"num_tickets\": {\n", + " \"type\": \"integer\",\n", + " \"description\": \"The number of tickets to book\",\n", + " },\n", + " \"ticket_class\": {\n", + " \"type\": \"string\",\n", + " \"description\": \"The class of the ticket (e.g., economy, business)\",\n", + " },\n", + "\n", + " \"mail_address\": {\n", + " \"type\": \"string\",\n", + " \"description\": \"Mail address to send the ticket to\",\n", + " },\n", + " },\n", + " \"required\": [\"destination_city\", \"num_tickets\", \"ticket_class\", \"mail_address\"],\n", + " \"additionalProperties\": False\n", + " }\n", + "}" + ] + }, + { + "cell_type": "code", + "execution_count": 7, + "id": "6d62733e", + "metadata": {}, + "outputs": [], + "source": [ + "# And this is included in a list of tools:\n", + "\n", + "tools = [{\"type\": \"function\", \"function\": price_function},\n", + " {\"type\": \"function\", \"function\": booking_function}]" + ] + }, + { + "cell_type": "markdown", + "id": "c3d3554f-b4e3-4ce7-af6f-68faa6dd2340", + "metadata": {}, + "source": [ + "## Creating Application" + ] + }, + { + "cell_type": "code", + "execution_count": 8, + "id": "f671dbb3", + "metadata": {}, + "outputs": [], + "source": [ + "# Creating ID for each booking\n", + "# Produces IDs like 'BK-000001', 'BK-000002'\n", + "\n", + "booking_counter = 0\n", + "\n", + "def generate_unique_booking_id():\n", + " global booking_counter\n", + " booking_counter += 1\n", + " return f\"BK-{booking_counter:06d}\" \n" + ] + }, + { + "cell_type": "code", + "execution_count": 9, + "id": "77ae8c75", + "metadata": {}, + "outputs": [], + "source": [ + "# Temporary storage for bookings\n", + "bookingDB = pd.DataFrame()" + ] + }, + { + "cell_type": "code", + "execution_count": 10, + "id": "b0992986-ea09-4912-a076-8e5603ee631f", + "metadata": {}, + "outputs": [], + "source": [ + "# We have to write that function handle_tool_call:\n", + "\n", + "def handle_tool_call(message):\n", + " global bookingDB\n", + " tool_call = message.tool_calls[0]\n", + " if tool_call.function.name == \"get_ticket_price\":\n", + " arguments = json.loads(tool_call.function.arguments)\n", + " destination_city = arguments.get('destination_city')\n", + " price = get_ticket_price(destination_city)\n", + " response = {\n", + " \"role\": \"tool\",\n", + " \"content\": json.dumps({\"destination_city\": destination_city,\"price\": price}),\n", + " \"tool_call_id\": message.tool_calls[0].id\n", + " }\n", + " return response\n", + " \n", + " elif tool_call.function.name == \"book_ticket\":\n", + " arguments = json.loads(tool_call.function.arguments)\n", + " destination_city = arguments.get('destination_city')\n", + " num_tickets = arguments.get('num_tickets')\n", + " ticket_class = arguments.get('ticket_class')\n", + " mail_address = arguments.get('mail_address')\n", + " response = {\n", + " \"role\": \"tool\",\n", + " \"content\": json.dumps({\"destination_city\": destination_city,\"num_tickets\": num_tickets, \"ticket_class\": ticket_class}),\n", + " \"tool_call_id\": message.tool_calls[0].id\n", + " } \n", + " # Get the ticket price\n", + " price = int(get_ticket_price(destination_city).replace(\"$\", \"\"))\n", + " total_price = price * num_tickets\n", + " # Generate a unique booking ID\n", + " booked_ID = generate_unique_booking_id()\n", + " # Define the booking data\n", + " data = {\n", + " \"booking_id\": [booked_ID],\n", + " \"mail_address\": [mail_address],\n", + " \"destination_city\": [destination_city],\n", + " \"num_tickets\": [num_tickets],\n", + " \"ticket_class\": [ticket_class],\n", + " \"total_price\": [total_price],\n", + " }\n", + " booking_temp = pd.DataFrame(data)\n", + " bookingDB = bookingDB._append(booking_temp)\n", + " # Can update with real booking system\n", + " bookingDB.to_csv('bookingDB.csv', index=False)\n", + " return response" + ] + }, + { + "cell_type": "code", + "execution_count": 11, + "id": "ce9b0744-9c78-408d-b9df-9f6fd9ed78cf", + "metadata": {}, + "outputs": [], + "source": [ + "def chat(message, history):\n", + " messages = [{\"role\": \"system\", \"content\": system_message}]\n", + " for human, assistant in history:\n", + " messages.append({\"role\": \"user\", \"content\": human})\n", + " messages.append({\"role\": \"assistant\", \"content\": assistant})\n", + " messages.append({\"role\": \"user\", \"content\": message})\n", + " response = openai.chat.completions.create(model=MODEL, messages=messages, tools=tools)\n", + "\n", + " if response.choices[0].finish_reason==\"tool_calls\":\n", + " message = response.choices[0].message\n", + " response = handle_tool_call(message)\n", + " messages.append(message)\n", + " messages.append(response)\n", + " response = openai.chat.completions.create(model=MODEL, messages=messages)\n", + " \n", + " return response.choices[0].message.content" + ] + }, + { + "cell_type": "code", + "execution_count": 12, + "id": "f4be8a71-b19e-4c2f-80df-f59ff2661f14", + "metadata": {}, + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "Running on local URL: http://127.0.0.1:7860\n", + "\n", + "To create a public link, set `share=True` in `launch()`.\n" + ] + }, + { + "data": { + "text/html": [ + "
" + ], + "text/plain": [ + "" + ] + }, + "metadata": {}, + "output_type": "display_data" + }, + { + "data": { + "text/plain": [] + }, + "execution_count": 12, + "metadata": {}, + "output_type": "execute_result" + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "/Users/abdul.meral/Desktop/courses/llm2/venv/lib/python3.10/site-packages/gradio/analytics.py:106: UserWarning: IMPORTANT: You are using gradio version 4.44.1, however version 5.0.1 is available, please upgrade. \n", + "--------\n", + " warnings.warn(\n" + ] + }, + { + "name": "stdout", + "output_type": "stream", + "text": [ + "Tool get_ticket_price called for Paris\n", + "Tool get_ticket_price called for Paris\n" + ] + } + ], + "source": [ + "gr.ChatInterface(fn=chat).launch()" + ] + }, + { + "cell_type": "markdown", + "id": "11c9da69-d0cf-4cf2-a49e-e5669deec47b", + "metadata": {}, + "source": [ + "## Enjoy, Happy Learning Thanks to Ed!" + ] + } + ], + "metadata": { + "kernelspec": { + "display_name": "Python 3", + "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.10.15" + } + }, + "nbformat": 4, + "nbformat_minor": 5 +} From 684242f8d0d5f266567dea6f3075d8ad68271fb3 Mon Sep 17 00:00:00 2001 From: Edward Donner Date: Mon, 28 Oct 2024 10:08:00 -0400 Subject: [PATCH 2/2] Added in student contribution --- .../{ => community-contributions}/task1.ipynb | 81 +++++-------------- 1 file changed, 19 insertions(+), 62 deletions(-) rename week2/{ => community-contributions}/task1.ipynb (85%) diff --git a/week2/task1.ipynb b/week2/community-contributions/task1.ipynb similarity index 85% rename from week2/task1.ipynb rename to week2/community-contributions/task1.ipynb index f278505..9ca08a3 100644 --- a/week2/task1.ipynb +++ b/week2/community-contributions/task1.ipynb @@ -5,6 +5,10 @@ "id": "ddfa9ae6-69fe-444a-b994-8c4c5970a7ec", "metadata": {}, "source": [ + "# Student Contribution\n", + "\n", + "An awesome variation that includes a tool to make a booking! Thank you! -- Ed\n", + "\n", "# Project - Airline AI Assistant with Booking Tool\n", "\n", "We'll now bring together what we've learned to make an AI Customer Support assistant for an Airline" @@ -12,7 +16,7 @@ }, { "cell_type": "code", - "execution_count": 1, + "execution_count": null, "id": "8b50bbe2-c0b1-49c3-9a5c-1ba7efa2bcb4", "metadata": {}, "outputs": [], @@ -29,7 +33,7 @@ }, { "cell_type": "code", - "execution_count": 2, + "execution_count": null, "id": "747e8786-9da8-4342-b6c9-f5f69c2e22ae", "metadata": {}, "outputs": [], @@ -44,7 +48,7 @@ }, { "cell_type": "code", - "execution_count": 3, + "execution_count": null, "id": "0a521d84-d07c-49ab-a0df-d6451499ed97", "metadata": {}, "outputs": [], @@ -66,7 +70,7 @@ }, { "cell_type": "code", - "execution_count": 4, + "execution_count": null, "id": "0696acb1-0b05-4dc2-80d5-771be04f1fb2", "metadata": {}, "outputs": [], @@ -83,7 +87,7 @@ }, { "cell_type": "code", - "execution_count": 5, + "execution_count": null, "id": "4afceded-7178-4c05-8fa6-9f2085e6a344", "metadata": {}, "outputs": [], @@ -109,7 +113,7 @@ }, { "cell_type": "code", - "execution_count": 6, + "execution_count": null, "id": "6996ab93", "metadata": {}, "outputs": [], @@ -148,7 +152,7 @@ }, { "cell_type": "code", - "execution_count": 7, + "execution_count": null, "id": "6d62733e", "metadata": {}, "outputs": [], @@ -169,7 +173,7 @@ }, { "cell_type": "code", - "execution_count": 8, + "execution_count": null, "id": "f671dbb3", "metadata": {}, "outputs": [], @@ -187,7 +191,7 @@ }, { "cell_type": "code", - "execution_count": 9, + "execution_count": null, "id": "77ae8c75", "metadata": {}, "outputs": [], @@ -198,7 +202,7 @@ }, { "cell_type": "code", - "execution_count": 10, + "execution_count": null, "id": "b0992986-ea09-4912-a076-8e5603ee631f", "metadata": {}, "outputs": [], @@ -253,7 +257,7 @@ }, { "cell_type": "code", - "execution_count": 11, + "execution_count": null, "id": "ce9b0744-9c78-408d-b9df-9f6fd9ed78cf", "metadata": {}, "outputs": [], @@ -278,57 +282,10 @@ }, { "cell_type": "code", - "execution_count": 12, + "execution_count": null, "id": "f4be8a71-b19e-4c2f-80df-f59ff2661f14", "metadata": {}, - "outputs": [ - { - "name": "stdout", - "output_type": "stream", - "text": [ - "Running on local URL: http://127.0.0.1:7860\n", - "\n", - "To create a public link, set `share=True` in `launch()`.\n" - ] - }, - { - "data": { - "text/html": [ - "
" - ], - "text/plain": [ - "" - ] - }, - "metadata": {}, - "output_type": "display_data" - }, - { - "data": { - "text/plain": [] - }, - "execution_count": 12, - "metadata": {}, - "output_type": "execute_result" - }, - { - "name": "stderr", - "output_type": "stream", - "text": [ - "/Users/abdul.meral/Desktop/courses/llm2/venv/lib/python3.10/site-packages/gradio/analytics.py:106: UserWarning: IMPORTANT: You are using gradio version 4.44.1, however version 5.0.1 is available, please upgrade. \n", - "--------\n", - " warnings.warn(\n" - ] - }, - { - "name": "stdout", - "output_type": "stream", - "text": [ - "Tool get_ticket_price called for Paris\n", - "Tool get_ticket_price called for Paris\n" - ] - } - ], + "outputs": [], "source": [ "gr.ChatInterface(fn=chat).launch()" ] @@ -344,7 +301,7 @@ ], "metadata": { "kernelspec": { - "display_name": "Python 3", + "display_name": "Python 3 (ipykernel)", "language": "python", "name": "python3" }, @@ -358,7 +315,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.10.15" + "version": "3.11.10" } }, "nbformat": 4,