@truefoundry/ai-gateway-sdk 0.0.7 → 0.0.8

package/dist/autogen/schema.json

@@ -5946,8 +5946,14 @@
                 "provider_accounts": {
                     "$ref": "#/definitions/BudgetV2InNotInFilter"
                 },
+                "virtual_models": {
+                    "$ref": "#/definitions/BudgetV2InNotInFilter"
+                },
+                "virtual_provider_accounts": {
+                    "$ref": "#/definitions/BudgetV2InNotInFilter"
+                },
                 "metadata": {
-                    "description": "+usage=Filter by metadata key-value pairs this budget applies to\n+label=Metadata\n+sort=4",
+                    "description": "+usage=Filter by metadata key-value pairs this budget applies to\n+label=Metadata\n+sort=6",
                     "type": "object",
                     "additionalProperties": {
                         "$ref": "#/definitions/BudgetV2InNotInFilter"
@@ -5969,8 +5975,14 @@
                 "provider_accounts": {
                     "$ref": "#/definitions/BudgetV2InNotInFilter"
                 },
+                "virtual_models": {
+                    "$ref": "#/definitions/BudgetV2InNotInFilter"
+                },
+                "virtual_provider_accounts": {
+                    "$ref": "#/definitions/BudgetV2InNotInFilter"
+                },
                 "metadata": {
-                    "description": "+usage=Filter by metadata key-value pairs this budget applies to\n+label=Metadata\n+sort=4",
+                    "description": "+usage=Filter by metadata key-value pairs this budget applies to\n+label=Metadata\n+sort=6",
                     "type": "object",
                     "additionalProperties": {
                         "$ref": "#/definitions/BudgetV2InNotInFilter"
@@ -9899,6 +9911,80 @@
             },
             "$schema": "http://json-schema.org/draft-04/schema#"
         },
+        "FireworksApiKeyAuth": {
+            "description": "+label=Fireworks API Key Auth\n+usage=Authentication using a Fireworks API key",
+            "type": "object",
+            "required": ["type", "api_key"],
+            "properties": {
+                "type": {
+                    "description": "+value=api-key",
+                    "type": "string",
+                    "enum": ["api-key"]
+                },
+                "api_key": {
+                    "description": "+sort=100\n+usage=Fireworks API key (sent as a Bearer token). Create one in the Fireworks console.\n+message=API key must not be empty\n+uiType=SecretSelect\n+uiProps={\"isPassword\": true, \"enableNew\": true}",
+                    "type": "string",
+                    "pattern": "^\\S+$"
+                }
+            },
+            "$schema": "http://json-schema.org/draft-04/schema#"
+        },
+        "FireworksGuardrailConfig": {
+            "description": "+label=Fireworks\n+icon=https://assets.production.truefoundry.com/fireworks.svg",
+            "type": "object",
+            "required": ["name", "type", "auth_data", "operation", "enforcing_strategy", "config"],
+            "properties": {
+                "name": {
+                    "description": "+label=Name\n+sort=50\n+usage=The name of the Guardrail Config.\n+message=3 to 36 lower case characters long alphanumeric word, may contain - in between, cannot start with a number\n+uiProps={\"disableEdit\":true}",
+                    "type": "string",
+                    "pattern": "^[a-z][a-z0-9\\-]{1,34}[a-z0-9]$"
+                },
+                "description": {
+                    "description": "+label=Description\n+sort=75\n+usage=Optional description for this Guardrail Config.",
+                    "type": "string",
+                    "default": "Fireworks-hosted safety model (gpt-oss-safeguard or Llama Guard) for policy-based content classification"
+                },
+                "type": {
+                    "description": "+uiType=Hidden\n+value=integration/guardrail-config/fireworks",
+                    "type": "string",
+                    "enum": ["integration/guardrail-config/fireworks"]
+                },
+                "auth_data": {
+                    "$ref": "#/definitions/FireworksApiKeyAuth"
+                },
+                "operation": {
+                    "description": "+label=Operation\n+usage=Fireworks safety models classify content without modifying it.\n+uiType=MultiSelectPills\n+uiProps={\"hideRadioIcon\": true}\n+sort=100",
+                    "type": "string",
+                    "enum": ["validate"]
+                },
+                "enforcing_strategy": {
+                    "$ref": "#/definitions/EnforcingStrategy"
+                },
+                "config": {
+                    "description": "+uiType=Ignore\n+uiProps={\"forwardJsonKey\": true}",
+                    "type": "object",
+                    "required": ["model", "model_type"],
+                    "properties": {
+                        "model": {
+                            "description": "+label=Model\n+usage=Deployment name from your Fireworks account (accounts/<account>/deployments/<deployment>). Safety models run on dedicated (on-demand) deployments only.\n+message=Model must not be empty\n+sort=50",
+                            "type": "string",
+                            "pattern": "^\\S+$"
+                        },
+                        "model_type": {
+                            "description": "+label=Model Type\n+usage=gpt-oss-safeguard classifies content against the policy below. llama-guard classifies against Meta's fixed hazard categories (S1 to S14 for Llama Guard 3 8B) and ignores the policy.\n+uiType=Select\n+sort=100",
+                            "type": "string",
+                            "enum": ["gpt-oss-safeguard", "llama-guard"],
+                            "default": "gpt-oss-safeguard"
+                        },
+                        "policy": {
+                            "description": "+label=Policy\n+usage=Safety policy the model applies (required for gpt-oss-safeguard, ignored for llama-guard). Describe what violates and what is safe, with examples. Do not put secrets in the policy.\n+uiType=TextArea\n+uiProps={\"descriptionInline\":true}\n+sort=150",
+                            "type": "string"
+                        }
+                    }
+                }
+            },
+            "$schema": "http://json-schema.org/draft-04/schema#"
+        },
         "FlyteLaunchPlan": {
             "type": "object",
             "required": ["id", "spec", "closure"],
@@ -12728,6 +12814,12 @@
                 },
                 {
                     "$ref": "#/definitions/HiddenLayerGuardrailConfig"
+                },
+                {
+                    "$ref": "#/definitions/FireworksGuardrailConfig"
+                },
+                {
+                    "$ref": "#/definitions/ZscalerAIGuardGuardrailConfig"
                 }
             ],
             "$schema": "http://json-schema.org/draft-04/schema#"
@@ -21336,8 +21428,14 @@
                 "provider_accounts": {
                     "$ref": "#/definitions/RateLimitV2InNotInFilter"
                 },
+                "virtual_models": {
+                    "$ref": "#/definitions/RateLimitV2InNotInFilter"
+                },
+                "virtual_provider_accounts": {
+                    "$ref": "#/definitions/RateLimitV2InNotInFilter"
+                },
                 "metadata": {
-                    "description": "+usage=Filter by metadata key-value pairs this rate limit applies to\n+label=Metadata\n+sort=4",
+                    "description": "+usage=Filter by metadata key-value pairs this rate limit applies to\n+label=Metadata\n+sort=6",
                     "type": "object",
                     "additionalProperties": {
                         "$ref": "#/definitions/RateLimitV2InNotInFilter"
@@ -28411,6 +28509,83 @@
             "type": "string",
             "enum": ["cloudpickle", "joblib", "pickle", "json"],
             "$schema": "http://json-schema.org/draft-04/schema#"
+        },
+        "ZscalerAIGuardApiKeyAuth": {
+            "description": "+label=Zscaler AI Guard API Key Auth",
+            "type": "object",
+            "required": ["type", "api_key"],
+            "properties": {
+                "type": {
+                    "description": "+value=api-key",
+                    "type": "string",
+                    "enum": ["api-key"]
+                },
+                "api_key": {
+                    "description": "+sort=100\n+usage=Zscaler AI Guard API key. Sent as a Bearer token against the AI Guard detection API.\n+message=API key must not be empty\n+uiType=SecretSelect\n+uiProps={\"isPassword\": true, \"enableNew\": true}",
+                    "type": "string",
+                    "pattern": "^\\S+$"
+                }
+            },
+            "$schema": "http://json-schema.org/draft-04/schema#"
+        },
+        "ZscalerAIGuardGuardrailConfig": {
+            "description": "+label=Zscaler AI Guard\n+icon=https://assets.production.truefoundry.com/zscaler.svg",
+            "type": "object",
+            "required": ["name", "type", "auth_data", "operation", "enforcing_strategy", "config"],
+            "properties": {
+                "name": {
+                    "description": "+label=Name\n+sort=50\n+usage=The name of the Guardrail Config.\n+message=3 to 36 lower case characters long alphanumeric word, may contain - in between, cannot start with a number\n+uiProps={\"disableEdit\":true}",
+                    "type": "string",
+                    "pattern": "^[a-z][a-z0-9\\-]{1,34}[a-z0-9]$"
+                },
+                "description": {
+                    "description": "+label=Description\n+sort=75\n+usage=Optional description for this Guardrail Config.",
+                    "type": "string",
+                    "default": "Zscaler AI Guard runtime protection for prompts and model responses"
+                },
+                "type": {
+                    "description": "+uiType=Hidden\n+value=integration/guardrail-config/zscaler-ai-guard",
+                    "type": "string",
+                    "enum": ["integration/guardrail-config/zscaler-ai-guard"]
+                },
+                "auth_data": {
+                    "$ref": "#/definitions/ZscalerAIGuardApiKeyAuth"
+                },
+                "operation": {
+                    "description": "+label=Operation\n+usage=Validate treats a Zscaler BLOCK action as a violation and leaves the payload unchanged. Mutate also replaces content with Zscaler maskedContent when returned.\nValidate guardrails are run in parallel while mutate guardrails are run sequentially.\n+uiType=MultiSelectPills\n+uiProps={\"hideRadioIcon\": true}\n+sort=100",
+                    "type": "string",
+                    "enum": ["validate", "mutate"],
+                    "default": "validate"
+                },
+                "priority": {
+                    "description": "+label=Priority\n+usage=Execution order for mutate guardrails. Lower values run first. Only applicable when operation is mutate.\n+sort=150",
+                    "type": "integer",
+                    "minimum": 0,
+                    "exclusiveMinimum": true,
+                    "default": 1
+                },
+                "enforcing_strategy": {
+                    "$ref": "#/definitions/EnforcingStrategy"
+                },
+                "config": {
+                    "description": "+uiType=Ignore\n+uiProps={\"forwardJsonKey\": true}",
+                    "type": "object",
+                    "properties": {
+                        "base_url": {
+                            "description": "+label=Base URL\n+usage=The AI Guard API base URL.\n+sort=100\n+message=Base URL must be an http or https URL",
+                            "type": "string",
+                            "pattern": "^https?:\\/\\/\\S+$"
+                        },
+                        "policy_id": {
+                            "description": "+label=Policy ID\n+usage=Optional AI Guard policy ID. When set, calls /v1/detection/execute-policy. When omitted, calls /v1/detection/resolve-and-execute-policy for automatic policy resolution.\n+sort=200",
+                            "type": "integer",
+                            "minimum": 0,
+                            "exclusiveMinimum": true
+                        }
+                    }
+                }
+            },
+            "$schema": "http://json-schema.org/draft-04/schema#"
         }
     },
     "properties": {
@@ -29338,6 +29513,12 @@
         "FiddlerKeyAuth": {
             "$ref": "#/definitions/FiddlerKeyAuth"
         },
+        "FireworksApiKeyAuth": {
+            "$ref": "#/definitions/FireworksApiKeyAuth"
+        },
+        "FireworksGuardrailConfig": {
+            "$ref": "#/definitions/FireworksGuardrailConfig"
+        },
         "FlyteLaunchPlan": {
             "$ref": "#/definitions/FlyteLaunchPlan"
         },
@@ -31323,6 +31504,12 @@
         },
         "XGBoostSerializationFormat": {
             "$ref": "#/definitions/XGBoostSerializationFormat"
+        },
+        "ZscalerAIGuardApiKeyAuth": {
+            "$ref": "#/definitions/ZscalerAIGuardApiKeyAuth"
+        },
+        "ZscalerAIGuardGuardrailConfig": {
+            "$ref": "#/definitions/ZscalerAIGuardGuardrailConfig"
         }
     }
 }

