Add new chat and generate_content templates.

Author: MarkDaoust

templates/aistudio_gemini_prompt_chat.ipynb

@@ -0,0 +1,276 @@
+{
+  "cells": [
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "Tce3stUlHN0L"
+      },
+      "source": [
+        "##### Copyright 2023 Google LLC"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "tuOe1ymfHZPu"
+      },
+      "outputs": [],
+      "source": [
+        "#@title Licensed under the Apache License, Version 2.0 (the \"License\");\n",
+        "# you may not use this file except in compliance with the License.\n",
+        "# You may obtain a copy of the License at\n",
+        "#\n",
+        "# https://www.apache.org/licenses/LICENSE-2.0\n",
+        "#\n",
+        "# Unless required by applicable law or agreed to in writing, software\n",
+        "# distributed under the License is distributed on an \"AS IS\" BASIS,\n",
+        "# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n",
+        "# See the License for the specific language governing permissions and\n",
+        "# limitations under the License."
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "source": [
+        "## Setup"
+      ],
+      "metadata": {
+        "id": "FKwyTRdwB8aW"
+      }
+    },
+    {
+      "cell_type": "markdown",
+      "source": [
+        "### Install & import\n"
+      ],
+      "metadata": {
+        "id": "rlE8UqxrDIez"
+      }
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "!pip install google-generativelanguage"
+      ],
+      "metadata": {
+        "id": "cZiU4TKzznh9"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "kWIuwKG2_oWE"
+      },
+      "outputs": [],
+      "source": [
+        "# Install the client library and import necessary modules.\n",
+        "#!pip install google-generativeai\n",
+        "import google.generativeai as genai\n",
+        "import json\n",
+        "import pathlib\n",
+        "import pprint\n",
+        "import requests\n",
+        "import mimetypes\n",
+        "from IPython.display import Markdown"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "source": [
+        "\n",
+        "\n",
+        "### Mount Google Drive"
+      ],
+      "metadata": {
+        "id": "qZsRPVv1ITbh"
+      }
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "from google.colab import drive\n",
+        "drive.mount('/gdrive')"
+      ],
+      "metadata": {
+        "id": "d9-t_OkGoLIP"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
+    {
+      "cell_type": "markdown",
+      "source": [
+        "## Set the API key"
+      ],
+      "metadata": {
+        "id": "Fet3lFjdKHEM"
+      }
+    },
+    {
+      "cell_type": "markdown",
+      "source": [
+        "Add your API_KEY to the secrets manager in the left pannel \"🔑\"."
+      ],
+      "metadata": {
+        "id": "ZoRWILAtCzBE"
+      }
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "from google.colab import userdata\n",
+        "\n",
+        "API_KEY=userdata.get('API_KEY')"
+      ],
+      "metadata": {
+        "id": "LaLCwNlkCyQd"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "_SvYoR3WCeKr"
+      },
+      "outputs": [],
+      "source": [
+        "# Configure the client library by providing your API key.\n",
+        "genai.configure(api_key=API_KEY)"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "source": [
+        "### Parse the arguments"
+      ],
+      "metadata": {
+        "id": "weo-o73WDpdm"
+      }
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "import json\n",
+        "\n",
+        "model = \"gemini-pro\" # @param {isTemplate: true}\n",
+        "contents = '[{\"role\":\"user\", \"parts\" : [{\"text\": \"hello\"}]}, {\"role\": \"model\", \"parts\": [{\"text\": \"Hello! How can I assist you today?\"}]}]' # @param {isTemplate: true}\n",
+        "generation_config = \"{}\" # @param {isTemplate: true}\n",
+        "safety_settings = \"{}\" # @param {isTemplate: true}\n",
+        "user_input = 'How does electricity work?' #@param {isTemplate: true}\n",
+        "\n",
+        "contents = json.loads(contents)\n",
+        "generation_config = json.loads(generation_config)\n",
+        "safety_settings = json.loads(safety_settings)\n",
+        "\n",
+        "\n",
+        "stream = False"
+      ],
+      "metadata": {
+        "id": "uIog-0SyDuIF"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "contents"
+      ],
+      "metadata": {
+        "id": "wBS8xNhN0x62"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
+    {
+      "cell_type": "markdown",
+      "source": [
+        "### Call the API"
+      ],
+      "metadata": {
+        "id": "E7zAD69vE92b"
+      }
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "LB2LxPmAB95V"
+      },
+      "outputs": [],
+      "source": [
+        "# Call the model and print the response.\n",
+        "gemini = genai.GenerativeModel(model_name=model)\n",
+        "\n",
+        "chat = gemini.start_chat(history=contents)\n",
+        "\n",
+        "response = chat.send_message(\n",
+        "    user_input,\n",
+        "    stream=False)"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "display(Markdown(response.text))"
+      ],
+      "metadata": {
+        "id": "Lm3RXwYuGtZK"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "response.prompt_feedback"
+      ],
+      "metadata": {
+        "id": "JbKuUc3NGxYD"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "response.candidates"
+      ],
+      "metadata": {
+        "id": "SLAaIq3kgwwJ"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
+    {
+      "cell_type": "code",
+      "source": [],
+      "metadata": {
+        "id": "8kS1DntQ2TKM"
+      },
+      "execution_count": null,
+      "outputs": []
+    }
+  ],
+  "metadata": {
+    "colab": {
+      "provenance": [],
+      "private_outputs": true,
+      "collapsed_sections": [
+        "Tce3stUlHN0L"
+      ]
+    },
+    "kernelspec": {
+      "display_name": "Python 3",
+      "name": "python3"
+    }
+  },
+  "nbformat": 4,
+  "nbformat_minor": 0
+}