@superblocksteam/vite-plugin-file-sync 2.0.51-next.1 → 2.0.52-next.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/ai-service/agent/prompts/build-base-system-prompt.d.ts.map +1 -1
- package/dist/ai-service/agent/prompts/build-base-system-prompt.js +36 -7
- package/dist/ai-service/agent/prompts/build-base-system-prompt.js.map +1 -1
- package/dist/ai-service/agent/subagents/apis/generate-api-source.d.ts +28 -3
- package/dist/ai-service/agent/subagents/apis/generate-api-source.d.ts.map +1 -1
- package/dist/ai-service/agent/subagents/apis/generate-api-source.js +102 -8
- package/dist/ai-service/agent/subagents/apis/generate-api-source.js.map +1 -1
- package/dist/ai-service/agent/subagents/apis/prompt-builder.d.ts +1 -1
- package/dist/ai-service/agent/subagents/apis/prompt-builder.js +2 -2
- package/dist/ai-service/agent/subagents/apis/state.d.ts +10 -0
- package/dist/ai-service/agent/subagents/apis/state.d.ts.map +1 -1
- package/dist/ai-service/agent/subagents/apis/system-prompt.d.ts +1 -1
- package/dist/ai-service/agent/subagents/apis/system-prompt.d.ts.map +1 -1
- package/dist/ai-service/agent/subagents/apis/system-prompt.js +294 -5
- package/dist/ai-service/agent/subagents/apis/system-prompt.js.map +1 -1
- package/dist/ai-service/agent/tools/apis/build-api.d.ts +2 -0
- package/dist/ai-service/agent/tools/apis/build-api.d.ts.map +1 -1
- package/dist/ai-service/agent/tools/apis/build-api.js +96 -13
- package/dist/ai-service/agent/tools/apis/build-api.js.map +1 -1
- package/dist/ai-service/agent/tools/apis/finalize-api.d.ts.map +1 -1
- package/dist/ai-service/agent/tools/apis/finalize-api.js +13 -4
- package/dist/ai-service/agent/tools/apis/finalize-api.js.map +1 -1
- package/dist/ai-service/agent/tools/build-debug.d.ts.map +1 -1
- package/dist/ai-service/agent/tools/build-debug.js +64 -2
- package/dist/ai-service/agent/tools/build-debug.js.map +1 -1
- package/dist/ai-service/agent/tools/build-edit-file.d.ts.map +1 -1
- package/dist/ai-service/agent/tools/build-edit-file.js +12 -0
- package/dist/ai-service/agent/tools/build-edit-file.js.map +1 -1
- package/dist/ai-service/agent/tools/build-finalize.d.ts.map +1 -1
- package/dist/ai-service/agent/tools/build-finalize.js +43 -0
- package/dist/ai-service/agent/tools/build-finalize.js.map +1 -1
- package/dist/ai-service/agent/tools/build-multi-edit-file.d.ts.map +1 -1
- package/dist/ai-service/agent/tools/build-multi-edit-file.js +6 -0
- package/dist/ai-service/agent/tools/build-multi-edit-file.js.map +1 -1
- package/dist/ai-service/agent/tools/build-write-file.d.ts.map +1 -1
- package/dist/ai-service/agent/tools/build-write-file.js +6 -0
- package/dist/ai-service/agent/tools/build-write-file.js.map +1 -1
- package/dist/ai-service/agent/tools/debug-cache.d.ts +42 -0
- package/dist/ai-service/agent/tools/debug-cache.d.ts.map +1 -1
- package/dist/ai-service/agent/tools/debug-cache.js +121 -14
- package/dist/ai-service/agent/tools/debug-cache.js.map +1 -1
- package/dist/ai-service/agent/tools/index.d.ts +1 -0
- package/dist/ai-service/agent/tools/index.d.ts.map +1 -1
- package/dist/ai-service/agent/tools/index.js +1 -0
- package/dist/ai-service/agent/tools/index.js.map +1 -1
- package/dist/ai-service/agent/tools/integrations/execute-request.d.ts +9 -5
- package/dist/ai-service/agent/tools/integrations/execute-request.d.ts.map +1 -1
- package/dist/ai-service/agent/tools/integrations/execute-request.js +105 -7
- package/dist/ai-service/agent/tools/integrations/execute-request.js.map +1 -1
- package/dist/ai-service/agent/tools/integrations/index.d.ts +1 -1
- package/dist/ai-service/agent/tools/integrations/index.d.ts.map +1 -1
- package/dist/ai-service/agent/tools/integrations/index.js +1 -1
- package/dist/ai-service/agent/tools/integrations/index.js.map +1 -1
- package/dist/ai-service/agent/tools/integrations/metadata.d.ts +0 -4
- package/dist/ai-service/agent/tools/integrations/metadata.d.ts.map +1 -1
- package/dist/ai-service/agent/tools/integrations/metadata.js +0 -44
- package/dist/ai-service/agent/tools/integrations/metadata.js.map +1 -1
- package/dist/ai-service/agent/tools.d.ts.map +1 -1
- package/dist/ai-service/agent/tools.js +45 -10
- package/dist/ai-service/agent/tools.js.map +1 -1
- package/dist/ai-service/agent/tools2/access-control.d.ts +1 -1
- package/dist/ai-service/agent/tools2/access-control.d.ts.map +1 -1
- package/dist/ai-service/agent/tools2/access-control.js +13 -5
- package/dist/ai-service/agent/tools2/access-control.js.map +1 -1
- package/dist/ai-service/agent/tools2/entity-permissions.d.ts +180 -0
- package/dist/ai-service/agent/tools2/entity-permissions.d.ts.map +1 -0
- package/dist/ai-service/agent/tools2/entity-permissions.js +154 -0
- package/dist/ai-service/agent/tools2/entity-permissions.js.map +1 -0
- package/dist/ai-service/agent/tools2/example.d.ts.map +1 -1
- package/dist/ai-service/agent/tools2/example.js +25 -6
- package/dist/ai-service/agent/tools2/example.js.map +1 -1
- package/dist/ai-service/agent/tools2/index.d.ts +4 -2
- package/dist/ai-service/agent/tools2/index.d.ts.map +1 -1
- package/dist/ai-service/agent/tools2/index.js +2 -1
- package/dist/ai-service/agent/tools2/index.js.map +1 -1
- package/dist/ai-service/agent/tools2/registry.d.ts +47 -0
- package/dist/ai-service/agent/tools2/registry.d.ts.map +1 -1
- package/dist/ai-service/agent/tools2/registry.js +122 -19
- package/dist/ai-service/agent/tools2/registry.js.map +1 -1
- package/dist/ai-service/agent/tools2/test-utils.d.ts +7 -0
- package/dist/ai-service/agent/tools2/test-utils.d.ts.map +1 -0
- package/dist/ai-service/agent/tools2/test-utils.js +10 -0
- package/dist/ai-service/agent/tools2/test-utils.js.map +1 -0
- package/dist/ai-service/agent/tools2/tools/ask-multi-choice.d.ts.map +1 -1
- package/dist/ai-service/agent/tools2/tools/ask-multi-choice.js +17 -9
- package/dist/ai-service/agent/tools2/tools/ask-multi-choice.js.map +1 -1
- package/dist/ai-service/agent/tools2/tools/exit-plan-mode.d.ts +2 -1
- package/dist/ai-service/agent/tools2/tools/exit-plan-mode.d.ts.map +1 -1
- package/dist/ai-service/agent/tools2/tools/exit-plan-mode.js +192 -27
- package/dist/ai-service/agent/tools2/tools/exit-plan-mode.js.map +1 -1
- package/dist/ai-service/agent/tools2/tools/grep-metadata-ripgrep.d.ts +18 -0
- package/dist/ai-service/agent/tools2/tools/grep-metadata-ripgrep.d.ts.map +1 -0
- package/dist/ai-service/agent/tools2/tools/grep-metadata-ripgrep.js +837 -0
- package/dist/ai-service/agent/tools2/tools/grep-metadata-ripgrep.js.map +1 -0
- package/dist/ai-service/agent/tools2/tools/grep-metadata.d.ts +96 -0
- package/dist/ai-service/agent/tools2/tools/grep-metadata.d.ts.map +1 -0
- package/dist/ai-service/agent/tools2/tools/grep-metadata.js +437 -0
- package/dist/ai-service/agent/tools2/tools/grep-metadata.js.map +1 -0
- package/dist/ai-service/agent/tools2/types.d.ts +113 -6
- package/dist/ai-service/agent/tools2/types.d.ts.map +1 -1
- package/dist/ai-service/agent/tools2/types.js +15 -0
- package/dist/ai-service/agent/tools2/types.js.map +1 -1
- package/dist/ai-service/agent/tools2/utils.d.ts +9 -0
- package/dist/ai-service/agent/tools2/utils.d.ts.map +1 -0
- package/dist/ai-service/agent/tools2/utils.js +11 -0
- package/dist/ai-service/agent/tools2/utils.js.map +1 -0
- package/dist/ai-service/chat/chat-session-store.d.ts.map +1 -1
- package/dist/ai-service/chat/chat-session-store.js +12 -0
- package/dist/ai-service/chat/chat-session-store.js.map +1 -1
- package/dist/ai-service/const.d.ts +0 -1
- package/dist/ai-service/const.d.ts.map +1 -1
- package/dist/ai-service/const.js +0 -1
- package/dist/ai-service/const.js.map +1 -1
- package/dist/ai-service/index.d.ts +11 -0
- package/dist/ai-service/index.d.ts.map +1 -1
- package/dist/ai-service/index.js +50 -3
- package/dist/ai-service/index.js.map +1 -1
- package/dist/ai-service/integrations/metadata-storage/index.d.ts +25 -0
- package/dist/ai-service/integrations/metadata-storage/index.d.ts.map +1 -0
- package/dist/ai-service/integrations/metadata-storage/index.js +2 -0
- package/dist/ai-service/integrations/metadata-storage/index.js.map +1 -0
- package/dist/ai-service/integrations/metadata-storage/local.d.ts +53 -0
- package/dist/ai-service/integrations/metadata-storage/local.d.ts.map +1 -0
- package/dist/ai-service/integrations/metadata-storage/local.js +315 -0
- package/dist/ai-service/integrations/metadata-storage/local.js.map +1 -0
- package/dist/ai-service/integrations/store.d.ts +17 -14
- package/dist/ai-service/integrations/store.d.ts.map +1 -1
- package/dist/ai-service/integrations/store.js +124 -131
- package/dist/ai-service/integrations/store.js.map +1 -1
- package/dist/ai-service/llm/utils.js +1 -1
- package/dist/ai-service/llm/utils.js.map +1 -1
- package/dist/ai-service/llmobs/helpers.d.ts.map +1 -1
- package/dist/ai-service/llmobs/helpers.js +17 -23
- package/dist/ai-service/llmobs/helpers.js.map +1 -1
- package/dist/ai-service/llmobs/middleware/stream-text.d.ts.map +1 -1
- package/dist/ai-service/llmobs/middleware/stream-text.js +1 -0
- package/dist/ai-service/llmobs/middleware/stream-text.js.map +1 -1
- package/dist/ai-service/state-machine/clark-fsm.d.ts +5 -2
- package/dist/ai-service/state-machine/clark-fsm.d.ts.map +1 -1
- package/dist/ai-service/state-machine/clark-fsm.js +2 -2
- package/dist/ai-service/state-machine/clark-fsm.js.map +1 -1
- package/dist/ai-service/state-machine/handlers/agent-planning.d.ts.map +1 -1
- package/dist/ai-service/state-machine/handlers/agent-planning.js +5 -2
- package/dist/ai-service/state-machine/handlers/agent-planning.js.map +1 -1
- package/dist/ai-service/state-machine/handlers/llm-generating.d.ts.map +1 -1
- package/dist/ai-service/state-machine/handlers/llm-generating.js +16 -7
- package/dist/ai-service/state-machine/handlers/llm-generating.js.map +1 -1
- package/dist/ai-service/state-machine/handlers/post-processing.d.ts.map +1 -1
- package/dist/ai-service/state-machine/handlers/post-processing.js +11 -2
- package/dist/ai-service/state-machine/handlers/post-processing.js.map +1 -1
- package/dist/ai-service/transform/api-builder/to-sdk-transformer.d.ts.map +1 -1
- package/dist/ai-service/transform/api-builder/to-sdk-transformer.js +0 -1
- package/dist/ai-service/transform/api-builder/to-sdk-transformer.js.map +1 -1
- package/dist/ai-service/types.d.ts +1 -0
- package/dist/ai-service/types.d.ts.map +1 -1
- package/dist/ai-service/types.js.map +1 -1
- package/dist/file-sync-vite-plugin.d.ts.map +1 -1
- package/dist/file-sync-vite-plugin.js +12 -0
- package/dist/file-sync-vite-plugin.js.map +1 -1
- package/dist/file-system-helpers.d.ts.map +1 -1
- package/dist/file-system-helpers.js +1 -0
- package/dist/file-system-helpers.js.map +1 -1
- package/dist/socket-manager.d.ts.map +1 -1
- package/dist/socket-manager.js +6 -0
- package/dist/socket-manager.js.map +1 -1
- package/dist/sync-service/index.d.ts +8 -0
- package/dist/sync-service/index.d.ts.map +1 -1
- package/dist/sync-service/index.js +15 -0
- package/dist/sync-service/index.js.map +1 -1
- package/dist/util/open-api.d.ts +2 -46
- package/dist/util/open-api.d.ts.map +1 -1
- package/dist/util/open-api.js +1 -114
- package/dist/util/open-api.js.map +1 -1
- package/package.json +7 -7
- package/dist/ai-service/integrations/metadata/database.d.ts +0 -15
- package/dist/ai-service/integrations/metadata/database.d.ts.map +0 -1
- package/dist/ai-service/integrations/metadata/database.js +0 -121
- package/dist/ai-service/integrations/metadata/database.js.map +0 -1
- package/dist/ai-service/integrations/metadata/databricks.d.ts +0 -25
- package/dist/ai-service/integrations/metadata/databricks.d.ts.map +0 -1
- package/dist/ai-service/integrations/metadata/databricks.js +0 -174
- package/dist/ai-service/integrations/metadata/databricks.js.map +0 -1
- package/dist/ai-service/integrations/metadata/graphql-based.d.ts +0 -16
- package/dist/ai-service/integrations/metadata/graphql-based.d.ts.map +0 -1
- package/dist/ai-service/integrations/metadata/graphql-based.js +0 -383
- package/dist/ai-service/integrations/metadata/graphql-based.js.map +0 -1
- package/dist/ai-service/integrations/metadata/index.d.ts +0 -9
- package/dist/ai-service/integrations/metadata/index.d.ts.map +0 -1
- package/dist/ai-service/integrations/metadata/index.js +0 -7
- package/dist/ai-service/integrations/metadata/index.js.map +0 -1
- package/dist/ai-service/integrations/metadata/llm-utils.d.ts +0 -24
- package/dist/ai-service/integrations/metadata/llm-utils.d.ts.map +0 -1
- package/dist/ai-service/integrations/metadata/llm-utils.js +0 -45
- package/dist/ai-service/integrations/metadata/llm-utils.js.map +0 -1
- package/dist/ai-service/integrations/metadata/open-api.d.ts +0 -17
- package/dist/ai-service/integrations/metadata/open-api.d.ts.map +0 -1
- package/dist/ai-service/integrations/metadata/open-api.js +0 -62
- package/dist/ai-service/integrations/metadata/open-api.js.map +0 -1
- package/dist/ai-service/integrations/metadata/table-based.d.ts +0 -10
- package/dist/ai-service/integrations/metadata/table-based.d.ts.map +0 -1
- package/dist/ai-service/integrations/metadata/table-based.js +0 -27
- package/dist/ai-service/integrations/metadata/table-based.js.map +0 -1
- package/dist/ai-service/prompt-builder-service/builders/dbschema-intent.d.ts +0 -19
- package/dist/ai-service/prompt-builder-service/builders/dbschema-intent.d.ts.map +0 -1
- package/dist/ai-service/prompt-builder-service/builders/dbschema-intent.js +0 -145
- package/dist/ai-service/prompt-builder-service/builders/dbschema-intent.js.map +0 -1
- package/dist/ai-service/prompt-builder-service/builders/openapi-intent.d.ts +0 -19
- package/dist/ai-service/prompt-builder-service/builders/openapi-intent.d.ts.map +0 -1
- package/dist/ai-service/prompt-builder-service/builders/openapi-intent.js +0 -41
- package/dist/ai-service/prompt-builder-service/builders/openapi-intent.js.map +0 -1
- package/dist/ai-service/prompt-builder-service/index.d.ts +0 -3
- package/dist/ai-service/prompt-builder-service/index.d.ts.map +0 -1
- package/dist/ai-service/prompt-builder-service/index.js +0 -3
- package/dist/ai-service/prompt-builder-service/index.js.map +0 -1
|
@@ -35,20 +35,12 @@ var __esDecorate = (this && this.__esDecorate) || function (ctor, descriptorIn,
|
|
|
35
35
|
import { DatabricksPlugin, GraphQLIntegrationPlugin, PluginType, RegisteredPlugins, SalesforcePlugin, } from "@superblocksteam/shared";
|
|
36
36
|
import { isEmpty } from "lodash-es";
|
|
37
37
|
import { getLogger } from "../../util/logger.js";
|
|
38
|
-
import { datasourceSdkClassByType } from "../const.js";
|
|
39
38
|
import { databaseIntegrationTypes } from "../const.js";
|
|
40
39
|
import llmobs from "../llmobs/index.js";
|
|
41
|
-
import {
|
|
42
|
-
import { DatabricksIntegrationMetadata } from "./metadata/databricks.js";
|
|
43
|
-
import { GraphQLIntegrationMetadata } from "./metadata/graphql-based.js";
|
|
44
|
-
import { IntegrationMetadata } from "./metadata/index.js";
|
|
45
|
-
import { OpenApiIntegrationMetadata } from "./metadata/open-api.js";
|
|
46
|
-
import { TableBasedIntegrationMetadata } from "./metadata/table-based.js";
|
|
40
|
+
import { LocalMetadataStorage } from "./metadata-storage/local.js";
|
|
47
41
|
let IntegrationStore = (() => {
|
|
48
42
|
let _instanceExtraInitializers = [];
|
|
49
43
|
let _getMetadata_decorators;
|
|
50
|
-
let _getIntegrationForPromptContext_decorators;
|
|
51
|
-
let _getRelevantMetadata_decorators;
|
|
52
44
|
let _getOpenApiSpec_decorators;
|
|
53
45
|
return class IntegrationStore {
|
|
54
46
|
static {
|
|
@@ -57,43 +49,48 @@ let IntegrationStore = (() => {
|
|
|
57
49
|
kind: "task",
|
|
58
50
|
name: "IntegrationStore.getMetadata",
|
|
59
51
|
})];
|
|
60
|
-
_getIntegrationForPromptContext_decorators = [llmobs.decorate({
|
|
61
|
-
kind: "task",
|
|
62
|
-
name: "IntegrationStore.getIntegrationForPromptContext",
|
|
63
|
-
})];
|
|
64
|
-
_getRelevantMetadata_decorators = [llmobs.decorate({
|
|
65
|
-
kind: "task",
|
|
66
|
-
name: "IntegrationStore.getRelevantMetadata",
|
|
67
|
-
})];
|
|
68
52
|
_getOpenApiSpec_decorators = [llmobs.decorate({
|
|
69
53
|
kind: "task",
|
|
70
54
|
name: "IntegrationStore.getOpenApiSpec",
|
|
71
55
|
})];
|
|
72
56
|
__esDecorate(this, null, _getMetadata_decorators, { kind: "method", name: "getMetadata", static: false, private: false, access: { has: obj => "getMetadata" in obj, get: obj => obj.getMetadata }, metadata: _metadata }, null, _instanceExtraInitializers);
|
|
73
|
-
__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);
|
|
74
|
-
__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);
|
|
75
57
|
__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);
|
|
76
58
|
if (_metadata) Object.defineProperty(this, Symbol.metadata, { enumerable: true, configurable: true, writable: true, value: _metadata });
|
|
77
59
|
}
|
|
78
60
|
openApiSpecs = (__runInitializers(this, _instanceExtraInitializers), {});
|
|
79
61
|
pluginRegistry;
|
|
80
62
|
integrations = [];
|
|
81
|
-
|
|
82
|
-
|
|
63
|
+
metadataStorage;
|
|
64
|
+
pendingMetadataFetches = new Map();
|
|
65
|
+
constructor(pluginRegistry = RegisteredPlugins("foo"), rootDir, maxCacheSizeMB, maxFileSizeMB) {
|
|
83
66
|
this.pluginRegistry = pluginRegistry;
|
|
67
|
+
this.metadataStorage = new LocalMetadataStorage({
|
|
68
|
+
rootDir: rootDir || process.cwd(),
|
|
69
|
+
maxTotalCacheMB: maxCacheSizeMB,
|
|
70
|
+
maxSingleFileMB: maxFileSizeMB,
|
|
71
|
+
});
|
|
72
|
+
}
|
|
73
|
+
async initialize() {
|
|
74
|
+
await this.metadataStorage.initialize();
|
|
75
|
+
}
|
|
76
|
+
async removeIntegrationCache() {
|
|
77
|
+
await this.metadataStorage.cleanup();
|
|
84
78
|
}
|
|
85
79
|
availableIntegrations() {
|
|
86
|
-
|
|
80
|
+
const available = this.integrations.filter((i) => i.plugin.type === PluginType.CODE || !i.plugin.isBuiltIn);
|
|
81
|
+
getLogger().debug(`[integration-store] availableIntegrations() returning ${available.length} integrations: ${available.map((i) => `${i.name}(${i.id})`).join(", ")}`);
|
|
82
|
+
return available;
|
|
87
83
|
}
|
|
88
84
|
getPlugin(pluginId) {
|
|
89
85
|
return this.pluginRegistry.get(pluginId);
|
|
90
86
|
}
|
|
91
87
|
insertFromPromptContext(integration, editorClient) {
|
|
88
|
+
getLogger().info(`[integration-store] Registering integration ${integration.id} (${integration.name}) type=${integration.type}`);
|
|
92
89
|
const pluginId = integration.type;
|
|
93
90
|
const plugin = this.pluginRegistry.get(pluginId);
|
|
94
91
|
if (!plugin) {
|
|
95
92
|
// this should never happen
|
|
96
|
-
getLogger().error(`No plugin found for integration type ${integration.type}`);
|
|
93
|
+
getLogger().error(`No plugin found for integration type ${integration.type} - skipping registration`);
|
|
97
94
|
return;
|
|
98
95
|
}
|
|
99
96
|
const pluginHeader = {
|
|
@@ -128,52 +125,108 @@ let IntegrationStore = (() => {
|
|
|
128
125
|
return this.integrations.find((i) => i.id === id);
|
|
129
126
|
}
|
|
130
127
|
async getMetadata(id, editorClient) {
|
|
128
|
+
// Check for in-flight fetch to avoid duplicate requests
|
|
129
|
+
const pending = this.pendingMetadataFetches.get(id);
|
|
130
|
+
if (pending) {
|
|
131
|
+
getLogger().debug(`[integration-store] Waiting for in-flight fetch of ${id}`);
|
|
132
|
+
return pending;
|
|
133
|
+
}
|
|
134
|
+
// Start new fetch and track it
|
|
135
|
+
const fetchPromise = this._fetchMetadata(id, editorClient);
|
|
136
|
+
this.pendingMetadataFetches.set(id, fetchPromise);
|
|
137
|
+
try {
|
|
138
|
+
const result = await fetchPromise;
|
|
139
|
+
return result;
|
|
140
|
+
}
|
|
141
|
+
finally {
|
|
142
|
+
// Clean up tracking
|
|
143
|
+
this.pendingMetadataFetches.delete(id);
|
|
144
|
+
}
|
|
145
|
+
}
|
|
146
|
+
async _fetchMetadata(id, editorClient) {
|
|
131
147
|
const header = this.getHeader(id);
|
|
132
148
|
if (!header) {
|
|
133
|
-
getLogger().warn(`[integration-store] No integration found for id ${id}
|
|
134
|
-
return
|
|
149
|
+
getLogger().warn(`[integration-store] No integration found for id ${id}`);
|
|
150
|
+
return undefined;
|
|
135
151
|
}
|
|
136
152
|
if (!editorClient) {
|
|
137
|
-
getLogger().warn(`[integration-store] No editor client provided, cannot fetch metadata for integration ${id}
|
|
138
|
-
return
|
|
153
|
+
getLogger().warn(`[integration-store] No editor client provided, cannot fetch metadata for integration ${id}`);
|
|
154
|
+
return undefined;
|
|
139
155
|
}
|
|
140
|
-
if
|
|
141
|
-
|
|
156
|
+
// Check if we have it cached in filesystem
|
|
157
|
+
const cachedPath = await this.metadataStorage.loadMetadata(id);
|
|
158
|
+
if (cachedPath) {
|
|
159
|
+
// Cache hit! Return metadata with special gron path marker
|
|
160
|
+
getLogger().debug(`[integration-store] Using filesystem cache for ${id}`);
|
|
161
|
+
return { _gronPath: cachedPath };
|
|
142
162
|
}
|
|
163
|
+
// Cache miss - fetch from server
|
|
164
|
+
getLogger().info(`[integration-store] Fetching metadata for ${id}`);
|
|
143
165
|
if (header.plugin.openApiSpecRef) {
|
|
144
|
-
//
|
|
145
|
-
const spec = await this.getOpenApiSpec(header.id, editorClient
|
|
146
|
-
|
|
147
|
-
|
|
166
|
+
// OpenAPI integration - get the spec
|
|
167
|
+
const spec = await this.getOpenApiSpec(header.id, editorClient);
|
|
168
|
+
const metadata = { openApiSpec: spec };
|
|
169
|
+
// Write cache and wait for it to complete
|
|
170
|
+
try {
|
|
171
|
+
await this.metadataStorage.saveMetadata(id, metadata);
|
|
172
|
+
// Return with gron path after successful cache write
|
|
173
|
+
const gronPath = await this.metadataStorage.loadMetadata(id);
|
|
174
|
+
if (gronPath) {
|
|
175
|
+
return { _gronPath: gronPath };
|
|
176
|
+
}
|
|
177
|
+
}
|
|
178
|
+
catch (err) {
|
|
179
|
+
getLogger().error(`[integration-store] Failed to cache metadata for ${id}: ${err.message}`);
|
|
180
|
+
// Continue anyway - return the metadata even if caching fails
|
|
148
181
|
}
|
|
149
|
-
|
|
150
|
-
return this.metadataCache.get(id);
|
|
182
|
+
return metadata;
|
|
151
183
|
}
|
|
184
|
+
// Regular database integration - get metadata
|
|
152
185
|
const integration = await editorClient?.call.aiGetIntegration({
|
|
153
186
|
integrationId: id,
|
|
154
187
|
});
|
|
155
188
|
// Check if the integration has an openApiSpecRef at the top level
|
|
156
|
-
// This is returned by the UI when a REST API integration has
|
|
157
|
-
|
|
158
|
-
|
|
159
|
-
const
|
|
160
|
-
|
|
161
|
-
|
|
189
|
+
// This is returned by the UI when a REST API integration has a custom OpenAPI spec
|
|
190
|
+
if (integration.openApiSpecRef) {
|
|
191
|
+
const spec = await this.getOpenApiSpec(header.id, editorClient, integration.openApiSpecRef);
|
|
192
|
+
const metadata = {
|
|
193
|
+
openApiSpec: spec,
|
|
194
|
+
openApiSpecRef: integration.openApiSpecRef,
|
|
195
|
+
};
|
|
196
|
+
// Write cache and wait for it to complete
|
|
197
|
+
try {
|
|
198
|
+
await this.metadataStorage.saveMetadata(id, metadata);
|
|
199
|
+
// Return with gron path after successful cache write
|
|
200
|
+
const gronPath = await this.metadataStorage.loadMetadata(id);
|
|
201
|
+
if (gronPath) {
|
|
202
|
+
return { _gronPath: gronPath };
|
|
203
|
+
}
|
|
162
204
|
}
|
|
163
|
-
|
|
164
|
-
|
|
205
|
+
catch (err) {
|
|
206
|
+
getLogger().error(`[integration-store] Failed to cache metadata for ${id}: ${err.message}`);
|
|
207
|
+
// Continue anyway - return the metadata even if caching fails
|
|
208
|
+
}
|
|
209
|
+
return metadata;
|
|
165
210
|
}
|
|
166
211
|
if (Object.keys(integration.metadata).length === 0) {
|
|
167
|
-
getLogger().warn(`[integration-store] No metadata found for integration ${id}
|
|
168
|
-
return
|
|
212
|
+
getLogger().warn(`[integration-store] No metadata found for integration ${id}`);
|
|
213
|
+
return undefined;
|
|
169
214
|
}
|
|
170
|
-
|
|
171
|
-
|
|
172
|
-
|
|
173
|
-
|
|
174
|
-
|
|
175
|
-
|
|
176
|
-
|
|
215
|
+
// Write cache and wait for it to complete
|
|
216
|
+
try {
|
|
217
|
+
await this.metadataStorage.saveMetadata(id, integration.metadata);
|
|
218
|
+
getLogger().info(`[integration-store] Successfully cached metadata for ${id}`);
|
|
219
|
+
// Return with gron path after successful cache write
|
|
220
|
+
const gronPath = await this.metadataStorage.loadMetadata(id);
|
|
221
|
+
if (gronPath) {
|
|
222
|
+
return { _gronPath: gronPath };
|
|
223
|
+
}
|
|
224
|
+
}
|
|
225
|
+
catch (err) {
|
|
226
|
+
getLogger().error(`[integration-store] Failed to cache metadata for ${id}: ${err.message}`);
|
|
227
|
+
// Continue anyway - return the metadata even if caching fails
|
|
228
|
+
}
|
|
229
|
+
return integration.metadata;
|
|
177
230
|
}
|
|
178
231
|
async getIntegration(id, editorClient) {
|
|
179
232
|
const header = this.getHeader(id);
|
|
@@ -186,69 +239,6 @@ let IntegrationStore = (() => {
|
|
|
186
239
|
metadata: metadata || {},
|
|
187
240
|
};
|
|
188
241
|
}
|
|
189
|
-
async getIntegrationForPromptContext(integration, prompt, clark, params) {
|
|
190
|
-
const { name, id } = integration;
|
|
191
|
-
const type = integration.type;
|
|
192
|
-
if (!name ||
|
|
193
|
-
!id ||
|
|
194
|
-
!isSdkIntegration({
|
|
195
|
-
type,
|
|
196
|
-
})) {
|
|
197
|
-
return null;
|
|
198
|
-
}
|
|
199
|
-
const metadata = await this.getRelevantMetadata({
|
|
200
|
-
clark,
|
|
201
|
-
integrationId: integration.id,
|
|
202
|
-
prompt,
|
|
203
|
-
llmProvider: params.llmProvider,
|
|
204
|
-
});
|
|
205
|
-
return {
|
|
206
|
-
name,
|
|
207
|
-
id,
|
|
208
|
-
type,
|
|
209
|
-
metadata,
|
|
210
|
-
};
|
|
211
|
-
}
|
|
212
|
-
async getRelevantMetadata({ clark, integrationId, prompt, llmProvider, }) {
|
|
213
|
-
getLogger().debug(`[integration-store] Fetching metadata for integration ${integrationId}
|
|
214
|
-
`);
|
|
215
|
-
// If we're operating on an openapi integration, we need to fetch its full spec.
|
|
216
|
-
// It would be nice to push this down, but the cache lives on this instance as this.openApiSpecs
|
|
217
|
-
const integration = await this.getIntegration(integrationId, clark.context.peer);
|
|
218
|
-
if (!integration) {
|
|
219
|
-
throw new Error(`No integration found for id ${integrationId}`);
|
|
220
|
-
}
|
|
221
|
-
const metadata = integration.metadata;
|
|
222
|
-
if (Object.keys(metadata).length === 0) {
|
|
223
|
-
// we couldn't fetch metadata for some reason, throw error
|
|
224
|
-
throw new Error(`No metadata found for integration ${integration.id}`);
|
|
225
|
-
}
|
|
226
|
-
const klass = this.getMetadataClassForIntegration(integration);
|
|
227
|
-
getLogger().info(`[integration-store] Using metadata class ${klass.name} for integration ${integration.id} of type ${integration.plugin.id}`);
|
|
228
|
-
return await klass.refineMetadataForPrompt({
|
|
229
|
-
metadata,
|
|
230
|
-
prompt,
|
|
231
|
-
llmProvider,
|
|
232
|
-
});
|
|
233
|
-
}
|
|
234
|
-
getMetadataClassForIntegration(integration) {
|
|
235
|
-
if (isOpenApiIntegration(integration)) {
|
|
236
|
-
return OpenApiIntegrationMetadata;
|
|
237
|
-
}
|
|
238
|
-
if (isDatabricksIntegration(integration)) {
|
|
239
|
-
return DatabricksIntegrationMetadata;
|
|
240
|
-
}
|
|
241
|
-
if (isDatabaseIntegration(integration)) {
|
|
242
|
-
return DatabaseIntegrationMetadata;
|
|
243
|
-
}
|
|
244
|
-
if (isTableBasedIntegration(integration)) {
|
|
245
|
-
return TableBasedIntegrationMetadata;
|
|
246
|
-
}
|
|
247
|
-
if (isGraphQlIntegration(integration)) {
|
|
248
|
-
return GraphQLIntegrationMetadata;
|
|
249
|
-
}
|
|
250
|
-
return IntegrationMetadata;
|
|
251
|
-
}
|
|
252
242
|
async getOpenApiSpec(integrationId, editorClient, specRef) {
|
|
253
243
|
if (this.openApiSpecs[integrationId]) {
|
|
254
244
|
return this.openApiSpecs[integrationId];
|
|
@@ -278,36 +268,39 @@ let IntegrationStore = (() => {
|
|
|
278
268
|
this.openApiSpecs[header.id] = result;
|
|
279
269
|
return result;
|
|
280
270
|
}
|
|
271
|
+
/**
|
|
272
|
+
* Get the entity tag for the integration
|
|
273
|
+
* eg @SuperblocksGithub[id=2ce0985a-342b-445d-b897-d17d80a26b8b]
|
|
274
|
+
*/
|
|
275
|
+
getEntityTag(id) {
|
|
276
|
+
const header = this.getHeader(id);
|
|
277
|
+
if (!header) {
|
|
278
|
+
return undefined;
|
|
279
|
+
}
|
|
280
|
+
return `@${header.name}[id=${id}]`;
|
|
281
|
+
}
|
|
281
282
|
};
|
|
282
283
|
})();
|
|
283
284
|
export { IntegrationStore };
|
|
284
|
-
function isOpenApiIntegration(header) {
|
|
285
|
-
// Check if it's an Integration (has metadata property)
|
|
286
|
-
if ("metadata" in header) {
|
|
287
|
-
return (header.plugin.openApiSpecRef !== undefined ||
|
|
288
|
-
header.metadata.openApiSpecRef !== undefined);
|
|
289
|
-
}
|
|
290
|
-
// Otherwise it's just an IntegrationHeader
|
|
285
|
+
export function isOpenApiIntegration(header) {
|
|
291
286
|
return header.plugin.openApiSpecRef !== undefined;
|
|
292
287
|
}
|
|
293
|
-
function
|
|
294
|
-
return (typeof integration.type === "string" &&
|
|
295
|
-
integration.type in datasourceSdkClassByType);
|
|
296
|
-
}
|
|
297
|
-
function isDatabricksIntegration(header) {
|
|
288
|
+
export function isDatabricksIntegration(header) {
|
|
298
289
|
return header.plugin.id === DatabricksPlugin.id;
|
|
299
290
|
}
|
|
300
|
-
function isDatabaseIntegration(header) {
|
|
291
|
+
export function isDatabaseIntegration(header) {
|
|
301
292
|
return (databaseIntegrationTypes.includes(header.plugin.id) || header.plugin.id === SalesforcePlugin.id);
|
|
302
293
|
}
|
|
303
|
-
function isTableBasedIntegration(integration) {
|
|
304
|
-
const schema = integration.metadata?.
|
|
294
|
+
export function isTableBasedIntegration(integration) {
|
|
295
|
+
const schema = integration.metadata?.databaseSchemaMetadata ||
|
|
296
|
+
integration.metadata?.dbSchema ||
|
|
297
|
+
integration.metadata?.schema;
|
|
305
298
|
if (schema && schema.tables && Array.isArray(schema.tables)) {
|
|
306
299
|
return true;
|
|
307
300
|
}
|
|
308
301
|
return false;
|
|
309
302
|
}
|
|
310
|
-
function isGraphQlIntegration(header) {
|
|
303
|
+
export function isGraphQlIntegration(header) {
|
|
311
304
|
return header.plugin.id === GraphQLIntegrationPlugin.id;
|
|
312
305
|
}
|
|
313
306
|
//# sourceMappingURL=store.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"store.js","sourceRoot":"","sources":["../../../src/ai-service/integrations/store.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,OAAO,EACL,gBAAgB,EAChB,wBAAwB,EACxB,UAAU,EACV,iBAAiB,EACjB,gBAAgB,GAGjB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,
|
|
1
|
+
{"version":3,"file":"store.js","sourceRoot":"","sources":["../../../src/ai-service/integrations/store.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,OAAO,EACL,gBAAgB,EAChB,wBAAwB,EACxB,UAAU,EACV,iBAAiB,EACjB,gBAAgB,GAGjB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;IAgCtD,gBAAgB;;;;iBAAhB,gBAAgB;;;uCAoG1B,MAAM,CAAC,QAAQ,CAAC;oBACf,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,8BAA8B;iBACrC,CAAC;0CA6JD,MAAM,CAAC,QAAQ,CAAC;oBACf,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,iCAAiC;iBACxC,CAAC;YA/JF,oLAAM,WAAW,6DAwBhB;YAwID,6LAAM,cAAc,6DA2CnB;;;QAlTO,YAAY,IADT,mDAAgB,EACyB,EAAE,EAAC;QAC/C,cAAc,CAAiB;QAC/B,YAAY,GAAwB,EAAE,CAAC;QACvC,eAAe,CAAkB;QACjC,sBAAsB,GAG1B,IAAI,GAAG,EAAE,CAAC;QAEd,YACE,iBAAiC,iBAAiB,CAAC,KAAK,CAAC,EACzD,OAAgB,EAChB,cAAuB,EACvB,aAAsB;YAEtB,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;YACrC,IAAI,CAAC,eAAe,GAAG,IAAI,oBAAoB,CAAC;gBAC9C,OAAO,EAAE,OAAO,IAAI,OAAO,CAAC,GAAG,EAAE;gBACjC,eAAe,EAAE,cAAc;gBAC/B,eAAe,EAAE,aAAa;aAC/B,CAAC,CAAC;QACL,CAAC;QAED,KAAK,CAAC,UAAU;YACd,MAAM,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,CAAC;QAC1C,CAAC;QAED,KAAK,CAAC,sBAAsB;YAC1B,MAAM,IAAI,CAAC,eAAe,CAAC,OAAO,EAAE,CAAC;QACvC,CAAC;QAED,qBAAqB;YACnB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,CACxC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,KAAK,UAAU,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAChE,CAAC;YACF,SAAS,EAAE,CAAC,KAAK,CACf,yDAAyD,SAAS,CAAC,MAAM,kBAAkB,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACnJ,CAAC;YACF,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,SAAS,CAAC,QAAkB;YAC1B,OAAO,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC3C,CAAC;QAED,uBAAuB,CACrB,WAAqC,EACrC,YAA2B;YAE3B,SAAS,EAAE,CAAC,IAAI,CACd,+CAA+C,WAAW,CAAC,EAAE,KAAK,WAAW,CAAC,IAAI,UAAU,WAAW,CAAC,IAAI,EAAE,CAC/G,CAAC;YAEF,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAgB,CAAC;YAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACjD,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,2BAA2B;gBAC3B,SAAS,EAAE,CAAC,KAAK,CACf,wCAAwC,WAAW,CAAC,IAAI,0BAA0B,CACnF,CAAC;gBACF,OAAO;YACT,CAAC;YACD,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,MAAM,EAAE,EAAE;gBACd,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,cAAc,EAAE,MAAM,CAAC,cAAc;gBACrC,SAAS,EAAE,IAAI,CAAC,cAAc;qBAC3B,cAAc,EAAE;qBAChB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qBAChB,QAAQ,CAAC,QAAQ,CAAC;aACtB,CAAC;YACF,MAAM,MAAM,GAAG;gBACb,EAAE,EAAE,WAAW,CAAC,EAAE;gBAClB,IAAI,EAAE,WAAW,CAAC,IAAI;gBACtB,MAAM,EAAE,YAAY;aACrB,CAAC;YAEF,MAAM,aAAa,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS,CAC/C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,MAAM,CAAC,EAAE,CAC1B,CAAC;YACF,IAAI,aAAa,KAAK,CAAC,CAAC,EAAE,CAAC;gBACzB,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,GAAG,MAAM,CAAC;YAC5C,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACjC,CAAC;YACD,IAAI,YAAY,EAAE,CAAC;gBACjB,oBAAoB;gBACpB,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE,EAAE,YAAY,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;oBACzD,SAAS,EAAE,CAAC,KAAK,CACf,8CAA8C,WAAW,CAAC,EAAE,KAAK,CAAC,EAAE,CACrE,CAAC;gBACJ,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,SAAS,CAAC,EAAiB;YACzB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;QACpD,CAAC;QAMD,KAAK,CAAC,WAAW,CACf,EAAiB,EACjB,YAA2B;YAE3B,wDAAwD;YACxD,MAAM,OAAO,GAAG,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACpD,IAAI,OAAO,EAAE,CAAC;gBACZ,SAAS,EAAE,CAAC,KAAK,CACf,sDAAsD,EAAE,EAAE,CAC3D,CAAC;gBACF,OAAO,OAAO,CAAC;YACjB,CAAC;YAED,+BAA+B;YAC/B,MAAM,YAAY,GAAG,IAAI,CAAC,cAAc,CAAC,EAAE,EAAE,YAAY,CAAC,CAAC;YAC3D,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,EAAE,EAAE,YAAY,CAAC,CAAC;YAElD,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;gBAClC,OAAO,MAAM,CAAC;YAChB,CAAC;oBAAS,CAAC;gBACT,oBAAoB;gBACpB,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YACzC,CAAC;QACH,CAAC;QAEO,KAAK,CAAC,cAAc,CAC1B,EAAiB,EACjB,YAA2B;YAE3B,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;YAClC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,SAAS,EAAE,CAAC,IAAI,CAAC,mDAAmD,EAAE,EAAE,CAAC,CAAC;gBAC1E,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,IAAI,CAAC,YAAY,EAAE,CAAC;gBAClB,SAAS,EAAE,CAAC,IAAI,CACd,wFAAwF,EAAE,EAAE,CAC7F,CAAC;gBACF,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,2CAA2C;YAC3C,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;YAC/D,IAAI,UAAU,EAAE,CAAC;gBACf,2DAA2D;gBAC3D,SAAS,EAAE,CAAC,KAAK,CAAC,kDAAkD,EAAE,EAAE,CAAC,CAAC;gBAC1E,OAAO,EAAE,SAAS,EAAE,UAAU,EAAS,CAAC;YAC1C,CAAC;YAED,iCAAiC;YACjC,SAAS,EAAE,CAAC,IAAI,CAAC,6CAA6C,EAAE,EAAE,CAAC,CAAC;YAEpE,IAAI,MAAM,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;gBACjC,qCAAqC;gBACrC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,EAAE,EAAE,YAAY,CAAC,CAAC;gBAChE,MAAM,QAAQ,GAAG,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;gBAEvC,0CAA0C;gBAC1C,IAAI,CAAC;oBACH,MAAM,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;oBACtD,qDAAqD;oBACrD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;oBAC7D,IAAI,QAAQ,EAAE,CAAC;wBACb,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAS,CAAC;oBACxC,CAAC;gBACH,CAAC;gBAAC,OAAO,GAAQ,EAAE,CAAC;oBAClB,SAAS,EAAE,CAAC,KAAK,CACf,oDAAoD,EAAE,KAAK,GAAG,CAAC,OAAO,EAAE,CACzE,CAAC;oBACF,8DAA8D;gBAChE,CAAC;gBAED,OAAO,QAAQ,CAAC;YAClB,CAAC;YAED,8CAA8C;YAC9C,MAAM,WAAW,GAAG,MAAM,YAAY,EAAE,IAAI,CAAC,gBAAgB,CAAC;gBAC5D,aAAa,EAAE,EAAE;aAClB,CAAC,CAAC;YAEH,kEAAkE;YAClE,mFAAmF;YACnF,IAAI,WAAW,CAAC,cAAc,EAAE,CAAC;gBAC/B,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,cAAc,CACpC,MAAM,CAAC,EAAE,EACT,YAAY,EACZ,WAAW,CAAC,cAAc,CAC3B,CAAC;gBACF,MAAM,QAAQ,GAAG;oBACf,WAAW,EAAE,IAAI;oBACjB,cAAc,EAAE,WAAW,CAAC,cAAc;iBAC3C,CAAC;gBAEF,0CAA0C;gBAC1C,IAAI,CAAC;oBACH,MAAM,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;oBACtD,qDAAqD;oBACrD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;oBAC7D,IAAI,QAAQ,EAAE,CAAC;wBACb,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAS,CAAC;oBACxC,CAAC;gBACH,CAAC;gBAAC,OAAO,GAAQ,EAAE,CAAC;oBAClB,SAAS,EAAE,CAAC,KAAK,CACf,oDAAoD,EAAE,KAAK,GAAG,CAAC,OAAO,EAAE,CACzE,CAAC;oBACF,8DAA8D;gBAChE,CAAC;gBAED,OAAO,QAAQ,CAAC;YAClB,CAAC;YAED,IAAI,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACnD,SAAS,EAAE,CAAC,IAAI,CACd,yDAAyD,EAAE,EAAE,CAC9D,CAAC;gBACF,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,0CAA0C;YAC1C,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,EAAE,EAAE,WAAW,CAAC,QAAQ,CAAC,CAAC;gBAClE,SAAS,EAAE,CAAC,IAAI,CACd,wDAAwD,EAAE,EAAE,CAC7D,CAAC;gBACF,qDAAqD;gBACrD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;gBAC7D,IAAI,QAAQ,EAAE,CAAC;oBACb,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAS,CAAC;gBACxC,CAAC;YACH,CAAC;YAAC,OAAO,GAAQ,EAAE,CAAC;gBAClB,SAAS,EAAE,CAAC,KAAK,CACf,oDAAoD,EAAE,KAAK,GAAG,CAAC,OAAO,EAAE,CACzE,CAAC;gBACF,8DAA8D;YAChE,CAAC;YAED,OAAO,WAAW,CAAC,QAAQ,CAAC;QAC9B,CAAC;QAEM,KAAK,CAAC,cAAc,CACzB,EAAiB,EACjB,YAA2B;YAE3B,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;YAClC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,YAAY,CAAC,CAAC;YAC1D,OAAO;gBACL,GAAG,MAAM;gBACT,QAAQ,EAAE,QAAQ,IAAI,EAAE;aACzB,CAAC;QACJ,CAAC;QAMD,KAAK,CAAC,cAAc,CAClB,aAA4B,EAC5B,YAA2B,EAC3B,OAAgB;YAEhB,IAAI,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,EAAE,CAAC;gBACrC,OAAO,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;YAC1C,CAAC;YAED,IAAI,CAAC,YAAY,EAAE,CAAC;gBAClB,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;YACnC,CAAC;YAED,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;YAC7C,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,KAAK,CAAC,+BAA+B,aAAa,EAAE,CAAC,CAAC;YAClE,CAAC;YAED,+EAA+E;YAC/E,MAAM,YAAY,GAAG,OAAO,IAAI,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC;YAE7D,IAAI,CAAC,YAAY,EAAE,CAAC;gBAClB,MAAM,IAAI,KAAK,CACb,eAAe,aAAa,gCAAgC,CAC7D,CAAC;YACJ,CAAC;YAED,oKAAoK;YACpK,iHAAiH;YACjH,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,gBAAgB,CAAC;gBACtD,aAAa,EAAE,MAAM,CAAC,EAAE;gBACxB,GAAG,EAAE,YAAY;aAClB,CAAC,CAAC;YACH,IAAI,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBACpB,SAAS,EAAE,CAAC,IAAI,CACd,oEAAoE,EACpE,kBAAkB,aAAa,EAAE,EACjC,YAAY,YAAY,EAAE,CAC3B,CAAC;gBACF,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC;YACtC,OAAO,MAAM,CAAC;QAChB,CAAC;QAED;;;WAGG;QACH,YAAY,CAAC,EAAiB;YAC5B,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;YAClC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,OAAO,IAAI,MAAM,CAAC,IAAI,OAAO,EAAE,GAAG,CAAC;QACrC,CAAC;;;SA/TU,gBAAgB;AAkU7B,MAAM,UAAU,oBAAoB,CAAC,MAAyB;IAC5D,OAAO,MAAM,CAAC,MAAM,CAAC,cAAc,KAAK,SAAS,CAAC;AACpD,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,MAAyB;IAC/D,OAAO,MAAM,CAAC,MAAM,CAAC,EAAE,KAAK,gBAAgB,CAAC,EAAE,CAAC;AAClD,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,MAAyB;IAC7D,OAAO,CACL,wBAAwB,CAAC,QAAQ,CAC/B,MAAM,CAAC,MAAM,CAAC,EAAgC,CAC/C,IAAI,MAAM,CAAC,MAAM,CAAC,EAAE,KAAK,gBAAgB,CAAC,EAAE,CAC9C,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,WAAwB;IAC9D,MAAM,MAAM,GACV,WAAW,CAAC,QAAQ,EAAE,sBAAsB;QAC5C,WAAW,CAAC,QAAQ,EAAE,QAAQ;QAC9B,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC/B,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,MAAM,UAAU,oBAAoB,CAAC,MAAyB;IAC5D,OAAO,MAAM,CAAC,MAAM,CAAC,EAAE,KAAK,wBAAwB,CAAC,EAAE,CAAC;AAC1D,CAAC"}
|
|
@@ -36,8 +36,8 @@ export const prioritySettingsMiddleware = ({ settings, }) => {
|
|
|
36
36
|
return {
|
|
37
37
|
transformParams: async ({ params }) => {
|
|
38
38
|
return {
|
|
39
|
-
...params,
|
|
40
39
|
...settings,
|
|
40
|
+
...params,
|
|
41
41
|
providerMetadata: {
|
|
42
42
|
...params.providerOptions,
|
|
43
43
|
...settings.providerOptions,
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/ai-service/llm/utils.ts"],"names":[],"mappings":"AAQA,MAAM,CAAC,MAAM,uBAAuB,GAAG,CACrC,YAAoD,EACpD,cAAoC,EACnB,EAAE;IACnB,QAAQ,cAAc,EAAE,CAAC;QACvB,KAAK,cAAc,CAAC;QACpB,KAAK,UAAU;YACb,OAAO,YAAY,CAAC,oBAAoB,CAAC,CAAC;QAC5C,KAAK,YAAY;YACf,OAAO,YAAY,CAAC,mBAAmB,CAAC,CAAC;QAC3C,KAAK,eAAe,CAAC;QACrB,KAAK,SAAS;YACZ,OAAO,YAAY,CAAC,sBAAsB,CAAC,CAAC;QAC9C;YACE,MAAM,IAAI,KAAK,CAAC,2BAA2B,cAAc,EAAE,CAAC,CAAC;IACjE,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,aAAa,GAAG,CAC3B,MAAiC,EACN,EAAE;IAC7B,OAAO;QACL,eAAe,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;YACpC,MAAM,GAAG,GAAG,MAAM,EAAE,EAAE,CAAC;YACvB,IAAI,GAAG,EAAE,CAAC;gBACR,OAAO;oBACL,GAAG,MAAM;oBACT,OAAO,EAAE;wBACP,GAAG,MAAM,CAAC,OAAO;wBACjB,aAAa,EAAE,UAAU,GAAG,EAAE;wBAC9B,6BAA6B,EAAE,UAAU,GAAG,EAAE;qBAC/C;iBACF,CAAC;YACJ,CAAC;YAED,OAAO,MAAM,CAAC;QAChB,CAAC;KACF,CAAC;AACJ,CAAC,CAAC;AAEF,yEAAyE;AACzE,+BAA+B;AAC/B,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,EACzC,QAAQ,GAGT,EAA6B,EAAE;IAC9B,OAAO;QACL,eAAe,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;YACpC,OAAO;gBACL,GAAG,
|
|
1
|
+
{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/ai-service/llm/utils.ts"],"names":[],"mappings":"AAQA,MAAM,CAAC,MAAM,uBAAuB,GAAG,CACrC,YAAoD,EACpD,cAAoC,EACnB,EAAE;IACnB,QAAQ,cAAc,EAAE,CAAC;QACvB,KAAK,cAAc,CAAC;QACpB,KAAK,UAAU;YACb,OAAO,YAAY,CAAC,oBAAoB,CAAC,CAAC;QAC5C,KAAK,YAAY;YACf,OAAO,YAAY,CAAC,mBAAmB,CAAC,CAAC;QAC3C,KAAK,eAAe,CAAC;QACrB,KAAK,SAAS;YACZ,OAAO,YAAY,CAAC,sBAAsB,CAAC,CAAC;QAC9C;YACE,MAAM,IAAI,KAAK,CAAC,2BAA2B,cAAc,EAAE,CAAC,CAAC;IACjE,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,aAAa,GAAG,CAC3B,MAAiC,EACN,EAAE;IAC7B,OAAO;QACL,eAAe,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;YACpC,MAAM,GAAG,GAAG,MAAM,EAAE,EAAE,CAAC;YACvB,IAAI,GAAG,EAAE,CAAC;gBACR,OAAO;oBACL,GAAG,MAAM;oBACT,OAAO,EAAE;wBACP,GAAG,MAAM,CAAC,OAAO;wBACjB,aAAa,EAAE,UAAU,GAAG,EAAE;wBAC9B,6BAA6B,EAAE,UAAU,GAAG,EAAE;qBAC/C;iBACF,CAAC;YACJ,CAAC;YAED,OAAO,MAAM,CAAC;QAChB,CAAC;KACF,CAAC;AACJ,CAAC,CAAC;AAEF,yEAAyE;AACzE,+BAA+B;AAC/B,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,EACzC,QAAQ,GAGT,EAA6B,EAAE;IAC9B,OAAO;QACL,eAAe,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;YACpC,OAAO;gBACL,GAAG,QAAQ;gBACX,GAAG,MAAM;gBACT,gBAAgB,EAAE;oBAChB,GAAG,MAAM,CAAC,eAAe;oBACzB,GAAG,QAAQ,CAAC,eAAe;iBAC5B;aACF,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
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;AAI9D,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,CAUxB;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;;;;;;;;;;;;GAYG;AACH,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,2CAa7B;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,
|
|
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;AAI9D,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,CAUxB;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;;;;;;;;;;;;GAYG;AACH,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,2CAa7B;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,CAyCF,CAAC"}
|
|
@@ -103,30 +103,24 @@ export const traceToolSet = (toolset, llmobs = LLMObsTracer.instance, hooks) =>
|
|
|
103
103
|
kind: "tool",
|
|
104
104
|
name: `tool.${name}`,
|
|
105
105
|
}, async (span) => {
|
|
106
|
-
|
|
107
|
-
|
|
108
|
-
throw new Error(`Tool ${name} has no execute method`);
|
|
109
|
-
}
|
|
110
|
-
const result = await tool.execute(input, options);
|
|
111
|
-
hooks?.afterExecute?.(name, span);
|
|
112
|
-
const metadata = {
|
|
113
|
-
toolCallid: options?.toolCallId,
|
|
114
|
-
...Object.fromEntries(Object.entries(tool).map(([key, value]) => [
|
|
115
|
-
key,
|
|
116
|
-
safeJsonStringify(value),
|
|
117
|
-
])),
|
|
118
|
-
};
|
|
119
|
-
llmobs.annotate(span, {
|
|
120
|
-
inputData: safeJsonStringify(input),
|
|
121
|
-
outputData: safeJsonStringify(result),
|
|
122
|
-
metadata,
|
|
123
|
-
});
|
|
124
|
-
return result;
|
|
125
|
-
}
|
|
126
|
-
catch (error) {
|
|
127
|
-
console.error(`Error executing tool ${name}:`, error);
|
|
128
|
-
throw error;
|
|
106
|
+
if (!tool.execute) {
|
|
107
|
+
throw new Error(`Tool ${name} has no execute method`);
|
|
129
108
|
}
|
|
109
|
+
const result = await tool.execute(input, options);
|
|
110
|
+
hooks?.afterExecute?.(name, span);
|
|
111
|
+
const metadata = {
|
|
112
|
+
toolCallid: options?.toolCallId,
|
|
113
|
+
...Object.fromEntries(Object.entries(tool).map(([key, value]) => [
|
|
114
|
+
key,
|
|
115
|
+
safeJsonStringify(value),
|
|
116
|
+
])),
|
|
117
|
+
};
|
|
118
|
+
llmobs.annotate(span, {
|
|
119
|
+
inputData: safeJsonStringify(input),
|
|
120
|
+
outputData: safeJsonStringify(result),
|
|
121
|
+
metadata,
|
|
122
|
+
});
|
|
123
|
+
return result;
|
|
130
124
|
});
|
|
131
125
|
},
|
|
132
126
|
};
|
|
@@ -1 +1 @@
|
|
|
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,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAChD,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;IAC1C,MAAM,MAAM,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC;IAE7B,IAAI,MAAM,CAAC,KAAK;QAAE,MAAM,CAAC,QAAQ,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC;IAClD,IAAI,MAAM,CAAC,OAAO;QAAE,MAAM,CAAC,UAAU,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC;IACxD,IAAI,MAAM,CAAC,MAAM;QAAE,MAAM,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC;IACrD,IAAI,MAAM,CAAC,SAAS;QAAE,MAAM,CAAC,YAAY,CAAC,GAAG,MAAM,CAAC,SAAS,CAAC;IAE9D,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;;;;;;;;;;;;GAYG;AACH,MAAM,UAAU,gBAAgB,CAC9B,MAA2C,EAC3C,SAAuB,YAAY,CAAC,QAAQ,EAC5C,SAAiB,SAAS,EAAE;IAE5B,uFAAuF;IACvF,MAAM,GAAG;QACP,GAAG,MAAM;QACT,UAAU,EAAE,CAAC;KACd,CAAC;IAEF,kCAAkC;IAClC,MAAM,gBAAgB,GAAG,IAAI,0BAA0B,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACxE,MAAM,gBAAgB,GAAG,gBAAgB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAE3D,OAAO,gBAAgB,CAAC,MAAM,CAAC,CAAC;AAClC,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
|
|
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,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAChD,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;IAC1C,MAAM,MAAM,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC;IAE7B,IAAI,MAAM,CAAC,KAAK;QAAE,MAAM,CAAC,QAAQ,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC;IAClD,IAAI,MAAM,CAAC,OAAO;QAAE,MAAM,CAAC,UAAU,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC;IACxD,IAAI,MAAM,CAAC,MAAM;QAAE,MAAM,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC;IACrD,IAAI,MAAM,CAAC,SAAS;QAAE,MAAM,CAAC,YAAY,CAAC,GAAG,MAAM,CAAC,SAAS,CAAC;IAE9D,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;;;;;;;;;;;;GAYG;AACH,MAAM,UAAU,gBAAgB,CAC9B,MAA2C,EAC3C,SAAuB,YAAY,CAAC,QAAQ,EAC5C,SAAiB,SAAS,EAAE;IAE5B,uFAAuF;IACvF,MAAM,GAAG;QACP,GAAG,MAAM;QACT,UAAU,EAAE,CAAC;KACd,CAAC;IAEF,kCAAkC;IAClC,MAAM,gBAAgB,GAAG,IAAI,0BAA0B,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACxE,MAAM,gBAAgB,GAAG,gBAAgB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAE3D,OAAO,gBAAgB,CAAC,MAAM,CAAC,CAAC;AAClC,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,IAAI,CAAC,OAAO,EAAE,CAAC;wBAClB,MAAM,IAAI,KAAK,CAAC,QAAQ,IAAI,wBAAwB,CAAC,CAAC;oBACxD,CAAC;oBAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;oBAClD,KAAK,EAAE,YAAY,EAAE,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;oBAClC,MAAM,QAAQ,GAAG;wBACf,UAAU,EAAE,OAAO,EAAE,UAAU;wBAC/B,GAAG,MAAM,CAAC,WAAW,CACnB,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC;4BACzC,GAAG;4BACH,iBAAiB,CAAC,KAAK,CAAC;yBACzB,CAAC,CACH;qBACF,CAAC;oBACF,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE;wBACpB,SAAS,EAAE,iBAAiB,CAAC,KAAK,CAAC;wBACnC,UAAU,EAAE,iBAAiB,CAAC,MAAM,CAAC;wBACrC,QAAQ;qBACT,CAAC,CAAC;oBACH,OAAO,MAAM,CAAC;gBAChB,CAAC,CACF,CAAC;YACJ,CAAC;SACF,CAAC;IACJ,CAAC;IACD,OAAO,aAAkB,CAAC;AAC5B,CAAC,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"stream-text.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/llmobs/middleware/stream-text.ts"],"names":[],"mappings":"AAAA,OAAO,EAA2B,KAAK,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAI/E,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;
|
|
1
|
+
{"version":3,"file":"stream-text.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/llmobs/middleware/stream-text.ts"],"names":[],"mappings":"AAAA,OAAO,EAA2B,KAAK,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAI/E,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;gBAoBrE,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;IAUzB,OAAO,CAAC,sBAAsB;IAI9B,OAAO,CAAC,cAAc;IAWtB,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;IA8BrB,OAAO,CAAC,aAAa;IAoBrB,OAAO,CAAC,cAAc;IAwBtB,OAAO,CAAC,aAAa;IAyBrB,OAAO,CAAC,cAAc;IA6BtB,OAAO,CAAC,QAAQ;CAOjB"}
|
|
@@ -19,6 +19,7 @@ export class LLMObsStreamTextMiddleware {
|
|
|
19
19
|
const result = fn(this.transform(params));
|
|
20
20
|
return {
|
|
21
21
|
...result,
|
|
22
|
+
textStream: wrapStream(result.textStream, async (error) => this.handleStreamError(error), async () => this.onFinally()),
|
|
22
23
|
fullStream: wrapStream(result.fullStream, async (error) => this.handleStreamError(error), async () => this.onFinally()),
|
|
23
24
|
};
|
|
24
25
|
};
|