ziiagentmemory 0.2.0 → 0.3.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/cli.mjs +4 -4
- package/dist/index.mjs +46 -16
- package/dist/index.mjs.map +1 -1
- package/dist/{src-Dl4Re_-n.mjs → src-Dl5_tJgD.mjs} +47 -17
- package/dist/src-Dl5_tJgD.mjs.map +1 -0
- package/package.json +1 -1
- package/dist/src-Dl4Re_-n.mjs.map +0 -1
|
@@ -608,25 +608,25 @@ function l2Normalize(vec) {
|
|
|
608
608
|
}
|
|
609
609
|
//#endregion
|
|
610
610
|
//#region src/providers/embedding/openai.ts
|
|
611
|
-
const DEFAULT_MODEL$
|
|
611
|
+
const DEFAULT_MODEL$2 = "text-embedding-3-small";
|
|
612
612
|
/**
|
|
613
613
|
* Known OpenAI embedding model dimensions. Extend as new models ship.
|
|
614
614
|
* Override in any case via OPENAI_EMBEDDING_DIMENSIONS for custom or
|
|
615
615
|
* self-hosted OpenAI-compatible endpoints returning non-standard sizes.
|
|
616
616
|
*/
|
|
617
|
-
const MODEL_DIMENSIONS = {
|
|
617
|
+
const MODEL_DIMENSIONS$1 = {
|
|
618
618
|
"text-embedding-3-small": 1536,
|
|
619
619
|
"text-embedding-3-large": 3072,
|
|
620
620
|
"text-embedding-ada-002": 1536
|
|
621
621
|
};
|
|
622
|
-
const DEFAULT_DIMENSIONS = MODEL_DIMENSIONS[DEFAULT_MODEL$
|
|
623
|
-
function resolveDimensions(model, override) {
|
|
622
|
+
const DEFAULT_DIMENSIONS$1 = MODEL_DIMENSIONS$1[DEFAULT_MODEL$2] ?? 1536;
|
|
623
|
+
function resolveDimensions$1(model, override) {
|
|
624
624
|
if (override !== void 0 && override.trim().length > 0) {
|
|
625
625
|
const parsed = parseInt(override, 10);
|
|
626
626
|
if (!Number.isFinite(parsed) || parsed <= 0) throw new Error(`OPENAI_EMBEDDING_DIMENSIONS must be a positive integer, got: ${override}`);
|
|
627
627
|
return parsed;
|
|
628
628
|
}
|
|
629
|
-
return MODEL_DIMENSIONS[model] ?? DEFAULT_DIMENSIONS;
|
|
629
|
+
return MODEL_DIMENSIONS$1[model] ?? DEFAULT_DIMENSIONS$1;
|
|
630
630
|
}
|
|
631
631
|
/**
|
|
632
632
|
* OpenAI-compatible embedding provider.
|
|
@@ -676,8 +676,8 @@ var OpenAIEmbeddingProvider = class {
|
|
|
676
676
|
this.apiKey = apiKey || getEnvVar("OPENAI_EMBEDDING_API_KEY") || getEnvVar("OPENAI_API_KEY") || "";
|
|
677
677
|
if (!this.apiKey) throw new Error("API key is required (via constructor, OPENAI_EMBEDDING_API_KEY, or OPENAI_API_KEY)");
|
|
678
678
|
this.baseUrl = normalizeBaseUrl(getEnvVar("OPENAI_EMBEDDING_BASE_URL") || getEnvVar("OPENAI_BASE_URL"));
|
|
679
|
-
this.model = getEnvVar("OPENAI_EMBEDDING_MODEL") || DEFAULT_MODEL$
|
|
680
|
-
this.dimensions = resolveDimensions(this.model, getEnvVar("OPENAI_EMBEDDING_DIMENSIONS"));
|
|
679
|
+
this.model = getEnvVar("OPENAI_EMBEDDING_MODEL") || DEFAULT_MODEL$2;
|
|
680
|
+
this.dimensions = resolveDimensions$1(this.model, getEnvVar("OPENAI_EMBEDDING_DIMENSIONS"));
|
|
681
681
|
this.isAzure = detectAzure(this.baseUrl);
|
|
682
682
|
this.azureApiVersion = getEnvVar("OPENAI_API_VERSION") || "2024-08-01-preview";
|
|
683
683
|
}
|
|
@@ -774,15 +774,36 @@ var CohereEmbeddingProvider = class {
|
|
|
774
774
|
//#endregion
|
|
775
775
|
//#region src/providers/embedding/openrouter.ts
|
|
776
776
|
const API_URL = "https://openrouter.ai/api/v1/embeddings";
|
|
777
|
+
/**
|
|
778
|
+
* Known OpenRouter embedding model dimensions. Extend as new models ship.
|
|
779
|
+
* Override via OPENROUTER_EMBEDDING_DIMENSIONS for models not listed here
|
|
780
|
+
* or custom OpenAI-compatible endpoints returning non-standard sizes.
|
|
781
|
+
*/
|
|
782
|
+
const MODEL_DIMENSIONS = {
|
|
783
|
+
"openai/text-embedding-3-small": 1536,
|
|
784
|
+
"openai/text-embedding-3-large": 3072,
|
|
785
|
+
"openai/text-embedding-ada-002": 1536
|
|
786
|
+
};
|
|
787
|
+
const DEFAULT_MODEL$1 = "openai/text-embedding-3-small";
|
|
788
|
+
const DEFAULT_DIMENSIONS = MODEL_DIMENSIONS[DEFAULT_MODEL$1] ?? 1536;
|
|
789
|
+
function resolveDimensions(model, override) {
|
|
790
|
+
if (override !== void 0 && override.trim().length > 0) {
|
|
791
|
+
const parsed = parseInt(override, 10);
|
|
792
|
+
if (!Number.isFinite(parsed) || parsed <= 0) throw new Error(`OPENROUTER_EMBEDDING_DIMENSIONS must be a positive integer, got: ${override}`);
|
|
793
|
+
return parsed;
|
|
794
|
+
}
|
|
795
|
+
return MODEL_DIMENSIONS[model] ?? DEFAULT_DIMENSIONS;
|
|
796
|
+
}
|
|
777
797
|
var OpenRouterEmbeddingProvider = class {
|
|
778
798
|
name = "openrouter";
|
|
779
|
-
dimensions
|
|
799
|
+
dimensions;
|
|
780
800
|
apiKey;
|
|
781
801
|
model;
|
|
782
802
|
constructor(apiKey) {
|
|
783
803
|
this.apiKey = apiKey || getEnvVar("OPENROUTER_API_KEY") || "";
|
|
784
804
|
if (!this.apiKey) throw new Error("OPENROUTER_API_KEY is required");
|
|
785
|
-
this.model = getEnvVar("OPENROUTER_EMBEDDING_MODEL") ||
|
|
805
|
+
this.model = getEnvVar("OPENROUTER_EMBEDDING_MODEL") || DEFAULT_MODEL$1;
|
|
806
|
+
this.dimensions = resolveDimensions(this.model, getEnvVar("OPENROUTER_EMBEDDING_DIMENSIONS"));
|
|
786
807
|
}
|
|
787
808
|
async embed(text) {
|
|
788
809
|
const [result] = await this.embedBatch([text]);
|
|
@@ -15812,13 +15833,22 @@ function registerApiTriggers(sdk, kv, secret, metricsStore, provider) {
|
|
|
15812
15833
|
sdk.registerFunction("api::consolidate", async (req) => {
|
|
15813
15834
|
const authErr = checkAuth(req, secret);
|
|
15814
15835
|
if (authErr) return authErr;
|
|
15815
|
-
|
|
15816
|
-
|
|
15817
|
-
|
|
15818
|
-
|
|
15819
|
-
|
|
15820
|
-
|
|
15821
|
-
|
|
15836
|
+
try {
|
|
15837
|
+
return {
|
|
15838
|
+
status_code: 200,
|
|
15839
|
+
body: await sdk.trigger({
|
|
15840
|
+
function_id: "mem::consolidate",
|
|
15841
|
+
payload: req.body
|
|
15842
|
+
})
|
|
15843
|
+
};
|
|
15844
|
+
} catch (err) {
|
|
15845
|
+
const message = err instanceof Error ? err.message : String(err);
|
|
15846
|
+
logger.warn("api::consolidate failed", { error: message });
|
|
15847
|
+
return {
|
|
15848
|
+
status_code: 500,
|
|
15849
|
+
body: { error: message }
|
|
15850
|
+
};
|
|
15851
|
+
}
|
|
15822
15852
|
});
|
|
15823
15853
|
sdk.registerTrigger({
|
|
15824
15854
|
type: "http",
|
|
@@ -20682,4 +20712,4 @@ main().catch((err) => {
|
|
|
20682
20712
|
//#endregion
|
|
20683
20713
|
export {};
|
|
20684
20714
|
|
|
20685
|
-
//# sourceMappingURL=src-
|
|
20715
|
+
//# sourceMappingURL=src-Dl5_tJgD.mjs.map
|