@axiom-lattice/client-sdk 4.5.2 → 4.6.0

package/dist/index.mjs

@@ -2889,6 +2889,70 @@ var AbstractClient = class {
         return await this.makeRequest(`/api/database-configs/${id}/test`, { method: "POST" });
       }
     };
+    this.modelProviders = {
+      list: async () => {
+        const response = await this.makeRequest("/api/model-providers");
+        return response.data.records;
+      },
+      get: async (id) => {
+        const response = await this.makeRequest(`/api/model-providers/${encodeURIComponent(id)}`);
+        if (!response.data)
+          throw new ApiError("Model provider not found", 404);
+        return response.data;
+      },
+      create: async (data) => {
+        const response = await this.makeRequest("/api/model-providers", { method: "POST", body: data });
+        if (!response.data)
+          throw new ApiError("Failed to create model provider", 500);
+        return response.data;
+      },
+      update: async (id, data) => {
+        const response = await this.makeRequest(`/api/model-providers/${encodeURIComponent(id)}`, { method: "PUT", body: data });
+        if (!response.data)
+          throw new ApiError("Failed to update model provider", 500);
+        return response.data;
+      },
+      delete: async (id) => {
+        await this.makeRequest(`/api/model-providers/${encodeURIComponent(id)}`, { method: "DELETE" });
+      },
+      // Probes a base URL + API key that may not be saved yet; used by the
+      // provider form to validate before saving.
+      probe: async (data) => {
+        const response = await this.makeRequest("/api/model-providers/probe", { method: "POST", body: data });
+        if (!response.data)
+          throw new ApiError("Failed to probe model provider", 500);
+        return response.data;
+      },
+      // Replace this provider's model set with exactly these models, picked
+      // from the live list a probe returned. Both states travel: a deselected
+      // model arrives as `enabled: false` rather than being dropped.
+      setModels: async (id, models) => {
+        const response = await this.makeRequest(`/api/model-providers/${encodeURIComponent(id)}/models`, { method: "PUT", body: { models } });
+        if (!response.data)
+          throw new ApiError("Failed to update models", 500);
+        return response.data;
+      },
+      discover: async (id) => {
+        const response = await this.makeRequest(`/api/model-providers/${encodeURIComponent(id)}/discover`, { method: "POST", body: {} });
+        if (!response.data)
+          throw new ApiError("Failed to discover models", 500);
+        return response.data;
+      },
+      // The model id travels in the body because upstream ids routinely
+      // contain "/" (e.g. Qwen/Qwen2.5-7B-Instruct).
+      setModelEnabled: async (id, modelId, enabled) => {
+        const response = await this.makeRequest(`/api/model-providers/${encodeURIComponent(id)}/models/enabled`, { method: "POST", body: { modelId, enabled } });
+        if (!response.data)
+          throw new ApiError("Failed to update model", 500);
+        return response.data;
+      }
+    };
+    this.tenantModels = {
+      list: async () => {
+        const response = await this.makeRequest("/api/tenant-models");
+        return response.data;
+      }
+    };
     this.metricsConfigs = {
       list: async () => {
         const response = await this.makeRequest("/api/metrics-configs");