Skip to content

Commit 6a0f4fa

Browse files
committed
feat: move to openrouter
1 parent 446d981 commit 6a0f4fa

1 file changed

Lines changed: 8 additions & 7 deletions

File tree

src/balatrollm/cli.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
import argparse
44
import asyncio
5+
import os
56
import sys
67
from pathlib import Path
78

@@ -192,7 +193,7 @@ def _create_argument_parser() -> argparse.ArgumentParser:
192193
Configured ArgumentParser instance with all commands and options.
193194
"""
194195
parser = argparse.ArgumentParser(
195-
description="LLM-powered Balatro bot using LiteLLM proxy",
196+
description="LLM-powered Balatro bot",
196197
formatter_class=argparse.RawDescriptionHelpFormatter,
197198
)
198199

@@ -204,13 +205,13 @@ def _create_argument_parser() -> argparse.ArgumentParser:
204205
"-m",
205206
"--model",
206207
default="openai/gpt-oss-20b",
207-
help="Model name to use from LiteLLM proxy (default: openai/gpt-oss-20b)",
208+
help="Model name to use from OpenRouter (default: openai/gpt-oss-20b)",
208209
)
209210
parser.add_argument(
210211
"-l",
211212
"--list-models",
212213
action="store_true",
213-
help="List available models from the proxy and exit",
214+
help="List available models from OpenRouter and exit",
214215
)
215216
parser.add_argument(
216217
"-s",
@@ -222,14 +223,14 @@ def _create_argument_parser() -> argparse.ArgumentParser:
222223
parser.add_argument(
223224
"-u",
224225
"--base-url",
225-
default="http://localhost:4000",
226-
help="LiteLLM base URL (default: http://localhost:4000)",
226+
default="https://openrouter.ai/api/v1",
227+
help="OpenAI API base URL (default: https://openrouter.ai/api/v1)",
227228
)
228229
parser.add_argument(
229230
"-k",
230231
"--api-key",
231-
default="sk-balatrollm-proxy-key",
232-
help="LiteLLM proxy API key (default: sk-balatrollm-proxy-key)",
232+
default=os.getenv("OPENROUTER_API_KEY"),
233+
help="API key (default: OPENROUTER_API_KEY env var)",
233234
)
234235
parser.add_argument(
235236
"-c",

0 commit comments

Comments
 (0)