generate_llm_function.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "fb4c86b0-8b97-40c8-8538-ef61b5d9aee2",
   "metadata": {},
   "source": [
    "<div style=\"\n",
    "    display: block;\n",
    "    padding: 12px 20px;\n",
    "    background-color: #1A73E8;\n",
    "    color: white;\n",
    "    border-radius: 30px;\n",
    "    font-family: 'Helvetica Neue', Arial, sans-serif;\n",
    "    font-size: 16px;\n",
    "    font-weight: 600;\n",
    "    margin: 15px auto;\n",
    "    width: fit-content;\n",
    "    box-shadow: 0 4px 8px rgba(0, 0, 0, 0.1);\n",
    "    text-align: center;\n",
    "    letter-spacing: 0.5px;\n",
    "\">\n",
    "    <strong>Text Generation with Various AI Models</strong>\n",
    "</div>\n",
    "\n",
    "<div style=\"\n",
    "    display: block;\n",
    "    padding: 12px 20px;\n",
    "    background-color: #66BB6A;\n",
    "    color: white;\n",
    "    border-radius: 30px;\n",
    "    font-family: 'Helvetica Neue', Arial, sans-serif;\n",
    "    font-size: 16px;\n",
    "    font-weight: 600;\n",
    "    margin: 15px auto;\n",
    "    width: fit-content;\n",
    "    box-shadow: 0 4px 8px rgba(0, 0, 0, 0.1);\n",
    "    text-align: center;\n",
    "    letter-spacing: 0.5px;\n",
    "\">\n",
    "    <strong>Simon-Pierre Boucher</strong>\n",
    "</div>\n",
    "\n",
    "<div style=\"\n",
    "    display: block;\n",
    "    padding: 12px 20px;\n",
    "    background-color: #FFA726;\n",
    "    color: white;\n",
    "    border-radius: 30px;\n",
    "    font-family: 'Helvetica Neue', Arial, sans-serif;\n",
    "    font-size: 16px;\n",
    "    font-weight: 600;\n",
    "    margin: 15px auto;\n",
    "    width: fit-content;\n",
    "    box-shadow: 0 4px 8px rgba(0, 0, 0, 0.1);\n",
    "    text-align: center;\n",
    "    letter-spacing: 0.5px;\n",
    "\">\n",
    "    <strong>2024-09-14</strong>\n",
    "</div>\n"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "1a0a9a01-e2f8-435b-842c-7d7c5913fbd2",
   "metadata": {},
   "source": [
    "This Python script generates text using multiple language models from OpenAI, Anthropic, and Mistral. Here's a breakdown of its components:\n",
    "\n",
    "### 1. **Environment Setup**:\n",
    "   - **`load_dotenv()`**: This loads environment variables (API keys) from a `.env` file, which securely stores sensitive data like API keys for OpenAI, Anthropic, and Mistral.\n",
    "\n",
    "### 2. **Text Generation Functions**:\n",
    "   - **`openai_generate_text()`**:\n",
    "     - This function sends a prompt to OpenAI's models (e.g., `gpt-4`).\n",
    "     - The function uses HTTP POST to communicate with OpenAI’s API, passing the prompt and other parameters like temperature, max tokens, and stop sequences.\n",
    "     - It returns the generated response or an error message if the request fails.\n",
    "   \n",
    "   - **`anthropic_generate_text()`**:\n",
    "     - Similar to OpenAI, this function sends a prompt to Anthropic's models (e.g., `claude-3-5-sonnet`).\n",
    "     - It also handles HTTP POST requests and returns the generated text from the Anthropic API or an error message if it fails.\n",
    "\n",
    "   - **`run_mistral()`**:\n",
    "     - This function sends a prompt to Mistral’s API (e.g., `mistral-medium-latest`) and retrieves the generated response.\n",
    "     - It includes configurable parameters like temperature, max tokens, and others, and processes the response returned from the Mistral API.\n",
    "   \n",
    "   - **`mistral_generate_text()`**:\n",
    "     - This function builds the prompt specifically for Mistral and calls `run_mistral()` to generate a response.\n",
    "\n",
    "### 3. **Aggregated Text Generation**:\n",
    "   - **`generate_text_with_all_models()`**:\n",
    "     - This function takes a prompt and loops through a list of OpenAI, Anthropic, and Mistral models to generate responses from each one.\n",
    "     - It stores the generated results from each model in a dictionary, keyed by the model name.\n",
    "     - The responses are generated using the three different APIs (OpenAI, Anthropic, Mistral) by calling the respective functions.\n",
    "\n",
    "### 4. **Main Program**:\n",
    "   - **API Keys and Prompt Setup**:\n",
    "     - API keys for OpenAI, Anthropic, and Mistral are loaded from environment variables.\n",
    "     - The user prompt in this case is to \"Create a simple, healthy recipe using chicken breast, spinach, and quinoa.\"\n",
    "\n",
    "   - **Model Lists**:\n",
    "     - Lists of models for OpenAI (`gpt-3.5-turbo`, `gpt-4`), Anthropic (`claude-3-5-sonnet`), and Mistral (`open-mistral-7b`, `mistral-medium-latest`) are defined.\n",
    "   \n",
    "   - **Generating Text**:\n",
    "     - The function `generate_text_with_all_models()` is called to generate responses from all models for the given prompt.\n",
    "\n",
    "   - **Results Output**:\n",
    "     - The results are printed for each model, showing the model name, word count, and the generated text.\n",
    "\n",
    "### Purpose:\n",
    "This script is used to compare outputs from different AI models for a single prompt, allowing the user to evaluate how different models from OpenAI, Anthropic, and Mistral respond to the same input. It helps in cross-model evaluation and benchmarking."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "8bb27245-b996-43bf-b21f-6a8aaf9cde2a",
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "/Users/simon-pierreboucher/Desktop/notebook/venv/lib/python3.9/site-packages/urllib3/__init__.py:35: NotOpenSSLWarning: urllib3 v2 only supports OpenSSL 1.1.1+, currently the 'ssl' module is compiled with 'LibreSSL 2.8.3'. See: https://github.com/urllib3/urllib3/issues/3020\n",
      "  warnings.warn(\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "True"
      ]
     },
     "execution_count": 1,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import os\n",
    "from dotenv import load_dotenv\n",
    "import requests\n",
    "import json\n",
    "\n",
    "# Charger les variables d'environnement\n",
    "load_dotenv()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "cd3e7b29-1067-4dc3-b6eb-0d8d70afd458",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Fonction pour générer du texte avec OpenAI\n",
    "def openai_generate_text(api_key, prompt, model, temperature=0.7, max_tokens=1024, stop=None):\n",
    "    headers = {\n",
    "        \"Content-Type\": \"application/json\",\n",
    "        \"Authorization\": f\"Bearer {api_key}\"\n",
    "    }\n",
    "\n",
    "    data = {\n",
    "        \"model\": model,\n",
    "        \"messages\": [\n",
    "            {\"role\": \"user\", \"content\": prompt}\n",
    "        ],\n",
    "        \"temperature\": temperature,\n",
    "        \"max_tokens\": max_tokens\n",
    "    }\n",
    "    \n",
    "    if stop:\n",
    "        data[\"stop\"] = stop\n",
    "\n",
    "    response = requests.post(\"https://api.openai.com/v1/chat/completions\", headers=headers, data=json.dumps(data))\n",
    "\n",
    "    if response.status_code == 200:\n",
    "        response_json = response.json()\n",
    "        generated_text = response_json[\"choices\"][0][\"message\"][\"content\"].strip()\n",
    "        return generated_text\n",
    "    else:\n",
    "        return f\"Error {response.status_code}: {response.text}\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "011237f6-7354-45b2-bdcf-59cf717abc48",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Fonction pour générer du texte avec Anthropic\n",
    "def anthropic_generate_text(api_key, prompt, model=\"claude-3-5-sonnet-20240620\", max_tokens=1024, temperature=0.7):\n",
    "    url = \"https://api.anthropic.com/v1/messages\"\n",
    "    \n",
    "    headers = {\n",
    "        \"x-api-key\": api_key,\n",
    "        \"anthropic-version\": \"2023-06-01\",\n",
    "        \"content-type\": \"application/json\"\n",
    "    }\n",
    "\n",
    "    data = {\n",
    "        \"model\": model,\n",
    "        \"max_tokens\": max_tokens,\n",
    "        \"temperature\": temperature,\n",
    "        \"messages\": [\n",
    "            {\"role\": \"user\", \"content\": prompt}\n",
    "        ]\n",
    "    }\n",
    "\n",
    "    response = requests.post(url, headers=headers, data=json.dumps(data))\n",
    "\n",
    "    if response.status_code == 200:\n",
    "        response_json = response.json()\n",
    "        generated_text = response_json[\"content\"][0][\"text\"].strip()\n",
    "        return generated_text\n",
    "    else:\n",
    "        return f\"Error {response.status_code}: {response.text}\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "174052a1-6680-430f-bd6e-0c52bd1955c0",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Fonction pour générer du texte avec Mistral\n",
    "def run_mistral(api_key, user_message, model=\"mistral-medium-latest\"):\n",
    "    url = \"https://api.mistral.ai/v1/chat/completions\"\n",
    "    \n",
    "    headers = {\n",
    "        \"Content-Type\": \"application/json\",\n",
    "        \"Authorization\": f\"Bearer {api_key}\"\n",
    "    }\n",
    "\n",
    "    data = {\n",
    "        \"model\": model,\n",
    "        \"messages\": [\n",
    "            {\"role\": \"user\", \"content\": user_message}\n",
    "        ],\n",
    "        \"temperature\": 0.7,\n",
    "        \"top_p\": 1.0,\n",
    "        \"max_tokens\": 1024,\n",
    "        \"stream\": False,\n",
    "        \"safe_prompt\": False,\n",
    "        \"random_seed\": 1337\n",
    "    }\n",
    "\n",
    "    response = requests.post(url, headers=headers, data=json.dumps(data))\n",
    "\n",
    "    if response.status_code == 200:\n",
    "        response_json = response.json()\n",
    "        return response_json[\"choices\"][0][\"message\"][\"content\"].strip()\n",
    "    else:\n",
    "        return f\"Error {response.status_code}: {response.text}\"\n",
    "\n",
    "def mistral_generate_text(api_key, prompt, model=\"mistral-medium-latest\"):\n",
    "    user_message = f\"Generate text based on the following prompt:\\n\\n{prompt}\"\n",
    "    return run_mistral(api_key, user_message, model=model)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "ca1c6cbb-3d9e-43ab-b27e-20f71f36214c",
   "metadata": {},
   "outputs": [],
   "source": [
    "def generate_text_with_all_models(openai_key, anthropic_key, mistral_key, prompt, openai_models, anthropic_models, mistral_models, temperature=0.7, max_tokens=100, stop=None):\n",
    "    results = {}\n",
    "\n",
    "    # Générer du texte avec tous les modèles OpenAI\n",
    "    for model in openai_models:\n",
    "        openai_result = openai_generate_text(openai_key, prompt, model, temperature, max_tokens, stop)\n",
    "        results[f'openai_{model}'] = openai_result\n",
    "\n",
    "    # Générer du texte avec tous les modèles Anthropic\n",
    "    for model in anthropic_models:\n",
    "        anthropic_result = anthropic_generate_text(anthropic_key, prompt, model, max_tokens, temperature)\n",
    "        results[f'anthropic_{model}'] = anthropic_result\n",
    "\n",
    "    # Générer du texte avec tous les modèles Mistral\n",
    "    for model in mistral_models:\n",
    "        mistral_result = mistral_generate_text(mistral_key, prompt, model)\n",
    "        results[f'mistral_{model}'] = mistral_result\n",
    "\n",
    "    return results"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "be825b9a-fb5e-42b5-b5e5-ee0348d3cb69",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\u001b[1mResult from openai_gpt-3.5-turbo (72 words):\u001b[0m\n",
      "Chicken, Spinach, and Quinoa Bowl\n",
      "\n",
      "Ingredients:\n",
      "- 1 chicken breast\n",
      "- 1 cup quinoa\n",
      "- 2 cups fresh spinach\n",
      "- Salt and pepper to taste\n",
      "- Olive oil\n",
      "\n",
      "Instructions:\n",
      "1. Cook the quinoa according to package instructions.\n",
      "2. Season the chicken breast with salt and pepper.\n",
      "3. In a large skillet, heat olive oil over medium heat. Add the chicken breast and cook for 6-8 minutes on each side, or until fully\n",
      "\n",
      "\u001b[1mResult from openai_gpt-4 (70 words):\u001b[0m\n",
      "Recipe: Spinach and Quinoa Stuffed Chicken Breast\n",
      "\n",
      "Ingredients:\n",
      "\n",
      "- 2 boneless, skinless chicken breasts\n",
      "- 1 cup of cooked quinoa\n",
      "- 2 cups of fresh spinach\n",
      "- 2 cloves of garlic, minced\n",
      "- 1 tablespoon of olive oil\n",
      "- Salt and pepper to taste\n",
      "\n",
      "Instructions:\n",
      "\n",
      "1. Preheat your oven to 375 degrees Fahrenheit (190 degrees Celsius).\n",
      "\n",
      "2. While the oven is heating, heat the olive oil in a skillet over\n",
      "\n",
      "\u001b[1mResult from openai_gpt-4-turbo (68 words):\u001b[0m\n",
      "Certainly! Here's a simple and healthy recipe using chicken breast, spinach, and quinoa:\n",
      "\n",
      "### Ingredients:\n",
      "- 2 chicken breasts (boneless and skinless)\n",
      "- 1 cup quinoa\n",
      "- 2 cups fresh spinach leaves\n",
      "- 2 cloves garlic, minced\n",
      "- 1 lemon (juice and zest)\n",
      "- 2 tablespoons olive oil\n",
      "- Salt and pepper, to taste\n",
      "- Optional: fresh herbs (like parsley or basil), cherry tomatoes, or a pinch of\n",
      "\n",
      "\u001b[1mResult from openai_gpt-4o-mini (72 words):\u001b[0m\n",
      "Here's a simple and healthy recipe using chicken breast, spinach, and quinoa:\n",
      "\n",
      "### Lemon Garlic Chicken with Spinach and Quinoa\n",
      "\n",
      "#### Ingredients:\n",
      "- 1 cup quinoa\n",
      "- 2 cups water or low-sodium chicken broth\n",
      "- 2 chicken breasts (about 6 oz each)\n",
      "- 2 cups fresh spinach, washed and chopped\n",
      "- 2 tablespoons olive oil\n",
      "- 2 cloves garlic, minced\n",
      "- Juice of 1 lemon\n",
      "- Salt and pepper to taste\n",
      "-\n",
      "\n",
      "\u001b[1mResult from openai_gpt-4o (69 words):\u001b[0m\n",
      "Sure, here's a simple, healthy recipe using chicken breast, spinach, and quinoa:\n",
      "\n",
      "### Lemon Garlic Chicken with Spinach and Quinoa\n",
      "\n",
      "#### Ingredients:\n",
      "- 2 chicken breasts\n",
      "- 1 cup quinoa\n",
      "- 2 cups fresh spinach\n",
      "- 2 cloves garlic, minced\n",
      "- 1 lemon (zested and juiced)\n",
      "- 2 tablespoons olive oil\n",
      "- 1 teaspoon dried oregano\n",
      "- Salt and pepper to taste\n",
      "- 2 cups low-sodium chicken broth (\n",
      "\n",
      "\u001b[1mResult from anthropic_claude-3-5-sonnet-20240620 (51 words):\u001b[0m\n",
      "Here's a simple and healthy recipe using chicken breast, spinach, and quinoa:\n",
      "\n",
      "Lemon Herb Chicken with Spinach Quinoa\n",
      "\n",
      "Ingredients:\n",
      "- 2 boneless, skinless chicken breasts\n",
      "- 1 cup quinoa\n",
      "- 2 cups water\n",
      "- 4 cups fresh spinach\n",
      "- 2 tbsp olive oil\n",
      "- 1 lemon, juiced and zested\n",
      "- 2\n",
      "\n",
      "\u001b[1mResult from anthropic_claude-3-opus-20240229 (47 words):\u001b[0m\n",
      "Here's a simple and healthy recipe using chicken breast, spinach, and quinoa:\n",
      "\n",
      "Spinach and Quinoa Stuffed Chicken Breast\n",
      "\n",
      "Ingredients:\n",
      "- 4 boneless, skinless chicken breasts\n",
      "- 1 cup cooked quinoa\n",
      "- 2 cups fresh spinach, chopped\n",
      "- 1/4 cup feta cheese, crumbled\n",
      "- 1 clove garlic, minced\n",
      "\n",
      "\u001b[1mResult from anthropic_claude-3-sonnet-20240229 (48 words):\u001b[0m\n",
      "Here's a simple and healthy recipe using chicken breast, spinach, and quinoa:\n",
      "\n",
      "Spinach and Quinoa Stuffed Chicken Breast\n",
      "\n",
      "Ingredients:\n",
      "\n",
      "- 4 boneless, skinless chicken breasts\n",
      "- 1 cup cooked quinoa\n",
      "- 2 cups fresh spinach, chopped\n",
      "- 1/4 cup crumbled feta cheese (optional)\n",
      "- 2 cloves garlic, min\n",
      "\n",
      "\u001b[1mResult from anthropic_claude-3-haiku-20240307 (51 words):\u001b[0m\n",
      "Here's a simple and healthy recipe using chicken breast, spinach, and quinoa:\n",
      "\n",
      "Baked Chicken with Spinach and Quinoa\n",
      "\n",
      "Ingredients:\n",
      "- 4 boneless, skinless chicken breasts\n",
      "- 2 cups fresh spinach, chopped\n",
      "- 1 cup cooked quinoa\n",
      "- 1 tbsp olive oil\n",
      "- 1 tsp garlic powder\n",
      "- 1 tsp dried oregano\n",
      "\n",
      "\u001b[1mResult from mistral_open-mistral-7b (222 words):\u001b[0m\n",
      "Title: Chicken, Spinach, and Quinoa Bowl\n",
      "\n",
      "Ingredients:\n",
      "1. 2 boneless, skinless chicken breasts\n",
      "2. 2 cups of fresh spinach\n",
      "3. 1 cup of uncooked quinoa\n",
      "4. 2 tablespoons of olive oil\n",
      "5. Salt and pepper to taste\n",
      "6. 1 lemon, juiced\n",
      "7. 1 clove of garlic, minced\n",
      "8. 1/2 cup of chopped onion\n",
      "9. 1 cup of chopped cherry tomatoes\n",
      "10. 1/4 cup of crumbled feta cheese (optional)\n",
      "\n",
      "Instructions:\n",
      "\n",
      "1. Cook the quinoa according to the package instructions. Once cooked, set aside and let it cool.\n",
      "\n",
      "2. Season the chicken breasts with salt, pepper, and garlic. Heat 1 tablespoon of olive oil in a skillet over medium heat. Add the chicken breasts and cook until fully cooked, about 6-8 minutes on each side. Once cooked, let it rest for a few minutes before slicing.\n",
      "\n",
      "3. In the same skillet, heat the remaining 1 tablespoon of olive oil. Add the chopped onion and sauté until softened. Add the spinach and cook until wilted. Season with salt and pepper.\n",
      "\n",
      "4. In a large bowl, combine the cooked quinoa, sautéed spinach and onion, sliced chicken breast, cherry tomatoes, and a squeeze of lemon juice. Mix well to combine.\n",
      "\n",
      "5. Top with crumbled feta cheese if desired.\n",
      "\n",
      "Enjoy your Chicken, Spinach, and Quinoa Bowl, a simple and healthy meal packed with protein, fiber, and essential nutrients.\n",
      "\n",
      "\u001b[1mResult from mistral_open-mixtral-8x7b (265 words):\u001b[0m\n",
      "Sure, I'd be happy to help! Here's a simple and healthy recipe that incorporates chicken breast, spinach, and quinoa.\n",
      "\n",
      "**Chicken and Spinach Quinoa Bowl**\n",
      "\n",
      "Ingredients:\n",
      "\n",
      "* 1 boneless, skinless chicken breast\n",
      "* 1 cup spinach\n",
      "* 1/2 cup quinoa\n",
      "* 1 cup low-sodium chicken broth\n",
      "* 1 tablespoon olive oil\n",
      "* Salt and pepper to taste\n",
      "* Optional toppings: lemon slices, chopped fresh herbs (such as parsley or cilantro), sliced almonds, crumbled feta cheese\n",
      "\n",
      "Instructions:\n",
      "\n",
      "1. Rinse the quinoa in a fine mesh strainer and drain well.\n",
      "2. Heat the olive oil in a medium saucepan over medium heat.\n",
      "3. Season the chicken breast with salt and pepper on both sides, then add it to the saucepan. Cook for 4-5 minutes on each side, until it's browned and cooked through. Remove the chicken from the saucepan and set it aside.\n",
      "4. Add the quinoa and chicken broth to the saucepan and bring to a boil. Reduce the heat to low, cover, and simmer for 15-20 minutes, or until the quinoa is tender and the liquid has been absorbed.\n",
      "5. While the quinoa is cooking, chop the cooked chicken into bite-sized pieces.\n",
      "6. When the quinoa is done, stir in the chopped spinach and chicken. Cook for an additional 2-3 minutes, until the spinach is wilted and the chicken is heated through.\n",
      "7. Season with salt and pepper to taste. Serve the quinoa bowl hot, with your choice of optional toppings.\n",
      "\n",
      "This recipe is not only simple and healthy, but also versatile. Feel free to customize it with your favorite herbs, spices, or other vegetables. Enjoy!\n",
      "\n",
      "\u001b[1mResult from mistral_open-mixtral-8x22b (343 words):\u001b[0m\n",
      "Title: Spinach and Quinoa Stuffed Chicken Breast\n",
      "\n",
      "Ingredients:\n",
      "- 2 boneless, skinless chicken breasts\n",
      "- 1 cup fresh spinach\n",
      "- 1/2 cup quinoa\n",
      "- 1 cup low-sodium chicken broth\n",
      "- 1 clove garlic, minced\n",
      "- 1 tablespoon olive oil\n",
      "- Salt and pepper to taste\n",
      "- Optional: Fresh lemon for garnish\n",
      "\n",
      "Instructions:\n",
      "\n",
      "1. Start by cooking the quinoa. Rinse the quinoa under cold water and drain. Heat a small pot over medium heat, add the quinoa and chicken broth. Bring to a boil, then reduce heat to low, cover and let it simmer for 15 minutes or until the quinoa has absorbed all the liquid. Fluff with a fork and set aside.\n",
      "\n",
      "2. While the quinoa is cooking, heat the olive oil in a pan over medium heat. Add the minced garlic and cook for about 1 minute until fragrant. Add the spinach and cook until wilted, about 3-4 minutes.\n",
      "\n",
      "3. Once the spinach is cooked, add it to the cooked quinoa and mix well. Season with salt and pepper to taste.\n",
      "\n",
      "4. Preheat your oven to 400°F (200°C).\n",
      "\n",
      "5. Place the chicken breasts on a cutting board. Using a sharp knife, cut a horizontal slit along the thickest part of the breast to create a pocket, being careful not to cut all the way through.\n",
      "\n",
      "6. Stuff each chicken breast with the quinoa and spinach mixture. Use toothpicks to secure the openings if necessary. Season the chicken with salt and pepper.\n",
      "\n",
      "7. Heat the same pan used for the spinach over medium-high heat. Add the stuffed chicken breasts and sear on both sides until golden brown, about 2-3 minutes per side.\n",
      "\n",
      "8. Transfer the seared chicken breasts to a baking dish and bake in the preheated oven for 20-25 minutes, or until the chicken is cooked through and reaches an internal temperature of 165°F (74°C).\n",
      "\n",
      "9. Remove from the oven and let the chicken rest for a few minutes before serving. Garnish with a squeeze of fresh lemon juice if desired.\n",
      "\n",
      "Enjoy your healthy and delicious Spinach and Quinoa Stuffed Chicken Breast!\n",
      "\n",
      "\u001b[1mResult from mistral_mistral-small-latest (170 words):\u001b[0m\n",
      "Sure, here's a simple, healthy recipe for you:\n",
      "\n",
      "**Quinoa and Spinach Stuffed Chicken Breast**\n",
      "\n",
      "Ingredients:\n",
      "- 2 boneless, skinless chicken breasts\n",
      "- 1 cup cooked quinoa\n",
      "- 2 cups fresh spinach\n",
      "- 1 clove garlic, minced\n",
      "- 1/2 onion, finely chopped\n",
      "- Salt and pepper to taste\n",
      "- 1 tablespoon olive oil\n",
      "\n",
      "Instructions:\n",
      "\n",
      "1. Preheat your oven to 375°F (190°C).\n",
      "\n",
      "2. Heat the olive oil in a pan over medium heat. Add the onion and garlic, sauté until softened.\n",
      "\n",
      "3. Add the spinach to the pan and cook until it wilts. Remove from heat and stir in the cooked quinoa. Season with salt and pepper to taste.\n",
      "\n",
      "4. Cut a pocket into each chicken breast. Stuff each pocket with the spinach and quinoa mixture.\n",
      "\n",
      "5. Place the stuffed chicken breasts in a baking dish. Bake for about 25-30 minutes, or until the chicken is cooked through and the internal temperature reaches 165°F (74°C).\n",
      "\n",
      "6. Let the chicken rest for a few minutes before serving. Enjoy your healthy and delicious meal!\n",
      "\n",
      "\u001b[1mResult from mistral_mistral-medium-latest (253 words):\u001b[0m\n",
      "Sure, here's a simple and healthy recipe that you can try using chicken breast, spinach, and quinoa:\n",
      "\n",
      "**Chicken and Spinach Quinoa Bowl**\n",
      "\n",
      "Ingredients:\n",
      "\n",
      "* 1 lb boneless, skinless chicken breast\n",
      "* 1 cup quinoa\n",
      "* 2 cups water or chicken broth\n",
      "* 3 cups fresh spinach\n",
      "* 1 tbsp olive oil\n",
      "* Salt and pepper, to taste\n",
      "* Optional: lemon juice, garlic powder, and crumbled feta cheese\n",
      "\n",
      "Instructions:\n",
      "\n",
      "1. Rinse the quinoa in a fine mesh strainer and drain well.\n",
      "2. In a medium saucepan, bring the water or chicken broth to a boil. Add the quinoa, reduce heat to low, cover, and let it simmer for 15-20 minutes or until the quinoa is tender and the liquid is absorbed.\n",
      "3. While the quinoa is cooking, heat the olive oil in a large skillet over medium-high heat. Season the chicken breast with salt and pepper, and cook for 5-7 minutes on each side or until it's cooked through and no longer pink in the center.\n",
      "4. Remove the chicken from the skillet and let it rest for a few minutes before slicing it into thin strips.\n",
      "5. In the same skillet, add the fresh spinach and cook for 2-3 minutes or until it's wilted.\n",
      "6. Fluff the cooked quinoa with a fork and divide it into bowls. Top with the sliced chicken breast and wilted spinach.\n",
      "7. If desired, drizzle with lemon juice, sprinkle with garlic powder, and top with crumbled feta cheese.\n",
      "\n",
      "Enjoy your healthy and delicious Chicken and Spinach Quinoa Bowl!\n",
      "\n",
      "\u001b[1mResult from mistral_mistral-large-latest (319 words):\u001b[0m\n",
      "**Spinach and Quinoa Stuffed Chicken Breast**\n",
      "\n",
      "**Ingredients:**\n",
      "- 2 boneless, skinless chicken breasts\n",
      "- 1 cup cooked quinoa\n",
      "- 2 cups fresh spinach\n",
      "- 2 cloves garlic, minced\n",
      "- 1 tbsp olive oil\n",
      "- 1/2 cup crumbled feta cheese (optional, for added flavor)\n",
      "- Salt and pepper to taste\n",
      "- 1 tsp lemon zest (optional, for extra freshness)\n",
      "\n",
      "**Instructions:**\n",
      "\n",
      "1. **Preheat** your oven to 400°F (200°C) and line a baking sheet with parchment paper.\n",
      "\n",
      "2. **Cook quinoa** according to package instructions if you haven't already. Set aside.\n",
      "\n",
      "3. **Heat** olive oil in a pan over medium heat. Add minced garlic and sauté for 1 minute until fragrant.\n",
      "\n",
      "4. **Add** spinach to the pan and cook until wilted, about 2-3 minutes. Season with salt and pepper.\n",
      "\n",
      "5. **In a bowl**, combine cooked quinoa, wilted spinach, feta cheese (if using), and lemon zest (if using). Mix well.\n",
      "\n",
      "6. **Butterfly** the chicken breasts by slicing them horizontally, almost but not all the way through. Open them up like a book.\n",
      "\n",
      "7. **Place** the chicken breasts between two pieces of plastic wrap and pound them with a meat mallet until they're about 1/4 inch thick.\n",
      "\n",
      "8. **Spoon** the quinoa and spinach mixture onto one half of each chicken breast, then fold the other half over to close, pressing gently to seal. Use toothpicks to secure if needed.\n",
      "\n",
      "9. **Place** the stuffed chicken breasts on the prepared baking sheet. Season the outside of the chicken with salt and pepper.\n",
      "\n",
      "10. **Bake** for 25-30 minutes, or until the chicken is cooked through and the internal temperature reaches 165°F (74°C).\n",
      "\n",
      "11. **Let** the chicken rest for a few minutes before slicing and serving. Enjoy your healthy and delicious meal!\n",
      "\n",
      "**Nutritional Information** (per serving, without feta):\n",
      "- Calories: 350\n",
      "- Total Fat: 12g\n",
      "- Saturated Fat: 1.5g\n",
      "- Cholesterol: 75mg\n",
      "- Sodium: 300mg\n",
      "- Total Carbohydrate: 27g\n",
      "- Dietary Fiber: 4g\n",
      "- Sugars: 1g\n",
      "- Protein: 35g\n",
      "\n"
     ]
    }
   ],
   "source": [
    "if __name__ == \"__main__\":\n",
    "    openai_key = os.getenv(\"OPENAI_API_KEY\")\n",
    "    anthropic_key = os.getenv(\"ANTHROPIC_API_KEY\")\n",
    "    mistral_key = os.getenv(\"MISTRAL_API_KEY\")\n",
    "    prompt = \"Create a simple, healthy recipe using the following ingredients: chicken breast, spinach, and quinoa.\"\n",
    "\n",
    "    openai_models = [\"gpt-3.5-turbo\", \"gpt-4\", \"gpt-4-turbo\", \"gpt-4o-mini\", \"gpt-4o\"]\n",
    "    anthropic_models = [\"claude-3-5-sonnet-20240620\", \"claude-3-opus-20240229\", \"claude-3-sonnet-20240229\", \"claude-3-haiku-20240307\"]\n",
    "    mistral_models = [\"open-mistral-7b\", \"open-mixtral-8x7b\", \"open-mixtral-8x22b\", \"mistral-small-latest\", \"mistral-medium-latest\", \"mistral-large-latest\"]\n",
    "\n",
    "    results = generate_text_with_all_models(openai_key, anthropic_key, mistral_key, prompt, openai_models, anthropic_models, mistral_models)\n",
    "    \n",
    "    for model_name, result in results.items():\n",
    "        word_count = len(result.split())\n",
    "        print(f\"\\033[1mResult from {model_name} ({word_count} words):\\033[0m\\n{result}\\n\")"
   ]
  }
 ],
 "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.9.6"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}