Working with folders

See folder.py on GitHub.

from pprint import pprint

from dotenv import load_dotenv

from genai.client import Client
from genai.credentials import Credentials

# make sure you have a .env file under genai root with
# GENAI_KEY=<your-genai-key>
# GENAI_API=<genai-api-endpoint>
load_dotenv()
client = Client(credentials=Credentials.from_env())


def heading(text: str) -> str:
    """Helper function for centering text."""
    return "\n" + f" {text} ".center(80, "=") + "\n"


print(heading("Create a new folder"))
create_response = client.folder.create(name="My folder")
folder_id = create_response.result.id
print(f"Folder ID: {folder_id}")

print(heading("Creating a prompt in the folder"))
prompt_id = client.prompt.create(
    name="Recipe Generator Prompt",
    model_id="google/flan-t5-xl",
    folder_id=folder_id,
    input="Make a short recipe for {{meal}} (use bullet points)",
).result.id
print(f"Prompt ID: {prompt_id}")

print(heading("Get the folder details"))
retrieve_response = client.folder.retrieve(id=folder_id)
pprint(retrieve_response.result.model_dump())

print(heading("Show all existing folders"))
list_response = client.folder.list(offset=0, limit=10)
print("Total Count: ", list_response.total_count)
print("Results: ", list_response.results)

print(heading("Delete the folder"))
client.folder.delete(id=folder_id)
print("OK")

print(heading("Deleting the prompt"))
client.prompt.delete(id=prompt_id)
print("OK")