@truefoundry/ai-gateway-sdk 0.0.12 → 0.0.13

package/dist/autogen/schema.json

@@ -9911,6 +9911,118 @@
             },
             "$schema": "http://json-schema.org/draft-04/schema#"
         },
+        "FireworksAIAPIKeyAuth": {
+            "description": "+label=Fireworks AI API Key Auth\n+icon=https://assets.production.truefoundry.com/fireworks.svg",
+            "type": "object",
+            "required": ["type", "api_key"],
+            "properties": {
+                "type": {
+                    "description": "+value=api-key",
+                    "type": "string",
+                    "enum": ["api-key"]
+                },
+                "api_key": {
+                    "description": "+label=API Key\n+sort=100\n+usage=The Fireworks AI 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#"
+        },
+        "FireworksAIIntegrations": {
+            "$ref": "#/definitions/FireworksAIModel",
+            "$schema": "http://json-schema.org/draft-04/schema#"
+        },
+        "FireworksAIModel": {
+            "description": "+label=Fireworks AI Model\n+icon=https://assets.production.truefoundry.com/fireworks.svg",
+            "type": "object",
+            "required": ["type", "name", "model_id", "model_types"],
+            "properties": {
+                "type": {
+                    "description": "+value=integration/model/fireworks-ai",
+                    "type": "string",
+                    "enum": ["integration/model/fireworks-ai"]
+                },
+                "name": {
+                    "description": "+label=Display Name\n+sort=1\n+usage=A descriptive name to identify this model integration in the UI\n+message=2 to 64 characters long alphanumeric word, may contain - , . or _ in between, cannot start with a number",
+                    "type": "string",
+                    "pattern": "^[a-zA-Z](?:[a-zA-Z0-9\\-._]{0,62}[a-zA-Z0-9])$"
+                },
+                "model_id": {
+                    "description": "+label=Model ID\n+sort=2\n+usage=The Fireworks AI model identifier, for example accounts/fireworks/models/kimi-k3. For a dedicated deployment use accounts/<account>/deployments/<deployment>.\n+message=Model ID must not be empty\n+uiProps={\"descriptionInline\":true}",
+                    "type": "string",
+                    "pattern": "^\\S+$"
+                },
+                "model_types": {
+                    "description": "+label=Model Types\n+sort=3\n+usage=Specify the type of the Fireworks AI model\n+uiProps={\"disableAllOption\": true}",
+                    "type": "array",
+                    "items": {
+                        "$ref": "#/definitions/ModelType"
+                    },
+                    "default": ["chat"]
+                },
+                "cost": {
+                    "$ref": "#/definitions/ModelCostMetric"
+                },
+                "authorized_subjects": {
+                    "description": "+label=Access Control\n+sort=5\n+usage=List of subjects that are authorized to access this integration. List of user fqn in format <user_type>:<username>.\n+uiType=Hidden",
+                    "type": "array",
+                    "items": {
+                        "type": "string"
+                    }
+                }
+            },
+            "$schema": "http://json-schema.org/draft-04/schema#"
+        },
+        "FireworksAIProviderAccount": {
+            "description": "+label=Fireworks AI Model Account\n+icon=https://assets.production.truefoundry.com/fireworks.svg",
+            "type": "object",
+            "required": ["type", "name", "auth_data", "integrations"],
+            "properties": {
+                "type": {
+                    "description": "+value=provider-account/fireworks-ai",
+                    "type": "string",
+                    "enum": ["provider-account/fireworks-ai"]
+                },
+                "name": {
+                    "description": "+label=Name\n+sort=100\n+usage=The name of the Fireworks AI model account\n+message=3 to 32 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,30}[a-z0-9]$"
+                },
+                "auth_data": {
+                    "$ref": "#/definitions/FireworksAIAPIKeyAuth"
+                },
+                "base_url": {
+                    "description": "+label=Base URL\n+sort=250\n+usage=Override the Fireworks AI API base URL. Defaults to https://api.fireworks.ai/inference/v1. For US-only serverless use https://us.api.fireworks.ai/inference/v1 together with the US router model IDs, for example accounts/fireworks/routers/kimi-k3-us.\n+message=Base URL must be an https URL without credentials, a query or a fragment\n+isAdvanced=true",
+                    "type": "string",
+                    "pattern": "^https://[^\\s/?#@]+(?:/[^\\s?#]*)?$"
+                },
+                "integrations": {
+                    "description": "+label=Integrations\n+sort=300\n+usage=List of integrations that are associated with the Fireworks AI model account\n+uiType=IntegrationsGroup",
+                    "type": "array",
+                    "items": {
+                        "$ref": "#/definitions/FireworksAIIntegrations"
+                    }
+                },
+                "collaborators": {
+                    "description": "+label=Collaborators\n+sort=400\n+usage=List of users who have access to this model account\n+uiType=Collaborators",
+                    "type": "array",
+                    "items": {
+                        "$ref": "#/definitions/Collaborator"
+                    }
+                },
+                "ownedBy": {
+                    "$ref": "#/definitions/OwnedBy"
+                },
+                "discount_percent": {
+                    "description": "+label=Discount %\n+sort=650\n+uiProps={\"allowDecimal\": true}\n+usage=Discount % applied to upstream list price for this model account.\n+placeholder=Value should be in the range of 0–100\n+isAdvanced=true",
+                    "type": "number",
+                    "minimum": 0,
+                    "maximum": 100
+                }
+            },
+            "$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",
@@ -17094,6 +17206,9 @@
                 {
                     "$ref": "#/definitions/WaferProviderAccount"
                 },
+                {
+                    "$ref": "#/definitions/FireworksAIProviderAccount"
+                },
                 {
                     "$ref": "#/definitions/TypeSafeProviderAccount"
                 },
@@ -20397,6 +20512,9 @@
                 {
                     "$ref": "#/definitions/WaferIntegrations"
                 },
+                {
+                    "$ref": "#/definitions/FireworksAIIntegrations"
+                },
                 {
                     "$ref": "#/definitions/TypeSafeIntegrations"
                 },
@@ -29517,6 +29635,18 @@
         "FiddlerKeyAuth": {
             "$ref": "#/definitions/FiddlerKeyAuth"
         },
+        "FireworksAIAPIKeyAuth": {
+            "$ref": "#/definitions/FireworksAIAPIKeyAuth"
+        },
+        "FireworksAIIntegrations": {
+            "$ref": "#/definitions/FireworksAIIntegrations"
+        },
+        "FireworksAIModel": {
+            "$ref": "#/definitions/FireworksAIModel"
+        },
+        "FireworksAIProviderAccount": {
+            "$ref": "#/definitions/FireworksAIProviderAccount"
+        },
         "FireworksApiKeyAuth": {
             "$ref": "#/definitions/FireworksApiKeyAuth"
         },

package/dist/entities/Provider.d.ts

@@ -30,6 +30,7 @@ export declare enum Provider {
     sambanova = "sambanova",
     cerebras = "cerebras",
     wafer = "wafer",
+    fireworks_ai = "fireworks-ai",
     typesafe = "typesafe",
     openrouter = "openrouter",
     snowflake_cortex = "snowflake-cortex",

package/dist/entities/Provider.js

@@ -31,6 +31,7 @@ export var Provider;
     Provider["sambanova"] = "sambanova";
     Provider["cerebras"] = "cerebras";
     Provider["wafer"] = "wafer";
+    Provider["fireworks_ai"] = "fireworks-ai";
     Provider["typesafe"] = "typesafe";
     Provider["openrouter"] = "openrouter";
     Provider["snowflake_cortex"] = "snowflake-cortex";

package/dist/entities/Provider.js.map

@@ -1 +1 @@
-{"version":3,"file":"Provider.js","sourceRoot":"","sources":["../../src/entities/Provider.ts"],"names":[],"mappings":"AAAA,MAAM,CAAN,IAAY,QA6CX;AA7CD,WAAY,QAAQ;IAClB,6BAAiB,CAAA;IACjB,kCAAsB,CAAA;IACtB,6BAAiB,CAAA;IACjB,8CAAkC,CAAA;IAClC,6BAAiB,CAAA;IACjB,uBAAW,CAAA;IACX,yBAAa,CAAA;IACb,mCAAuB,CAAA;IACvB,uDAA2C,CAAA;IAC3C,iCAAqB,CAAA;IACrB,mCAAuB,CAAA;IACvB,0BAAc,CAAA;IACd,yBAAa,CAAA;IACb,qCAAyB,CAAA;IACzB,2BAAe,CAAA;IACf,6BAAiB,CAAA;IACjB,yBAAa,CAAA;IACb,2CAA+B,CAAA;IAC/B,uCAA2B,CAAA;IAC3B,uCAA2B,CAAA;IAC3B,qDAAyC,CAAA;IACzC,2CAA+B,CAAA;IAC/B,mDAAuC,CAAA;IACvC,oCAAwB,CAAA;IACxB,oCAAwB,CAAA;IACxB,4CAAgC,CAAA;IAChC,qCAAyB,CAAA;IACzB,2CAA+B,CAAA;IAC/B,mCAAuB,CAAA;IACvB,iCAAqB,CAAA;IACrB,2BAAe,CAAA;IACf,iCAAqB,CAAA;IACrB,qCAAyB,CAAA;IACzB,iDAAqC,CAAA;IACrC,iCAAqB,CAAA;IACrB,uBAAW,CAAA;IACX,+BAAmB,CAAA;IACnB,iCAAqB,CAAA;IACrB,iCAAqB,CAAA;IACrB,2CAA+B,CAAA;IAC/B,qCAAyB,CAAA;IACzB,+CAAmC,CAAA;IACnC,uCAA2B,CAAA;IAC3B,uDAA2C,CAAA;AAC7C,CAAC,EA7CW,QAAQ,KAAR,QAAQ,QA6CnB"}
+{"version":3,"file":"Provider.js","sourceRoot":"","sources":["../../src/entities/Provider.ts"],"names":[],"mappings":"AAAA,MAAM,CAAN,IAAY,QA8CX;AA9CD,WAAY,QAAQ;IAClB,6BAAiB,CAAA;IACjB,kCAAsB,CAAA;IACtB,6BAAiB,CAAA;IACjB,8CAAkC,CAAA;IAClC,6BAAiB,CAAA;IACjB,uBAAW,CAAA;IACX,yBAAa,CAAA;IACb,mCAAuB,CAAA;IACvB,uDAA2C,CAAA;IAC3C,iCAAqB,CAAA;IACrB,mCAAuB,CAAA;IACvB,0BAAc,CAAA;IACd,yBAAa,CAAA;IACb,qCAAyB,CAAA;IACzB,2BAAe,CAAA;IACf,6BAAiB,CAAA;IACjB,yBAAa,CAAA;IACb,2CAA+B,CAAA;IAC/B,uCAA2B,CAAA;IAC3B,uCAA2B,CAAA;IAC3B,qDAAyC,CAAA;IACzC,2CAA+B,CAAA;IAC/B,mDAAuC,CAAA;IACvC,oCAAwB,CAAA;IACxB,oCAAwB,CAAA;IACxB,4CAAgC,CAAA;IAChC,qCAAyB,CAAA;IACzB,2CAA+B,CAAA;IAC/B,mCAAuB,CAAA;IACvB,iCAAqB,CAAA;IACrB,2BAAe,CAAA;IACf,yCAA6B,CAAA;IAC7B,iCAAqB,CAAA;IACrB,qCAAyB,CAAA;IACzB,iDAAqC,CAAA;IACrC,iCAAqB,CAAA;IACrB,uBAAW,CAAA;IACX,+BAAmB,CAAA;IACnB,iCAAqB,CAAA;IACrB,iCAAqB,CAAA;IACrB,2CAA+B,CAAA;IAC/B,qCAAyB,CAAA;IACzB,+CAAmC,CAAA;IACnC,uCAA2B,CAAA;IAC3B,uDAA2C,CAAA;AAC7C,CAAC,EA9CW,QAAQ,KAAR,QAAQ,QA8CnB"}

package/dist/handlers/tfyUtils.js

@@ -22,6 +22,7 @@ import DatabricksConfig from '../providers/databricks/index.js';
 import DeepgramConfig from '../providers/deepgram/index.js';
 import DeepInfraConfig from '../providers/deepinfra/index.js';
 import ElevenLabsConfig from '../providers/elevenlabs/index.js';
+import FireworksAIConfig from '../providers/fireworks-ai/index.js';
 import GoogleConfig from '../providers/google/index.js';
 import VertexConfig from '../providers/google-vertex-ai/index.js';
 import GroqConfig from '../providers/groq/index.js';
@@ -188,6 +189,9 @@ export const getProviderConfig = (providerAccount, model) => {
     if (model.manifest.type === 'integration/model/wafer') {
         return WaferConfig;
     }
+    if (model.manifest.type === 'integration/model/fireworks-ai') {
+        return FireworksAIConfig;
+    }
     if (model.manifest.type === 'integration/model/typesafe') {
         return TypeSafeConfig;
     }

package/dist/handlers/tfyUtils.js.map

@@ -1 +1 @@
-{"version":3,"file":"tfyUtils.js","sourceRoot":"","sources":["../../src/handlers/tfyUtils.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,OAAO,EAAS,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAE3D,OAAO,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC;AAC5C,OAAO,SAAS,MAAM,cAAc,CAAC;AACrC,OAAO,UAAU,MAAM,mBAAmB,CAAC;AAC3C,OAAO,eAAe,MAAM,wBAAwB,CAAC;AACrD,OAAO,sBAAsB,MAAM,iCAAiC,CAAC;AACrE,OAAO,uBAAuB,MAAM,kCAAkC,CAAC;AACvE,OAAO,EAAE,yBAAyB,EAAE,MAAM,iCAAiC,CAAC;AAC5E,OAAO,iBAAiB,MAAM,2BAA2B,CAAC;AAC1D,OAAO,aAAa,MAAM,sBAAsB,CAAC;AACjD,OAAO,aAAa,MAAM,sBAAsB,CAAC;AACjD,OAAO,cAAc,MAAM,uBAAuB,CAAC;AACnD,OAAO,cAAc,MAAM,uBAAuB,CAAC;AACnD,OAAO,cAAc,MAAM,uBAAuB,CAAC;AACnD,OAAO,YAAY,MAAM,qBAAqB,CAAC;AAC/C,OAAO,qBAAqB,MAAM,qBAAqB,CAAC;AACxD,OAAO,gBAAgB,MAAM,yBAAyB,CAAC;AACvD,OAAO,cAAc,MAAM,uBAAuB,CAAC;AACnD,OAAO,eAAe,MAAM,wBAAwB,CAAC;AACrD,OAAO,gBAAgB,MAAM,yBAAyB,CAAC;AACvD,OAAO,YAAY,MAAM,qBAAqB,CAAC;AAC/C,OAAO,YAAY,MAAM,+BAA+B,CAAC;AACzD,OAAO,UAAU,MAAM,mBAAmB,CAAC;AAC3C,OAAO,sBAAsB,MAAM,gCAAgC,CAAC;AACpE,OAAO,eAAe,MAAM,yBAAyB,CAAC;AACtD,OAAO,WAAW,MAAM,oBAAoB,CAAC;AAC7C,OAAO,YAAY,MAAM,qBAAqB,CAAC;AAC/C,OAAO,YAAY,MAAM,qBAAqB,CAAC;AAC/C,OAAO,kBAAkB,MAAM,4BAA4B,CAAC;AAC5D,OAAO,eAAe,MAAM,wBAAwB,CAAC;AACrD,OAAO,gBAAgB,MAAM,0BAA0B,CAAC;AACxD,OAAO,gBAAgB,MAAM,0BAA0B,CAAC;AACxD,OAAO,wBAAwB,MAAM,kCAAkC,CAAC;AAExE,OAAO,cAAc,MAAM,uBAAuB,CAAC;AACnD,OAAO,WAAW,MAAM,oBAAoB,CAAC;AAC7C,OAAO,SAAS,MAAM,kBAAkB,CAAC;AACzC,OAAO,EACL,oBAAoB,EACpB,aAAa,EACb,cAAc,EACd,mBAAmB,EACnB,sBAAsB,EACtB,SAAS,EACT,UAAU,GACX,MAAM,yBAAyB,CAAC;AAEjC;;;GAGG;AACH,MAAM,UAAU,gBAAgB,CAAC,KAA+B;IAC9D,IAAI,CAAC,KAAK;QAAE,OAAO,EAAE,CAAC;IACtB,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,2BAA2B,EAAE,CAAC;QACxD,MAAM,IAAI,eAAe,CAAC,kBAAkB,KAAK,CAAC,GAAG,qBAAqB,CAAC,CAAC;IAC9E,CAAC;IACD,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,mCAAmC,EAAE,CAAC;QAChE,MAAM,IAAI,eAAe,CAAC,kBAAkB,KAAK,CAAC,GAAG,uBAAuB,CAAC,CAAC;IAChF,CAAC;IACD,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,qCAAqC,EAAE,CAAC;QAClE,OAAO,KAAK,CAAC,QAAQ,CAAC,iBAAiB,IAAI,EAAE,CAAC;IAChD,CAAC;IACD,OAAO,KAAK,CAAC,QAAQ,CAAC,QAAQ,IAAI,EAAE,CAAC;AACvC,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,oBAAoB,CAAC,KAA+B;IAClE,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAC7B,MAAM,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC;IAC3B,IAAI,QAAQ,CAAC,IAAI,KAAK,gCAAgC,IAAI,QAAQ,CAAC,IAAI,KAAK,qCAAqC,EAAE,CAAC;QAClH,OAAO,QAAQ,CAAC,gBAAgB,IAAI,QAAQ,CAAC,QAAQ,IAAI,SAAS,CAAC;IACrE,CAAC;IACD,MAAM,aAAa,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;IAC9C,OAAO,aAAa,IAAI,SAAS,CAAC;AACpC,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,kBAAkB,CAAC,KAA+B;IAChE,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAC7B,MAAM,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC;IAC3B,IAAI,QAAQ,CAAC,IAAI,KAAK,gCAAgC,EAAE,CAAC;QACvD,OAAO,QAAQ,CAAC,WAAW,IAAI,SAAS,CAAC;IAC3C,CAAC;IACD,IAAI,QAAQ,CAAC,IAAI,KAAK,qCAAqC,EAAE,CAAC;QAC5D,OAAO,QAAQ,CAAC,WAAW,CAAC;IAC9B,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAC/B,eAAsC,EACtC,KAAwB,EACP,EAAE;IACnB,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,cAAc,GAAG,SAAS,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC3D,OAAO,cAAc,CAAC;IACxB,CAAC;IACD,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,wBAAwB,EAAE,CAAC;QACrD,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,6BAA6B,EAAE,CAAC;QAC1D,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,uCAAuC,EAAE,CAAC;QACpE,OAAO,uBAAuB,CAAC;IACjC,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,gCAAgC,EAAE,CAAC;QAC7D,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,0BAA0B,EAAE,CAAC;QACvD,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,iCAAiC,EAAE,CAAC;QAC9D,OAAO,yBAAyB,CAAC;IACnC,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,qCAAqC,EAAE,CAAC;QAClE,OAAO,sBAAsB,CAAC;IAChC,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,qCAAqC,EAAE,CAAC;QAClE,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,KAAK,eAAe,CAAC,WAAW,EAAE,CAAC;YAChE,OAAO,UAAU,CAAC;QACpB,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,KAAK,eAAe,CAAC,GAAG,EAAE,CAAC;YAC/D,OAAO,SAAS,CAAC;QACnB,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,KAAK,eAAe,CAAC,QAAQ,EAAE,CAAC;YACpE,OAAO,cAAc,CAAC;QACxB,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,KAAK,eAAe,CAAC,cAAc,EAAE,CAAC;YAC1E,OAAO,mBAAmB,CAAC;QAC7B,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,KAAK,eAAe,CAAC,iBAAiB,EAAE,CAAC;YAC7E,OAAO,sBAAsB,CAAC;QAChC,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,KAAK,eAAe,CAAC,gBAAgB,EAAE,CAAC;YAC5E,OAAO,oBAAoB,CAAC;QAC9B,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,KAAK,eAAe,CAAC,OAAO,EAAE,CAAC;YACnE,OAAO,aAAa,CAAC;QACvB,CAAC;IACH,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,6BAA6B,EAAE,CAAC;QAC1D,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,wBAAwB,EAAE,CAAC;QACrD,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,8BAA8B,EAAE,CAAC;QAC3D,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,yBAAyB,EAAE,CAAC;QACtD,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,0BAA0B,EAAE,CAAC;QACvD,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,0BAA0B,EAAE,CAAC;QACvD,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,iCAAiC,EAAE,CAAC;QAC9D,OAAO,kBAAkB,CAAC;IAC5B,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,+BAA+B,EAAE,CAAC;QAC5D,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,8BAA8B,EAAE,CAAC;QAC3D,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,4BAA4B,EAAE,CAAC;QACzD,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,2BAA2B,EAAE,CAAC;QACxD,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,kCAAkC,EAAE,CAAC;QAC/D,OAAO,sBAAsB,CAAC;IAChC,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,0BAA0B,EAAE,CAAC;QACvD,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,0BAA0B,EAAE,CAAC;QACvD,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,6BAA6B,EAAE,CAAC;QAC1D,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,4BAA4B,EAAE,CAAC;QACzD,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,yBAAyB,EAAE,CAAC;QACtD,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,4BAA4B,EAAE,CAAC;QACzD,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,oCAAoC,EAAE,CAAC;QACjE,OAAO,qBAAqB,CAAC;IAC/B,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,uBAAuB,EAAE,CAAC;QACpD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,2BAA2B,EAAE,CAAC;QACxD,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,uCAAuC,EAAE,CAAC;QACpE,OAAO,wBAAwB,CAAC;IAClC,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,8BAA8B,EAAE,CAAC;QAC3D,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,4BAA4B,EAAE,CAAC;QACzD,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,4BAA4B,EAAE,CAAC;QACzD,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,+BAA+B,EAAE,CAAC;QAC5D,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAED,MAAM,cAAc,GAAG,SAAS,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;IAC3D,OAAO,cAAc,CAAC;AACxB,CAAC,CAAC"}
+{"version":3,"file":"tfyUtils.js","sourceRoot":"","sources":["../../src/handlers/tfyUtils.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,OAAO,EAAS,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAE3D,OAAO,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC;AAC5C,OAAO,SAAS,MAAM,cAAc,CAAC;AACrC,OAAO,UAAU,MAAM,mBAAmB,CAAC;AAC3C,OAAO,eAAe,MAAM,wBAAwB,CAAC;AACrD,OAAO,sBAAsB,MAAM,iCAAiC,CAAC;AACrE,OAAO,uBAAuB,MAAM,kCAAkC,CAAC;AACvE,OAAO,EAAE,yBAAyB,EAAE,MAAM,iCAAiC,CAAC;AAC5E,OAAO,iBAAiB,MAAM,2BAA2B,CAAC;AAC1D,OAAO,aAAa,MAAM,sBAAsB,CAAC;AACjD,OAAO,aAAa,MAAM,sBAAsB,CAAC;AACjD,OAAO,cAAc,MAAM,uBAAuB,CAAC;AACnD,OAAO,cAAc,MAAM,uBAAuB,CAAC;AACnD,OAAO,cAAc,MAAM,uBAAuB,CAAC;AACnD,OAAO,YAAY,MAAM,qBAAqB,CAAC;AAC/C,OAAO,qBAAqB,MAAM,qBAAqB,CAAC;AACxD,OAAO,gBAAgB,MAAM,yBAAyB,CAAC;AACvD,OAAO,cAAc,MAAM,uBAAuB,CAAC;AACnD,OAAO,eAAe,MAAM,wBAAwB,CAAC;AACrD,OAAO,gBAAgB,MAAM,yBAAyB,CAAC;AACvD,OAAO,iBAAiB,MAAM,2BAA2B,CAAC;AAC1D,OAAO,YAAY,MAAM,qBAAqB,CAAC;AAC/C,OAAO,YAAY,MAAM,+BAA+B,CAAC;AACzD,OAAO,UAAU,MAAM,mBAAmB,CAAC;AAC3C,OAAO,sBAAsB,MAAM,gCAAgC,CAAC;AACpE,OAAO,eAAe,MAAM,yBAAyB,CAAC;AACtD,OAAO,WAAW,MAAM,oBAAoB,CAAC;AAC7C,OAAO,YAAY,MAAM,qBAAqB,CAAC;AAC/C,OAAO,YAAY,MAAM,qBAAqB,CAAC;AAC/C,OAAO,kBAAkB,MAAM,4BAA4B,CAAC;AAC5D,OAAO,eAAe,MAAM,wBAAwB,CAAC;AACrD,OAAO,gBAAgB,MAAM,0BAA0B,CAAC;AACxD,OAAO,gBAAgB,MAAM,0BAA0B,CAAC;AACxD,OAAO,wBAAwB,MAAM,kCAAkC,CAAC;AAExE,OAAO,cAAc,MAAM,uBAAuB,CAAC;AACnD,OAAO,WAAW,MAAM,oBAAoB,CAAC;AAC7C,OAAO,SAAS,MAAM,kBAAkB,CAAC;AACzC,OAAO,EACL,oBAAoB,EACpB,aAAa,EACb,cAAc,EACd,mBAAmB,EACnB,sBAAsB,EACtB,SAAS,EACT,UAAU,GACX,MAAM,yBAAyB,CAAC;AAEjC;;;GAGG;AACH,MAAM,UAAU,gBAAgB,CAAC,KAA+B;IAC9D,IAAI,CAAC,KAAK;QAAE,OAAO,EAAE,CAAC;IACtB,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,2BAA2B,EAAE,CAAC;QACxD,MAAM,IAAI,eAAe,CAAC,kBAAkB,KAAK,CAAC,GAAG,qBAAqB,CAAC,CAAC;IAC9E,CAAC;IACD,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,mCAAmC,EAAE,CAAC;QAChE,MAAM,IAAI,eAAe,CAAC,kBAAkB,KAAK,CAAC,GAAG,uBAAuB,CAAC,CAAC;IAChF,CAAC;IACD,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,qCAAqC,EAAE,CAAC;QAClE,OAAO,KAAK,CAAC,QAAQ,CAAC,iBAAiB,IAAI,EAAE,CAAC;IAChD,CAAC;IACD,OAAO,KAAK,CAAC,QAAQ,CAAC,QAAQ,IAAI,EAAE,CAAC;AACvC,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,oBAAoB,CAAC,KAA+B;IAClE,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAC7B,MAAM,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC;IAC3B,IAAI,QAAQ,CAAC,IAAI,KAAK,gCAAgC,IAAI,QAAQ,CAAC,IAAI,KAAK,qCAAqC,EAAE,CAAC;QAClH,OAAO,QAAQ,CAAC,gBAAgB,IAAI,QAAQ,CAAC,QAAQ,IAAI,SAAS,CAAC;IACrE,CAAC;IACD,MAAM,aAAa,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;IAC9C,OAAO,aAAa,IAAI,SAAS,CAAC;AACpC,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,kBAAkB,CAAC,KAA+B;IAChE,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAC7B,MAAM,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC;IAC3B,IAAI,QAAQ,CAAC,IAAI,KAAK,gCAAgC,EAAE,CAAC;QACvD,OAAO,QAAQ,CAAC,WAAW,IAAI,SAAS,CAAC;IAC3C,CAAC;IACD,IAAI,QAAQ,CAAC,IAAI,KAAK,qCAAqC,EAAE,CAAC;QAC5D,OAAO,QAAQ,CAAC,WAAW,CAAC;IAC9B,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAC/B,eAAsC,EACtC,KAAwB,EACP,EAAE;IACnB,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,cAAc,GAAG,SAAS,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC3D,OAAO,cAAc,CAAC;IACxB,CAAC;IACD,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,wBAAwB,EAAE,CAAC;QACrD,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,6BAA6B,EAAE,CAAC;QAC1D,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,uCAAuC,EAAE,CAAC;QACpE,OAAO,uBAAuB,CAAC;IACjC,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,gCAAgC,EAAE,CAAC;QAC7D,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,0BAA0B,EAAE,CAAC;QACvD,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,iCAAiC,EAAE,CAAC;QAC9D,OAAO,yBAAyB,CAAC;IACnC,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,qCAAqC,EAAE,CAAC;QAClE,OAAO,sBAAsB,CAAC;IAChC,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,qCAAqC,EAAE,CAAC;QAClE,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,KAAK,eAAe,CAAC,WAAW,EAAE,CAAC;YAChE,OAAO,UAAU,CAAC;QACpB,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,KAAK,eAAe,CAAC,GAAG,EAAE,CAAC;YAC/D,OAAO,SAAS,CAAC;QACnB,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,KAAK,eAAe,CAAC,QAAQ,EAAE,CAAC;YACpE,OAAO,cAAc,CAAC;QACxB,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,KAAK,eAAe,CAAC,cAAc,EAAE,CAAC;YAC1E,OAAO,mBAAmB,CAAC;QAC7B,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,KAAK,eAAe,CAAC,iBAAiB,EAAE,CAAC;YAC7E,OAAO,sBAAsB,CAAC;QAChC,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,KAAK,eAAe,CAAC,gBAAgB,EAAE,CAAC;YAC5E,OAAO,oBAAoB,CAAC;QAC9B,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,KAAK,eAAe,CAAC,OAAO,EAAE,CAAC;YACnE,OAAO,aAAa,CAAC;QACvB,CAAC;IACH,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,6BAA6B,EAAE,CAAC;QAC1D,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,wBAAwB,EAAE,CAAC;QACrD,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,8BAA8B,EAAE,CAAC;QAC3D,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,yBAAyB,EAAE,CAAC;QACtD,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,0BAA0B,EAAE,CAAC;QACvD,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,0BAA0B,EAAE,CAAC;QACvD,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,iCAAiC,EAAE,CAAC;QAC9D,OAAO,kBAAkB,CAAC;IAC5B,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,+BAA+B,EAAE,CAAC;QAC5D,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,8BAA8B,EAAE,CAAC;QAC3D,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,4BAA4B,EAAE,CAAC;QACzD,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,2BAA2B,EAAE,CAAC;QACxD,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,kCAAkC,EAAE,CAAC;QAC/D,OAAO,sBAAsB,CAAC;IAChC,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,0BAA0B,EAAE,CAAC;QACvD,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,0BAA0B,EAAE,CAAC;QACvD,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,6BAA6B,EAAE,CAAC;QAC1D,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,4BAA4B,EAAE,CAAC;QACzD,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,yBAAyB,EAAE,CAAC;QACtD,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,gCAAgC,EAAE,CAAC;QAC7D,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,4BAA4B,EAAE,CAAC;QACzD,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,oCAAoC,EAAE,CAAC;QACjE,OAAO,qBAAqB,CAAC;IAC/B,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,uBAAuB,EAAE,CAAC;QACpD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,2BAA2B,EAAE,CAAC;QACxD,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,uCAAuC,EAAE,CAAC;QACpE,OAAO,wBAAwB,CAAC;IAClC,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,8BAA8B,EAAE,CAAC;QAC3D,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,4BAA4B,EAAE,CAAC;QACzD,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,4BAA4B,EAAE,CAAC;QACzD,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,+BAA+B,EAAE,CAAC;QAC5D,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAED,MAAM,cAAc,GAAG,SAAS,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;IAC3D,OAAO,cAAc,CAAC;AACxB,CAAC,CAAC"}

package/dist/index.d.ts

@@ -4,6 +4,7 @@
  */
 export { initAdapters } from './adapters/index.js';
 export type { ContentItem, ExtractedMessages, GuardrailHelpers, GuardrailsRunConfig, GuardrailsScope, HookEventType, MessageLocation, } from './adapters/index.js';
+export { runWithProviderResponseCapture, type ProviderResponseCaptureHandler, type ProviderResponseCaptureMeta, } from './parity/providerResponseCapture.js';
 export { transformCapturedProviderResponse, type CapturedProviderResponse, type TransformCapturedProviderResponseInput, type TransformedClientResponse, } from './parity/transformCapturedProviderResponse.js';
 export { getSdkRequestTenantName, runWithSdkRequestContext } from './requestContext.js';
 export { LlmGatewaySdk } from './sdk.js';

package/dist/index.js

@@ -3,6 +3,7 @@
  * Internal modules (providers, bridges, transforms, stream pumps internals) are not exported.
  */
 export { initAdapters } from './adapters/index.js';
+export { runWithProviderResponseCapture, } from './parity/providerResponseCapture.js';
 export { transformCapturedProviderResponse, } from './parity/transformCapturedProviderResponse.js';
 export { getSdkRequestTenantName, runWithSdkRequestContext } from './requestContext.js';
 export { LlmGatewaySdk } from './sdk.js';

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAU1C,OAAO,EACL,iCAAiC,GAIlC,MAAM,4CAA4C,CAAC;AACpD,OAAO,EAAE,uBAAuB,EAAE,wBAAwB,EAAE,MAAM,kBAAkB,CAAC;AACrF,OAAO,EAAE,aAAa,EAAE,MAAM,OAAO,CAAC;AACtC,OAAO,EACL,yBAAyB,EACzB,yBAAyB,EACzB,0BAA0B,EAC1B,qBAAqB,EACrB,aAAa,EACb,wBAAwB,GACzB,MAAM,UAAU,CAAC;AAkElB,OAAO,EAAE,2BAA2B,EAAE,MAAM,qCAAqC,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAU1C,OAAO,EACL,8BAA8B,GAG/B,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EACL,iCAAiC,GAIlC,MAAM,4CAA4C,CAAC;AACpD,OAAO,EAAE,uBAAuB,EAAE,wBAAwB,EAAE,MAAM,kBAAkB,CAAC;AACrF,OAAO,EAAE,aAAa,EAAE,MAAM,OAAO,CAAC;AACtC,OAAO,EACL,yBAAyB,EACzB,yBAAyB,EACzB,0BAA0B,EAC1B,qBAAqB,EACrB,aAAa,EACb,wBAAwB,GACzB,MAAM,UAAU,CAAC;AAkElB,OAAO,EAAE,2BAA2B,EAAE,MAAM,qCAAqC,CAAC"}

package/dist/parity/providerResponseCapture.d.ts

@@ -0,0 +1,18 @@
+import type { Endpoint } from '../providers/types.js';
+import type { SdkRequestBody } from '../types/contracts.js';
+export interface ProviderResponseCaptureMeta {
+    url: string;
+    endpoint: Endpoint;
+    postAdapterBody: SdkRequestBody;
+    needsResponseTranslation: boolean;
+    openaiUseResponsesApi?: boolean;
+    contentType: string;
+    /** Aborted when this attempt fails. One cancelled tee branch does not cancel the upstream. */
+    signal: AbortSignal;
+}
+export type ProviderResponseCaptureHandler = (body: ReadableStream<Uint8Array>, meta: ProviderResponseCaptureMeta) => void;
+export declare function runWithProviderResponseCapture<T>(handler: ProviderResponseCaptureHandler, fn: () => Promise<T>): Promise<T>;
+export declare function abortProviderResponseCapture(): void;
+/** This attempt's abort. A later signal listener must call this, not look the store up again. */
+export declare function providerResponseCaptureAbort(): (() => void) | undefined;
+export declare function emitProviderResponseCapture(response: Response, meta: Omit<ProviderResponseCaptureMeta, 'contentType' | 'signal'>): Response;

package/dist/parity/providerResponseCapture.js

@@ -0,0 +1,35 @@
+import { AsyncLocalStorage } from 'node:async_hooks';
+import { logger } from '../logger.js';
+const providerResponseCaptureAls = new AsyncLocalStorage();
+export function runWithProviderResponseCapture(handler, fn) {
+    const attempt = new AbortController();
+    return providerResponseCaptureAls.run({ handler, signal: attempt.signal, abort: () => attempt.abort() }, fn);
+}
+export function abortProviderResponseCapture() {
+    providerResponseCaptureAls.getStore()?.abort();
+}
+/** This attempt's abort. A later signal listener must call this, not look the store up again. */
+export function providerResponseCaptureAbort() {
+    return providerResponseCaptureAls.getStore()?.abort;
+}
+export function emitProviderResponseCapture(response, meta) {
+    const store = providerResponseCaptureAls.getStore();
+    if (!store || response.status !== 200 || !response.body)
+        return response;
+    const [live, capture] = response.body.tee();
+    try {
+        store.handler(capture, {
+            ...meta,
+            contentType: response.headers.get('content-type') || '',
+            signal: store.signal,
+        });
+    }
+    catch (err) {
+        void capture.cancel();
+        logger.warn('provider response capture handler failed', {
+            error: err instanceof Error ? err.name : 'unknown',
+        });
+    }
+    return new Response(live, response);
+}
+//# sourceMappingURL=providerResponseCapture.js.map

package/dist/parity/providerResponseCapture.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"providerResponseCapture.js","sourceRoot":"","sources":["../../src/parity/providerResponseCapture.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AA0BnC,MAAM,0BAA0B,GAAG,IAAI,iBAAiB,EAAgC,CAAC;AAEzF,MAAM,UAAU,8BAA8B,CAC5C,OAAuC,EACvC,EAAoB;IAEpB,MAAM,OAAO,GAAG,IAAI,eAAe,EAAE,CAAC;IACtC,OAAO,0BAA0B,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;AAC/G,CAAC;AAED,MAAM,UAAU,4BAA4B;IAC1C,0BAA0B,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,CAAC;AACjD,CAAC;AAED,iGAAiG;AACjG,MAAM,UAAU,4BAA4B;IAC1C,OAAO,0BAA0B,CAAC,QAAQ,EAAE,EAAE,KAAK,CAAC;AACtD,CAAC;AAED,MAAM,UAAU,2BAA2B,CACzC,QAAkB,EAClB,IAAiE;IAEjE,MAAM,KAAK,GAAG,0BAA0B,CAAC,QAAQ,EAAE,CAAC;IACpD,IAAI,CAAC,KAAK,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI;QAAE,OAAO,QAAQ,CAAC;IACzE,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,QAAQ,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;IAC5C,IAAI,CAAC;QACH,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE;YACrB,GAAG,IAAI;YACP,WAAW,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE;YACvD,MAAM,EAAE,KAAK,CAAC,MAAM;SACrB,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC;QACtB,MAAM,CAAC,IAAI,CAAC,0CAA0C,EAAE;YACtD,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;SACnD,CAAC,CAAC;IACL,CAAC;IACD,OAAO,IAAI,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;AACtC,CAAC"}

package/dist/providers/fireworks-ai/api.d.ts

@@ -0,0 +1,2 @@
+import type { ProviderAPIConfig } from '../types.js';
+export declare const fireworksAIAPIConfig: ProviderAPIConfig;

package/dist/providers/fireworks-ai/api.js

@@ -0,0 +1,25 @@
+const DEFAULT_BASE_URL = 'https://api.fireworks.ai/inference/v1';
+const TRAILING_SLASHES_REGEX = /\/+$/;
+export const fireworksAIAPIConfig = {
+    getBaseURL: ({ providerOptions }) => {
+        const account = providerOptions.providerAccount.manifest;
+        return (account.base_url || DEFAULT_BASE_URL).replace(TRAILING_SLASHES_REGEX, '');
+    },
+    headers: ({ providerOptions }) => {
+        const account = providerOptions.providerAccount.manifest;
+        return { Authorization: `Bearer ${account.auth_data.api_key}` };
+    },
+    getEndpoint: ({ endpoint }) => {
+        switch (endpoint) {
+            case 'chatComplete':
+                return '/chat/completions';
+            case 'complete':
+                return '/completions';
+            case 'embed':
+                return '/embeddings';
+            default:
+                return '';
+        }
+    },
+};
+//# sourceMappingURL=api.js.map

package/dist/providers/fireworks-ai/api.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../src/providers/fireworks-ai/api.ts"],"names":[],"mappings":"AAGA,MAAM,gBAAgB,GAAG,uCAAuC,CAAC;AACjE,MAAM,sBAAsB,GAAG,MAAM,CAAC;AAEtC,MAAM,CAAC,MAAM,oBAAoB,GAAsB;IACrD,UAAU,EAAE,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE;QAClC,MAAM,OAAO,GAAG,eAAe,CAAC,eAAe,CAAC,QAAsC,CAAC;QACvF,OAAO,CAAC,OAAO,CAAC,QAAQ,IAAI,gBAAgB,CAAC,CAAC,OAAO,CAAC,sBAAsB,EAAE,EAAE,CAAC,CAAC;IACpF,CAAC;IACD,OAAO,EAAE,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE;QAC/B,MAAM,OAAO,GAAG,eAAe,CAAC,eAAe,CAAC,QAAsC,CAAC;QACvF,OAAO,EAAE,aAAa,EAAE,UAAU,OAAO,CAAC,SAAS,CAAC,OAAO,EAAE,EAAE,CAAC;IAClE,CAAC;IACD,WAAW,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE;QAC5B,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,cAAc;gBACjB,OAAO,mBAAmB,CAAC;YAC7B,KAAK,UAAU;gBACb,OAAO,cAAc,CAAC;YACxB,KAAK,OAAO;gBACV,OAAO,aAAa,CAAC;YACvB;gBACE,OAAO,EAAE,CAAC;QACd,CAAC;IACH,CAAC;CACF,CAAC"}

package/dist/providers/fireworks-ai/index.d.ts

@@ -0,0 +1,3 @@
+import { ProviderConfigs } from '../types.js';
+export declare const FireworksAIConfig: ProviderConfigs;
+export default FireworksAIConfig;

package/dist/providers/fireworks-ai/index.js

@@ -0,0 +1,23 @@
+import { FIREWORKS_AI } from '../../globals.js';
+import { chatCompleteParams, completeParams, embedParams, responseTransformers } from '../open-ai-base/index.js';
+import { OpenAILikeChatCompleteResponseTransform, OpenAILikeChatCompleteStreamChunkTransform, } from '../openai-like/chatComplete.js';
+import { OpenAILikeEmbedResponseTransform } from '../openai-like/embed.js';
+import { fireworksAIAPIConfig } from './api.js';
+const { complete: completeResponseTransform } = responseTransformers(FIREWORKS_AI, { complete: true });
+// Fields without a mapping here (top_k, reasoning_effort, prompt_cache_key, ...) are forwarded as sent.
+// Fireworks rejects fields it does not define with a 400 ("Extra inputs are not permitted"), so
+// OpenAI-only fields such as web_search_options or suffix surface as that error instead of being dropped.
+export const FireworksAIConfig = {
+    chatComplete: chatCompleteParams([]),
+    complete: completeParams([]),
+    embed: embedParams([]),
+    api: fireworksAIAPIConfig,
+    responseTransforms: {
+        complete: completeResponseTransform,
+        chatComplete: OpenAILikeChatCompleteResponseTransform(FIREWORKS_AI),
+        'stream-chatComplete': OpenAILikeChatCompleteStreamChunkTransform(FIREWORKS_AI),
+        embed: OpenAILikeEmbedResponseTransform(FIREWORKS_AI),
+    },
+};
+export default FireworksAIConfig;
+//# sourceMappingURL=index.js.map

package/dist/providers/fireworks-ai/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/providers/fireworks-ai/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,kBAAkB,EAAE,cAAc,EAAE,WAAW,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AACxG,OAAO,EACL,uCAAuC,EACvC,0CAA0C,GAC3C,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,gCAAgC,EAAE,MAAM,sBAAsB,CAAC;AAExE,OAAO,EAAE,oBAAoB,EAAE,MAAM,OAAO,CAAC;AAE7C,MAAM,EAAE,QAAQ,EAAE,yBAAyB,EAAE,GAAG,oBAAoB,CAAC,YAAY,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;AAEvG,wGAAwG;AACxG,gGAAgG;AAChG,0GAA0G;AAC1G,MAAM,CAAC,MAAM,iBAAiB,GAAoB;IAChD,YAAY,EAAE,kBAAkB,CAAC,EAAE,CAAC;IACpC,QAAQ,EAAE,cAAc,CAAC,EAAE,CAAC;IAC5B,KAAK,EAAE,WAAW,CAAC,EAAE,CAAC;IACtB,GAAG,EAAE,oBAAoB;IACzB,kBAAkB,EAAE;QAClB,QAAQ,EAAE,yBAAyB;QACnC,YAAY,EAAE,uCAAuC,CAAC,YAAY,CAAC;QACnE,qBAAqB,EAAE,0CAA0C,CAAC,YAAY,CAAC;QAC/E,KAAK,EAAE,gCAAgC,CAAC,YAAY,CAAC;KACtD;CACF,CAAC;AAEF,eAAe,iBAAiB,CAAC"}

package/dist/providers/index.js

@@ -23,6 +23,7 @@ import CerebrasConfig from './cerebras/index.js';
 import CortexConfig from './cortex/index.js';
 import DeepgramConfig from './deepgram/index.js';
 import ElevenLabsConfig from './elevenlabs/index.js';
+import FireworksAIConfig from './fireworks-ai/index.js';
 import GoogleConfig from './google/index.js';
 import VertexConfig from './google-vertex-ai/index.js';
 import GroqConfig from './groq/index.js';
@@ -73,6 +74,7 @@ const Providers = {
     sambanova: SambaNovaConfig,
     cerebras: CerebrasConfig,
     wafer: WaferConfig,
+    'fireworks-ai': FireworksAIConfig,
     typesafe: TypeSafeConfig,
     'snowflake-cortex': CortexConfig,
     cloudera: ClouderaConfig,

package/dist/providers/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,UAAU,MAAM,QAAQ,CAAC;AAChC,OAAO,eAAe,MAAM,aAAa,CAAC;AAC1C,OAAO,cAAc,MAAM,YAAY,CAAC;AACxC,OAAO,sBAAsB,MAAM,sBAAsB,CAAC;AAC1D,OAAO,uBAAuB,MAAM,uBAAuB,CAAC;AAC5D,OAAO,iBAAiB,MAAM,gBAAgB,CAAC;AAC/C,OAAO,aAAa,MAAM,WAAW,CAAC;AACtC,OAAO,cAAc,MAAM,YAAY,CAAC;AACxC,OAAO,YAAY,MAAM,UAAU,CAAC;AACpC,OAAO,eAAe,MAAM,aAAa,CAAC;AAC1C,OAAO,eAAe,MAAM,cAAc,CAAC;AAC3C,OAAO,WAAW,MAAM,SAAS,CAAC;AAClC,OAAO,YAAY,MAAM,UAAU,CAAC;AACpC,qCAAqC;AACrC,OAAO,aAAa,MAAM,WAAW,CAAC;AACtC,OAAO,cAAc,MAAM,YAAY,CAAC;AACxC,OAAO,cAAc,MAAM,YAAY,CAAC;AACxC,OAAO,YAAY,MAAM,UAAU,CAAC;AACpC,OAAO,cAAc,MAAM,YAAY,CAAC;AACxC,OAAO,gBAAgB,MAAM,cAAc,CAAC;AAC5C,OAAO,YAAY,MAAM,UAAU,CAAC;AACpC,OAAO,YAAY,MAAM,oBAAoB,CAAC;AAC9C,OAAO,UAAU,MAAM,QAAQ,CAAC;AAChC,OAAO,sBAAsB,MAAM,qBAAqB,CAAC;AACzD,OAAO,eAAe,MAAM,UAAU,CAAC;AACvC,OAAO,gBAAgB,MAAM,cAAc,CAAC;AAC5C,OAAO,kBAAkB,MAAM,iBAAiB,CAAC;AACjD,OAAO,eAAe,MAAM,aAAa,CAAC;AAC1C,OAAO,eAAe,MAAM,aAAa,CAAC;AAC1C,OAAO,aAAa,MAAM,WAAW,CAAC;AACtC,OAAO,gBAAgB,MAAM,eAAe,CAAC;AAC7C,OAAO,iBAAiB,MAAM,gBAAgB,CAAC;AAC/C,OAAO,gBAAgB,MAAM,eAAe,CAAC;AAC7C,OAAO,wBAAwB,MAAM,uBAAuB,CAAC;AAE7D,OAAO,cAAc,MAAM,YAAY,CAAC;AACxC,OAAO,WAAW,MAAM,SAAS,CAAC;AAClC,OAAO,SAAS,MAAM,OAAO,CAAC;AAE9B,MAAM,SAAS,GAAuC;IACpD,MAAM,EAAE,YAAY;IACpB,MAAM,EAAE,YAAY;IACpB,SAAS,EAAE,eAAe;IAC1B,qBAAqB,EAAE,uBAAuB;IAC9C,KAAK,EAAE,iBAAiB;IACxB,QAAQ,EAAE,cAAc;IACxB,sBAAsB;IACtB,aAAa,EAAE,gBAAgB;IAC/B,GAAG,EAAE,YAAY;IACjB,eAAe,EAAE,kBAAkB;IACnC,YAAY,EAAE,eAAe;IAC7B,SAAS,EAAE,eAAe;IAC1B,cAAc,EAAE,iBAAiB;IACjC,KAAK,EAAE,WAAW;IAClB,MAAM,EAAE,eAAe;IACvB,IAAI,EAAE,UAAU;IAChB,GAAG,EAAE,aAAa;IAClB,IAAI,EAAE,UAAU;IAChB,OAAO,EAAE,aAAa;IACtB,MAAM,EAAE,YAAY;IACpB,eAAe,EAAE,YAAY;IAC7B,eAAe,EAAE,YAAY;IAC7B,aAAa,EAAE,aAAa;IAC5B,oBAAoB,EAAE,sBAAsB;IAC5C,cAAc,EAAE,iBAAiB;IACjC,eAAe,EAAE,iBAAiB;IAClC,mBAAmB,EAAE,sBAAsB;IAC3C,UAAU,EAAE,gBAAgB;IAC5B,eAAe,EAAE,eAAe;IAChC,SAAS,EAAE,eAAe;IAC1B,QAAQ,EAAE,cAAc;IACxB,KAAK,EAAE,WAAW;IAClB,QAAQ,EAAE,cAAc;IACxB,kBAAkB,EAAE,YAAY;IAChC,QAAQ,EAAE,cAAc;IACxB,QAAQ,EAAE,cAAc;IACxB,QAAQ,EAAE,cAAc;IACxB,GAAG,EAAE,SAAS;IACd,OAAO,EAAE,aAAa;IACtB,qBAAqB,EAAE,wBAAwB;IAC/C,UAAU,EAAE,gBAAgB;IAC5B,aAAa,EAAE,gBAAgB;CAChC,CAAC;AAEF,eAAe,SAAS,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,UAAU,MAAM,QAAQ,CAAC;AAChC,OAAO,eAAe,MAAM,aAAa,CAAC;AAC1C,OAAO,cAAc,MAAM,YAAY,CAAC;AACxC,OAAO,sBAAsB,MAAM,sBAAsB,CAAC;AAC1D,OAAO,uBAAuB,MAAM,uBAAuB,CAAC;AAC5D,OAAO,iBAAiB,MAAM,gBAAgB,CAAC;AAC/C,OAAO,aAAa,MAAM,WAAW,CAAC;AACtC,OAAO,cAAc,MAAM,YAAY,CAAC;AACxC,OAAO,YAAY,MAAM,UAAU,CAAC;AACpC,OAAO,eAAe,MAAM,aAAa,CAAC;AAC1C,OAAO,eAAe,MAAM,cAAc,CAAC;AAC3C,OAAO,WAAW,MAAM,SAAS,CAAC;AAClC,OAAO,YAAY,MAAM,UAAU,CAAC;AACpC,qCAAqC;AACrC,OAAO,aAAa,MAAM,WAAW,CAAC;AACtC,OAAO,cAAc,MAAM,YAAY,CAAC;AACxC,OAAO,cAAc,MAAM,YAAY,CAAC;AACxC,OAAO,YAAY,MAAM,UAAU,CAAC;AACpC,OAAO,cAAc,MAAM,YAAY,CAAC;AACxC,OAAO,gBAAgB,MAAM,cAAc,CAAC;AAC5C,OAAO,iBAAiB,MAAM,gBAAgB,CAAC;AAC/C,OAAO,YAAY,MAAM,UAAU,CAAC;AACpC,OAAO,YAAY,MAAM,oBAAoB,CAAC;AAC9C,OAAO,UAAU,MAAM,QAAQ,CAAC;AAChC,OAAO,sBAAsB,MAAM,qBAAqB,CAAC;AACzD,OAAO,eAAe,MAAM,UAAU,CAAC;AACvC,OAAO,gBAAgB,MAAM,cAAc,CAAC;AAC5C,OAAO,kBAAkB,MAAM,iBAAiB,CAAC;AACjD,OAAO,eAAe,MAAM,aAAa,CAAC;AAC1C,OAAO,eAAe,MAAM,aAAa,CAAC;AAC1C,OAAO,aAAa,MAAM,WAAW,CAAC;AACtC,OAAO,gBAAgB,MAAM,eAAe,CAAC;AAC7C,OAAO,iBAAiB,MAAM,gBAAgB,CAAC;AAC/C,OAAO,gBAAgB,MAAM,eAAe,CAAC;AAC7C,OAAO,wBAAwB,MAAM,uBAAuB,CAAC;AAE7D,OAAO,cAAc,MAAM,YAAY,CAAC;AACxC,OAAO,WAAW,MAAM,SAAS,CAAC;AAClC,OAAO,SAAS,MAAM,OAAO,CAAC;AAE9B,MAAM,SAAS,GAAuC;IACpD,MAAM,EAAE,YAAY;IACpB,MAAM,EAAE,YAAY;IACpB,SAAS,EAAE,eAAe;IAC1B,qBAAqB,EAAE,uBAAuB;IAC9C,KAAK,EAAE,iBAAiB;IACxB,QAAQ,EAAE,cAAc;IACxB,sBAAsB;IACtB,aAAa,EAAE,gBAAgB;IAC/B,GAAG,EAAE,YAAY;IACjB,eAAe,EAAE,kBAAkB;IACnC,YAAY,EAAE,eAAe;IAC7B,SAAS,EAAE,eAAe;IAC1B,cAAc,EAAE,iBAAiB;IACjC,KAAK,EAAE,WAAW;IAClB,MAAM,EAAE,eAAe;IACvB,IAAI,EAAE,UAAU;IAChB,GAAG,EAAE,aAAa;IAClB,IAAI,EAAE,UAAU;IAChB,OAAO,EAAE,aAAa;IACtB,MAAM,EAAE,YAAY;IACpB,eAAe,EAAE,YAAY;IAC7B,eAAe,EAAE,YAAY;IAC7B,aAAa,EAAE,aAAa;IAC5B,oBAAoB,EAAE,sBAAsB;IAC5C,cAAc,EAAE,iBAAiB;IACjC,eAAe,EAAE,iBAAiB;IAClC,mBAAmB,EAAE,sBAAsB;IAC3C,UAAU,EAAE,gBAAgB;IAC5B,eAAe,EAAE,eAAe;IAChC,SAAS,EAAE,eAAe;IAC1B,QAAQ,EAAE,cAAc;IACxB,KAAK,EAAE,WAAW;IAClB,cAAc,EAAE,iBAAiB;IACjC,QAAQ,EAAE,cAAc;IACxB,kBAAkB,EAAE,YAAY;IAChC,QAAQ,EAAE,cAAc;IACxB,QAAQ,EAAE,cAAc;IACxB,QAAQ,EAAE,cAAc;IACxB,GAAG,EAAE,SAAS;IACd,OAAO,EAAE,aAAa;IACtB,qBAAqB,EAAE,wBAAwB;IAC/C,UAAU,EAAE,gBAAgB;IAC5B,aAAa,EAAE,gBAAgB;CAChC,CAAC;AAEF,eAAe,SAAS,CAAC"}