@openbkn/bkn-sdk 0.1.1-alpha.12 → 0.1.1-alpha.14

package/dist/{chunk-3FVB6VFQ.js → chunk-2GNQU4XH.js}

@@ -3745,6 +3745,23 @@ function deleteModels(ctx, kind, modelIds) {
 function testModel(ctx, kind, body) {
   return request(ctx, `${MANAGER}/${kind}/test`, { method: "POST", body });
 }
+function setDefaultLlm(ctx, modelId, isDefault = true) {
+  return request(ctx, `${MANAGER}/llm/default/edit`, {
+    method: "POST",
+    body: { model_id: modelId, default: isDefault }
+  });
+}
+function setDefaultSmallModel(ctx, modelId, isDefault = true) {
+  return request(ctx, `${MANAGER}/small-model/set-default`, {
+    method: "POST",
+    body: { model_id: modelId, default: isDefault }
+  });
+}
+function getDefaultSmallModel(ctx, modelType = "embedding") {
+  return request(ctx, `${MANAGER}/small-model/get_default`, {
+    query: { model_type: modelType }
+  });
+}
 function rerank(ctx, model, query, documents) {
   return request(ctx, `${API}/small-model/reranker`, {
     method: "POST",
@@ -3763,7 +3780,9 @@ function models(ctx) {
       add: (body) => addModel(ctx, "llm", body),
       edit: (body) => editModel(ctx, "llm", body),
       delete: (modelIds) => deleteModels(ctx, "llm", modelIds),
-      test: (body) => testModel(ctx, "llm", body)
+      test: (body) => testModel(ctx, "llm", body),
+      /** Set (or clear) the system default LLM. */
+      setDefault: (modelId, isDefault = true) => setDefaultLlm(ctx, modelId, isDefault)
     },
     small: {
       list: (opts) => listSmallModels(ctx, opts),
@@ -3773,7 +3792,11 @@ function models(ctx) {
       add: (body) => addModel(ctx, "small-model", body),
       edit: (body) => editModel(ctx, "small-model", body),
       delete: (modelIds) => deleteModels(ctx, "small-model", modelIds),
-      test: (body) => testModel(ctx, "small-model", body)
+      test: (body) => testModel(ctx, "small-model", body),
+      /** Set (or clear) the system default small model (type inferred from the model). */
+      setDefault: (modelId, isDefault = true) => setDefaultSmallModel(ctx, modelId, isDefault),
+      /** Get the system default small model for a type (default "embedding"). */
+      getDefault: (modelType) => getDefaultSmallModel(ctx, modelType)
     }
   };
 }
@@ -5308,4 +5331,4 @@ export {
   exportCreds,
   auth_exports
 };
-//# sourceMappingURL=chunk-3FVB6VFQ.js.map
+//# sourceMappingURL=chunk-2GNQU4XH.js.map