mirror of
https://github.com/trycua/lume.git
synced 2026-01-06 04:20:03 -06:00
add a more specific env CUA_INFERENCE_API_KEY
This commit is contained in:
@@ -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>"
|
||||
|
||||
Reference in New Issue
Block a user