first commit

src/create_litellm_keys.py

@@ -0,0 +1,33 @@
+from __future__ import annotations
+
+import httpx
+
+from settings import Env, load_config
+
+
+def main() -> None:
+    env = Env()
+    cfg = load_config(env.config_file)
+    headers = {'Authorization': f'Bearer {env.litellm_master_key}'}
+
+    with httpx.Client(timeout=30) as client:
+        for api in cfg.apis:
+            payload = {
+                'key': api.virtual_key,
+                'aliases': {cfg.model.alias: cfg.model.alias},
+                'models': [cfg.model.alias],
+                'metadata': {'api_id': api.id, 'api_name': api.name, 'owner': api.owner},
+                'team_id': api.owner or api.id,
+                'user_id': api.id,
+            }
+            r = client.post(f'{env.litellm_base_url.replace("/v1", "")}/key/generate', headers=headers, json=payload)
+            if r.status_code in (200, 201):
+                print(f'OK key criada/registrada: {api.id}')
+            elif r.status_code == 400 and 'already' in r.text.lower():
+                print(f'OK key já existe: {api.id}')
+            else:
+                print(f'ERRO {api.id}: {r.status_code} {r.text}')
+
+
+if __name__ == '__main__':
+    main()