05-extras

# /// script
# requires-python = ">=3.12"
# dependencies = [
#     "llm==0.27.1",
#     "llm-openrouter==0.5",
#     "marimo",
#     "python-dotenv==1.2.1",
# ]
# ///

import marimo

__generated_with = "0.17.8"
app = marimo.App(sql_output="polars")


@app.cell
def _(mo):
    mo.md(r"""
    ## OpenRouter
    """)
    return


@app.cell
def _():
    import marimo as mo
    return (mo,)


@app.cell
def _():
    import os
    from dotenv import load_dotenv

    load_dotenv(".env")
    return (os,)


@app.cell
def _(llm):
    llm.get_model_aliases();
    return


@app.cell
def _(os):
    import llm

    model = llm.get_model("openrouter/moonshotai/kimi-k2-0905")
    # key= is optional, you can configure the key in other ways
    response = model.prompt(
        "Five surprising names for a pet pelican",
        key=os.environ.get("OPENROUTER_KEY")
    )
    print(response.text())
    return (llm,)


@app.cell
def _(llm):
    new_model = llm.get_model("gpt-4o-mini")
    print(new_model.attachment_types)
    return (new_model,)


@app.cell
def _(mo):
    mo.image("grump.jpg")
    return


@app.cell
def _(llm, new_model):
    new_resp = new_model.prompt(
        "Describe these images",
        attachments=[llm.Attachment(path="grump.jpg")]
    )
    return (new_resp,)


@app.cell
def _(new_resp):
    print(new_resp.text())
    return


if __name__ == "__main__":
    app.run()