From 0a98049f21425de1d5def73d70cd6bb90e277597 Mon Sep 17 00:00:00 2001
From: Jeff Epler <jepler@gmail.com>
Date: Sun, 23 Jun 2024 14:45:24 -0500
Subject: [PATCH] Update get_api to not have any required arguments

---
 src/chap/core.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/chap/core.py b/src/chap/core.py
index 9a3df7f..67ed9f7 100644
--- a/src/chap/core.py
+++ b/src/chap/core.py
@@ -129,7 +129,11 @@ def configure_api_from_environment(
         setattr(api.parameters, field.name, tv)
 
 
-def get_api(ctx: click.Context, name: str = "openai_chatgpt") -> Backend:
+def get_api(ctx: click.Context | None = None, name: str | None = None) -> Backend:
+    if ctx is None:
+        ctx = click.Context(click.Command("chap"))
+    if name is None:
+        name = os.environ.get("CHAP_BACKEND", "openai_chatgpt")
     name = name.replace("-", "_")
     backend = cast(
         Backend, importlib.import_module(f"{__package__}.backends.{name}").factory()