add a more specific env CUA_INFERENCE_API_KEY

This commit is contained in:
Dillon DuPont
2025-11-11 10:13:14 -05:00
parent cfedce2806
commit ac1511210b

View File

@@ -10,7 +10,7 @@ class CUAAdapter(CustomLLM):
def __init__(self, base_url: str | None = None, api_key: str | None = None, **_: Any):
super().__init__()
self.base_url = base_url or os.environ.get("CUA_BASE_URL") or "https://inference.cua.ai/v1"
self.api_key = api_key or os.environ.get("CUA_API_KEY")
self.api_key = api_key or os.environ.get("CUA_INFERENCE_API_KEY") or os.environ.get("CUA_API_KEY")
def _normalize_model(self, model: str) -> str:
# Accept either "cua/<model>" or raw "<model>"