@mistralai/mistralai 1.9.18 → 1.10.0

package/packages/mistralai-azure/src/models/components/ocrresponse.ts

@@ -0,0 +1,106 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod";
+import { remap as remap$ } from "../../lib/primitives.js";
+import { safeParse } from "../../lib/schemas.js";
+import { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+import {
+  OCRPageObject,
+  OCRPageObject$inboundSchema,
+  OCRPageObject$Outbound,
+  OCRPageObject$outboundSchema,
+} from "./ocrpageobject.js";
+import {
+  OCRUsageInfo,
+  OCRUsageInfo$inboundSchema,
+  OCRUsageInfo$Outbound,
+  OCRUsageInfo$outboundSchema,
+} from "./ocrusageinfo.js";
+
+export type OCRResponse = {
+  /**
+   * List of OCR info for pages.
+   */
+  pages: Array<OCRPageObject>;
+  /**
+   * The model used to generate the OCR.
+   */
+  model: string;
+  /**
+   * Formatted response in the request_format if provided in json str
+   */
+  documentAnnotation?: string | null | undefined;
+  usageInfo: OCRUsageInfo;
+};
+
+/** @internal */
+export const OCRResponse$inboundSchema: z.ZodType<
+  OCRResponse,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  pages: z.array(OCRPageObject$inboundSchema),
+  model: z.string(),
+  document_annotation: z.nullable(z.string()).optional(),
+  usage_info: OCRUsageInfo$inboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    "document_annotation": "documentAnnotation",
+    "usage_info": "usageInfo",
+  });
+});
+
+/** @internal */
+export type OCRResponse$Outbound = {
+  pages: Array<OCRPageObject$Outbound>;
+  model: string;
+  document_annotation?: string | null | undefined;
+  usage_info: OCRUsageInfo$Outbound;
+};
+
+/** @internal */
+export const OCRResponse$outboundSchema: z.ZodType<
+  OCRResponse$Outbound,
+  z.ZodTypeDef,
+  OCRResponse
+> = z.object({
+  pages: z.array(OCRPageObject$outboundSchema),
+  model: z.string(),
+  documentAnnotation: z.nullable(z.string()).optional(),
+  usageInfo: OCRUsageInfo$outboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    documentAnnotation: "document_annotation",
+    usageInfo: "usage_info",
+  });
+});
+
+/**
+ * @internal
+ * @deprecated This namespace will be removed in future versions. Use schemas and types that are exported directly from this module.
+ */
+export namespace OCRResponse$ {
+  /** @deprecated use `OCRResponse$inboundSchema` instead. */
+  export const inboundSchema = OCRResponse$inboundSchema;
+  /** @deprecated use `OCRResponse$outboundSchema` instead. */
+  export const outboundSchema = OCRResponse$outboundSchema;
+  /** @deprecated use `OCRResponse$Outbound` instead. */
+  export type Outbound = OCRResponse$Outbound;
+}
+
+export function ocrResponseToJSON(ocrResponse: OCRResponse): string {
+  return JSON.stringify(OCRResponse$outboundSchema.parse(ocrResponse));
+}
+
+export function ocrResponseFromJSON(
+  jsonString: string,
+): SafeParseResult<OCRResponse, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => OCRResponse$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'OCRResponse' from JSON`,
+  );
+}

package/packages/mistralai-azure/src/models/components/ocrusageinfo.ts

@@ -0,0 +1,83 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod";
+import { remap as remap$ } from "../../lib/primitives.js";
+import { safeParse } from "../../lib/schemas.js";
+import { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+export type OCRUsageInfo = {
+  /**
+   * Number of pages processed
+   */
+  pagesProcessed: number;
+  /**
+   * Document size in bytes
+   */
+  docSizeBytes?: number | null | undefined;
+};
+
+/** @internal */
+export const OCRUsageInfo$inboundSchema: z.ZodType<
+  OCRUsageInfo,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  pages_processed: z.number().int(),
+  doc_size_bytes: z.nullable(z.number().int()).optional(),
+}).transform((v) => {
+  return remap$(v, {
+    "pages_processed": "pagesProcessed",
+    "doc_size_bytes": "docSizeBytes",
+  });
+});
+
+/** @internal */
+export type OCRUsageInfo$Outbound = {
+  pages_processed: number;
+  doc_size_bytes?: number | null | undefined;
+};
+
+/** @internal */
+export const OCRUsageInfo$outboundSchema: z.ZodType<
+  OCRUsageInfo$Outbound,
+  z.ZodTypeDef,
+  OCRUsageInfo
+> = z.object({
+  pagesProcessed: z.number().int(),
+  docSizeBytes: z.nullable(z.number().int()).optional(),
+}).transform((v) => {
+  return remap$(v, {
+    pagesProcessed: "pages_processed",
+    docSizeBytes: "doc_size_bytes",
+  });
+});
+
+/**
+ * @internal
+ * @deprecated This namespace will be removed in future versions. Use schemas and types that are exported directly from this module.
+ */
+export namespace OCRUsageInfo$ {
+  /** @deprecated use `OCRUsageInfo$inboundSchema` instead. */
+  export const inboundSchema = OCRUsageInfo$inboundSchema;
+  /** @deprecated use `OCRUsageInfo$outboundSchema` instead. */
+  export const outboundSchema = OCRUsageInfo$outboundSchema;
+  /** @deprecated use `OCRUsageInfo$Outbound` instead. */
+  export type Outbound = OCRUsageInfo$Outbound;
+}
+
+export function ocrUsageInfoToJSON(ocrUsageInfo: OCRUsageInfo): string {
+  return JSON.stringify(OCRUsageInfo$outboundSchema.parse(ocrUsageInfo));
+}
+
+export function ocrUsageInfoFromJSON(
+  jsonString: string,
+): SafeParseResult<OCRUsageInfo, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => OCRUsageInfo$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'OCRUsageInfo' from JSON`,
+  );
+}

