@nemesis-oss/ollama-sdk 1.8.0 → 1.10.0

package/dist/generated-runtime.d.cts

@@ -1,6 +1,6 @@
-export { I as InvokeRequest, O as OllamaRuntime, b as OllamaRuntimeOptions, a as OperationDefinition, S as SchemaRef } from './runtime-DTmUeU6K.cjs';
+export { I as InvokeRequest, O as OllamaRuntime, b as OllamaRuntimeOptions, a as OperationDefinition, S as SchemaRef } from './runtime-BsdBbuOv.cjs';
 import { z } from 'zod';
-import './types-5PAjLoCp.cjs';
+import './types-BtQexFpT.cjs';
 
 /**
  * AUTO-GENERATED by `npm run contract:generate`. DO NOT EDIT.

package/dist/generated-runtime.d.ts

@@ -1,6 +1,6 @@
-export { I as InvokeRequest, O as OllamaRuntime, b as OllamaRuntimeOptions, a as OperationDefinition, S as SchemaRef } from './runtime-C6UYtBRB.js';
+export { I as InvokeRequest, O as OllamaRuntime, b as OllamaRuntimeOptions, a as OperationDefinition, S as SchemaRef } from './runtime-BV9SK_cf.js';
 import { z } from 'zod';
-import './types-5PAjLoCp.js';
+import './types-BtQexFpT.js';
 
 /**
  * AUTO-GENERATED by `npm run contract:generate`. DO NOT EDIT.

package/dist/generated-runtime.js

@@ -63,6 +63,14 @@ var OllamaClientError = class extends Error {
     this.response = options.response;
   }
 };
+var OllamaStreamError = class extends OllamaClientError {
+  /** Content accumulated before the in-band error frame, or '' if none. */
+  partialContent;
+  constructor(message, options) {
+    super(message, { ...options, code: "stream_error", retryable: false });
+    this.partialContent = options?.partialContent ?? "";
+  }
+};
 var OllamaRequestValidationError = class extends OllamaClientError {
   operationId;
   issues;
@@ -121,6 +129,12 @@ function disposableReader(reader) {
 }
 
 // src/streaming/ndjson.ts
+function isInBandErrorFrame(value) {
+  if (typeof value !== "object" || value === null) return false;
+  const record = value;
+  if (typeof record.error !== "string" || record.error.length === 0) return false;
+  return !("model" in record) && !("response" in record) && !("message" in record) && !("status" in record) && !("done" in record);
+}
 async function* parseNdjsonStream(stream) {
   var _stack = [];
   try {
@@ -137,17 +151,34 @@ async function* parseNdjsonStream(stream) {
       for (const line of lines) {
         const trimmed = line.trim();
         if (!trimmed) continue;
+        let json;
         try {
-          yield JSON.parse(trimmed);
+          json = JSON.parse(trimmed);
         } catch (err) {
           throw new OllamaGenericClientError(`Failed to parse stream JSON chunk: ${trimmed}`, {
             cause: err
           });
         }
+        if (isInBandErrorFrame(json)) {
+          throw new OllamaStreamError(json.error);
+        }
+        yield json;
       }
     }
     if (buffer.trim()) {
-      yield JSON.parse(buffer.trim());
+      const trimmed = buffer.trim();
+      let json;
+      try {
+        json = JSON.parse(trimmed);
+      } catch (err) {
+        throw new OllamaGenericClientError(`Failed to parse stream JSON chunk: ${trimmed}`, {
+          cause: err
+        });
+      }
+      if (isInBandErrorFrame(json)) {
+        throw new OllamaStreamError(json.error);
+      }
+      yield json;
     }
   } catch (_) {
     var _error = _, _hasError = true;
@@ -248,6 +279,32 @@ async function* parseSseStream(stream) {
     _promise && await _promise;
   }
 }
+var IncludedBalanceSchema = z.object({
+  balance_usd: z.number(),
+  allowance_usd: z.number(),
+  period: z.object({
+    from: z.string(),
+    until: z.string()
+  })
+});
+var LegacyBalanceLimitSchema = z.object({
+  remaining_percent: z.number().min(0).max(100),
+  resets_at: z.string()
+});
+
+// src/generated/models/LegacyIncludedBalance.schema.ts
+var LegacyIncludedBalanceSchema = z.object({
+  session: LegacyBalanceLimitSchema,
+  weekly: LegacyBalanceLimitSchema
+});
+
+// src/generated/models/BalanceResponse.schema.ts
+var BalanceResponseSchema = z.object({
+  included: z.union([IncludedBalanceSchema, LegacyIncludedBalanceSchema]),
+  purchased: z.object({
+    balance_usd: z.number()
+  })
+});
 var ToolCallSchema = z.object({
   function: z.object({
     name: z.string(),
@@ -351,6 +408,7 @@ var CreateRequestSchema = z.object({
   quantize: z.string().optional(),
   draft_quantize: z.string().optional(),
   requires: z.string().optional(),
+  capabilities: z.array(z.string()).optional(),
   stream: z.boolean().optional()
 });
 var DeleteRequestSchema = z.object({
@@ -413,7 +471,7 @@ var ModelSummarySchema = z.object({
   details: z.object({
     format: z.string().optional(),
     family: z.string().optional(),
-    families: z.array(z.string()).optional(),
+    families: z.union([z.array(z.string()), z.null()]).optional(),
     parameter_size: z.string().optional(),
     quantization_level: z.string().optional()
   }).optional()
@@ -548,6 +606,34 @@ var SystemOneResponseSchema = z.object({
     output_tokens: z.number().min(0)
   })
 });
+var UsageMetricsSchema = z.object({
+  request_count: z.number().min(0),
+  usage_usd: z.number().optional(),
+  input_tokens: z.number().optional(),
+  cached_input_tokens: z.number().optional(),
+  output_tokens: z.number().optional()
+});
+
+// src/generated/models/UsageBucket.schema.ts
+var UsageBucketSchema = z.intersection(
+  UsageMetricsSchema,
+  z.object({
+    from: z.string(),
+    until: z.string(),
+    partial: z.boolean().optional()
+  })
+);
+
+// src/generated/models/UsageResponse.schema.ts
+var UsageResponseSchema = z.object({
+  range: z.enum(["24h", "7d", "30d"]),
+  scope: z.enum(["self", "team"]),
+  granularity: z.enum(["hour", "day"]),
+  from: z.string(),
+  until: z.string(),
+  totals: UsageMetricsSchema,
+  buckets: z.array(UsageBucketSchema)
+});
 var VersionResponseSchema = z.object({
   version: z.string().optional()
 });
@@ -590,6 +676,7 @@ var requestSchemas = {
   webSearch: WebSearchRequestSchema
 };
 var responseSchemas = {
+  balance: BalanceResponseSchema,
   chat: ChatResponseSchema,
   create: StatusResponseSchema,
   embed: EmbedResponseSchema,
@@ -600,6 +687,7 @@ var responseSchemas = {
   show: ShowResponseSchema,
   systemOne: SystemOneResponseSchema,
   tags: ListResponseSchema,
+  usage: UsageResponseSchema,
   version: VersionResponseSchema,
   webFetch: WebFetchResponseSchema,
   webSearch: WebSearchResponseSchema
@@ -650,7 +738,6 @@ var OllamaRuntime = class {
   constructor(options) {
     this.options = options;
   }
-  options;
   /**
    * Cached promise of the Ollama server version, populated lazily the first
    * time a version-gated operation is invoked. `undefined` until then.
@@ -727,11 +814,19 @@ var OllamaRuntime = class {
     }
     const streaming = shouldStream(req);
     const resolvedPath = resolvePathParams(req.operation.path, req.pathParams);
+    const pathWithQuery = applyQueryParams(resolvedPath, req.queryParams);
     const model = req.model ?? extractModelFromBody(body);
+    const headers = {};
+    if (req.headerParams) {
+      for (const [key, value] of Object.entries(req.headerParams)) {
+        headers[key] = value;
+      }
+    }
     const httpReq = {
-      path: resolvedPath,
+      path: pathWithQuery,
       method: req.operation.method,
       ...body !== void 0 ? { body } : {},
+      ...Object.keys(headers).length > 0 ? { headers } : {},
       ...req.signal !== void 0 ? { signal: req.signal } : {},
       ...streaming ? { raw: true } : {},
       ...model !== void 0 ? { model } : {}
@@ -872,6 +967,11 @@ function extractModelFromBody(body) {
   const model = obj.model;
   return typeof model === "string" ? model : void 0;
 }
+function applyQueryParams(path, queryParams) {
+  if (!queryParams || Object.keys(queryParams).length === 0) return path;
+  const pairs = Object.entries(queryParams).map(([key, value]) => `${encodeURIComponent(key)}=${encodeURIComponent(value)}`).join("&");
+  return `${path}?${pairs}`;
+}
 
 export { OllamaRuntime, getRequestSchema, getResponseSchema, requestSchemas, responseSchemas };
 //# sourceMappingURL=generated-runtime.js.map