@superblocksteam/vite-plugin-file-sync 2.0.41-next.4 → 2.0.41-next.41

package/dist/ai-service/integrations/store.js

@@ -1,92 +1,153 @@
+var __runInitializers = (this && this.__runInitializers) || function (thisArg, initializers, value) {
+    var useValue = arguments.length > 2;
+    for (var i = 0; i < initializers.length; i++) {
+        value = useValue ? initializers[i].call(thisArg, value) : initializers[i].call(thisArg);
+    }
+    return useValue ? value : void 0;
+};
+var __esDecorate = (this && this.__esDecorate) || function (ctor, descriptorIn, decorators, contextIn, initializers, extraInitializers) {
+    function accept(f) { if (f !== void 0 && typeof f !== "function") throw new TypeError("Function expected"); return f; }
+    var kind = contextIn.kind, key = kind === "getter" ? "get" : kind === "setter" ? "set" : "value";
+    var target = !descriptorIn && ctor ? contextIn["static"] ? ctor : ctor.prototype : null;
+    var descriptor = descriptorIn || (target ? Object.getOwnPropertyDescriptor(target, contextIn.name) : {});
+    var _, done = false;
+    for (var i = decorators.length - 1; i >= 0; i--) {
+        var context = {};
+        for (var p in contextIn) context[p] = p === "access" ? {} : contextIn[p];
+        for (var p in contextIn.access) context.access[p] = contextIn.access[p];
+        context.addInitializer = function (f) { if (done) throw new TypeError("Cannot add initializers after decoration has completed"); extraInitializers.push(accept(f || null)); };
+        var result = (0, decorators[i])(kind === "accessor" ? { get: descriptor.get, set: descriptor.set } : descriptor[key], context);
+        if (kind === "accessor") {
+            if (result === void 0) continue;
+            if (result === null || typeof result !== "object") throw new TypeError("Object expected");
+            if (_ = accept(result.get)) descriptor.get = _;
+            if (_ = accept(result.set)) descriptor.set = _;
+            if (_ = accept(result.init)) initializers.unshift(_);
+        }
+        else if (_ = accept(result)) {
+            if (kind === "field") initializers.unshift(_);
+            else descriptor[key] = _;
+        }
+    }
+    if (target) Object.defineProperty(target, contextIn.name, descriptor);
+    done = true;
+};
 import { isEmpty } from "lodash-es";
 import { datasourceSdkClassByType } from "../const.js";
 import { databaseIntegrationTypes } from "../const.js";
+import llmobs from "../llmobs/index.js";
 import { DatabaseIntegrationMetadata } from "./metadata/database.js";
 import { GraphQLIntegrationMetadata } from "./metadata/graphql-based.js";
 import { IntegrationMetadata } from "./metadata/index.js";
 import { OpenApiIntegrationMetadata } from "./metadata/open-api.js";
 import { TableBasedIntegrationMetadata } from "./metadata/table-based.js";
