@mistralai/mistralai 2.2.5 → 2.2.6

package/src/extra/observability/streaming.ts

@@ -0,0 +1,174 @@
+/**
+ * Streaming response helpers for OTEL tracing.
+ *
+ * Pure functions that parse SSE byte streams and accumulate CompletionChunk
+ * deltas into a ChatCompletionResponse-shaped dict suitable for span enrichment.
+ *
+ * NOTE: The SSE bytes are re-parsed here even though EventStream already
+ * parsed them during iteration.
+ * TracedResponse sits below EventStream and can only accumulate raw bytes; it
+ * has no access to the decoded events. Hooking into EventStream could eliminate
+ * this double-parse, but EventStream is Speakeasy-generated code.
+ */
+
+import {
+  CompletionChunk,
+  completionChunkFromJSON,
+} from "../../models/components/completionchunk.js";
+import {
+  UsageInfo,
+  UsageInfo$outboundSchema,
+} from "../../models/components/usageinfo.js";
+
+type AccumulatedChoice = {
+  message: {
+    role: string;
+    content: string;
+    tool_calls?: Array<{
+      id: string | null;
+      function: {
+        name: string;
+        arguments: string;
+      };
+    }>;
+  };
+  finish_reason: string;
+};
+
+function normalizeCompletionChunkPayload(payload: string): string {
+  const parsed = JSON.parse(payload) as unknown;
+  if (typeof parsed !== "object" || parsed === null || Array.isArray(parsed)) {
+    return payload;
+  }
+
+  const chunk = parsed as Record<string, unknown>;
+
+  // Python accepts null for these optional fields; the TS schema expects omission.
+  for (const field of ["object", "created", "usage"]) {
+    if (chunk[field] === null) {
+      delete chunk[field];
+    }
+  }
+
+  const choices = chunk["choices"];
+  if (Array.isArray(choices)) {
+    for (const choice of choices) {
+      if (typeof choice === "object" && choice !== null && !Array.isArray(choice)) {
+        const choiceObj = choice as Record<string, unknown>;
+        if (!("finish_reason" in choiceObj)) {
+          choiceObj["finish_reason"] = null;
+        }
+      }
+    }
+  }
+
+  return JSON.stringify(chunk);
+}
+
+/**
+ * Parse raw SSE text into a list of typed CompletionChunk models.
+ *
+ * Only CompletionChunk is handled. If new SSE-streamed response types
+ * are added, parsing and typing here will need updating.
+ */
+export function parseSseChunks(rawSseText: string): CompletionChunk[] {
+  const chunks: CompletionChunk[] = [];
+  const lines = rawSseText.split("\n");
+  for (const line of lines) {
+    const trimmed = line.trim();
+    if (!trimmed.startsWith("data: ")) {
+      continue;
+    }
+    const payload = trimmed.slice(6);
+    if (payload === "[DONE]") {
+      continue;
+    }
+    try {
+      const result = completionChunkFromJSON(normalizeCompletionChunkPayload(payload));
+      if (result.ok) {
+        chunks.push(result.value);
+      }
+    } catch {
+      continue;
+    }
+  }
+  return chunks;
+}
+
+/**
+ * Accumulate streaming CompletionChunk deltas into a ChatCompletionResponse-shaped dict.
+ */
+export function accumulateChunksToResponseDict(
+  chunks: CompletionChunk[]
+): Record<string, unknown> {
+  let responseId: string | undefined;
+  let model: string | undefined;
+  let usage: UsageInfo | undefined;
+  const choices: Map<number, AccumulatedChoice> = new Map();
+
+  for (const chunk of chunks) {
+    responseId = responseId || chunk.id;
+    model = model || chunk.model;
+    usage = usage || chunk.usage;
+
+    for (const choice of chunk.choices) {
+      let accumulated = choices.get(choice.index);
+      if (!accumulated) {
+        accumulated = {
+          message: { role: "assistant", content: "" },
+          finish_reason: "",
+        };
+        choices.set(choice.index, accumulated);
+      }
+      const msg = accumulated.message;
+      const delta = choice.delta;
+      if (typeof delta.role === "string") {
+        msg.role = delta.role;
+      }
+      if (typeof delta.content === "string" && delta.content) {
+        msg.content += delta.content;
+      }
+      if (typeof choice.finishReason === "string") {
+        accumulated.finish_reason = choice.finishReason;
+      }
+      if (Array.isArray(delta.toolCalls)) {
+        if (!msg.tool_calls) {
+          msg.tool_calls = [];
+        }
+        const toolCalls = msg.tool_calls;
+        for (const tc of delta.toolCalls) {
+          const tcIdx = tc.index != null ? tc.index : toolCalls.length;
+          while (toolCalls.length <= tcIdx) {
+            toolCalls.push({
+              id: null,
+              function: { name: "", arguments: "" },
+            });
+          }
+          const existing = toolCalls[tcIdx]!;
+          // ToolCall.id defaults to "null" string in some cases (Speakeasy codegen quirk)
+          if (tc.id != null && tc.id !== "null") {
+            existing.id = tc.id;
+          }
+          if (tc.function.name) {
+            existing.function.name += tc.function.name;
+          }
+          if (typeof tc.function.arguments === "string" && tc.function.arguments) {
+            existing.function.arguments += tc.function.arguments;
+          }
+        }
+      }
+    }
+  }
+
+  const sortedIndices = Array.from(choices.keys()).sort((a, b) => a - b);
+  const result: Record<string, unknown> = {
+    id: responseId,
+    model,
+    choices: sortedIndices.map((idx) => choices.get(idx)),
+  };
+  if (usage != null) {
+    // Convert to snake_case JSON format for OTEL attributes
+    result["usage"] = UsageInfo$outboundSchema.parse(usage);
+  }
+  return result;
+}

