@mastra/server 0.0.0-commonjs-20250414101718
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/LICENSE.md +46 -0
- package/README.md +159 -0
- package/dist/_tsup-dts-rollup.d.cts +571 -0
- package/dist/_tsup-dts-rollup.d.ts +571 -0
- package/dist/chunk-2FJURXCL.cjs +213 -0
- package/dist/chunk-3AHQ5RGN.js +11 -0
- package/dist/chunk-3EJZQ6TQ.js +49 -0
- package/dist/chunk-3RVHWGWO.js +95 -0
- package/dist/chunk-4C3EPMMF.cjs +122 -0
- package/dist/chunk-4JINXASC.js +120 -0
- package/dist/chunk-67WTHYAV.js +156 -0
- package/dist/chunk-A7DF4ETD.cjs +100 -0
- package/dist/chunk-B64YROKU.cjs +317 -0
- package/dist/chunk-CWSDZEZG.cjs +139 -0
- package/dist/chunk-DVPP5S6I.js +135 -0
- package/dist/chunk-FV45V6WC.cjs +43 -0
- package/dist/chunk-JLDXUWK7.cjs +126 -0
- package/dist/chunk-L7XE5QTW.js +16 -0
- package/dist/chunk-M56ECCHK.cjs +128 -0
- package/dist/chunk-OPP7H5TW.js +305 -0
- package/dist/chunk-QH6XWSXP.cjs +5579 -0
- package/dist/chunk-QN4KF3BH.cjs +18 -0
- package/dist/chunk-RBQASTUP.js +203 -0
- package/dist/chunk-SKBVVI24.cjs +54 -0
- package/dist/chunk-TRDNDNGQ.js +40 -0
- package/dist/chunk-VK6FX47H.js +5576 -0
- package/dist/chunk-YANVFOYA.js +120 -0
- package/dist/chunk-YNSGUC2O.js +116 -0
- package/dist/chunk-Z46X3YCB.cjs +164 -0
- package/dist/chunk-ZLBRQFDD.cjs +13 -0
- package/dist/index.cjs +2 -0
- package/dist/index.d.cts +1 -0
- package/dist/index.d.ts +1 -0
- package/dist/index.js +1 -0
- package/dist/server/handlers/agents.cjs +30 -0
- package/dist/server/handlers/agents.d.cts +6 -0
- package/dist/server/handlers/agents.d.ts +6 -0
- package/dist/server/handlers/agents.js +1 -0
- package/dist/server/handlers/error.cjs +10 -0
- package/dist/server/handlers/error.d.cts +1 -0
- package/dist/server/handlers/error.d.ts +1 -0
- package/dist/server/handlers/error.js +1 -0
- package/dist/server/handlers/logs.cjs +18 -0
- package/dist/server/handlers/logs.d.cts +3 -0
- package/dist/server/handlers/logs.d.ts +3 -0
- package/dist/server/handlers/logs.js +1 -0
- package/dist/server/handlers/memory.cjs +38 -0
- package/dist/server/handlers/memory.d.cts +8 -0
- package/dist/server/handlers/memory.d.ts +8 -0
- package/dist/server/handlers/memory.js +1 -0
- package/dist/server/handlers/network.cjs +22 -0
- package/dist/server/handlers/network.d.cts +4 -0
- package/dist/server/handlers/network.d.ts +4 -0
- package/dist/server/handlers/network.js +1 -0
- package/dist/server/handlers/telemetry.cjs +14 -0
- package/dist/server/handlers/telemetry.d.cts +2 -0
- package/dist/server/handlers/telemetry.d.ts +2 -0
- package/dist/server/handlers/telemetry.js +1 -0
- package/dist/server/handlers/tools.cjs +22 -0
- package/dist/server/handlers/tools.d.cts +4 -0
- package/dist/server/handlers/tools.d.ts +4 -0
- package/dist/server/handlers/tools.js +1 -0
- package/dist/server/handlers/utils.cjs +10 -0
- package/dist/server/handlers/utils.d.cts +1 -0
- package/dist/server/handlers/utils.d.ts +1 -0
- package/dist/server/handlers/utils.js +1 -0
- package/dist/server/handlers/vector.cjs +30 -0
- package/dist/server/handlers/vector.d.cts +6 -0
- package/dist/server/handlers/vector.d.ts +6 -0
- package/dist/server/handlers/vector.js +1 -0
- package/dist/server/handlers/voice.cjs +18 -0
- package/dist/server/handlers/voice.d.cts +3 -0
- package/dist/server/handlers/voice.d.ts +3 -0
- package/dist/server/handlers/voice.js +1 -0
- package/dist/server/handlers/workflows.cjs +46 -0
- package/dist/server/handlers/workflows.d.cts +10 -0
- package/dist/server/handlers/workflows.d.ts +10 -0
- package/dist/server/handlers/workflows.js +1 -0
- package/dist/server/handlers.cjs +50 -0
- package/dist/server/handlers.d.cts +9 -0
- package/dist/server/handlers.d.ts +9 -0
- package/dist/server/handlers.js +9 -0
- package/package.json +71 -0
|
@@ -0,0 +1,135 @@
|
|
|
1
|
+
import { handleError } from './chunk-3AHQ5RGN.js';
|
|
2
|
+
import { __export, HTTPException } from './chunk-TRDNDNGQ.js';
|
|
3
|
+
|
|
4
|
+
// src/server/handlers/telemetry.ts
|
|
5
|
+
var telemetry_exports = {};
|
|
6
|
+
__export(telemetry_exports, {
|
|
7
|
+
getTelemetryHandler: () => getTelemetryHandler,
|
|
8
|
+
storeTelemetryHandler: () => storeTelemetryHandler
|
|
9
|
+
});
|
|
10
|
+
async function getTelemetryHandler({ mastra, body }) {
|
|
11
|
+
try {
|
|
12
|
+
const telemetry = mastra.getTelemetry();
|
|
13
|
+
const storage = mastra.getStorage();
|
|
14
|
+
if (!telemetry) {
|
|
15
|
+
throw new HTTPException(400, { message: "Telemetry is not initialized" });
|
|
16
|
+
}
|
|
17
|
+
if (!storage) {
|
|
18
|
+
throw new HTTPException(400, { message: "Storage is not initialized" });
|
|
19
|
+
}
|
|
20
|
+
if (!body) {
|
|
21
|
+
throw new HTTPException(400, { message: "Body is required" });
|
|
22
|
+
}
|
|
23
|
+
const { name, scope, page, perPage, attribute } = body;
|
|
24
|
+
const attributes = attribute ? Object.fromEntries(
|
|
25
|
+
(Array.isArray(attribute) ? attribute : [attribute]).map((attr) => {
|
|
26
|
+
const [key, value] = attr.split(":");
|
|
27
|
+
return [key, value];
|
|
28
|
+
})
|
|
29
|
+
) : void 0;
|
|
30
|
+
const traces = await storage.__getTraces({
|
|
31
|
+
name,
|
|
32
|
+
scope,
|
|
33
|
+
page: Number(page ?? 0),
|
|
34
|
+
perPage: Number(perPage ?? 100),
|
|
35
|
+
attributes
|
|
36
|
+
});
|
|
37
|
+
return traces;
|
|
38
|
+
} catch (error2) {
|
|
39
|
+
return handleError(error2, "Error getting telemetry");
|
|
40
|
+
}
|
|
41
|
+
}
|
|
42
|
+
async function storeTelemetryHandler({ mastra, body }) {
|
|
43
|
+
try {
|
|
44
|
+
const storage = mastra.getStorage();
|
|
45
|
+
const logger = mastra.getLogger();
|
|
46
|
+
if (!storage) {
|
|
47
|
+
throw new HTTPException(400, { message: "Storage is not initialized" });
|
|
48
|
+
}
|
|
49
|
+
const now = /* @__PURE__ */ new Date();
|
|
50
|
+
const items = body?.resourceSpans?.[0]?.scopeSpans;
|
|
51
|
+
logger.debug("[Telemetry Handler] Received spans:", {
|
|
52
|
+
totalSpans: items?.reduce((acc, scope) => acc + scope.spans.length, 0) || 0,
|
|
53
|
+
timestamp: now.toISOString()
|
|
54
|
+
});
|
|
55
|
+
if (!items?.length) {
|
|
56
|
+
return {
|
|
57
|
+
status: "success",
|
|
58
|
+
message: "No spans to process",
|
|
59
|
+
traceCount: 0
|
|
60
|
+
};
|
|
61
|
+
}
|
|
62
|
+
const allSpans = items.reduce((acc, scopedSpans) => {
|
|
63
|
+
const { scope, spans } = scopedSpans;
|
|
64
|
+
for (const span of spans) {
|
|
65
|
+
const {
|
|
66
|
+
spanId,
|
|
67
|
+
parentSpanId,
|
|
68
|
+
traceId,
|
|
69
|
+
name,
|
|
70
|
+
kind,
|
|
71
|
+
attributes,
|
|
72
|
+
status,
|
|
73
|
+
events,
|
|
74
|
+
links,
|
|
75
|
+
startTimeUnixNano,
|
|
76
|
+
endTimeUnixNano,
|
|
77
|
+
...rest
|
|
78
|
+
} = span;
|
|
79
|
+
const startTime = Number(BigInt(startTimeUnixNano) / 1000n);
|
|
80
|
+
const endTime = Number(BigInt(endTimeUnixNano) / 1000n);
|
|
81
|
+
acc.push({
|
|
82
|
+
id: spanId,
|
|
83
|
+
parentSpanId,
|
|
84
|
+
traceId,
|
|
85
|
+
name,
|
|
86
|
+
scope: scope.name,
|
|
87
|
+
kind,
|
|
88
|
+
status: JSON.stringify(status),
|
|
89
|
+
events: JSON.stringify(events),
|
|
90
|
+
links: JSON.stringify(links),
|
|
91
|
+
attributes: JSON.stringify(
|
|
92
|
+
attributes.reduce((acc2, attr) => {
|
|
93
|
+
const valueKey = Object.keys(attr.value)[0];
|
|
94
|
+
if (valueKey) {
|
|
95
|
+
acc2[attr.key] = attr.value[valueKey];
|
|
96
|
+
}
|
|
97
|
+
return acc2;
|
|
98
|
+
}, {})
|
|
99
|
+
),
|
|
100
|
+
startTime,
|
|
101
|
+
endTime,
|
|
102
|
+
other: JSON.stringify(rest),
|
|
103
|
+
createdAt: now
|
|
104
|
+
});
|
|
105
|
+
}
|
|
106
|
+
return acc;
|
|
107
|
+
}, []);
|
|
108
|
+
return storage.__batchTraceInsert({
|
|
109
|
+
records: allSpans
|
|
110
|
+
}).then(() => {
|
|
111
|
+
return {
|
|
112
|
+
status: "success",
|
|
113
|
+
message: "Traces received and processed successfully",
|
|
114
|
+
traceCount: body.resourceSpans?.length || 0
|
|
115
|
+
};
|
|
116
|
+
}).catch(() => {
|
|
117
|
+
return {
|
|
118
|
+
status: "error",
|
|
119
|
+
message: "Failed to process traces",
|
|
120
|
+
// @ts-ignore
|
|
121
|
+
error: error.message
|
|
122
|
+
};
|
|
123
|
+
});
|
|
124
|
+
} catch (error2) {
|
|
125
|
+
console.error("Error processing traces:", error2);
|
|
126
|
+
return {
|
|
127
|
+
status: "error",
|
|
128
|
+
message: "Failed to process traces",
|
|
129
|
+
// @ts-ignore
|
|
130
|
+
error: error2.message
|
|
131
|
+
};
|
|
132
|
+
}
|
|
133
|
+
}
|
|
134
|
+
|
|
135
|
+
export { getTelemetryHandler, storeTelemetryHandler, telemetry_exports };
|
|
@@ -0,0 +1,43 @@
|
|
|
1
|
+
'use strict';
|
|
2
|
+
|
|
3
|
+
var __defProp = Object.defineProperty;
|
|
4
|
+
var __export = (target, all) => {
|
|
5
|
+
for (var name in all)
|
|
6
|
+
__defProp(target, name, { get: all[name], enumerable: true });
|
|
7
|
+
};
|
|
8
|
+
|
|
9
|
+
// src/server/http-exception.ts
|
|
10
|
+
var HTTPException = class extends Error {
|
|
11
|
+
res;
|
|
12
|
+
status;
|
|
13
|
+
/**
|
|
14
|
+
* Creates an instance of `HTTPException`.
|
|
15
|
+
* @param status - HTTP status code for the exception. Defaults to 500.
|
|
16
|
+
* @param options - Additional options for the exception.
|
|
17
|
+
*/
|
|
18
|
+
constructor(status = 500, options) {
|
|
19
|
+
super(options?.message, { cause: options?.cause });
|
|
20
|
+
this.res = options?.res;
|
|
21
|
+
this.status = status;
|
|
22
|
+
}
|
|
23
|
+
/**
|
|
24
|
+
* Returns the response object associated with the exception.
|
|
25
|
+
* If a response object is not provided, a new response is created with the error message and status code.
|
|
26
|
+
* @returns The response object.
|
|
27
|
+
*/
|
|
28
|
+
getResponse() {
|
|
29
|
+
if (this.res) {
|
|
30
|
+
const newResponse = new Response(this.res.body, {
|
|
31
|
+
status: this.status,
|
|
32
|
+
headers: this.res.headers
|
|
33
|
+
});
|
|
34
|
+
return newResponse;
|
|
35
|
+
}
|
|
36
|
+
return new Response(this.message, {
|
|
37
|
+
status: this.status
|
|
38
|
+
});
|
|
39
|
+
}
|
|
40
|
+
};
|
|
41
|
+
|
|
42
|
+
exports.HTTPException = HTTPException;
|
|
43
|
+
exports.__export = __export;
|
|
@@ -0,0 +1,126 @@
|
|
|
1
|
+
'use strict';
|
|
2
|
+
|
|
3
|
+
var chunkQN4KF3BH_cjs = require('./chunk-QN4KF3BH.cjs');
|
|
4
|
+
var chunkZLBRQFDD_cjs = require('./chunk-ZLBRQFDD.cjs');
|
|
5
|
+
var chunkFV45V6WC_cjs = require('./chunk-FV45V6WC.cjs');
|
|
6
|
+
|
|
7
|
+
// src/server/handlers/network.ts
|
|
8
|
+
var network_exports = {};
|
|
9
|
+
chunkFV45V6WC_cjs.__export(network_exports, {
|
|
10
|
+
generateHandler: () => generateHandler,
|
|
11
|
+
getNetworkByIdHandler: () => getNetworkByIdHandler,
|
|
12
|
+
getNetworksHandler: () => getNetworksHandler,
|
|
13
|
+
streamGenerateHandler: () => streamGenerateHandler
|
|
14
|
+
});
|
|
15
|
+
async function getNetworksHandler({ mastra }) {
|
|
16
|
+
try {
|
|
17
|
+
const networks = mastra.getNetworks();
|
|
18
|
+
const serializedNetworks = networks.map((network) => {
|
|
19
|
+
const routingAgent = network.getRoutingAgent();
|
|
20
|
+
const agents = network.getAgents();
|
|
21
|
+
return {
|
|
22
|
+
id: network.formatAgentId(routingAgent.name),
|
|
23
|
+
name: routingAgent.name,
|
|
24
|
+
instructions: routingAgent.instructions,
|
|
25
|
+
agents: agents.map((agent) => ({
|
|
26
|
+
name: agent.name,
|
|
27
|
+
provider: agent.llm?.getProvider(),
|
|
28
|
+
modelId: agent.llm?.getModelId()
|
|
29
|
+
})),
|
|
30
|
+
routingModel: {
|
|
31
|
+
provider: routingAgent.llm?.getProvider(),
|
|
32
|
+
modelId: routingAgent.llm?.getModelId()
|
|
33
|
+
}
|
|
34
|
+
};
|
|
35
|
+
});
|
|
36
|
+
return serializedNetworks;
|
|
37
|
+
} catch (error) {
|
|
38
|
+
return chunkZLBRQFDD_cjs.handleError(error, "Error getting networks");
|
|
39
|
+
}
|
|
40
|
+
}
|
|
41
|
+
async function getNetworkByIdHandler({ mastra, networkId }) {
|
|
42
|
+
try {
|
|
43
|
+
const networks = mastra.getNetworks();
|
|
44
|
+
const network = networks.find((network2) => {
|
|
45
|
+
const routingAgent2 = network2.getRoutingAgent();
|
|
46
|
+
return network2.formatAgentId(routingAgent2.name) === networkId;
|
|
47
|
+
});
|
|
48
|
+
if (!network) {
|
|
49
|
+
throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Network not found" });
|
|
50
|
+
}
|
|
51
|
+
const routingAgent = network.getRoutingAgent();
|
|
52
|
+
const agents = network.getAgents();
|
|
53
|
+
const serializedNetwork = {
|
|
54
|
+
id: network.formatAgentId(routingAgent.name),
|
|
55
|
+
name: routingAgent.name,
|
|
56
|
+
instructions: routingAgent.instructions,
|
|
57
|
+
agents: agents.map((agent) => ({
|
|
58
|
+
name: agent.name,
|
|
59
|
+
provider: agent.llm?.getProvider(),
|
|
60
|
+
modelId: agent.llm?.getModelId()
|
|
61
|
+
})),
|
|
62
|
+
routingModel: {
|
|
63
|
+
provider: routingAgent.llm?.getProvider(),
|
|
64
|
+
modelId: routingAgent.llm?.getModelId()
|
|
65
|
+
}
|
|
66
|
+
};
|
|
67
|
+
return serializedNetwork;
|
|
68
|
+
} catch (error) {
|
|
69
|
+
return chunkZLBRQFDD_cjs.handleError(error, "Error getting network by ID");
|
|
70
|
+
}
|
|
71
|
+
}
|
|
72
|
+
async function generateHandler({
|
|
73
|
+
mastra,
|
|
74
|
+
container,
|
|
75
|
+
networkId,
|
|
76
|
+
body
|
|
77
|
+
}) {
|
|
78
|
+
try {
|
|
79
|
+
const network = mastra.getNetwork(networkId);
|
|
80
|
+
if (!network) {
|
|
81
|
+
throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Network not found" });
|
|
82
|
+
}
|
|
83
|
+
chunkQN4KF3BH_cjs.validateBody({ messages: body.messages });
|
|
84
|
+
const { messages, ...rest } = body;
|
|
85
|
+
const result = await network.generate(messages, { ...rest, container });
|
|
86
|
+
return result;
|
|
87
|
+
} catch (error) {
|
|
88
|
+
return chunkZLBRQFDD_cjs.handleError(error, "Error generating from network");
|
|
89
|
+
}
|
|
90
|
+
}
|
|
91
|
+
async function streamGenerateHandler({
|
|
92
|
+
mastra,
|
|
93
|
+
networkId,
|
|
94
|
+
body,
|
|
95
|
+
container
|
|
96
|
+
}) {
|
|
97
|
+
try {
|
|
98
|
+
const network = mastra.getNetwork(networkId);
|
|
99
|
+
if (!network) {
|
|
100
|
+
throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Network not found" });
|
|
101
|
+
}
|
|
102
|
+
chunkQN4KF3BH_cjs.validateBody({ messages: body.messages });
|
|
103
|
+
const { messages, output, ...rest } = body;
|
|
104
|
+
const streamResult = await network.stream(messages, {
|
|
105
|
+
output,
|
|
106
|
+
...rest,
|
|
107
|
+
container
|
|
108
|
+
});
|
|
109
|
+
const streamResponse = output ? streamResult.toTextStreamResponse() : streamResult.toDataStreamResponse({
|
|
110
|
+
sendUsage: true,
|
|
111
|
+
sendReasoning: true,
|
|
112
|
+
getErrorMessage: (error) => {
|
|
113
|
+
return `An error occurred while processing your request. ${error instanceof Error ? error.message : JSON.stringify(error)}`;
|
|
114
|
+
}
|
|
115
|
+
});
|
|
116
|
+
return streamResponse;
|
|
117
|
+
} catch (error) {
|
|
118
|
+
return chunkZLBRQFDD_cjs.handleError(error, "Error streaming from network");
|
|
119
|
+
}
|
|
120
|
+
}
|
|
121
|
+
|
|
122
|
+
exports.generateHandler = generateHandler;
|
|
123
|
+
exports.getNetworkByIdHandler = getNetworkByIdHandler;
|
|
124
|
+
exports.getNetworksHandler = getNetworksHandler;
|
|
125
|
+
exports.network_exports = network_exports;
|
|
126
|
+
exports.streamGenerateHandler = streamGenerateHandler;
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
import { HTTPException } from './chunk-TRDNDNGQ.js';
|
|
2
|
+
|
|
3
|
+
// src/server/handlers/utils.ts
|
|
4
|
+
function validateBody(body) {
|
|
5
|
+
const errorResponse = Object.entries(body).reduce((acc, [key, value]) => {
|
|
6
|
+
if (!value) {
|
|
7
|
+
acc[key] = `Argument "${key}" is required`;
|
|
8
|
+
}
|
|
9
|
+
return acc;
|
|
10
|
+
}, {});
|
|
11
|
+
if (Object.keys(errorResponse).length > 0) {
|
|
12
|
+
throw new HTTPException(400, { message: Object.values(errorResponse)[0] });
|
|
13
|
+
}
|
|
14
|
+
}
|
|
15
|
+
|
|
16
|
+
export { validateBody };
|
|
@@ -0,0 +1,128 @@
|
|
|
1
|
+
'use strict';
|
|
2
|
+
|
|
3
|
+
var chunkZLBRQFDD_cjs = require('./chunk-ZLBRQFDD.cjs');
|
|
4
|
+
var chunkFV45V6WC_cjs = require('./chunk-FV45V6WC.cjs');
|
|
5
|
+
|
|
6
|
+
// src/server/handlers/vector.ts
|
|
7
|
+
var vector_exports = {};
|
|
8
|
+
chunkFV45V6WC_cjs.__export(vector_exports, {
|
|
9
|
+
createIndex: () => createIndex,
|
|
10
|
+
deleteIndex: () => deleteIndex,
|
|
11
|
+
describeIndex: () => describeIndex,
|
|
12
|
+
listIndexes: () => listIndexes,
|
|
13
|
+
queryVectors: () => queryVectors,
|
|
14
|
+
upsertVectors: () => upsertVectors
|
|
15
|
+
});
|
|
16
|
+
function getVector(mastra, vectorName) {
|
|
17
|
+
if (!vectorName) {
|
|
18
|
+
throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Vector name is required" });
|
|
19
|
+
}
|
|
20
|
+
const vector = mastra.getVector(vectorName);
|
|
21
|
+
if (!vector) {
|
|
22
|
+
throw new chunkFV45V6WC_cjs.HTTPException(404, { message: `Vector store ${vectorName} not found` });
|
|
23
|
+
}
|
|
24
|
+
return vector;
|
|
25
|
+
}
|
|
26
|
+
async function upsertVectors({ mastra, vectorName, index }) {
|
|
27
|
+
try {
|
|
28
|
+
if (!index?.indexName || !index?.vectors || !Array.isArray(index.vectors)) {
|
|
29
|
+
throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Invalid request index. indexName and vectors array are required." });
|
|
30
|
+
}
|
|
31
|
+
const vector = getVector(mastra, vectorName);
|
|
32
|
+
const result = await vector.upsert(index);
|
|
33
|
+
return { ids: result };
|
|
34
|
+
} catch (error) {
|
|
35
|
+
return chunkZLBRQFDD_cjs.handleError(error, "Error upserting vectors");
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
async function createIndex({
|
|
39
|
+
mastra,
|
|
40
|
+
vectorName,
|
|
41
|
+
index
|
|
42
|
+
}) {
|
|
43
|
+
try {
|
|
44
|
+
const { indexName, dimension, metric } = index;
|
|
45
|
+
if (!indexName || typeof dimension !== "number" || dimension <= 0) {
|
|
46
|
+
throw new chunkFV45V6WC_cjs.HTTPException(400, {
|
|
47
|
+
message: "Invalid request index, indexName and positive dimension number are required."
|
|
48
|
+
});
|
|
49
|
+
}
|
|
50
|
+
if (metric && !["cosine", "euclidean", "dotproduct"].includes(metric)) {
|
|
51
|
+
throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Invalid metric. Must be one of: cosine, euclidean, dotproduct" });
|
|
52
|
+
}
|
|
53
|
+
const vector = getVector(mastra, vectorName);
|
|
54
|
+
await vector.createIndex({ indexName, dimension, metric });
|
|
55
|
+
return { success: true };
|
|
56
|
+
} catch (error) {
|
|
57
|
+
return chunkZLBRQFDD_cjs.handleError(error, "Error creating index");
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
async function queryVectors({
|
|
61
|
+
mastra,
|
|
62
|
+
vectorName,
|
|
63
|
+
query
|
|
64
|
+
}) {
|
|
65
|
+
try {
|
|
66
|
+
if (!query?.indexName || !query?.queryVector || !Array.isArray(query.queryVector)) {
|
|
67
|
+
throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Invalid request query. indexName and queryVector array are required." });
|
|
68
|
+
}
|
|
69
|
+
const vector = getVector(mastra, vectorName);
|
|
70
|
+
const results = await vector.query(query);
|
|
71
|
+
return results;
|
|
72
|
+
} catch (error) {
|
|
73
|
+
return chunkZLBRQFDD_cjs.handleError(error, "Error querying vectors");
|
|
74
|
+
}
|
|
75
|
+
}
|
|
76
|
+
async function listIndexes({ mastra, vectorName }) {
|
|
77
|
+
try {
|
|
78
|
+
const vector = getVector(mastra, vectorName);
|
|
79
|
+
const indexes = await vector.listIndexes();
|
|
80
|
+
return indexes.filter(Boolean);
|
|
81
|
+
} catch (error) {
|
|
82
|
+
return chunkZLBRQFDD_cjs.handleError(error, "Error listing indexes");
|
|
83
|
+
}
|
|
84
|
+
}
|
|
85
|
+
async function describeIndex({
|
|
86
|
+
mastra,
|
|
87
|
+
vectorName,
|
|
88
|
+
indexName
|
|
89
|
+
}) {
|
|
90
|
+
try {
|
|
91
|
+
if (!indexName) {
|
|
92
|
+
throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Index name is required" });
|
|
93
|
+
}
|
|
94
|
+
const vector = getVector(mastra, vectorName);
|
|
95
|
+
const stats = await vector.describeIndex(indexName);
|
|
96
|
+
return {
|
|
97
|
+
dimension: stats.dimension,
|
|
98
|
+
count: stats.count,
|
|
99
|
+
metric: stats.metric?.toLowerCase()
|
|
100
|
+
};
|
|
101
|
+
} catch (error) {
|
|
102
|
+
return chunkZLBRQFDD_cjs.handleError(error, "Error describing index");
|
|
103
|
+
}
|
|
104
|
+
}
|
|
105
|
+
async function deleteIndex({
|
|
106
|
+
mastra,
|
|
107
|
+
vectorName,
|
|
108
|
+
indexName
|
|
109
|
+
}) {
|
|
110
|
+
try {
|
|
111
|
+
if (!indexName) {
|
|
112
|
+
throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Index name is required" });
|
|
113
|
+
}
|
|
114
|
+
const vector = getVector(mastra, vectorName);
|
|
115
|
+
await vector.deleteIndex(indexName);
|
|
116
|
+
return { success: true };
|
|
117
|
+
} catch (error) {
|
|
118
|
+
return chunkZLBRQFDD_cjs.handleError(error, "Error deleting index");
|
|
119
|
+
}
|
|
120
|
+
}
|
|
121
|
+
|
|
122
|
+
exports.createIndex = createIndex;
|
|
123
|
+
exports.deleteIndex = deleteIndex;
|
|
124
|
+
exports.describeIndex = describeIndex;
|
|
125
|
+
exports.listIndexes = listIndexes;
|
|
126
|
+
exports.queryVectors = queryVectors;
|
|
127
|
+
exports.upsertVectors = upsertVectors;
|
|
128
|
+
exports.vector_exports = vector_exports;
|