-export class IntegrationStore {
-    openApiSpecs = {};
-    async getIntegrationForPromptContext(integration, prompt, clark, params) {
-        const { name, id } = integration;
-        const type = integration.type;
-        if (!name ||
-            !id ||
-            !isSdkIntegration({
-                type,
-            })) {
-            return null;
+let IntegrationStore = (() => {
+    let _instanceExtraInitializers = [];
+    let _getIntegrationForPromptContext_decorators;
+    let _getRelevantMetadata_decorators;
+    let _getOpenApiSpec_decorators;
+    return class IntegrationStore {
+        static {
+            const _metadata = typeof Symbol === "function" && Symbol.metadata ? Object.create(null) : void 0;
+            _getIntegrationForPromptContext_decorators = [llmobs.decorate({
+                    kind: "task",
+                    name: "IntegrationStore.getIntegrationForPromptContext",
+                })];
+            _getRelevantMetadata_decorators = [llmobs.decorate({
+                    kind: "task",
+                    name: "IntegrationStore.getRelevantMetadata",
+                })];
+            _getOpenApiSpec_decorators = [llmobs.decorate({
+                    kind: "task",
+                    name: "IntegrationStore.getOpenApiSpec",
+                })];
+            __esDecorate(this, null, _getIntegrationForPromptContext_decorators, { kind: "method", name: "getIntegrationForPromptContext", static: false, private: false, access: { has: obj => "getIntegrationForPromptContext" in obj, get: obj => obj.getIntegrationForPromptContext }, metadata: _metadata }, null, _instanceExtraInitializers);
+            __esDecorate(this, null, _getRelevantMetadata_decorators, { kind: "method", name: "getRelevantMetadata", static: false, private: false, access: { has: obj => "getRelevantMetadata" in obj, get: obj => obj.getRelevantMetadata }, metadata: _metadata }, null, _instanceExtraInitializers);
+            __esDecorate(this, null, _getOpenApiSpec_decorators, { kind: "method", name: "getOpenApiSpec", static: false, private: false, access: { has: obj => "getOpenApiSpec" in obj, get: obj => obj.getOpenApiSpec }, metadata: _metadata }, null, _instanceExtraInitializers);
+            if (_metadata) Object.defineProperty(this, Symbol.metadata, { enumerable: true, configurable: true, writable: true, value: _metadata });
         }
-        const metadata = await this.getRelevantMetadata({
-            clark,
-            integration,
-            prompt,
-            llmProvider: params.llmProvider,
-        });
-        return {
-            name,
-            id,
-            type,
-            metadata,
-        };
-    }
-    async getRelevantMetadata({ clark, integration, prompt, llmProvider, }) {
-        // If we're operating on an openapi integration, we need to fetch its full spec.
-        // It would be nice to push this down, but the cache lives on this instance as this.openApiSpecs
-        let metadata = integration.metadata;
-        if (isOpenApiIntegration(integration)) {
-            const openApiSpec = await this.getOpenApiSpec(integration, clark);
-            metadata = {
-                ...(integration.metadata || {}),
-                openApiSpec,
+        openApiSpecs = (__runInitializers(this, _instanceExtraInitializers), {});
+        async getIntegrationForPromptContext(integration, prompt, clark, params) {
+            const { name, id } = integration;
+            const type = integration.type;
+            if (!name ||
+                !id ||
+                !isSdkIntegration({
+                    type,
+                })) {
+                return null;
+            }
+            const metadata = await this.getRelevantMetadata({
+                clark,
+                integration,
+                prompt,
+                llmProvider: params.llmProvider,
+            });
+            return {
+                name,
+                id,
+                type,
+                metadata,
             };
         }
-        const klass = this.getMetadataClassForIntegration(integration);
-        return await klass.refineMetadataForPrompt({
-            metadata,
-            prompt,
-            llmProvider,
-        });
-    }
-    getMetadataClassForIntegration(integration) {
-        if (isOpenApiIntegration(integration)) {
-            return OpenApiIntegrationMetadata;
-        }
-        if (isDatabaseIntegration(integration)) {
-            return DatabaseIntegrationMetadata;
-        }
-        if (isTableBasedIntegration(integration)) {
-            return TableBasedIntegrationMetadata;
+        async getRelevantMetadata({ clark, integration, prompt, llmProvider, }) {
+            // If we're operating on an openapi integration, we need to fetch its full spec.
+            // It would be nice to push this down, but the cache lives on this instance as this.openApiSpecs
+            let metadata = integration.metadata;
+            if (isOpenApiIntegration(integration)) {
+                const openApiSpec = await this.getOpenApiSpec(integration, clark);
+                metadata = {
+                    ...(integration.metadata || {}),
+                    openApiSpec,
+                };
+            }
+            const klass = this.getMetadataClassForIntegration(integration);
+            return await klass.refineMetadataForPrompt({
+                metadata,
+                prompt,
+                llmProvider,
+            });
         }
-        if (isGraphQlIntegration(integration)) {
-            return GraphQLIntegrationMetadata;
+        getMetadataClassForIntegration(integration) {
+            if (isOpenApiIntegration(integration)) {
+                return OpenApiIntegrationMetadata;
+            }
+            if (isDatabaseIntegration(integration)) {
+                return DatabaseIntegrationMetadata;
+            }
+            if (isTableBasedIntegration(integration)) {
+                return TableBasedIntegrationMetadata;
+            }
+            if (isGraphQlIntegration(integration)) {
+                return GraphQLIntegrationMetadata;
+            }
+            return IntegrationMetadata;
         }
-        return IntegrationMetadata;
-    }
-    async getOpenApiSpec(integration, clark) {
-        if (this.openApiSpecs[integration.id]) {
+        async getOpenApiSpec(integration, clark) {
+            if (this.openApiSpecs[integration.id]) {
+                return this.openApiSpecs[integration.id];
+            }
+            if (!clark.context.peer) {
+                throw new Error("No peer found");
+            }
+            if (!integration.metadata.openApiSpecRef) {
+                return undefined;
+            }
+            // the ideal case is actually requesting this ourselves, not calling back to the UI to make the http request on our behalf (why two network hops when we can do one)
+            // but bucketeer requires bearer tokens to access, and we don't have that here (yet), so we will do this for now.
+            const result = await clark.context.peer.call.aiGetOpenApiSpec({
+                integrationId: integration.id,
+                ref: integration.metadata.openApiSpecRef,
+            });
+            if (isEmpty(result)) {
+                return undefined;
+            }
+            this.openApiSpecs[integration.id] = result;
             return this.openApiSpecs[integration.id];
         }
-        if (!clark.context.peer) {
-            throw new Error("No peer found");
-        }
-        if (!integration.metadata.openApiSpecRef) {
-            return undefined;
-        }
-        // the ideal case is actually requesting this ourselves, not calling back to the UI to make the http request on our behalf (why two network hops when we can do one)
-        // but bucketeer requires bearer tokens to access, and we don't have that here (yet), so we will do this for now.
-        const result = await clark.context.peer.call.aiGetOpenApiSpec({
-            integrationId: integration.id,
-            ref: integration.metadata.openApiSpecRef,
-        });
-        if (isEmpty(result)) {
-            return undefined;
-        }
-        this.openApiSpecs[integration.id] = result;
-        return this.openApiSpecs[integration.id];
-    }
-}
+    };
+})();
+export { IntegrationStore };
 function isOpenApiIntegration(integration) {
     return integration.metadata?.openApiSpecRef !== undefined;
 }

package/dist/ai-service/integrations/store.js.map

@@ -1 +1 @@
-{"version":3,"file":"store.js","sourceRoot":"","sources":["../../../src/ai-service/integrations/store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,EAAE,2BAA2B,EAAE,MAAM,wBAAwB,CAAC;AACrE,OAAO,EAAE,0BAA0B,EAAE,MAAM,6BAA6B,CAAC;AACzE,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,EAAE,0BAA0B,EAAE,MAAM,wBAAwB,CAAC;AACpE,OAAO,EAAE,6BAA6B,EAAE,MAAM,2BAA2B,CAAC;AAc1E,MAAM,OAAO,gBAAgB;IACnB,YAAY,GAAgC,EAAE,CAAC;IAEvD,KAAK,CAAC,8BAA8B,CAClC,WAAqC,EACrC,MAAc,EACd,KAAY,EACZ,MAA+B;QAE/B,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,GAAG,WAAW,CAAC;QACjC,MAAM,IAAI,GAAG,WAAW,CAAC,IAA0B,CAAC;QAEpD,IACE,CAAC,IAAI;YACL,CAAC,EAAE;YACH,CAAC,gBAAgB,CAAC;gBAChB,IAAI;aACL,CAAC,EACF,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC;YAC9C,KAAK;YACL,WAAW;YACX,MAAM;YACN,WAAW,EAAE,MAAM,CAAC,WAAW;SAChC,CAAC,CAAC;QAEH,OAAO;YACL,IAAI;YACJ,EAAE;YACF,IAAI;YACJ,QAAQ;SACT,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,EACxB,KAAK,EACL,WAAW,EACX,MAAM,EACN,WAAW,GAMZ;QACC,gFAAgF;QAChF,gGAAgG;QAChG,IAAI,QAAQ,GAAG,WAAW,CAAC,QAAQ,CAAC;QACpC,IAAI,oBAAoB,CAAC,WAAW,CAAC,EAAE,CAAC;YACtC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;YAElE,QAAQ,GAAG;gBACT,GAAG,CAAC,WAAW,CAAC,QAAQ,IAAI,EAAE,CAAC;gBAC/B,WAAW;aACZ,CAAC;QACJ,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,8BAA8B,CAAC,WAAW,CAAC,CAAC;QAC/D,OAAO,MAAM,KAAK,CAAC,uBAAuB,CAAC;YACzC,QAAQ;YACR,MAAM;YACN,WAAW;SACZ,CAAC,CAAC;IACL,CAAC;IAED,8BAA8B,CAC5B,WAAqC;QAErC,IAAI,oBAAoB,CAAC,WAAW,CAAC,EAAE,CAAC;YACtC,OAAO,0BAA0B,CAAC;QACpC,CAAC;QACD,IAAI,qBAAqB,CAAC,WAAW,CAAC,EAAE,CAAC;YACvC,OAAO,2BAA2B,CAAC;QACrC,CAAC;QACD,IAAI,uBAAuB,CAAC,WAAW,CAAC,EAAE,CAAC;YACzC,OAAO,6BAA6B,CAAC;QACvC,CAAC;QACD,IAAI,oBAAoB,CAAC,WAAW,CAAC,EAAE,CAAC;YACtC,OAAO,0BAA0B,CAAC;QACpC,CAAC;QACD,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,cAAc,CAClB,WAAqC,EACrC,KAAY;QAEZ,IAAI,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,CAAC;YACtC,OAAO,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;QACnC,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC;YACzC,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,oKAAoK;QACpK,iHAAiH;QACjH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC;YAC5D,aAAa,EAAE,WAAW,CAAC,EAAE;YAC7B,GAAG,EAAE,WAAW,CAAC,QAAQ,CAAC,cAAc;SACzC,CAAC,CAAC;QACH,IAAI,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YACpB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,GAAG,MAAqB,CAAC;QAC1D,OAAO,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;IAC3C,CAAC;CACF;AAED,SAAS,oBAAoB,CAAC,WAAqC;IACjE,OAAO,WAAW,CAAC,QAAQ,EAAE,cAAc,KAAK,SAAS,CAAC;AAC5D,CAAC;AAED,SAAS,gBAAgB,CACvB,WAAoC;IAEpC,OAAO,CACL,OAAO,WAAW,CAAC,IAAI,KAAK,QAAQ;QACpC,WAAW,CAAC,IAAI,IAAI,wBAAwB,CAC7C,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,WAAqC;IAClE,OAAO,CACL,wBAAwB,CAAC,QAAQ,CAC/B,WAAW,CAAC,IAAkC,CAC/C,IAAI,WAAW,CAAC,IAAI,KAAK,YAAY,CACvC,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB,CAAC,WAAqC;IACpE,MAAM,MAAM,GAAG,WAAW,CAAC,QAAQ,EAAE,MAAM,IAAI,WAAW,CAAC,QAAQ,EAAE,QAAQ,CAAC;IAC9E,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC5D,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,oBAAoB,CAAC,WAAqC;IACjE,OAAO,WAAW,CAAC,IAAI,KAAK,oBAAoB,CAAC;AACnD,CAAC"}
+{"version":3,"file":"store.js","sourceRoot":"","sources":["../../../src/ai-service/integrations/store.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,2BAA2B,EAAE,MAAM,wBAAwB,CAAC;AACrE,OAAO,EAAE,0BAA0B,EAAE,MAAM,6BAA6B,CAAC;AACzE,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,EAAE,0BAA0B,EAAE,MAAM,wBAAwB,CAAC;AACpE,OAAO,EAAE,6BAA6B,EAAE,MAAM,2BAA2B,CAAC;IAc7D,gBAAgB;;;;;iBAAhB,gBAAgB;;;0DAG1B,MAAM,CAAC,QAAQ,CAAC;oBACf,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,iDAAiD;iBACxD,CAAC;+CAmCD,MAAM,CAAC,QAAQ,CAAC;oBACf,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,sCAAsC;iBAC7C,CAAC;0CAkDD,MAAM,CAAC,QAAQ,CAAC;oBACf,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,iCAAiC;iBACxC,CAAC;YA1FF,6OAAM,8BAA8B,6DAgCnC;YAMD,4MAAM,mBAAmB,6DA6BxB;YAwBD,6LAAM,cAAc,6DA2BnB;;;QA5HO,YAAY,IADT,mDAAgB,EACyB,EAAE,EAAC;QAMvD,KAAK,CAAC,8BAA8B,CAClC,WAAqC,EACrC,MAAc,EACd,KAAY,EACZ,MAA+B;YAE/B,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,GAAG,WAAW,CAAC;YACjC,MAAM,IAAI,GAAG,WAAW,CAAC,IAA0B,CAAC;YAEpD,IACE,CAAC,IAAI;gBACL,CAAC,EAAE;gBACH,CAAC,gBAAgB,CAAC;oBAChB,IAAI;iBACL,CAAC,EACF,CAAC;gBACD,OAAO,IAAI,CAAC;YACd,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC;gBAC9C,KAAK;gBACL,WAAW;gBACX,MAAM;gBACN,WAAW,EAAE,MAAM,CAAC,WAAW;aAChC,CAAC,CAAC;YAEH,OAAO;gBACL,IAAI;gBACJ,EAAE;gBACF,IAAI;gBACJ,QAAQ;aACT,CAAC;QACJ,CAAC;QAMD,KAAK,CAAC,mBAAmB,CAAC,EACxB,KAAK,EACL,WAAW,EACX,MAAM,EACN,WAAW,GAMZ;YACC,gFAAgF;YAChF,gGAAgG;YAChG,IAAI,QAAQ,GAAG,WAAW,CAAC,QAAQ,CAAC;YACpC,IAAI,oBAAoB,CAAC,WAAW,CAAC,EAAE,CAAC;gBACtC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;gBAElE,QAAQ,GAAG;oBACT,GAAG,CAAC,WAAW,CAAC,QAAQ,IAAI,EAAE,CAAC;oBAC/B,WAAW;iBACZ,CAAC;YACJ,CAAC;YAED,MAAM,KAAK,GAAG,IAAI,CAAC,8BAA8B,CAAC,WAAW,CAAC,CAAC;YAC/D,OAAO,MAAM,KAAK,CAAC,uBAAuB,CAAC;gBACzC,QAAQ;gBACR,MAAM;gBACN,WAAW;aACZ,CAAC,CAAC;QACL,CAAC;QAED,8BAA8B,CAC5B,WAAqC;YAErC,IAAI,oBAAoB,CAAC,WAAW,CAAC,EAAE,CAAC;gBACtC,OAAO,0BAA0B,CAAC;YACpC,CAAC;YACD,IAAI,qBAAqB,CAAC,WAAW,CAAC,EAAE,CAAC;gBACvC,OAAO,2BAA2B,CAAC;YACrC,CAAC;YACD,IAAI,uBAAuB,CAAC,WAAW,CAAC,EAAE,CAAC;gBACzC,OAAO,6BAA6B,CAAC;YACvC,CAAC;YACD,IAAI,oBAAoB,CAAC,WAAW,CAAC,EAAE,CAAC;gBACtC,OAAO,0BAA0B,CAAC;YACpC,CAAC;YACD,OAAO,mBAAmB,CAAC;QAC7B,CAAC;QAMD,KAAK,CAAC,cAAc,CAClB,WAAqC,EACrC,KAAY;YAEZ,IAAI,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,CAAC;gBACtC,OAAO,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;YAC3C,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;gBACxB,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;YACnC,CAAC;YAED,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC;gBACzC,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,oKAAoK;YACpK,iHAAiH;YACjH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC;gBAC5D,aAAa,EAAE,WAAW,CAAC,EAAE;gBAC7B,GAAG,EAAE,WAAW,CAAC,QAAQ,CAAC,cAAc;aACzC,CAAC,CAAC;YACH,IAAI,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBACpB,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,GAAG,MAAqB,CAAC;YAC1D,OAAO,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAC3C,CAAC;;;SA7HU,gBAAgB;AAgI7B,SAAS,oBAAoB,CAAC,WAAqC;IACjE,OAAO,WAAW,CAAC,QAAQ,EAAE,cAAc,KAAK,SAAS,CAAC;AAC5D,CAAC;AAED,SAAS,gBAAgB,CACvB,WAAoC;IAEpC,OAAO,CACL,OAAO,WAAW,CAAC,IAAI,KAAK,QAAQ;QACpC,WAAW,CAAC,IAAI,IAAI,wBAAwB,CAC7C,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,WAAqC;IAClE,OAAO,CACL,wBAAwB,CAAC,QAAQ,CAC/B,WAAW,CAAC,IAAkC,CAC/C,IAAI,WAAW,CAAC,IAAI,KAAK,YAAY,CACvC,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB,CAAC,WAAqC;IACpE,MAAM,MAAM,GAAG,WAAW,CAAC,QAAQ,EAAE,MAAM,IAAI,WAAW,CAAC,QAAQ,EAAE,QAAQ,CAAC;IAC9E,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC5D,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,oBAAoB,CAAC,WAAqC;IACjE,OAAO,WAAW,CAAC,IAAI,KAAK,oBAAoB,CAAC;AACnD,CAAC"}

package/dist/ai-service/llmobs/helpers.d.ts

@@ -0,0 +1,20 @@
+import { streamText, type ToolSet } from "ai";
+import { type Logger } from "../../util/logger.js";
+import { LLMObsTracer } from "./tracer.js";
+import type { LLMObsSpan } from "./types.js";
+export declare function traceWorkflow<T>(name: string, operation: (span: any) => Promise<T>, tags?: Record<string, string | boolean>, llmobs?: LLMObsTracer): Promise<T>;
+export declare function traceTask<T>(name: string, operation: (span: any) => Promise<T>, tags?: Record<string, string | boolean>, llmobs?: LLMObsTracer): Promise<T>;
+export declare function traceLLM<T>(name: string, modelName: string, modelProvider: string, operation: (span: LLMObsSpan) => Promise<T>, tags?: Record<string, string | boolean>, llmobs?: LLMObsTracer): Promise<T>;
+export declare function getJwtTraceTags(jwt: string | undefined): Record<string, string>;
+export declare function getCommonTraceTags(jwt: string | undefined, applicationId: string, organizationId: string, additionalTags?: Record<string, string | boolean>): Record<string, string | boolean>;
+/**
+ * Instruments ai.streamText with LLMObs tracing.
+ * Wraps user-provided callbacks to ensure spans are finished correctly.
+ * Captures input/output data, metrics, errors, and metadata for each step and the overall call.
+ */
+export declare function tracedStreamText<T extends ToolSet>(params: Parameters<typeof streamText<T>>[0], llmobs?: LLMObsTracer, logger?: Logger): import("ai").StreamTextResult<T, never>;
+export declare const traceToolSet: <T extends ToolSet>(toolset: T, llmobs?: LLMObsTracer, hooks?: {
+    beforeExecute?: (name: string) => void;
+    afterExecute?: (name: string, span: LLMObsSpan) => void;
+}) => T;
+//# sourceMappingURL=helpers.d.ts.map

package/dist/ai-service/llmobs/helpers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"helpers.d.ts","sourceRoot":"","sources":["../../../src/ai-service/llmobs/helpers.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,KAAK,OAAO,EAAE,MAAM,IAAI,CAAC;AAC9C,OAAO,EAAE,KAAK,MAAM,EAAa,MAAM,sBAAsB,CAAC;AAG9D,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAG7C,wBAAgB,aAAa,CAAC,CAAC,EAC7B,IAAI,EAAE,MAAM,EACZ,SAAS,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,CAAC,CAAC,EACpC,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,EACvC,MAAM,GAAE,YAAoC,GAC3C,OAAO,CAAC,CAAC,CAAC,CAaZ;AAGD,wBAAgB,SAAS,CAAC,CAAC,EACzB,IAAI,EAAE,MAAM,EACZ,SAAS,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,CAAC,CAAC,EACpC,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,EACvC,MAAM,GAAE,YAAoC,GAC3C,OAAO,CAAC,CAAC,CAAC,CAaZ;AAGD,wBAAgB,QAAQ,CAAC,CAAC,EACxB,IAAI,EAAE,MAAM,EACZ,SAAS,EAAE,MAAM,EACjB,aAAa,EAAE,MAAM,EACrB,SAAS,EAAE,CAAC,IAAI,EAAE,UAAU,KAAK,OAAO,CAAC,CAAC,CAAC,EAC3C,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,EACvC,MAAM,GAAE,YAAoC,GAC3C,OAAO,CAAC,CAAC,CAAC,CAeZ;AAGD,wBAAgB,eAAe,CAC7B,GAAG,EAAE,MAAM,GAAG,SAAS,GACtB,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAoBxB;AAGD,wBAAgB,kBAAkB,CAChC,GAAG,EAAE,MAAM,GAAG,SAAS,EACvB,aAAa,EAAE,MAAM,EACrB,cAAc,EAAE,MAAM,EACtB,cAAc,GAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAM,GACpD,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,CAclC;AAED;;;;GAIG;AAEH,wBAAgB,gBAAgB,CAAC,CAAC,SAAS,OAAO,EAChD,MAAM,EAAE,UAAU,CAAC,OAAO,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAC3C,MAAM,GAAE,YAAoC,EAC5C,MAAM,GAAE,MAAoB,2CAI7B;AAED,eAAO,MAAM,YAAY,GAAI,CAAC,SAAS,OAAO,EAC5C,SAAS,CAAC,EACV,SAAQ,YAAoC,EAC5C,QAAQ;IACN,aAAa,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;IACvC,YAAY,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,KAAK,IAAI,CAAC;CACzD,KACA,CA8CF,CAAC"}

package/dist/ai-service/llmobs/helpers.js

@@ -0,0 +1,130 @@
+import { streamText } from "ai";
+import { getLogger } from "../../util/logger.js";
+import { safeJsonStringify } from "../util/safe-stringify.js";
+import { LLMObsStreamTextMiddleware } from "./middleware/stream-text.js";
+import { LLMObsTracer } from "./tracer.js";
+// Helper function for workflow tracing with common patterns
+export function traceWorkflow(name, operation, tags, llmobs = LLMObsTracer.instance) {
+    return llmobs.trace({
+        kind: "workflow",
+        name,
+    }, async (span) => {
+        if (tags) {
+            llmobs.annotate({ tags });
+        }
+        return operation(span);
+    });
+}
+// Helper function for task tracing with common patterns
+export function traceTask(name, operation, tags, llmobs = LLMObsTracer.instance) {
+    return llmobs.trace({
+        kind: "task",
+        name,
+    }, async (span) => {
+        if (tags) {
+            llmobs.annotate({ tags });
+        }
+        return operation(span);
+    });
+}
+// Helper function for LLM tracing with common patterns
+export function traceLLM(name, modelName, modelProvider, operation, tags, llmobs = LLMObsTracer.instance) {
+    return llmobs.trace({
+        kind: "llm",
+        name,
+        modelName,
+        modelProvider,
+    }, async (span) => {
+        if (tags) {
+            llmobs.annotate({ tags });
+        }
+        return operation(span);
+    });
+}
+// Helper to extract JWT trace tags
+export function getJwtTraceTags(jwt) {
+    const result = {};
+    if (jwt) {
+        try {
+            const parts = jwt.split(".");
+            if (parts.length === 3 && parts[1]) {
+                const payload = JSON.parse(Buffer.from(parts[1], "base64").toString());
+                if (payload.org_id)
+                    result["org.id"] = payload.org_id;
+                if (payload.org_name)
+                    result["org.name"] = payload.org_name;
+                if (payload.user_id)
+                    result["user.id"] = payload.user_id;
+                if (payload.user_email)
+                    result["user.email"] = payload.user_email;
+            }
+        }
+        catch {
+            // Silently handle JWT parsing errors
+        }
+    }
+    return result;
+}
+// Helper to combine common tags with enhanced user information
+export function getCommonTraceTags(jwt, applicationId, organizationId, additionalTags = {}) {
+    const result = {};
+    // Extract and add JWT tags using the extracted function
+    Object.assign(result, getJwtTraceTags(jwt));
+    // Add required tags (no if checks needed since they're required)
+    result["application.id"] = applicationId;
+    result["organization.id"] = organizationId;
+    // Add additional tags
+    Object.assign(result, additionalTags);
+    return result;
+}
+/**
+ * Instruments ai.streamText with LLMObs tracing.
+ * Wraps user-provided callbacks to ensure spans are finished correctly.
+ * Captures input/output data, metrics, errors, and metadata for each step and the overall call.
+ */
+export function tracedStreamText(params, llmobs = LLMObsTracer.instance, logger = getLogger()) {
+    const middleware = new LLMObsStreamTextMiddleware(llmobs, logger);
+    return middleware.wrap(streamText)(params);
+}
+export const traceToolSet = (toolset, llmobs = LLMObsTracer.instance, hooks) => {
+    const tracedToolset = {};
+    for (const [name, tool] of Object.entries(toolset)) {
+        tracedToolset[name] = {
+            ...tool,
+            execute: async (input, options) => {
+                hooks?.beforeExecute?.(name);
+                return await llmobs.trace({
+                    kind: "tool",
+                    name: `tool.${name}`,
+                }, async (span) => {
+                    try {
+                        if (!tool.execute) {
+                            throw new Error(`Tool ${name} has no execute method`);
+                        }
+                        const result = await tool.execute(input, options);
+                        hooks?.afterExecute?.(name, span);
+                        const metadata = {
+                            toolCallid: options?.toolCallId,
+                            ...Object.fromEntries(Object.entries(tool).map(([key, value]) => [
+                                key,
+                                safeJsonStringify(value),
+                            ])),
+                        };
+                        llmobs.annotate(span, {
+                            inputData: safeJsonStringify(input),
+                            outputData: safeJsonStringify(result),
+                            metadata,
+                        });
+                        return result;
+                    }
+                    catch (error) {
+                        console.error(`Error executing tool ${name}:`, error);
+                        throw error;
+                    }
+                });
+            },
+        };
+    }
+    return tracedToolset;
+};
+//# sourceMappingURL=helpers.js.map

package/dist/ai-service/llmobs/helpers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"helpers.js","sourceRoot":"","sources":["../../../src/ai-service/llmobs/helpers.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAgB,MAAM,IAAI,CAAC;AAC9C,OAAO,EAAe,SAAS,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,EAAE,0BAA0B,EAAE,MAAM,6BAA6B,CAAC;AACzE,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAG3C,4DAA4D;AAC5D,MAAM,UAAU,aAAa,CAC3B,IAAY,EACZ,SAAoC,EACpC,IAAuC,EACvC,SAAuB,YAAY,CAAC,QAAQ;IAE5C,OAAO,MAAM,CAAC,KAAK,CACjB;QACE,IAAI,EAAE,UAAU;QAChB,IAAI;KACL,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,IAAI,EAAE,CAAC;YACT,MAAM,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5B,CAAC;QACD,OAAO,SAAS,CAAC,IAAI,CAAC,CAAC;IACzB,CAAC,CACF,CAAC;AACJ,CAAC;AAED,wDAAwD;AACxD,MAAM,UAAU,SAAS,CACvB,IAAY,EACZ,SAAoC,EACpC,IAAuC,EACvC,SAAuB,YAAY,CAAC,QAAQ;IAE5C,OAAO,MAAM,CAAC,KAAK,CACjB;QACE,IAAI,EAAE,MAAM;QACZ,IAAI;KACL,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,IAAI,EAAE,CAAC;YACT,MAAM,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5B,CAAC;QACD,OAAO,SAAS,CAAC,IAAI,CAAC,CAAC;IACzB,CAAC,CACF,CAAC;AACJ,CAAC;AAED,uDAAuD;AACvD,MAAM,UAAU,QAAQ,CACtB,IAAY,EACZ,SAAiB,EACjB,aAAqB,EACrB,SAA2C,EAC3C,IAAuC,EACvC,SAAuB,YAAY,CAAC,QAAQ;IAE5C,OAAO,MAAM,CAAC,KAAK,CACjB;QACE,IAAI,EAAE,KAAK;QACX,IAAI;QACJ,SAAS;QACT,aAAa;KACd,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,IAAI,EAAE,CAAC;YACT,MAAM,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5B,CAAC;QACD,OAAO,SAAS,CAAC,IAAI,CAAC,CAAC;IACzB,CAAC,CACF,CAAC;AACJ,CAAC;AAED,mCAAmC;AACnC,MAAM,UAAU,eAAe,CAC7B,GAAuB;IAEvB,MAAM,MAAM,GAA2B,EAAE,CAAC;IAE1C,IAAI,GAAG,EAAE,CAAC;QACR,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC7B,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;gBACnC,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC;gBAEvE,IAAI,OAAO,CAAC,MAAM;oBAAE,MAAM,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;gBACtD,IAAI,OAAO,CAAC,QAAQ;oBAAE,MAAM,CAAC,UAAU,CAAC,GAAG,OAAO,CAAC,QAAQ,CAAC;gBAC5D,IAAI,OAAO,CAAC,OAAO;oBAAE,MAAM,CAAC,SAAS,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;gBACzD,IAAI,OAAO,CAAC,UAAU;oBAAE,MAAM,CAAC,YAAY,CAAC,GAAG,OAAO,CAAC,UAAU,CAAC;YACpE,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,qCAAqC;QACvC,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,+DAA+D;AAC/D,MAAM,UAAU,kBAAkB,CAChC,GAAuB,EACvB,aAAqB,EACrB,cAAsB,EACtB,iBAAmD,EAAE;IAErD,MAAM,MAAM,GAAqC,EAAE,CAAC;IAEpD,wDAAwD;IACxD,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC;IAE5C,iEAAiE;IACjE,MAAM,CAAC,gBAAgB,CAAC,GAAG,aAAa,CAAC;IACzC,MAAM,CAAC,iBAAiB,CAAC,GAAG,cAAc,CAAC;IAE3C,sBAAsB;IACtB,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;IAEtC,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;GAIG;AAEH,MAAM,UAAU,gBAAgB,CAC9B,MAA2C,EAC3C,SAAuB,YAAY,CAAC,QAAQ,EAC5C,SAAiB,SAAS,EAAE;IAE5B,MAAM,UAAU,GAAG,IAAI,0BAA0B,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAClE,OAAO,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC,CAAC;AAC7C,CAAC;AAED,MAAM,CAAC,MAAM,YAAY,GAAG,CAC1B,OAAU,EACV,SAAuB,YAAY,CAAC,QAAQ,EAC5C,KAGC,EACE,EAAE;IACL,MAAM,aAAa,GAAY,EAAE,CAAC;IAElC,KAAK,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACnD,aAAa,CAAC,IAAI,CAAC,GAAG;YACpB,GAAG,IAAI;YACP,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;gBAChC,KAAK,EAAE,aAAa,EAAE,CAAC,IAAI,CAAC,CAAC;gBAC7B,OAAO,MAAM,MAAM,CAAC,KAAK,CACvB;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,QAAQ,IAAI,EAAE;iBACrB,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;oBACb,IAAI,CAAC;wBACH,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;4BAClB,MAAM,IAAI,KAAK,CAAC,QAAQ,IAAI,wBAAwB,CAAC,CAAC;wBACxD,CAAC;wBAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;wBAClD,KAAK,EAAE,YAAY,EAAE,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;wBAClC,MAAM,QAAQ,GAAG;4BACf,UAAU,EAAE,OAAO,EAAE,UAAU;4BAC/B,GAAG,MAAM,CAAC,WAAW,CACnB,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC;gCACzC,GAAG;gCACH,iBAAiB,CAAC,KAAK,CAAC;6BACzB,CAAC,CACH;yBACF,CAAC;wBACF,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE;4BACpB,SAAS,EAAE,iBAAiB,CAAC,KAAK,CAAC;4BACnC,UAAU,EAAE,iBAAiB,CAAC,MAAM,CAAC;4BACrC,QAAQ;yBACT,CAAC,CAAC;wBACH,OAAO,MAAM,CAAC;oBAChB,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,OAAO,CAAC,KAAK,CAAC,wBAAwB,IAAI,GAAG,EAAE,KAAK,CAAC,CAAC;wBACtD,MAAM,KAAK,CAAC;oBACd,CAAC;gBACH,CAAC,CACF,CAAC;YACJ,CAAC;SACF,CAAC;IACJ,CAAC;IACD,OAAO,aAAkB,CAAC;AAC5B,CAAC,CAAC"}

package/dist/ai-service/llmobs/index.d.ts

@@ -0,0 +1,8 @@
+import tracer, { LLMObsTracer } from "./tracer.js";
+export * from "./types.js";
+declare const llmobs: typeof tracer & {
+    LLMObsTracer: typeof LLMObsTracer;
+};
+export { LLMObsTracer, tracer };
+export default llmobs;
+//# sourceMappingURL=index.d.ts.map

package/dist/ai-service/llmobs/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/ai-service/llmobs/index.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,EAAE,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAEnD,cAAc,YAAY,CAAC;AAE3B,QAAA,MAAM,MAAM,EAAa,OAAO,MAAM,GAAG;IACvC,YAAY,EAAE,OAAO,YAAY,CAAC;CACnC,CAAC;AAGF,OAAO,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC;AAChC,eAAe,MAAM,CAAC"}

package/dist/ai-service/llmobs/index.js

@@ -0,0 +1,7 @@
+import tracer, { LLMObsTracer } from "./tracer.js";
+export * from "./types.js";
+const llmobs = tracer;
+llmobs.LLMObsTracer = LLMObsTracer;
+export { LLMObsTracer, tracer };
+export default llmobs;
+//# sourceMappingURL=index.js.map

package/dist/ai-service/llmobs/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/ai-service/llmobs/index.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,EAAE,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAEnD,cAAc,YAAY,CAAC;AAE3B,MAAM,MAAM,GAAG,MAEd,CAAC;AAEF,MAAM,CAAC,YAAY,GAAG,YAAY,CAAC;AACnC,OAAO,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC;AAChC,eAAe,MAAM,CAAC"}

package/dist/ai-service/llmobs/middleware/stream-text.d.ts

@@ -0,0 +1,60 @@
+import { type Logger } from "../../../util/logger.js";
+import { LLMObsTracer, type LLMObsSpan } from "../index.js";
+import type { ToolSet, StreamTextOnStepFinishCallback, LanguageModel, StreamTextOnFinishCallback, StreamTextOnErrorCallback, StepResult, ModelMessage, PrepareStepFunction, PrepareStepResult, StreamTextOnChunkCallback } from "ai";
+import type { streamText } from "ai";
+type StreamAbortEvent<T extends ToolSet> = {
+    readonly steps: StepResult<T>[];
+};
+type StreamTextOnAbortCallback<T extends ToolSet> = (event: StreamAbortEvent<T>) => PromiseLike<void> | void;
+type StreamSegment = {
+    type: "text" | "reasoning" | "tool-call" | "tool-result" | "tool-input" | "raw" | "thinking" | "source";
+    span: LLMObsSpan;
+    content: string | {
+        [key: string]: any;
+    } | undefined;
+};
+export declare class LLMObsStreamTextMiddleware {
+    fullSpan?: LLMObsSpan;
+    stepSpan?: LLMObsSpan;
+    currentSegment?: StreamSegment;
+    modelName: string;
+    modelProvider?: string;
+    modelProviderDetailed?: string;
+    tools?: ToolSet;
+    stepResults: StepResult<any>[];
+    llmobs: LLMObsTracer;
+    logger: Logger;
+    private errorLogged;
+    wrap<T extends ToolSet>(fn: typeof streamText<T>): typeof streamText<T>;
+    constructor(llmobs?: LLMObsTracer, logger?: Logger);
+    private transform;
+    prepareStep<T extends ToolSet>(options: Parameters<PrepareStepFunction<T>>[0], next?: PrepareStepFunction<T>): Promise<{
+        steps: StepResult<NoInfer<T>>[];
+        stepNumber: number;
+        model: LanguageModel;
+        messages: Array<ModelMessage>;
+    } | PrepareStepResult<T>>;
+    onChunk<T extends ToolSet>(event: Parameters<StreamTextOnChunkCallback<T>>[0], next?: StreamTextOnChunkCallback<T>): Promise<void>;
+    onStepFinish<T extends ToolSet>(result: Parameters<StreamTextOnStepFinishCallback<T>>[0], next?: StreamTextOnStepFinishCallback<T>): Promise<void>;
+    onError(event: Parameters<StreamTextOnErrorCallback>[0], next?: StreamTextOnErrorCallback): Promise<void>;
+    onAbort<T extends ToolSet>(event: StreamAbortEvent<T>, next?: StreamTextOnAbortCallback<T>): Promise<void>;
+    onFinish<T extends ToolSet>(event: Parameters<StreamTextOnFinishCallback<T>>[0], next?: StreamTextOnFinishCallback<T>): Promise<void>;
+    onFinally(): Promise<void>;
+    private handleStreamError;
+    private handleToolExecuteStart;
+    private logStreamError;
+    private ensureSegment;
+    private appendTextToCurrentSegment;
+    private appendToolInputDelta;
+    private isTextualSegment;
+    private getSegmentType;
+    private startSegment;
+    private finishSegment;
+    private startStepSpan;
+    private finishStepSpan;
+    private startFullSpan;
+    private finishFullSpan;
+    private setModel;
+}
+export {};
+//# sourceMappingURL=stream-text.d.ts.map

package/dist/ai-service/llmobs/middleware/stream-text.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream-text.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/llmobs/middleware/stream-text.ts"],"names":[],"mappings":"AAAA,OAAO,EAAa,KAAK,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAGjE,OAAO,EACL,YAAY,EAGZ,KAAK,UAAU,EAChB,MAAM,aAAa,CAAC;AAGrB,OAAO,KAAK,EACV,OAAO,EACP,8BAA8B,EAC9B,aAAa,EACb,0BAA0B,EAC1B,yBAAyB,EAEzB,UAAU,EAGV,YAAY,EACZ,mBAAmB,EACnB,iBAAiB,EACjB,yBAAyB,EAK1B,MAAM,IAAI,CAAC;AACZ,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AAErC,KAAK,gBAAgB,CAAC,CAAC,SAAS,OAAO,IAAI;IAAE,QAAQ,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,CAAA;CAAE,CAAC;AAE/E,KAAK,yBAAyB,CAAC,CAAC,SAAS,OAAO,IAAI,CAClD,KAAK,EAAE,gBAAgB,CAAC,CAAC,CAAC,KACvB,WAAW,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;AAE9B,KAAK,aAAa,GAAG;IACnB,IAAI,EACA,MAAM,GACN,WAAW,GACX,WAAW,GACX,aAAa,GACb,YAAY,GACZ,KAAK,GACL,UAAU,GACV,QAAQ,CAAC;IACb,IAAI,EAAE,UAAU,CAAC;IACjB,OAAO,EAAE,MAAM,GAAG;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,GAAG,SAAS,CAAC;CACtD,CAAC;AAEF,qBAAa,0BAA0B;IACrC,QAAQ,CAAC,EAAE,UAAU,CAAC;IACtB,QAAQ,CAAC,EAAE,UAAU,CAAC;IACtB,cAAc,CAAC,EAAE,aAAa,CAAC;IAC/B,SAAS,SAAa;IACtB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,WAAW,EAAE,UAAU,CAAC,GAAG,CAAC,EAAE,CAAM;IACpC,MAAM,EAAE,YAAY,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,WAAW,CAAS;IAE5B,IAAI,CAAC,CAAC,SAAS,OAAO,EAAE,EAAE,EAAE,OAAO,UAAU,CAAC,CAAC,CAAC,GAAG,OAAO,UAAU,CAAC,CAAC,CAAC;gBAerE,MAAM,GAAE,YAAoC,EAC5C,MAAM,GAAE,MAAoB;IAM9B,OAAO,CAAC,SAAS;IAyBX,WAAW,CAAC,CAAC,SAAS,OAAO,EACjC,OAAO,EAAE,UAAU,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAC9C,IAAI,CAAC,EAAE,mBAAmB,CAAC,CAAC,CAAC;;;;;;IAyBzB,OAAO,CAAC,CAAC,SAAS,OAAO,EAC7B,KAAK,EAAE,UAAU,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAClD,IAAI,CAAC,EAAE,yBAAyB,CAAC,CAAC,CAAC;IAmC/B,YAAY,CAAC,CAAC,SAAS,OAAO,EAClC,MAAM,EAAE,UAAU,CAAC,8BAA8B,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EACxD,IAAI,CAAC,EAAE,8BAA8B,CAAC,CAAC,CAAC;IAsBpC,OAAO,CACX,KAAK,EAAE,UAAU,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC,EAC/C,IAAI,CAAC,EAAE,yBAAyB;IAa5B,OAAO,CAAC,CAAC,SAAS,OAAO,EAC7B,KAAK,EAAE,gBAAgB,CAAC,CAAC,CAAC,EAC1B,IAAI,CAAC,EAAE,yBAAyB,CAAC,CAAC,CAAC;IAY/B,QAAQ,CAAC,CAAC,SAAS,OAAO,EAC9B,KAAK,EAAE,UAAU,CAAC,0BAA0B,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EACnD,IAAI,CAAC,EAAE,0BAA0B,CAAC,CAAC,CAAC;IAWhC,SAAS;IAIf,OAAO,CAAC,iBAAiB;IAKzB,OAAO,CAAC,sBAAsB;IAI9B,OAAO,CAAC,cAAc;IAYtB,OAAO,CAAC,aAAa;IAWrB,OAAO,CAAC,0BAA0B;IASlC,OAAO,CAAC,oBAAoB;IAS5B,OAAO,CAAC,gBAAgB;IAUxB,OAAO,CAAC,cAAc;IAqBtB,OAAO,CAAC,YAAY;IAgBpB,OAAO,CAAC,aAAa;IA6BrB,OAAO,CAAC,aAAa;IAoBrB,OAAO,CAAC,cAAc;IAwBtB,OAAO,CAAC,aAAa;IAyBrB,OAAO,CAAC,cAAc;IA6BtB,OAAO,CAAC,QAAQ;CAOjB"}