@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.js
CHANGED
|
@@ -735,7 +735,7 @@ var AxBaseAI = class {
|
|
|
735
735
|
this.models = models;
|
|
736
736
|
this.id = crypto.randomUUID();
|
|
737
737
|
const model = this.getModel(defaults.model) ?? defaults.model;
|
|
738
|
-
const embedModel = this.getEmbedModel(defaults.embedModel);
|
|
738
|
+
const embedModel = this.getEmbedModel(defaults.embedModel) ?? defaults.embedModel;
|
|
739
739
|
this.defaults = { model, embedModel };
|
|
740
740
|
if (!defaults.model || typeof defaults.model !== "string" || defaults.model === "") {
|
|
741
741
|
throw new Error("No model defined");
|
|
@@ -1122,13 +1122,7 @@ var AxBaseAI = class {
|
|
|
1122
1122
|
}
|
|
1123
1123
|
this.embedModelUsage = res.modelUsage;
|
|
1124
1124
|
if (span?.isRecording()) {
|
|
1125
|
-
|
|
1126
|
-
this.embedModelUsage = res.modelUsage;
|
|
1127
|
-
span.setAttributes({
|
|
1128
|
-
[axSpanAttributes.LLM_USAGE_COMPLETION_TOKENS]: res.modelUsage.tokens?.completionTokens ?? 0,
|
|
1129
|
-
[axSpanAttributes.LLM_USAGE_PROMPT_TOKENS]: res.modelUsage.tokens?.promptTokens
|
|
1130
|
-
});
|
|
1131
|
-
}
|
|
1125
|
+
setResponseAttr(res, span);
|
|
1132
1126
|
}
|
|
1133
1127
|
span?.end();
|
|
1134
1128
|
return res;
|