Release notes writer

"""Write release notes with a versioned skill."""

import base64
import io
import os
import zipfile
from pathlib import Path

import openai

client = openai.OpenAI(api_key=os.environ["ORCA_API_KEY"], base_url=os.environ["ORCA_BASE_URL"])
agents = client.beta.agents

SKILL_MD = """---
name: release-notes
description: Turn a raw changelog into customer-facing release notes. Use when asked for release notes or a changelog summary.
---
Read the changelog you are given. Use only the newest version's entries.

1. Sort each entry into Features, Fixes, or Breaking changes. Drop internal-only entries
   such as refactors, tests, and CI changes.
2. Rewrite each entry as one sentence for customers: what they can now do, or what no
   longer goes wrong. No ticket numbers, no internal names.
3. Fill in template.md and save the result to /workspace/outputs/release_notes.md.
4. If an entry is unclear, keep it out of the notes and list it under "Needs a human"
   in your reply.
"""

TEMPLATE_MD = """# What's new in {version}

## Features
- ...

## Fixes
- ...

## Breaking changes
- ... (leave this section out if there are none)
"""

CHANGELOG = b"""# Changelog

## 2.4.0
- Export reports as PDF (#812)
- Fix totals rounding on invoices with more than 100 lines (#820)
- Refactor billing service tests
- The /v1/reports endpoint now requires a date range (#801)
"""


def bundle(files):
    """Zip a skill folder from {path: text}."""
    out = io.BytesIO()
    with zipfile.ZipFile(out, "w") as archive:
        for path, text in files.items():
            archive.writestr(path, text)
    return out.getvalue()


skill = template = agent = session = None
try:
    # 1. Upload the skill. Version 1 becomes its default.
    skill = client.skills.create(files=[("release-notes.zip", bundle({
        "release-notes/SKILL.md": SKILL_MD,
        "release-notes/template.md": TEMPLATE_MD,
    }))])

    # 2. Save a template that installs exactly this version, with no internet.
    template = agents.environments.templates.create(
        name="release-workspace",
        network={"access": "disabled"},
        skills=[{"type": "skill_reference", "skill_id": skill.id, "version": skill.default_version}],
    )

    agent = agents.create(
        model=os.environ["ORCA_MODEL"],
        name="release-writer",
        instructions=(
            "You write release notes. Before you write anything, load the release-notes skill\n"
            "with the skill tool, read its template.md, and follow both exactly.\n"
            "Create /workspace/outputs if it does not exist.\n"
            "The changelog is at /workspace/inputs/CHANGELOG.md. Do not use emojis."
        ),
    )

    # 3. Each session builds its sandbox from the template, plus this run's changelog.
    session = agents.sessions.create(
        agent_id=agent.id,
        environment={
            "type": "openai_hosted",
            "environment_template_id": template.id,
            "files": [{
                "type": "inline",
                "path": "/workspace/inputs/CHANGELOG.md",
                "data": base64.b64encode(CHANGELOG).decode(),
            }],
        },
    )

    with agents.sessions.stream(session.id, input="Write the release notes for 2.4.0.") as stream:
        for event in stream:
            if event.type == "agent.session.turn.completed":
                print("turn:", event.turn.status)

    for artifact in agents.sessions.artifacts.list(session.id):
        content = agents.sessions.artifacts.content(artifact.id, session_id=session.id)
        Path(Path(artifact.path).name).write_bytes(content.content)
        print("saved:", Path(artifact.path).name)
finally:
    # Delete everything this run created, each on its own, so one failed
    # delete does not skip the rest. To reuse the skill and template, keep them.
    if session:
        try:
            agents.sessions.delete(session.id)
        except openai.APIError:
            pass
    if agent:
        try:
            agents.delete(agent.id)
        except openai.APIError:
            pass
    if template:
        try:
            agents.environments.templates.delete(template.id)
        except openai.APIError:
            pass
    if skill:
        try:
            client.skills.delete(skill.id)
        except openai.APIError:
            pass
    client.close()