package/packages/mistralai-azure/src/models/components/usageinfo.ts

@@ -4,14 +4,19 @@
 
 import * as z from "zod";
 import { remap as remap$ } from "../../lib/primitives.js";
-import { safeParse } from "../../lib/schemas.js";
+import {
+  collectExtraKeys as collectExtraKeys$,
+  safeParse,
+} from "../../lib/schemas.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
 
 export type UsageInfo = {
-  promptTokens: number;
-  completionTokens: number;
-  totalTokens: number;
+  promptTokens?: number | undefined;
+  completionTokens?: number | undefined;
+  totalTokens?: number | undefined;
+  promptAudioSeconds?: number | null | undefined;
+  additionalProperties?: { [k: string]: any };
 };
 
 /** @internal */
@@ -19,15 +24,21 @@ export const UsageInfo$inboundSchema: z.ZodType<
   UsageInfo,
   z.ZodTypeDef,
   unknown
-> = z.object({
-  prompt_tokens: z.number().int(),
-  completion_tokens: z.number().int(),
-  total_tokens: z.number().int(),
-}).transform((v) => {
+> = collectExtraKeys$(
+  z.object({
+    prompt_tokens: z.number().int().default(0),
+    completion_tokens: z.number().int().default(0),
+    total_tokens: z.number().int().default(0),
+    prompt_audio_seconds: z.nullable(z.number().int()).optional(),
+  }).catchall(z.any()),
+  "additionalProperties",
+  true,
+).transform((v) => {
   return remap$(v, {
     "prompt_tokens": "promptTokens",
     "completion_tokens": "completionTokens",
     "total_tokens": "totalTokens",
+    "prompt_audio_seconds": "promptAudioSeconds",
   });
 });
 
@@ -36,6 +47,8 @@ export type UsageInfo$Outbound = {
   prompt_tokens: number;
   completion_tokens: number;
   total_tokens: number;
+  prompt_audio_seconds?: number | null | undefined;
+  [additionalProperties: string]: unknown;
 };
 
 /** @internal */
@@ -44,15 +57,22 @@ export const UsageInfo$outboundSchema: z.ZodType<
   z.ZodTypeDef,
   UsageInfo
 > = z.object({
-  promptTokens: z.number().int(),
-  completionTokens: z.number().int(),
-  totalTokens: z.number().int(),
+  promptTokens: z.number().int().default(0),
+  completionTokens: z.number().int().default(0),
+  totalTokens: z.number().int().default(0),
+  promptAudioSeconds: z.nullable(z.number().int()).optional(),
+  additionalProperties: z.record(z.any()),
 }).transform((v) => {
-  return remap$(v, {
-    promptTokens: "prompt_tokens",
-    completionTokens: "completion_tokens",
-    totalTokens: "total_tokens",
-  });
+  return {
+    ...v.additionalProperties,
+    ...remap$(v, {
+      promptTokens: "prompt_tokens",
+      completionTokens: "completion_tokens",
+      totalTokens: "total_tokens",
+      promptAudioSeconds: "prompt_audio_seconds",
+      additionalProperties: null,
+    }),
+  };
 });
 
 /**

package/packages/mistralai-azure/src/sdk/ocr.ts

@@ -0,0 +1,24 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { ocrProcess } from "../funcs/ocrProcess.js";
+import { ClientSDK, RequestOptions } from "../lib/sdks.js";
+import * as components from "../models/components/index.js";
+import { unwrapAsync } from "../types/fp.js";
+
+export class Ocr extends ClientSDK {
+  /**
+   * OCR
+   */
+  async process(
+    request: components.OCRRequest,
+    options?: RequestOptions,
+  ): Promise<components.OCRResponse> {
+    return unwrapAsync(ocrProcess(
+      this,
+      request,
+      options,
+    ));
+  }
+}

package/src/funcs/agentsComplete.ts

@@ -19,7 +19,8 @@ import {
   UnexpectedClientError,
 } from "../models/errors/httpclienterrors.js";
 import * as errors from "../models/errors/index.js";
-import { SDKError } from "../models/errors/sdkerror.js";
+import { MistralError } from "../models/errors/mistralerror.js";
+import { ResponseValidationError } from "../models/errors/responsevalidationerror.js";
 import { SDKValidationError } from "../models/errors/sdkvalidationerror.js";
 import { APICall, APIPromise } from "../types/async.js";
 import { Result } from "../types/fp.js";
@@ -35,13 +36,14 @@ export function agentsComplete(
   Result<
     components.ChatCompletionResponse,
     | errors.HTTPValidationError
-    | SDKError
-    | SDKValidationError
-    | UnexpectedClientError
-    | InvalidRequestError
+    | MistralError
+    | ResponseValidationError
+    | ConnectionError
     | RequestAbortedError
     | RequestTimeoutError
-    | ConnectionError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
   >
 > {
   return new APIPromise($do(
@@ -60,13 +62,14 @@ async function $do(
     Result<
       components.ChatCompletionResponse,
       | errors.HTTPValidationError
-      | SDKError
-      | SDKValidationError
-      | UnexpectedClientError
-      | InvalidRequestError
+      | MistralError
+      | ResponseValidationError
+      | ConnectionError
       | RequestAbortedError
       | RequestTimeoutError
-      | ConnectionError
+      | InvalidRequestError
+      | UnexpectedClientError
+      | SDKValidationError
     >,
     APICall,
   ]
@@ -94,6 +97,7 @@ async function $do(
   const requestSecurity = resolveGlobalSecurity(securityInput);
 
   const context = {
+    options: client._options,
     baseURL: options?.serverURL ?? client._baseURL ?? "",
     operationID: "agents_completion_v1_agents_completions_post",
     oAuth2Scopes: [],
@@ -114,6 +118,7 @@ async function $do(
     path: path,
     headers: headers,
     body: body,
+    userAgent: client._options.userAgent,
     timeoutMs: options?.timeoutMs || client._options.timeoutMs || -1,
   }, options);
   if (!requestRes.ok) {
@@ -139,19 +144,20 @@ async function $do(
   const [result] = await M.match<
     components.ChatCompletionResponse,
     | errors.HTTPValidationError
-    | SDKError
-    | SDKValidationError
-    | UnexpectedClientError
-    | InvalidRequestError
+    | MistralError
+    | ResponseValidationError
+    | ConnectionError
     | RequestAbortedError
     | RequestTimeoutError
-    | ConnectionError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
   >(
     M.json(200, components.ChatCompletionResponse$inboundSchema),
     M.jsonErr(422, errors.HTTPValidationError$inboundSchema),
     M.fail("4XX"),
     M.fail("5XX"),
-  )(response, { extraFields: responseFields });
+  )(response, req, { extraFields: responseFields });
   if (!result.ok) {
     return [result, { status: "complete", request: req, response }];
   }

package/src/funcs/agentsStream.ts

@@ -21,7 +21,8 @@ import {
   UnexpectedClientError,
 } from "../models/errors/httpclienterrors.js";
 import * as errors from "../models/errors/index.js";
-import { SDKError } from "../models/errors/sdkerror.js";
+import { MistralError } from "../models/errors/mistralerror.js";
+import { ResponseValidationError } from "../models/errors/responsevalidationerror.js";
 import { SDKValidationError } from "../models/errors/sdkvalidationerror.js";
 import { APICall, APIPromise } from "../types/async.js";
 import { Result } from "../types/fp.js";
@@ -40,13 +41,14 @@ export function agentsStream(
   Result<
     EventStream<components.CompletionEvent>,
     | errors.HTTPValidationError
-    | SDKError
-    | SDKValidationError
-    | UnexpectedClientError
-    | InvalidRequestError
+    | MistralError
+    | ResponseValidationError
+    | ConnectionError
     | RequestAbortedError
     | RequestTimeoutError
-    | ConnectionError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
   >
 > {
   return new APIPromise($do(
@@ -65,13 +67,14 @@ async function $do(
     Result<
       EventStream<components.CompletionEvent>,
       | errors.HTTPValidationError
-      | SDKError
-      | SDKValidationError
-      | UnexpectedClientError
-      | InvalidRequestError
+      | MistralError
+      | ResponseValidationError
+      | ConnectionError
       | RequestAbortedError
       | RequestTimeoutError
-      | ConnectionError
+      | InvalidRequestError
+      | UnexpectedClientError
+      | SDKValidationError
     >,
     APICall,
   ]
@@ -100,6 +103,7 @@ async function $do(
   const requestSecurity = resolveGlobalSecurity(securityInput);
 
   const context = {
+    options: client._options,
     baseURL: options?.serverURL ?? client._baseURL ?? "",
     operationID: "stream_agents",
     oAuth2Scopes: [],
@@ -120,6 +124,7 @@ async function $do(
     path: path,
     headers: headers,
     body: body,
+    userAgent: client._options.userAgent,
     timeoutMs: options?.timeoutMs || client._options.timeoutMs || -1,
   }, options);
   if (!requestRes.ok) {
@@ -145,31 +150,30 @@ async function $do(
   const [result] = await M.match<
     EventStream<components.CompletionEvent>,
     | errors.HTTPValidationError
-    | SDKError
-    | SDKValidationError
-    | UnexpectedClientError
-    | InvalidRequestError
+    | MistralError
+    | ResponseValidationError
+    | ConnectionError
     | RequestAbortedError
     | RequestTimeoutError
-    | ConnectionError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
   >(
     M.sse(
       200,
       z.instanceof(ReadableStream<Uint8Array>).transform(stream => {
-        return new EventStream({
-          stream,
-          decoder(rawEvent) {
-            const schema = components.CompletionEvent$inboundSchema;
-            return schema.parse(rawEvent);
-          },
+        return new EventStream(stream, rawEvent => {
+          if (rawEvent.data === "[DONE]") return { done: true };
+          return {
+            value: components.CompletionEvent$inboundSchema.parse(rawEvent),
+          };
         });
       }),
-      { sseSentinel: "[DONE]" },
     ),
     M.jsonErr(422, errors.HTTPValidationError$inboundSchema),
     M.fail("4XX"),
     M.fail("5XX"),
-  )(response, { extraFields: responseFields });
+  )(response, req, { extraFields: responseFields });
   if (!result.ok) {
     return [result, { status: "complete", request: req, response }];
   }

package/src/funcs/audioTranscriptionsComplete.ts

@@ -4,7 +4,10 @@
 
 import { MistralCore } from "../core.js";
 import { appendForm } from "../lib/encodings.js";
-import { readableStreamToArrayBuffer } from "../lib/files.js";
+import {
+  getContentTypeFromFileName,
+  readableStreamToArrayBuffer,
+} from "../lib/files.js";
 import * as M from "../lib/matchers.js";
 import { compactMap } from "../lib/primitives.js";
 import { safeParse } from "../lib/schemas.js";
@@ -19,7 +22,8 @@ import {
   RequestTimeoutError,
   UnexpectedClientError,
 } from "../models/errors/httpclienterrors.js";
-import { SDKError } from "../models/errors/sdkerror.js";
+import { MistralError } from "../models/errors/mistralerror.js";
+import { ResponseValidationError } from "../models/errors/responsevalidationerror.js";
 import { SDKValidationError } from "../models/errors/sdkvalidationerror.js";
 import { APICall, APIPromise } from "../types/async.js";
 import { isBlobLike } from "../types/blobs.js";
@@ -36,13 +40,14 @@ export function audioTranscriptionsComplete(
 ): APIPromise<
   Result<
     components.TranscriptionResponse,
-    | SDKError
-    | SDKValidationError
-    | UnexpectedClientError
-    | InvalidRequestError
+    | MistralError
+    | ResponseValidationError
+    | ConnectionError
     | RequestAbortedError
     | RequestTimeoutError
-    | ConnectionError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
   >
 > {
   return new APIPromise($do(
@@ -60,13 +65,14 @@ async function $do(
   [
     Result<
       components.TranscriptionResponse,
-      | SDKError
-      | SDKValidationError
-      | UnexpectedClientError
-      | InvalidRequestError
+      | MistralError
+      | ResponseValidationError
+      | ConnectionError
       | RequestAbortedError
       | RequestTimeoutError
-      | ConnectionError
+      | InvalidRequestError
+      | UnexpectedClientError
+      | SDKValidationError
     >,
     APICall,
   ]
@@ -88,13 +94,17 @@ async function $do(
       appendForm(body, "file", payload.file);
     } else if (isReadableStream(payload.file.content)) {
       const buffer = await readableStreamToArrayBuffer(payload.file.content);
-      const blob = new Blob([buffer], { type: "application/octet-stream" });
-      appendForm(body, "file", blob);
+      const contentType = getContentTypeFromFileName(payload.file.fileName)
+        || "application/octet-stream";
+      const blob = new Blob([buffer], { type: contentType });
+      appendForm(body, "file", blob, payload.file.fileName);
     } else {
+      const contentType = getContentTypeFromFileName(payload.file.fileName)
+        || "application/octet-stream";
       appendForm(
         body,
         "file",
-        new Blob([payload.file.content], { type: "application/octet-stream" }),
+        new Blob([payload.file.content], { type: contentType }),
         payload.file.fileName,
       );
     }
@@ -133,6 +143,7 @@ async function $do(
   const requestSecurity = resolveGlobalSecurity(securityInput);
 
   const context = {
+    options: client._options,
     baseURL: options?.serverURL ?? client._baseURL ?? "",
     operationID: "audio_api_v1_transcriptions_post",
     oAuth2Scopes: [],
@@ -153,6 +164,7 @@ async function $do(
     path: path,
     headers: headers,
     body: body,
+    userAgent: client._options.userAgent,
     timeoutMs: options?.timeoutMs || client._options.timeoutMs || -1,
   }, options);
   if (!requestRes.ok) {
@@ -173,18 +185,19 @@ async function $do(
 
   const [result] = await M.match<
     components.TranscriptionResponse,
-    | SDKError
-    | SDKValidationError
-    | UnexpectedClientError
-    | InvalidRequestError
+    | MistralError
+    | ResponseValidationError
+    | ConnectionError
     | RequestAbortedError
     | RequestTimeoutError
-    | ConnectionError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
   >(
     M.json(200, components.TranscriptionResponse$inboundSchema),
     M.fail("4XX"),
     M.fail("5XX"),
-  )(response);
+  )(response, req);
   if (!result.ok) {
     return [result, { status: "complete", request: req, response }];
   }