package/src/funcs/betaConnectorsGetAuthUrl.ts

@@ -106,6 +106,8 @@ async function $do(
   const query = encodeFormQuery({
     "app_return_url": payload.app_return_url,
     "credentials_name": payload.credentials_name,
+    "github_installation_link": payload.github_installation_link,
+    "method_type": payload.method_type,
   });
 
   const headers = new Headers(compactMap({

package/src/funcs/betaObservabilityLogsFetchOptions.ts

@@ -0,0 +1,192 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ * @generated-id: 6b70d7ba709f
+ */
+
+import { MistralCore } from "../core.js";
+import { encodeFormQuery, encodeSimple } from "../lib/encodings.js";
+import { matchStatusCode } from "../lib/http.js";
+import * as M from "../lib/matchers.js";
+import { compactMap } from "../lib/primitives.js";
+import { safeParse } from "../lib/schemas.js";
+import { RequestOptions } from "../lib/sdks.js";
+import { extractSecurity, resolveGlobalSecurity } from "../lib/security.js";
+import { pathToFunc } from "../lib/url.js";
+import * as components from "../models/components/index.js";
+import {
+  ConnectionError,
+  InvalidRequestError,
+  RequestAbortedError,
+  RequestTimeoutError,
+  UnexpectedClientError,
+} from "../models/errors/httpclienterrors.js";
+import * as errors from "../models/errors/index.js";
+import { MistralError } from "../models/errors/mistralerror.js";
+import { ResponseValidationError } from "../models/errors/responsevalidationerror.js";
+import { SDKValidationError } from "../models/errors/sdkvalidationerror.js";
+import * as operations from "../models/operations/index.js";
+import { APICall, APIPromise } from "../types/async.js";
+import { Result } from "../types/fp.js";
+
+/**
+ * Get options for a log field
+ */
+export function betaObservabilityLogsFetchOptions(
+  client: MistralCore,
+  request:
+    operations.GetLogFieldOptionsV1ObservabilityLogsFieldsFieldNameOptionsGetRequest,
+  options?: RequestOptions,
+): APIPromise<
+  Result<
+    components.GetLogFieldOptions,
+    | errors.ObservabilityError
+    | MistralError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >
+> {
+  return new APIPromise($do(
+    client,
+    request,
+    options,
+  ));
+}
+
+async function $do(
+  client: MistralCore,
+  request:
+    operations.GetLogFieldOptionsV1ObservabilityLogsFieldsFieldNameOptionsGetRequest,
+  options?: RequestOptions,
+): Promise<
+  [
+    Result<
+      components.GetLogFieldOptions,
+      | errors.ObservabilityError
+      | MistralError
+      | ResponseValidationError
+      | ConnectionError
+      | RequestAbortedError
+      | RequestTimeoutError
+      | InvalidRequestError
+      | UnexpectedClientError
+      | SDKValidationError
+    >,
+    APICall,
+  ]
+> {
+  const parsed = safeParse(
+    request,
+    (value) =>
+      operations
+        .GetLogFieldOptionsV1ObservabilityLogsFieldsFieldNameOptionsGetRequest$outboundSchema
+        .parse(value),
+    "Input validation failed",
+  );
+  if (!parsed.ok) {
+    return [parsed, { status: "invalid" }];
+  }
+  const payload = parsed.value;
+  const body = null;
+
+  const pathParams = {
+    field_name: encodeSimple("field_name", payload.field_name, {
+      explode: false,
+      charEncoding: "percent",
+    }),
+  };
+  const path = pathToFunc("/v1/observability/logs/fields/{field_name}/options")(
+    pathParams,
+  );
+
+  const query = encodeFormQuery({
+    "from": payload.from,
+    "to": payload.to,
+  });
+
+  const headers = new Headers(compactMap({
+    Accept: "application/json",
+  }));
+
+  const secConfig = await extractSecurity(client._options.apiKey);
+  const securityInput = secConfig == null ? {} : { apiKey: secConfig };
+  const requestSecurity = resolveGlobalSecurity(securityInput);
+
+  const context = {
+    options: client._options,
+    baseURL: options?.serverURL ?? client._baseURL ?? "",
+    operationID:
+      "get_log_field_options_v1_observability_logs_fields__field_name__options_get",
+    oAuth2Scopes: null,
+
+    resolvedSecurity: requestSecurity,
+
+    securitySource: client._options.apiKey,
+    retryConfig: options?.retries
+      || client._options.retryConfig
+      || { strategy: "none" },
+    retryCodes: options?.retryCodes || ["429", "500", "502", "503", "504"],
+  };
+
+  const requestRes = client._createRequest(context, {
+    security: requestSecurity,
+    method: "GET",
+    baseURL: options?.serverURL,
+    path: path,
+    headers: headers,
+    query: query,
+    body: body,
+    userAgent: client._options.userAgent,
+    timeoutMs: options?.timeoutMs || client._options.timeoutMs || 60000,
+  }, options);
+  if (!requestRes.ok) {
+    return [requestRes, { status: "invalid" }];
+  }
+  const req = requestRes.value;
+
+  const doResult = await client._do(req, {
+    context,
+    isErrorStatusCode: (statusCode: number) =>
+      matchStatusCode({ status: statusCode } as Response, ["4XX", "5XX"]),
+    retryConfig: context.retryConfig,
+    retryCodes: context.retryCodes,
+  });
+  if (!doResult.ok) {
+    return [doResult, { status: "request-error", request: req }];
+  }
+  const response = doResult.value;
+
+  const responseFields = {
+    HttpMeta: { Response: response, Request: req },
+  };
+
+  const [result] = await M.match<
+    components.GetLogFieldOptions,
+    | errors.ObservabilityError
+    | MistralError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >(
+    M.json(200, components.GetLogFieldOptions$inboundSchema),
+    M.jsonErr(
+      [400, 404, 408, 409, 422],
+      errors.ObservabilityError$inboundSchema,
+    ),
+    M.fail("4XX"),
+    M.fail("5XX"),
+  )(response, req, { extraFields: responseFields });
+  if (!result.ok) {
+    return [result, { status: "complete", request: req, response }];
+  }
+
+  return [result, { status: "complete", request: req, response }];
+}

package/src/funcs/betaObservabilityLogsList.ts

@@ -0,0 +1,154 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ * @generated-id: dffe2262abc3
+ */
+
+import { MistralCore } from "../core.js";
+import { matchStatusCode } from "../lib/http.js";
+import * as M from "../lib/matchers.js";
+import { compactMap } from "../lib/primitives.js";
+import { RequestOptions } from "../lib/sdks.js";
+import { extractSecurity, resolveGlobalSecurity } from "../lib/security.js";
+import { pathToFunc } from "../lib/url.js";
+import * as components from "../models/components/index.js";
+import {
+  ConnectionError,
+  InvalidRequestError,
+  RequestAbortedError,
+  RequestTimeoutError,
+  UnexpectedClientError,
+} from "../models/errors/httpclienterrors.js";
+import * as errors from "../models/errors/index.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";
+
+/**
+ * Get log field definitions
+ */
+export function betaObservabilityLogsList(
+  client: MistralCore,
+  options?: RequestOptions,
+): APIPromise<
+  Result<
+    components.GetLogFields,
+    | errors.ObservabilityError
+    | MistralError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >
+> {
+  return new APIPromise($do(
+    client,
+    options,
+  ));
+}
+
+async function $do(
+  client: MistralCore,
+  options?: RequestOptions,
+): Promise<
+  [
+    Result<
+      components.GetLogFields,
+      | errors.ObservabilityError
+      | MistralError
+      | ResponseValidationError
+      | ConnectionError
+      | RequestAbortedError
+      | RequestTimeoutError
+      | InvalidRequestError
+      | UnexpectedClientError
+      | SDKValidationError
+    >,
+    APICall,
+  ]
+> {
+  const path = pathToFunc("/v1/observability/logs/fields")();
+
+  const headers = new Headers(compactMap({
+    Accept: "application/json",
+  }));
+
+  const secConfig = await extractSecurity(client._options.apiKey);
+  const securityInput = secConfig == null ? {} : { apiKey: secConfig };
+  const requestSecurity = resolveGlobalSecurity(securityInput);
+
+  const context = {
+    options: client._options,
+    baseURL: options?.serverURL ?? client._baseURL ?? "",
+    operationID: "get_log_fields_v1_observability_logs_fields_get",
+    oAuth2Scopes: null,
+
+    resolvedSecurity: requestSecurity,
+
+    securitySource: client._options.apiKey,
+    retryConfig: options?.retries
+      || client._options.retryConfig
+      || { strategy: "none" },
+    retryCodes: options?.retryCodes || ["429", "500", "502", "503", "504"],
+  };
+
+  const requestRes = client._createRequest(context, {
+    security: requestSecurity,
+    method: "GET",
+    baseURL: options?.serverURL,
+    path: path,
+    headers: headers,
+    userAgent: client._options.userAgent,
+    timeoutMs: options?.timeoutMs || client._options.timeoutMs || 60000,
+  }, options);
+  if (!requestRes.ok) {
+    return [requestRes, { status: "invalid" }];
+  }
+  const req = requestRes.value;
+
+  const doResult = await client._do(req, {
+    context,
+    isErrorStatusCode: (statusCode: number) =>
+      matchStatusCode({ status: statusCode } as Response, ["4XX", "5XX"]),
+    retryConfig: context.retryConfig,
+    retryCodes: context.retryCodes,
+  });
+  if (!doResult.ok) {
+    return [doResult, { status: "request-error", request: req }];
+  }
+  const response = doResult.value;
+
+  const responseFields = {
+    HttpMeta: { Response: response, Request: req },
+  };
+
+  const [result] = await M.match<
+    components.GetLogFields,
+    | errors.ObservabilityError
+    | MistralError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >(
+    M.json(200, components.GetLogFields$inboundSchema),
+    M.jsonErr(
+      [400, 404, 408, 409, 422],
+      errors.ObservabilityError$inboundSchema,
+    ),
+    M.fail("4XX"),
+    M.fail("5XX"),
+  )(response, req, { extraFields: responseFields });
+  if (!result.ok) {
+    return [result, { status: "complete", request: req, response }];
+  }
+
+  return [result, { status: "complete", request: req, response }];
+}

package/src/funcs/betaObservabilityLogsSearch.ts

@@ -0,0 +1,183 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ * @generated-id: 82187ff38f29
+ */
+
+import { MistralCore } from "../core.js";
+import { encodeFormQuery, encodeJSON } from "../lib/encodings.js";
+import { matchStatusCode } from "../lib/http.js";
+import * as M from "../lib/matchers.js";
+import { compactMap } from "../lib/primitives.js";
+import { safeParse } from "../lib/schemas.js";
+import { RequestOptions } from "../lib/sdks.js";
+import { extractSecurity, resolveGlobalSecurity } from "../lib/security.js";
+import { pathToFunc } from "../lib/url.js";
+import * as components from "../models/components/index.js";
+import {
+  ConnectionError,
+  InvalidRequestError,
+  RequestAbortedError,
+  RequestTimeoutError,
+  UnexpectedClientError,
+} from "../models/errors/httpclienterrors.js";
+import * as errors from "../models/errors/index.js";
+import { MistralError } from "../models/errors/mistralerror.js";
+import { ResponseValidationError } from "../models/errors/responsevalidationerror.js";
+import { SDKValidationError } from "../models/errors/sdkvalidationerror.js";
+import * as operations from "../models/operations/index.js";
+import { APICall, APIPromise } from "../types/async.js";
+import { Result } from "../types/fp.js";
+
+/**
+ * Search logs
+ */
+export function betaObservabilityLogsSearch(
+  client: MistralCore,
+  request: operations.SearchLogsV1ObservabilityLogsSearchPostRequest,
+  options?: RequestOptions,
+): APIPromise<
+  Result<
+    components.GetLogs,
+    | errors.ObservabilityError
+    | MistralError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >
+> {
+  return new APIPromise($do(
+    client,
+    request,
+    options,
+  ));
+}
+
+async function $do(
+  client: MistralCore,
+  request: operations.SearchLogsV1ObservabilityLogsSearchPostRequest,
+  options?: RequestOptions,
+): Promise<
+  [
+    Result<
+      components.GetLogs,
+      | errors.ObservabilityError
+      | MistralError
+      | ResponseValidationError
+      | ConnectionError
+      | RequestAbortedError
+      | RequestTimeoutError
+      | InvalidRequestError
+      | UnexpectedClientError
+      | SDKValidationError
+    >,
+    APICall,
+  ]
+> {
+  const parsed = safeParse(
+    request,
+    (value) =>
+      operations.SearchLogsV1ObservabilityLogsSearchPostRequest$outboundSchema
+        .parse(value),
+    "Input validation failed",
+  );
+  if (!parsed.ok) {
+    return [parsed, { status: "invalid" }];
+  }
+  const payload = parsed.value;
+  const body = encodeJSON("body", payload.LogsRequest, { explode: true });
+
+  const path = pathToFunc("/v1/observability/logs/search")();
+
+  const query = encodeFormQuery({
+    "cursor": payload.cursor,
+    "from": payload.from,
+    "page_size": payload.page_size,
+    "to": payload.to,
+  });
+
+  const headers = new Headers(compactMap({
+    "Content-Type": "application/json",
+    Accept: "application/json",
+  }));
+
+  const secConfig = await extractSecurity(client._options.apiKey);
+  const securityInput = secConfig == null ? {} : { apiKey: secConfig };
+  const requestSecurity = resolveGlobalSecurity(securityInput);
+
+  const context = {
+    options: client._options,
+    baseURL: options?.serverURL ?? client._baseURL ?? "",
+    operationID: "search_logs_v1_observability_logs_search_post",
+    oAuth2Scopes: null,
+
+    resolvedSecurity: requestSecurity,
+
+    securitySource: client._options.apiKey,
+    retryConfig: options?.retries
+      || client._options.retryConfig
+      || { strategy: "none" },
+    retryCodes: options?.retryCodes || ["429", "500", "502", "503", "504"],
+  };
+
+  const requestRes = client._createRequest(context, {
+    security: requestSecurity,
+    method: "POST",
+    baseURL: options?.serverURL,
+    path: path,
+    headers: headers,
+    query: query,
+    body: body,
+    userAgent: client._options.userAgent,
+    timeoutMs: options?.timeoutMs || client._options.timeoutMs || 60000,
+  }, options);
+  if (!requestRes.ok) {
+    return [requestRes, { status: "invalid" }];
+  }
+  const req = requestRes.value;
+
+  const doResult = await client._do(req, {
+    context,
+    isErrorStatusCode: (statusCode: number) =>
+      matchStatusCode({ status: statusCode } as Response, ["4XX", "5XX"]),
+    retryConfig: context.retryConfig,
+    retryCodes: context.retryCodes,
+  });
+  if (!doResult.ok) {
+    return [doResult, { status: "request-error", request: req }];
+  }
+  const response = doResult.value;
+
+  const responseFields = {
+    HttpMeta: { Response: response, Request: req },
+  };
+
+  const [result] = await M.match<
+    components.GetLogs,
+    | errors.ObservabilityError
+    | MistralError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >(
+    M.json(200, components.GetLogs$inboundSchema),
+    M.jsonErr(
+      [400, 404, 408, 409, 422],
+      errors.ObservabilityError$inboundSchema,
+    ),
+    M.fail("4XX"),
+    M.fail("5XX"),
+  )(response, req, { extraFields: responseFields });
+  if (!result.ok) {
+    return [result, { status: "complete", request: req, response }];
+  }
+
+  return [result, { status: "complete", request: req, response }];
+}