AI_Function_Calling.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "57404460-b084-4a84-9734-a1efc75fba76",
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "[nltk_data] Downloading package punkt_tab to /Users/vaibhav.pawar/Libr\n",
      "[nltk_data]     ary/Caches/pypoetry/virtualenvs/genai-A6sECUHx-\n",
      "[nltk_data]     py3.11/lib/python3.11/site-\n",
      "[nltk_data]     packages/llama_index/core/_static/nltk_cache...\n",
      "[nltk_data]   Package punkt_tab is already up-to-date!\n"
     ]
    }
   ],
   "source": [
    "from llama_index.llms.openai import OpenAI\n",
    "from llama_index.core.prompts import ChatMessage, MessageRole\n",
    "from llama_index.core.tools import FunctionTool\n",
    "\n",
    "import dotenv\n",
    "import requests\n",
    "import json\n",
    "\n",
    "from llama_index.agent.openai import OpenAIAgent"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "fce83a3d-50f7-4f7f-884c-739671347d2d",
   "metadata": {},
   "outputs": [],
   "source": [
    "import os\n",
    "\n",
    "os.environ['OPENAI_API_KEY'] = ''"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "0c4e94cb-92fa-4255-877e-5d00a62f4914",
   "metadata": {},
   "outputs": [],
   "source": [
    "systemPrompt = \"\"\"\n",
    "You are a Chatbot \n",
    "\"\"\"\n",
    "sysMessage = ChatMessage(role=MessageRole.SYSTEM, content = systemPrompt)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "id": "c6efb304-1a64-42fe-b33e-1882b361652e",
   "metadata": {},
   "outputs": [],
   "source": [
    "import os\n",
    "os.environ['OPENAI_API_KEY'] = ''\n",
    "\n",
    "llm = OpenAI(model=\"gpt-4o-mini\", temperature=0)\n",
    "agent = OpenAIAgent.from_tools(llm=llm,tools=[], prefix_messages=[sysMessage], verbose=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "id": "07157cb6-cbc3-4af4-a6c6-da9f6c2df18f",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Added user message to memory: hey\n",
      "Hello! How can I assist you today?\n"
     ]
    }
   ],
   "source": [
    "print(agent.chat('hey').response)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "af23c1f3-75c8-4d85-a2f0-a3cb3cac01eb",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Added user message to memory: What is (121 * 3)/97 + (42+12)*71?\n",
      "Let's break down the calculation step by step.\n",
      "\n",
      "1. Calculate \\( 121 \\times 3 \\):\n",
      "   \\[\n",
      "   121 \\times 3 = 363\n",
      "   \\]\n",
      "\n",
      "2. Divide that result by 97:\n",
      "   \\[\n",
      "   \\frac{363}{97} \\approx 3.7433\n",
      "   \\]\n",
      "\n",
      "3. Calculate \\( 42 + 12 \\):\n",
      "   \\[\n",
      "   42 + 12 = 54\n",
      "   \\]\n",
      "\n",
      "4. Multiply that result by 71:\n",
      "   \\[\n",
      "   54 \\times 71 = 3834\n",
      "   \\]\n",
      "\n",
      "5. Now, add the two results together:\n",
      "   \\[\n",
      "   3.7433 + 3834 \\approx 3837.7433\n",
      "   \\]\n",
      "\n",
      "So, the final result is approximately \\( 3837.74 \\).\n"
     ]
    }
   ],
   "source": [
    "print(agent.chat(\"What is (121 * 3)/97 + (42+12)*71?\").response)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "3549b763-b5d8-40fb-9da0-5aa429b27613",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Added user message to memory: is this a correct GSTIN : 52AAAAA0000A1Z\n",
      "The format of a GSTIN (Goods and Services Tax Identification Number) in India is typically 15 characters long, consisting of:\n",
      "\n",
      "1. The first two digits represent the state code.\n",
      "2. The next ten characters are the PAN (Permanent Account Number) of the business.\n",
      "3. The 13th character is the entity code.\n",
      "4. The 14th character is usually 'Z' by default.\n",
      "5. The last character is a check digit.\n",
      "\n",
      "In your example, \"52AAAAA0000A1Z\", the first two digits \"52\" correspond to the state code for Goa. The rest of the format appears to follow the structure of a GSTIN. However, to verify if it is a valid GSTIN, you would need to check it against the official GST portal or database, as I cannot verify real-time data or validate specific GSTINs.\n"
     ]
    }
   ],
   "source": [
    "print(agent.chat('is this a correct GSTIN : 52AAAAA0000A1Z').response)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "id": "f339cd91-7383-49d9-83fa-0c52025c3e39",
   "metadata": {},
   "outputs": [],
   "source": [
    "import re\n",
    "\n",
    "def checksum(gst):\n",
    "    \"\"\"\n",
    "    Validates the checksum character of a GSTIN.\n",
    "    \n",
    "    Args:\n",
    "        gst (str): GSTIN string to validate.\n",
    "\n",
    "    Returns:\n",
    "        tuple: (bool, expected_checksum, provided_checksum)\n",
    "               - bool indicates if the checksum matches.\n",
    "               - expected_checksum is the calculated checksum.\n",
    "               - provided_checksum is the checksum from the input GSTIN.\n",
    "    \"\"\"\n",
    "    check = gst[-1]  # Extract the provided checksum.\n",
    "    gst = gst[:-1]   # Remove the checksum character for validation.\n",
    "    \n",
    "    # Map each character to its numeric value (digits as is, letters as position in alphabet + 10)\n",
    "    l = [int(c) if c.isdigit() else ord(c) - 55 for c in gst]\n",
    "    \n",
    "    # Multiply each value alternately by 1 or 2\n",
    "    l = [val * (ind % 2 + 1) for ind, val in enumerate(l)]\n",
    "\n",
    "    # Convert each product to its sum of digits within the base-36 system\n",
    "    l = [(x // 36 + x % 36) for x in l]\n",
    "\n",
    "    # Calculate the checksum\n",
    "    csum = (36 - sum(l) % 36) % 36\n",
    "    csum = str(csum) if csum < 10 else chr(csum + 55)  # Convert checksum to base-36 character\n",
    "\n",
    "    return (check == csum, csum, check)  # Return whether the checksum matches\n",
    "\n",
    "def validate_pan(gstin):\n",
    "    \"\"\"\n",
    "    Validates the PAN part of a GSTIN.\n",
    "    \n",
    "    Args:\n",
    "        gstin (str): GSTIN string to validate.\n",
    "\n",
    "    Returns:\n",
    "        bool: True if the PAN is valid, False otherwise.\n",
    "    \"\"\"\n",
    "    # Extract the PAN part (6th to 15th characters inclusive)\n",
    "    pan = gstin[2:12]\n",
    "\n",
    "    # PAN should have 10 characters: first 5 alphabets, next 4 digits, last alphabet\n",
    "    pan_pattern = r\"^[A-Z]{5}[0-9]{4}[A-Z]$\"\n",
    "\n",
    "    return bool(re.match(pan_pattern, pan))\n",
    "\n",
    "def validate_statecode(gstin):\n",
    "    \"\"\"\n",
    "    Validates the state code part of a GSTIN.\n",
    "    \n",
    "    Args:\n",
    "        gstin (str): GSTIN string to validate.\n",
    "\n",
    "    Returns:\n",
    "        bool: True if the state code is valid, False otherwise.\n",
    "    \"\"\"\n",
    "    # Extract the state code (first two characters)\n",
    "    state_code = gstin[:2]\n",
    "\n",
    "    # State code should be two digits\n",
    "    state_code_pattern = r\"^[0-9]{2}$\"\n",
    "\n",
    "    return bool(re.match(state_code_pattern, state_code))\n",
    "\n",
    "def validate_gstin(gstin):\n",
    "    \"\"\"\n",
    "    Validates the entire GSTIN format.\n",
    "    \n",
    "    Args:\n",
    "        gstin (str): GSTIN string to validate.\n",
    "\n",
    "    Returns:\n",
    "        bool: True if the GSTIN is valid, False otherwise.\n",
    "    \"\"\"\n",
    "    # GSTIN format: [state code (2 digits), PAN (10 characters), digit (1), 'Z', checksum (1 character)]\n",
    "    gstin_pattern = r\"^[0-9]{2}[A-Z]{5}[0-9]{4}[A-Z][0-9Z][A-Z0-9]$\"\n",
    "\n",
    "    if not re.match(gstin_pattern, gstin):\n",
    "        return False  # Basic format validation failed\n",
    "\n",
    "    # Validate individual components\n",
    "    if not validate_statecode(gstin):\n",
    "        return False\n",
    "\n",
    "    if not validate_pan(gstin):\n",
    "        return False\n",
    "\n",
    "    # Validate checksum\n",
    "    is_valid, _, _ = checksum(gstin)\n",
    "    return is_valid\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "id": "7bfe805c-a45f-4f76-a7f6-10c0d19c65e8",
   "metadata": {},
   "outputs": [],
   "source": [
    "validateGSTIN = FunctionTool.from_defaults(fn=validate_gstin)\n",
    "checkSum = FunctionTool.from_defaults(fn=checksum)\n",
    "validatePAN = FunctionTool.from_defaults(fn=validate_pan)\n",
    "validateStateCode = FunctionTool.from_defaults(fn=validate_statecode)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "id": "296ea90d-c293-4e15-bbdd-a5d8c57bb749",
   "metadata": {},
   "outputs": [],
   "source": [
    "agent = OpenAIAgent.from_tools(llm=llm,tools=[validateGSTIN, checkSum, validatePAN , validateStateCode], prefix_messages=[sysMessage], verbose=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "id": "f287f026-7a4b-4507-9a2a-e9ed9a6f5080",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Added user message to memory: is this a correct GSTIN : 52AAAAA0000A1Z9\n",
      "=== Calling Function ===\n",
      "Calling function: validate_gstin with args: {\"gstin\":\"52AAAAA0000A1Z9\"}\n",
      "Got output: False\n",
      "========================\n",
      "\n",
      "The GSTIN \"52AAAAA0000A1Z9\" is not valid.\n"
     ]
    }
   ],
   "source": [
    "print(agent.chat('is this a correct GSTIN : 52AAAAA0000A1Z9').response)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "0d652e7a-1914-4be4-8cea-d4c0192b73c0",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.11.11"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}