@acarmisc/backstage-plugin-litellm 0.13.0 → 0.14.0

package/dist/index.cjs.js

@@ -1353,10 +1353,18 @@ var init_format = __esm({
 function trimSlash(url) {
   return url.replace(/\/+$/, "");
 }
+function isModelAllowedByTeam(model, teamModels) {
+  if (!teamModels || teamModels.length === 0) return true;
+  if (teamModels.includes(ALL_PROXY_MODELS)) return true;
+  if (teamModels.includes(model.model_name)) return true;
+  return !!model.access_groups?.some((group) => teamModels.includes(group));
+}
 function buildSnippets(baseUrl, key, model) {
   const base = trimSlash(baseUrl);
+  const apiBase = `${base}/v1`;
   return {
-    curl: `curl ${base}/v1/chat/completions \\
+    publicEndpoint: apiBase,
+    curl: `curl ${apiBase}/chat/completions \\
   -H "Authorization: Bearer ${key}" \\
   -H "Content-Type: application/json" \\
   -d '{
@@ -1367,14 +1375,40 @@ function buildSnippets(baseUrl, key, model) {
 
 client = OpenAI(
   api_key="${key}",
-  base_url="${base}/v1",
+  base_url="${apiBase}",
 )
 
 response = client.chat.completions.create(
   model="${model}",
   messages=[{ "role": "user", "content": "Hello!" }],
 )
-print(response.choices[0].message.content)`
+print(response.choices[0].message.content)`,
+    opencode: `{
+  "$schema": "https://opencode.ai/config.json",
+  "provider": {
+    "litellm": {
+      "npm": "@ai-sdk/openai-compatible",
+      "name": "LiteLLM",
+      "options": {
+        "baseURL": "${apiBase}",
+        "apiKey": "${key}"
+      },
+      "models": {
+        "${model}": {}
+      }
+    }
+  }
+}`,
+    pi: `{
+  "litellm": {
+    "baseUrl": "${apiBase}",
+    "apiKey": "${key}",
+    "api": "openai-completions",
+    "models": [
+      { "id": "${model}", "name": "${model}" }
+    ]
+  }
+}`
   };
 }
 function aliasHelperText(aliasError, aliasDuplicate) {
@@ -1412,7 +1446,7 @@ function formatContextWindow2(maxInput, maxOutput) {
   if (outPart) return `ctx ${outPart} out`;
   return null;
 }
-var import_react4, import_Box4, import_Typography4, import_Dialog2, import_DialogTitle2, import_DialogContent2, import_DialogActions2, import_TextField2, import_MenuItem, import_Button3, import_IconButton2, import_CircularProgress2, import_Autocomplete2, import_Checkbox, import_FormControlLabel, import_Alert, import_Tabs, import_Tab, import_icons_material3, generateDefaultAlias, emptyForm, SnippetTabs, GenerateKeyDialog;
+var import_react4, import_Box4, import_Typography4, import_Dialog2, import_DialogTitle2, import_DialogContent2, import_DialogActions2, import_TextField2, import_MenuItem, import_Button3, import_IconButton2, import_CircularProgress2, import_Autocomplete2, import_Checkbox, import_FormControlLabel, import_Alert, import_Tabs, import_Tab, import_icons_material3, generateDefaultAlias, emptyForm, ALL_PROXY_MODELS, SNIPPET_FILE_HINTS, SnippetTabs, GenerateKeyDialog;
 var init_GenerateKeyDialog = __esm({
   "src/components/GenerateKeyDialog.tsx"() {
     "use strict";
@@ -1451,10 +1485,16 @@ var init_GenerateKeyDialog = __esm({
       team_id: void 0,
       key_type: "llm_api"
     });
+    ALL_PROXY_MODELS = "all-proxy-models";
+    SNIPPET_FILE_HINTS = {
+      opencode: "Add to ~/.config/opencode/opencode.json",
+      pi: "Add to ~/.pi/agent/models.json"
+    };
     SnippetTabs = ({ snippets, model, onCopy }) => {
       const [tab, setTab] = (0, import_react4.useState)("curl");
-      const code = tab === "curl" ? snippets.curl : snippets.openai;
-      return /* @__PURE__ */ import_react4.default.createElement(import_Box4.default, null, /* @__PURE__ */ import_react4.default.createElement(import_Tabs.default, { value: tab, onChange: (_, v) => setTab(v), sx: { mb: 1 } }, /* @__PURE__ */ import_react4.default.createElement(import_Tab.default, { label: "curl", value: "curl" }), /* @__PURE__ */ import_react4.default.createElement(import_Tab.default, { label: "OpenAI SDK", value: "openai" })), /* @__PURE__ */ import_react4.default.createElement(
+      const code = snippets[tab];
+      const fileHint = SNIPPET_FILE_HINTS[tab];
+      return /* @__PURE__ */ import_react4.default.createElement(import_Box4.default, null, /* @__PURE__ */ import_react4.default.createElement(import_Tabs.default, { value: tab, onChange: (_, v) => setTab(v), sx: { mb: 1 }, variant: "scrollable" }, /* @__PURE__ */ import_react4.default.createElement(import_Tab.default, { label: "curl", value: "curl" }), /* @__PURE__ */ import_react4.default.createElement(import_Tab.default, { label: "OpenAI SDK", value: "openai" }), /* @__PURE__ */ import_react4.default.createElement(import_Tab.default, { label: "opencode", value: "opencode" }), /* @__PURE__ */ import_react4.default.createElement(import_Tab.default, { label: "pi", value: "pi" })), fileHint && /* @__PURE__ */ import_react4.default.createElement(import_Typography4.default, { variant: "caption", color: "text.secondary", display: "block", mb: 0.5 }, fileHint), /* @__PURE__ */ import_react4.default.createElement(
         import_Box4.default,
         {
           position: "relative",
@@ -1521,12 +1561,7 @@ var init_GenerateKeyDialog = __esm({
       const teamRequired = keyGenerationSettings?.teamRequired ?? true;
       const selectedTeam = teams.find((t) => t.team_id === formData.team_id) ?? null;
       const availableModels = (0, import_react4.useMemo)(() => {
-        const teamModels = selectedTeam?.models;
-        if (teamModels && teamModels.length > 0) {
-          const allowed = new Set(teamModels);
-          return models.filter((m) => allowed.has(m.model_name));
-        }
-        return models;
+        return models.filter((m) => isModelAllowedByTeam(m, selectedTeam?.models));
       }, [models, selectedTeam]);
       const selectedModels = availableModels.filter((m) => (formData.models || []).includes(m.model_name));
       const aliasError = !(formData.alias || "").trim();
@@ -1550,12 +1585,12 @@ var init_GenerateKeyDialog = __esm({
             max_budget: unlimitedBudget ? null : formData.max_budget
           };
           const response = await onGenerateKey(request);
+          const model = formData.models?.[0] ?? availableModels[0]?.model_name ?? "";
           setNewKeyValue(response.key);
-          setNewKeyModel(formData.models?.[0] ?? "");
+          setNewKeyModel(model);
           setNewKeySnippets(null);
           try {
             const config = await onGetConfig();
-            const model = formData.models?.[0] ?? "";
             setNewKeySnippets(buildSnippets(config.baseUrl, response.key, model));
           } catch {
           }
@@ -1593,8 +1628,10 @@ var init_GenerateKeyDialog = __esm({
               getOptionLabel: (t) => t.team_alias || t.team_id,
               value: selectedTeam,
               onChange: (_e, team) => {
-                const teamModels = team?.models;
-                const restrictedModels = teamModels && teamModels.length > 0 ? (formData.models || []).filter((m) => teamModels.includes(m)) : formData.models;
+                const restrictedModels = (formData.models || []).filter((name) => {
+                  const model = models.find((m) => m.model_name === name);
+                  return model ? isModelAllowedByTeam(model, team?.models) : false;
+                });
                 setFormData({ ...formData, team_id: team?.team_id, models: restrictedModels });
               },
               renderInput: (params) => /* @__PURE__ */ import_react4.default.createElement(
@@ -1641,7 +1678,31 @@ var init_GenerateKeyDialog = __esm({
           newKeyValue
         ),
         /* @__PURE__ */ import_react4.default.createElement(import_IconButton2.default, { onClick: () => copyToClipboard(newKeyValue) }, /* @__PURE__ */ import_react4.default.createElement(import_icons_material3.ContentCopy, null))
-      ), newKeySnippets && /* @__PURE__ */ import_react4.default.createElement(import_Box4.default, { mt: 3 }, /* @__PURE__ */ import_react4.default.createElement(import_Box4.default, { display: "flex", alignItems: "center", gap: 1, mb: 1 }, /* @__PURE__ */ import_react4.default.createElement(import_icons_material3.Code, { fontSize: "small", color: "action" }), /* @__PURE__ */ import_react4.default.createElement(import_Typography4.default, { variant: "subtitle2" }, "Start calling the proxy \u2014 paste and run")), /* @__PURE__ */ import_react4.default.createElement(SnippetTabs, { snippets: newKeySnippets, model: newKeyModel, onCopy: copyToClipboard }))) : /* @__PURE__ */ import_react4.default.createElement(import_Box4.default, { display: "flex", flexDirection: "column", gap: 2, mt: 1 }, generateError && /* @__PURE__ */ import_react4.default.createElement(import_Alert.default, { severity: "error", onClose: () => setGenerateError(null) }, generateError), /* @__PURE__ */ import_react4.default.createElement(
+      ), newKeySnippets && /* @__PURE__ */ import_react4.default.createElement(import_Box4.default, { mt: 2 }, /* @__PURE__ */ import_react4.default.createElement(import_Typography4.default, { variant: "caption", color: "text.secondary", display: "block", gutterBottom: true }, "Public endpoint \u2014 paste into any tool's base URL / API base field"), /* @__PURE__ */ import_react4.default.createElement(
+        import_Box4.default,
+        {
+          display: "flex",
+          alignItems: "center",
+          gap: 1,
+          p: 1.5,
+          sx: {
+            backgroundColor: "action.hover",
+            border: "1px solid",
+            borderColor: "divider",
+            borderRadius: 1
+          }
+        },
+        /* @__PURE__ */ import_react4.default.createElement(
+          import_Typography4.default,
+          {
+            component: "code",
+            color: "text.primary",
+            sx: { fontFamily: "monospace", fontSize: 13, wordBreak: "break-all", flex: 1 }
+          },
+          newKeySnippets.publicEndpoint
+        ),
+        /* @__PURE__ */ import_react4.default.createElement(import_IconButton2.default, { size: "small", onClick: () => copyToClipboard(newKeySnippets.publicEndpoint) }, /* @__PURE__ */ import_react4.default.createElement(import_icons_material3.ContentCopy, { fontSize: "small" }))
+      )), newKeySnippets && /* @__PURE__ */ import_react4.default.createElement(import_Box4.default, { mt: 3 }, /* @__PURE__ */ import_react4.default.createElement(import_Box4.default, { display: "flex", alignItems: "center", gap: 1, mb: 1 }, /* @__PURE__ */ import_react4.default.createElement(import_icons_material3.Code, { fontSize: "small", color: "action" }), /* @__PURE__ */ import_react4.default.createElement(import_Typography4.default, { variant: "subtitle2" }, "Start calling the proxy \u2014 paste and run")), /* @__PURE__ */ import_react4.default.createElement(SnippetTabs, { snippets: newKeySnippets, model: newKeyModel, onCopy: copyToClipboard }))) : /* @__PURE__ */ import_react4.default.createElement(import_Box4.default, { display: "flex", flexDirection: "column", gap: 2, mt: 1 }, generateError && /* @__PURE__ */ import_react4.default.createElement(import_Alert.default, { severity: "error", onClose: () => setGenerateError(null) }, generateError), /* @__PURE__ */ import_react4.default.createElement(
         import_TextField2.default,
         {
           label: "Alias",