Impatient-OpenRouter

impatient.py

import os
import openai

openai.api_base = "https://openrouter.ai/api/v1"
openai.api_key = os.getenv("OPENAI_API_KEY")
OPENROUTER_REFERRER = "https://github.com/alonsosilvaallende/chatplotlib-openrouter"

def llm(text: str) -> str:
    return openai.ChatCompletion.create(
        model='openai/gpt-4-32K',
        headers={
            "HTTP-Referer": OPENROUTER_REFERRER
        },  # Not needed if you already have access to gpt-4
        messages=[{
            'role': 'user',
            'content': f'{text}'
        }],
        temperature=0.9).choices[0].message.content

print(llm("Tell me a joke"))