From Dad Jokes To Sad Jokes: Function Calling with GPTAssistantAgent#

%pip install autogen

import requests

import autogen
from autogen import UserProxyAgent
from autogen.agentchat.contrib.gpt_assistant_agent import GPTAssistantAgent
from autogen.tools import get_function_schema

config_list = autogen.config_list_from_json(
    env_or_file="OAI_CONFIG_LIST",
)

def get_dad_jokes(search_term: str, page: int = 1, limit: int = 10) -> str:
    """Fetches a list of dad jokes based on a search term.

    Parameters:
    - search_term: The search term to find jokes about.
    - page: The page number of results to fetch (default is 1).
    - limit: The number of results to return per page (default is 20, max is 30).

    Returns:
    A list of dad jokes.
    """
    url = "https://icanhazdadjoke.com/search"
    headers = {"Accept": "application/json"}
    params = {"term": search_term, "page": page, "limit": limit}

    response = requests.get(url, headers=headers, params=params)

    if response.status_code == 200:
        data = response.json()
        jokes = [joke["joke"] for joke in data["results"]]
        return jokes
    else:
        return f"Failed to fetch jokes, status code: {response.status_code}"

# Example Dad Jokes Function Usage:
jokes = get_dad_jokes("cats")
print(jokes)

def write_to_txt(content: str, filename: str = "dad_jokes.txt"):
    """Writes a formatted string to a text file.
    Parameters:

    - content: The formatted string to write.
    - filename: The name of the file to write to. Defaults to "output.txt".
    """
    with open(filename, "w") as file:
        file.write(content)

# Example Write to TXT Function Usage:
content = "\n".join(jokes)  # Format the jokes from the above example
write_to_txt(content)

# Assistant API Tool Schema for get_dad_jokes
get_dad_jokes_schema = get_function_schema(
    get_dad_jokes,
    name="get_dad_jokes",
    description="Fetches a list of dad jokes based on a search term. Allows pagination with page and limit parameters.",
)

# Assistant API Tool Schema for write_to_txt
write_to_txt_schema = get_function_schema(
    write_to_txt,
    name="write_to_txt",
    description="Writes a formatted string to a text file. If the file does not exist, it will be created. If the file does exist, it will be overwritten.",
)

user_proxy = UserProxyAgent(
    name="user_proxy",
    is_termination_msg=lambda msg: "TERMINATE" in msg["content"],
    human_input_mode="NEVER",
    max_consecutive_auto_reply=1,
)

the_dad = GPTAssistantAgent(
    name="the_dad",
    instructions="""
    As 'The Dad', your primary role is to entertain by fetching dad jokes which the sad joker will transform into 'sad jokes' based on a given theme. When provided with a theme, such as 'plants' or 'animals', your task is as follows:

    1. Use the 'get_dad_jokes' function to search for dad jokes related to the provided theme by providing a search term related to the theme. Fetch a list of jokes that are relevant to the theme.
    2. Present these jokes to the sad joker in a format that is clear and easy to read, preparing them for transformation.

    Remember, the team's goal is to creatively adapt the essence of each dad joke to fit the 'sad joke' format, all while staying true to the theme provided by the user.
    """,
    overwrite_instructions=True,  # overwrite any existing instructions with the ones provided
    overwrite_tools=True,  # overwrite any existing tools with the ones provided
    llm_config={
        "config_list": config_list,
        "tools": [get_dad_jokes_schema],
    },
)

# Register get_dad_jokes with the_dad GPTAssistantAgent
the_dad.register_function(
    function_map={
        "get_dad_jokes": get_dad_jokes,
    },
)

the_sad_joker = GPTAssistantAgent(
    name="the_sad_joker",
    instructions="""
    As 'The Sad Joker', your unique role is to take dad jokes and creatively transform them into 'sad jokes'. When you receive a list of dad jokes, themed around topics like 'plants' or 'animals', you should:

    1. Read through each dad joke carefully, understanding its theme and punchline.
    2. Creatively alter the joke to change its mood from humorous to somber or melancholic. This may involve tweaking the punchline, modifying the setup, or even completely reimagining the joke while keeping it relevant to the original theme.
    3. Ensure your transformations maintain a clear connection to the original theme and are understandable as adaptations of the dad jokes provided.
    4. Write your transformed sad jokes to a text file using the 'write_to_txt' function. Use meaningful file names that reflect the theme or the nature of the jokes within, unless a specific filename is requested.

    Your goal is not just to alter the mood of the jokes but to do so in a way that is creative, thoughtful, and respects the essence of the original humor. Remember, while the themes might be light-hearted, your transformations should offer a melancholic twist that makes them uniquely 'sad jokes'.
    """,
    overwrite_instructions=True,  # overwrite any existing instructions with the ones provided
    overwrite_tools=True,  # overwrite any existing tools with the ones provided
    llm_config={
        "config_list": config_list,
        "tools": [write_to_txt_schema],
    },
)

# Register get_dad_jokes with the_dad GPTAssistantAgent
the_sad_joker.register_function(
    function_map={
        "write_to_txt": write_to_txt,
    },
)

groupchat = autogen.GroupChat(agents=[user_proxy, the_dad, the_sad_joker], messages=[], max_round=15)
group_chat_manager = autogen.GroupChatManager(groupchat=groupchat, llm_config={"config_list": config_list})

user_proxy.initiate_chat(group_chat_manager, message="Jokes about cats")