@ax-llm/ax 11.0.36 → 11.0.37
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/index.cjs +2 -8
- package/index.cjs.map +1 -1
- package/index.js +2 -8
- package/index.js.map +1 -1
- package/package.json +1 -1
package/index.cjs
CHANGED
|
@@ -886,7 +886,7 @@ var AxBaseAI = class {
|
|
|
886
886
|
this.models = models;
|
|
887
887
|
this.id = crypto.randomUUID();
|
|
888
888
|
const model = this.getModel(defaults.model) ?? defaults.model;
|
|
889
|
-
const embedModel = this.getEmbedModel(defaults.embedModel);
|
|
889
|
+
const embedModel = this.getEmbedModel(defaults.embedModel) ?? defaults.embedModel;
|
|
890
890
|
this.defaults = { model, embedModel };
|
|
891
891
|
if (!defaults.model || typeof defaults.model !== "string" || defaults.model === "") {
|
|
892
892
|
throw new Error("No model defined");
|
|
@@ -1273,13 +1273,7 @@ var AxBaseAI = class {
|
|
|
1273
1273
|
}
|
|
1274
1274
|
this.embedModelUsage = res.modelUsage;
|
|
1275
1275
|
if (span?.isRecording()) {
|
|
1276
|
-
|
|
1277
|
-
this.embedModelUsage = res.modelUsage;
|
|
1278
|
-
span.setAttributes({
|
|
1279
|
-
[axSpanAttributes.LLM_USAGE_COMPLETION_TOKENS]: res.modelUsage.tokens?.completionTokens ?? 0,
|
|
1280
|
-
[axSpanAttributes.LLM_USAGE_PROMPT_TOKENS]: res.modelUsage.tokens?.promptTokens
|
|
1281
|
-
});
|
|
1282
|
-
}
|
|
1276
|
+
setResponseAttr(res, span);
|
|
1283
1277
|
}
|
|
1284
1278
|
span?.end();
|
|
1285
1279
|
return res;
|