lv_recipe_chatbot/vegan_recipe_assistant.py

# AUTOGENERATED! DO NOT EDIT! File to edit: ../nbs/02_vegan_recipe_assistant.ipynb.

# %% auto 0
__all__ = ['VEGAN_RECIPE_SEARCH_TOOL_SCHEMA', 'SYSTEM_PROMPT', 'get_vegan_recipes_edamam_api', 'vegan_recipe_edamam_search']

# %% ../nbs/02_vegan_recipe_assistant.ipynb 3
import os
from typing import Dict
import requests
from openai import OpenAI
import json
from .utils import load_json, dump_json
import constants
from tenacity import retry, wait_random_exponential, stop_after_attempt
import logging

# %% ../nbs/02_vegan_recipe_assistant.ipynb 10
def get_vegan_recipes_edamam_api(params: Dict) -> requests.Response:
    """
    type is required and can be "any", "public", "user"
    """
    if "health" in params:
        params["health"].append("vegan")
    else:
        params["health"] = ["vegan"]
    params["app_id"] = os.environ["EDAMAM_APP_ID"]
    params["app_key"] = os.environ["EDAMAM_APP_KEY"]
    params["type"] = "public"
    query = params["q"]
    if "vegan" not in query.lower():
        params["q"] = "vegan " + query
    return requests.get("https://api.edamam.com/api/recipes/v2", params=params)

# %% ../nbs/02_vegan_recipe_assistant.ipynb 14
def vegan_recipe_edamam_search(query: str) -> str:
    """
    Searches for vegan recipes based on a query.
    If the request fails an explanation should be returned.
    If the cause of the failure was due to no recipes found, prompt the user to try again with a provided shorter query with one word removed.
    """
    max_chars = 45  # 5 chars per word * 9 max words
    if len(query) > max_chars:
        return json.dumps(
            {
                "ok": False,
                "msg": f"The query is too long, politetly tell the user to try again with a query that is under {max_chars} characters in length.",
            }
        )

    params = {
        "q": query,
        "field": ["label", "url", "totalTime", "ingredientLines"],
    }

    response = get_vegan_recipes_edamam_api(params)
    if not response.ok:
        return json.dumps(
            {
                "ok": False,
                "msg": f"Received an error from Edamam API: {response.status_code} {response.text}",
            }
        )

    if response.json()["count"] <= 0:
        return json.dumps(
            {
                "ok": False,
                "msg": f"""No recipes found for query {query}.
This usually occurs when there are too many keywords or ingredients that are not commonly found together in recipes.
Politetly tell the user to try again with fewer words in the query.""",
            }
        )

    return json.dumps(
        {"ok": True, "recipes": [r["recipe"] for r in response.json()["hits"][0:3]]}
    )

# %% ../nbs/02_vegan_recipe_assistant.ipynb 18
VEGAN_RECIPE_SEARCH_TOOL_SCHEMA = {
    "type": "function",
    "function": {
        "name": "vegan_recipe_edamam_search",
        "description": "Searches an external API for vegan recipes based on a provided query.",
        "parameters": {
            "type": "object",
            "properties": {
                "query": {
                    "type": "string",
                    "description": "Query that includes a few vegan ingredients or recipe keywords.",
                },
            },
            "required": ["query"],
        },
    },
}

# %% ../nbs/02_vegan_recipe_assistant.ipynb 24
SYSTEM_PROMPT = """The following is a conversation between a human and a friendly vegan AI. 
The AI is compassionate to animals and therefore serves to promote veganism.
Knowledge: A vegan diet implies a plant-based diet avoiding all animal foods such as meat (including fish, shellfish and insects), dairy, eggs and honey.
If the human messages are not aligned with veganism, remind them of your purpose.
The AI NEVER generates vegan recipes itself but instead uses a tool.
If an image image URL is included by the recipe search output, keep the full URL with all params."""