package/dist/providers/anthropic/api.js

@@ -37,10 +37,10 @@ const AnthropicAPIConfig = {
         const account = providerOptions.providerAccount.manifest;
         const storedApiKey = account.auth_data?.api_key;
         const apiKey = storedApiKey ?? callConfig.bearerTokenOverride;
-        // pick auth header based on scheme(bearer or api-key). Match live: stored key → X-API-Key; subscription passthrough → Bearer.
-        const headers = storedApiKey
-            ? { 'X-API-Key': `${apiKey}` }
-            : { Authorization: `Bearer ${apiKey}` };
+        // Anthropic documents `Authorization: Bearer` as the auth header and `x-api-key` as a legacy
+        // fallback. Bearer accepts both Console API keys and the short-lived Workload Identity
+        // Federation tokens from `POST /v1/oauth/token`; `x-api-key` rejects the latter outright.
+        const headers = { Authorization: `Bearer ${apiKey}` };
         const body = request.body;
         const { betaHeader, version } = resolveAnthropicBetaAndVersion({
             endpoint,

package/dist/providers/anthropic/api.js.map

@@ -1 +1 @@
-{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../src/providers/anthropic/api.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,mBAAmB,EAAE,MAAM,UAAU,CAAC;AAC/C,OAAO,EAAE,+BAA+B,EAAoC,MAAM,SAAS,CAAC;AAE5F,MAAM,cAAc,GAAG,CAAC,YAAY,EAAE,WAAW,EAAE,cAAc,EAAE,YAAY,EAAE,qBAAqB,CAAC,CAAC;AAExG,MAAM,CAAC,MAAM,yBAAyB,GAAG,YAAY,CAAC;AACtD,MAAM,CAAC,MAAM,6BAA6B,GAAG,qBAAqB,CAAC;AACnE,MAAM,CAAC,MAAM,qBAAqB,GAAG,sBAAsB,CAAC;AAI5D;;;;GAIG;AACH,MAAM,UAAU,8BAA8B,CAAC,EAC7C,QAAQ,EACR,eAAe,EACf,WAAW,GAKZ;IACC,8CAA8C;IAC9C,IAAI,iBAAiB,GAAG,cAAc,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,6BAA6B,CAAC;IAElH,6FAA6F;IAC7F,IAAI,QAAQ,KAAK,cAAc,IAAI,WAAW,IAAI,mBAAmB,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC5F,iBAAiB,GAAG,qBAAqB,CAAC;IAC5C,CAAC;IAED,8GAA8G;IAC9G,oFAAoF;IACpF,8DAA8D;IAC9D,MAAM,WAAW,GAAG,+BAA+B,CAAC,eAAe,EAAE,WAAW,EAAE,CAAC,gBAAgB,CAAC,CAAC,CAAC;IACtG,MAAM,UAAU,GAAG,WAAW,EAAE,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC;IACnF,MAAM,OAAO,GAAG,eAAe,EAAE,gBAAgB,IAAI,WAAW,EAAE,CAAC,mBAAmB,CAAC,IAAI,yBAAyB,CAAC;IAErH,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC;AACjC,CAAC;AAED,MAAM,kBAAkB,GAAsB;IAC5C,UAAU,EAAE,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE;QAClC,MAAM,OAAO,GAAG,eAAe,CAAC,eAAe,CAAC,QAAoC,CAAC;QACrF,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC;QACjC,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACpC,CAAC;QACD,OAAO,8BAA8B,CAAC;IACxC,CAAC;IACD,OAAO,EAAE,CAAC,EAAE,eAAe,EAAE,UAAU,EAAE,aAAa,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE;QAC7E,MAAM,OAAO,GAAG,eAAe,CAAC,eAAe,CAAC,QAAoC,CAAC;QACrF,MAAM,YAAY,GAAG,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC;QAChD,MAAM,MAAM,GAAG,YAAY,IAAI,UAAU,CAAC,mBAAmB,CAAC;QAC9D,8HAA8H;QAC9H,MAAM,OAAO,GAA2B,YAAY;YAClD,CAAC,CAAC,EAAE,WAAW,EAAE,GAAG,MAAM,EAAE,EAAE;YAC9B,CAAC,CAAC,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE,CAAC;QAE1C,MAAM,IAAI,GAAG,OAAO,CAAC,IAA0B,CAAC;QAChD,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,8BAA8B,CAAC;YAC7D,QAAQ;YACR,eAAe,EAAE;gBACf,aAAa,EAAE,aAAa,CAAC,SAAS,EAAE,aAAa;gBACrD,qBAAqB,EAAE,aAAa,CAAC,SAAS,EAAE,qBAAqB;gBACrE,uBAAuB,EAAE,UAAU,CAAC,uBAAuB;gBAC3D,gBAAgB,EAAE,aAAa,CAAC,SAAS,EAAE,gBAAgB;aAC5D;YACD,WAAW,EAAE,IAAI;SAClB,CAAC,CAAC;QACH,OAAO,CAAC,gBAAgB,CAAC,GAAG,UAAU,CAAC;QACvC,OAAO,CAAC,mBAAmB,CAAC,GAAG,OAAO,CAAC;QACvC,OAAO,OAAO,CAAC;IACjB,CAAC;IACD,WAAW,EAAE,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE;QACrC,MAAM,UAAU,GAAG,CAAC,qBAAqB,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACxE,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC;QAEtD,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,UAAU;gBACb,OAAO,WAAW,CAAC;YACrB,KAAK,cAAc;gBACjB,OAAO,WAAW,CAAC;YACrB,KAAK,UAAU;gBACb,OAAO,WAAW,CAAC;YACrB,KAAK,qBAAqB;gBACxB,OAAO,wBAAwB,CAAC;YAClC,KAAK,YAAY;gBACf,OAAO,QAAQ,CAAC;YAClB,KAAK,WAAW;gBACd,OAAO,QAAQ,CAAC;YAClB,KAAK,cAAc;gBACjB,OAAO,UAAU,MAAM,EAAE,CAAC;YAC5B,KAAK,YAAY;gBACf,OAAO,UAAU,MAAM,EAAE,CAAC;YAC5B,KAAK,qBAAqB;gBACxB,OAAO,UAAU,MAAM,UAAU,CAAC;YACpC;gBACE,OAAO,EAAE,CAAC;QACd,CAAC;IACH,CAAC;CACF,CAAC;AAEF,eAAe,kBAAkB,CAAC"}
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../src/providers/anthropic/api.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,mBAAmB,EAAE,MAAM,UAAU,CAAC;AAC/C,OAAO,EAAE,+BAA+B,EAAoC,MAAM,SAAS,CAAC;AAE5F,MAAM,cAAc,GAAG,CAAC,YAAY,EAAE,WAAW,EAAE,cAAc,EAAE,YAAY,EAAE,qBAAqB,CAAC,CAAC;AAExG,MAAM,CAAC,MAAM,yBAAyB,GAAG,YAAY,CAAC;AACtD,MAAM,CAAC,MAAM,6BAA6B,GAAG,qBAAqB,CAAC;AACnE,MAAM,CAAC,MAAM,qBAAqB,GAAG,sBAAsB,CAAC;AAI5D;;;;GAIG;AACH,MAAM,UAAU,8BAA8B,CAAC,EAC7C,QAAQ,EACR,eAAe,EACf,WAAW,GAKZ;IACC,8CAA8C;IAC9C,IAAI,iBAAiB,GAAG,cAAc,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,6BAA6B,CAAC;IAElH,6FAA6F;IAC7F,IAAI,QAAQ,KAAK,cAAc,IAAI,WAAW,IAAI,mBAAmB,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC5F,iBAAiB,GAAG,qBAAqB,CAAC;IAC5C,CAAC;IAED,8GAA8G;IAC9G,oFAAoF;IACpF,8DAA8D;IAC9D,MAAM,WAAW,GAAG,+BAA+B,CAAC,eAAe,EAAE,WAAW,EAAE,CAAC,gBAAgB,CAAC,CAAC,CAAC;IACtG,MAAM,UAAU,GAAG,WAAW,EAAE,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC;IACnF,MAAM,OAAO,GAAG,eAAe,EAAE,gBAAgB,IAAI,WAAW,EAAE,CAAC,mBAAmB,CAAC,IAAI,yBAAyB,CAAC;IAErH,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC;AACjC,CAAC;AAED,MAAM,kBAAkB,GAAsB;IAC5C,UAAU,EAAE,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE;QAClC,MAAM,OAAO,GAAG,eAAe,CAAC,eAAe,CAAC,QAAoC,CAAC;QACrF,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC;QACjC,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACpC,CAAC;QACD,OAAO,8BAA8B,CAAC;IACxC,CAAC;IACD,OAAO,EAAE,CAAC,EAAE,eAAe,EAAE,UAAU,EAAE,aAAa,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE;QAC7E,MAAM,OAAO,GAAG,eAAe,CAAC,eAAe,CAAC,QAAoC,CAAC;QACrF,MAAM,YAAY,GAAG,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC;QAChD,MAAM,MAAM,GAAG,YAAY,IAAI,UAAU,CAAC,mBAAmB,CAAC;QAC9D,6FAA6F;QAC7F,uFAAuF;QACvF,0FAA0F;QAC1F,MAAM,OAAO,GAA2B,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE,CAAC;QAE9E,MAAM,IAAI,GAAG,OAAO,CAAC,IAA0B,CAAC;QAChD,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,8BAA8B,CAAC;YAC7D,QAAQ;YACR,eAAe,EAAE;gBACf,aAAa,EAAE,aAAa,CAAC,SAAS,EAAE,aAAa;gBACrD,qBAAqB,EAAE,aAAa,CAAC,SAAS,EAAE,qBAAqB;gBACrE,uBAAuB,EAAE,UAAU,CAAC,uBAAuB;gBAC3D,gBAAgB,EAAE,aAAa,CAAC,SAAS,EAAE,gBAAgB;aAC5D;YACD,WAAW,EAAE,IAAI;SAClB,CAAC,CAAC;QACH,OAAO,CAAC,gBAAgB,CAAC,GAAG,UAAU,CAAC;QACvC,OAAO,CAAC,mBAAmB,CAAC,GAAG,OAAO,CAAC;QACvC,OAAO,OAAO,CAAC;IACjB,CAAC;IACD,WAAW,EAAE,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE;QACrC,MAAM,UAAU,GAAG,CAAC,qBAAqB,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACxE,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC;QAEtD,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,UAAU;gBACb,OAAO,WAAW,CAAC;YACrB,KAAK,cAAc;gBACjB,OAAO,WAAW,CAAC;YACrB,KAAK,UAAU;gBACb,OAAO,WAAW,CAAC;YACrB,KAAK,qBAAqB;gBACxB,OAAO,wBAAwB,CAAC;YAClC,KAAK,YAAY;gBACf,OAAO,QAAQ,CAAC;YAClB,KAAK,WAAW;gBACd,OAAO,QAAQ,CAAC;YAClB,KAAK,cAAc;gBACjB,OAAO,UAAU,MAAM,EAAE,CAAC;YAC5B,KAAK,YAAY;gBACf,OAAO,UAAU,MAAM,EAAE,CAAC;YAC5B,KAAK,qBAAqB;gBACxB,OAAO,UAAU,MAAM,UAAU,CAAC;YACpC;gBACE,OAAO,EAAE,CAAC;QACd,CAAC;IACH,CAAC;CACF,CAAC;AAEF,eAAe,kBAAkB,CAAC"}

package/dist/providers/truefoundry-managed/index.js

@@ -9,6 +9,8 @@ import { chatCompleteParams, completeParams, createModelResponseParams, embedPar
 import { OpenAILikeChatCompleteResponseTransform, OpenAILikeChatCompleteStreamChunkTransform, } from '../openai-like/chatComplete.js';
 import { OpenAILikeEmbedResponseTransform } from '../openai-like/embed.js';
 import TrueFoundryManagedAPIConfig from './api.js';
+import { withInsufficientCreditError } from './insufficientCredit.js';
+const { complete, ...baseResponseTransforms } = responseTransformers(TRUEFOUNDRY_MANAGED, { complete: true });
 const TrueFoundryManagedConfig = {
     chatComplete: chatCompleteParams([]),
     complete: completeParams([]),
@@ -16,10 +18,11 @@ const TrueFoundryManagedConfig = {
     embed: embedParams([]),
     api: TrueFoundryManagedAPIConfig,
     responseTransforms: {
-        ...responseTransformers(TRUEFOUNDRY_MANAGED, { complete: true }),
-        chatComplete: OpenAILikeChatCompleteResponseTransform(TRUEFOUNDRY_MANAGED),
-        embed: OpenAILikeEmbedResponseTransform(TRUEFOUNDRY_MANAGED),
-        createModelResponse: OpenAICreateModelResponseTransformer(TRUEFOUNDRY_MANAGED),
+        ...baseResponseTransforms,
+        complete: complete && withInsufficientCreditError(complete),
+        chatComplete: withInsufficientCreditError(OpenAILikeChatCompleteResponseTransform(TRUEFOUNDRY_MANAGED)),
+        embed: withInsufficientCreditError(OpenAILikeEmbedResponseTransform(TRUEFOUNDRY_MANAGED)),
+        createModelResponse: withInsufficientCreditError(OpenAICreateModelResponseTransformer(TRUEFOUNDRY_MANAGED)),
         'stream-chatComplete': OpenAILikeChatCompleteStreamChunkTransform(TRUEFOUNDRY_MANAGED),
     },
 };

package/dist/providers/truefoundry-managed/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/providers/truefoundry-managed/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AACH,OAAO,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AACpD,OAAO,EACL,kBAAkB,EAClB,cAAc,EACd,yBAAyB,EACzB,WAAW,EACX,oCAAoC,EACpC,oBAAoB,GACrB,MAAM,iBAAiB,CAAC;AACzB,OAAO,EACL,uCAAuC,EACvC,0CAA0C,GAC3C,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,gCAAgC,EAAE,MAAM,sBAAsB,CAAC;AAExE,OAAO,2BAA2B,MAAM,OAAO,CAAC;AAEhD,MAAM,wBAAwB,GAAoB;IAChD,YAAY,EAAE,kBAAkB,CAAC,EAAE,CAAC;IACpC,QAAQ,EAAE,cAAc,CAAC,EAAE,CAAC;IAC5B,mBAAmB,EAAE,yBAAyB,CAAC,EAAE,CAAC;IAClD,KAAK,EAAE,WAAW,CAAC,EAAE,CAAC;IACtB,GAAG,EAAE,2BAA2B;IAChC,kBAAkB,EAAE;QAClB,GAAG,oBAAoB,CAAC,mBAAmB,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;QAChE,YAAY,EAAE,uCAAuC,CAAC,mBAAmB,CAAC;QAC1E,KAAK,EAAE,gCAAgC,CAAC,mBAAmB,CAAC;QAC5D,mBAAmB,EAAE,oCAAoC,CAAC,mBAAmB,CAAC;QAC9E,qBAAqB,EAAE,0CAA0C,CAAC,mBAAmB,CAAC;KACvF;CACF,CAAC;AAEF,eAAe,wBAAwB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/providers/truefoundry-managed/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AACH,OAAO,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AACpD,OAAO,EACL,kBAAkB,EAClB,cAAc,EACd,yBAAyB,EACzB,WAAW,EACX,oCAAoC,EACpC,oBAAoB,GACrB,MAAM,iBAAiB,CAAC;AACzB,OAAO,EACL,uCAAuC,EACvC,0CAA0C,GAC3C,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,gCAAgC,EAAE,MAAM,sBAAsB,CAAC;AAExE,OAAO,2BAA2B,MAAM,OAAO,CAAC;AAChD,OAAO,EAAE,2BAA2B,EAAE,MAAM,sBAAsB,CAAC;AAEnE,MAAM,EAAE,QAAQ,EAAE,GAAG,sBAAsB,EAAE,GAAG,oBAAoB,CAAC,mBAAmB,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;AAE9G,MAAM,wBAAwB,GAAoB;IAChD,YAAY,EAAE,kBAAkB,CAAC,EAAE,CAAC;IACpC,QAAQ,EAAE,cAAc,CAAC,EAAE,CAAC;IAC5B,mBAAmB,EAAE,yBAAyB,CAAC,EAAE,CAAC;IAClD,KAAK,EAAE,WAAW,CAAC,EAAE,CAAC;IACtB,GAAG,EAAE,2BAA2B;IAChC,kBAAkB,EAAE;QAClB,GAAG,sBAAsB;QACzB,QAAQ,EAAE,QAAQ,IAAI,2BAA2B,CAAC,QAAQ,CAAC;QAC3D,YAAY,EAAE,2BAA2B,CAAC,uCAAuC,CAAC,mBAAmB,CAAC,CAAC;QACvG,KAAK,EAAE,2BAA2B,CAAC,gCAAgC,CAAC,mBAAmB,CAAC,CAAC;QACzF,mBAAmB,EAAE,2BAA2B,CAAC,oCAAoC,CAAC,mBAAmB,CAAC,CAAC;QAC3G,qBAAqB,EAAE,0CAA0C,CAAC,mBAAmB,CAAC;KACvF;CACF,CAAC;AAEF,eAAe,wBAAwB,CAAC"}

package/dist/providers/truefoundry-managed/insufficientCredit.d.ts

@@ -0,0 +1,6 @@
+export declare const INSUFFICIENT_CREDIT_ERROR: {
+    message: string;
+    type: string;
+    code: string;
+};
+export declare function withInsufficientCreditError(transform: Function): (response: unknown, responseStatus: number, ...rest: unknown[]) => any;

package/dist/providers/truefoundry-managed/insufficientCredit.js

@@ -0,0 +1,20 @@
+import { TRUEFOUNDRY_MANAGED } from '../../globals.js';
+// The upstream budget is the tenant's TrueFoundry credit, so its rule and reset window stay hidden.
+export const INSUFFICIENT_CREDIT_ERROR = {
+    message: "You've run out of credit for TrueFoundry Managed Models. Add credits to resume requests.",
+    type: 'insufficient_credit',
+    code: 'insufficient_credit',
+};
+const UPSTREAM_BUDGET_LIMIT_ORIGIN = 'budget_limit';
+function isUpstreamBudgetLimit(response, responseStatus) {
+    return (responseStatus !== 200 &&
+        typeof response === 'object' &&
+        response !== null &&
+        response.error_origin_level === UPSTREAM_BUDGET_LIMIT_ORIGIN);
+}
+export function withInsufficientCreditError(transform) {
+    return (response, responseStatus, ...rest) => isUpstreamBudgetLimit(response, responseStatus)
+        ? { error: { ...INSUFFICIENT_CREDIT_ERROR }, provider: TRUEFOUNDRY_MANAGED }
+        : transform(response, responseStatus, ...rest);
+}
+//# sourceMappingURL=insufficientCredit.js.map

package/dist/providers/truefoundry-managed/insufficientCredit.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"insufficientCredit.js","sourceRoot":"","sources":["../../../src/providers/truefoundry-managed/insufficientCredit.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AAEpD,oGAAoG;AACpG,MAAM,CAAC,MAAM,yBAAyB,GAAG;IACvC,OAAO,EAAE,0FAA0F;IACnG,IAAI,EAAE,qBAAqB;IAC3B,IAAI,EAAE,qBAAqB;CAC5B,CAAC;AAEF,MAAM,4BAA4B,GAAG,cAAc,CAAC;AAEpD,SAAS,qBAAqB,CAAC,QAAiB,EAAE,cAAsB;IACtE,OAAO,CACL,cAAc,KAAK,GAAG;QACtB,OAAO,QAAQ,KAAK,QAAQ;QAC5B,QAAQ,KAAK,IAAI;QAChB,QAA6C,CAAC,kBAAkB,KAAK,4BAA4B,CACnG,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,2BAA2B,CAAC,SAAmB;IAC7D,OAAO,CAAC,QAAiB,EAAE,cAAsB,EAAE,GAAG,IAAe,EAAE,EAAE,CACvE,qBAAqB,CAAC,QAAQ,EAAE,cAAc,CAAC;QAC7C,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,GAAG,yBAAyB,EAAE,EAAE,QAAQ,EAAE,mBAAmB,EAAE;QAC5E,CAAC,CAAC,SAAS,CAAC,QAAQ,EAAE,cAAc,EAAE,GAAG,IAAI,CAAC,CAAC;AACrD,CAAC"}

package/dist/types/requestBody.d.ts

@@ -190,8 +190,6 @@ export interface Options {
     anthropicBetaSuppressed?: boolean;
     anthropicVersion?: string;
     anthropicApiKey?: string;
-    /** `'api-key'` -> `X-API-Key`, `'bearer'` -> `Authorization: Bearer`. */
-    anthropicAuthScheme?: 'api-key' | 'bearer';
     /** Claude Platform on AWS workspace id, sent as the `anthropic-workspace-id` header. */
     awsClaudePlatformWorkspaceId?: string;
     /** Fireworks finetune required fields */

package/dist/types/requestBody.js.map

@@ -1 +1 @@
-{"version":3,"file":"requestBody.js","sourceRoot":"","sources":["../../src/types/requestBody.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAwBH,MAAM,CAAN,IAAY,aAKX;AALD,WAAY,aAAa;IACvB,4CAA2B,CAAA;IAC3B,sCAAqB,CAAA;IACrB,kCAAiB,CAAA;IACjB,4CAA2B,CAAA;AAC7B,CAAC,EALW,aAAa,KAAb,aAAa,QAKxB;AAyVD,MAAM,CAAN,IAAY,aAOX;AAPD,WAAY,aAAa;IACvB,kCAAiB,CAAA;IACjB,8BAAa,CAAA;IACb,wCAAuB,CAAA;IACvB,sCAAqB,CAAA;IACrB,8BAAa,CAAA;IACb,wCAAuB,CAAA;AACzB,CAAC,EAPW,aAAa,KAAb,aAAa,QAOxB;AAED,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC"}
+{"version":3,"file":"requestBody.js","sourceRoot":"","sources":["../../src/types/requestBody.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAwBH,MAAM,CAAN,IAAY,aAKX;AALD,WAAY,aAAa;IACvB,4CAA2B,CAAA;IAC3B,sCAAqB,CAAA;IACrB,kCAAiB,CAAA;IACjB,4CAA2B,CAAA;AAC7B,CAAC,EALW,aAAa,KAAb,aAAa,QAKxB;AAuVD,MAAM,CAAN,IAAY,aAOX;AAPD,WAAY,aAAa;IACvB,kCAAiB,CAAA;IACjB,8BAAa,CAAA;IACb,wCAAuB,CAAA;IACvB,sCAAqB,CAAA;IACrB,8BAAa,CAAA;IACb,wCAAuB,CAAA;AACzB,CAAC,EAPW,aAAa,KAAb,aAAa,QAOxB;AAED,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@truefoundry/ai-gateway-sdk",
-  "version": "0.0.7",
+  "version": "0.0.8",
   "description": "LLM provider translation SDK",
   "author": "Truefoundry",
   "license": "UNLICENSED",