@mozaik-ai/core 4.0.0-beta.1 → 4.0.0-beta.2

package/dist/index.d.mts

@@ -422,4 +422,98 @@ declare function defineRuntime<TRuntimeState extends RuntimeState>(): {
     runLoop: (agentId: string, message: string, inferenceInput: InferenceInput) => void;
 };
 
-export { ContextItem, DeveloperMessageItem, type Endpoint, FunctionCallItem, FunctionCallOutputItem, type InferenceInput, type InferenceOutput, InputTokenDetails, McpClient, type McpServerConfig, McpToolRegistry, type McpToolSpec, ModelContext, type ModelContextRepository, ModelMessageItem, OutputTokenDetails, Participant, ReasoningItem, RuntimeState, SemanticEvent, type StructuredOutputFormat, SystemMessageItem, TokenUsage, type Tool, UserMessageItem, createAgent, createHuman, defineRuntime };
+declare const supportedModels: GenerativeModel[];
+
+declare class OpenAIResponses implements Endpoint {
+    endpointMapper: InferenceEndpointMapper;
+    private readonly client;
+    constructor(endpointMapper?: InferenceEndpointMapper);
+    infer(inferenceInput: InferenceInput): Promise<InferenceOutput>;
+    stream(inferenceInput: InferenceInput): AsyncIterable<SemanticEvent>;
+}
+
+/**
+ * Optional connection config. When omitted, the `openai` SDK reads
+ * `OPENAI_API_KEY` and `OPENAI_BASE_URL` from the environment, so the
+ * default-constructed runtime targets whatever `OPENAI_BASE_URL` points
+ * at (real OpenAI when unset). Provider presets (e.g. DeepSeek) pass an
+ * explicit base URL + credential.
+ */
+interface OpenAICompatibleConfig {
+    baseURL?: string;
+    apiKey?: string;
+    /**
+     * Extra request-body fields merged into every `chat.completions`
+     * call. This is how provider-specific quirks are handled **without
+     * a per-provider subclass in mozaik** — the consumer supplies the
+     * vendor-only fields (e.g. DeepSeek's `{ thinking: { type } }`,
+     * safety flags, routing hints). Standard fields the runtime already
+     * sets (`model`, `messages`, `tools`, `reasoning_effort`) take
+     * precedence and are not overwritten.
+     */
+    extraBody?: Record<string, unknown>;
+}
+/**
+ * Generic adapter for any **OpenAI Chat Completions**-compatible
+ * endpoint — real OpenAI, DeepSeek, Xiaomi MiMo, OpenRouter, vLLM,
+ * Ollama, etc. It speaks `/chat/completions` (not the Responses API),
+ * which is the dialect third-party OpenAI-compatible providers expose.
+ *
+ * The base URL and credential are configurable; everything else (the
+ * `ModelContext` ⇄ chat-message conversion, tool-call round-trip, token
+ * usage extraction) is provider-agnostic. Provider-specific request
+ * shaping (e.g. DeepSeek's `thinking` field) is supplied by the
+ * consumer via {@link OpenAICompatibleConfig.extraBody} — mozaik stays
+ * generic and gains no per-provider subclasses.
+ *
+ * Was `DeepSeekChatCompletions`; generalized so consumers can point it
+ * at any OpenAI-compatible endpoint.
+ */
+declare class OpenAIChatCompletions implements Endpoint {
+    endpointMapper: InferenceEndpointMapper;
+    private readonly client;
+    private readonly extraBody;
+    constructor(endpointMapper?: InferenceEndpointMapper, config?: OpenAICompatibleConfig);
+    private buildRequest;
+    infer(inferenceInput: InferenceInput): Promise<InferenceOutput>;
+    stream(inferenceInput: InferenceInput): AsyncIterable<SemanticEvent>;
+}
+
+interface AnthropicConnectionConfig {
+    baseURL?: string;
+    apiKey?: string;
+}
+/**
+ * Native Anthropic adapter on the `@anthropic-ai/sdk` (`messages.create`).
+ * Maps domain context to Anthropic's `messages`/`content` blocks shape,
+ * system prompt, tools, adaptive thinking, and structured output config.
+ */
+declare class AnthropicMessages implements Endpoint {
+    endpointMapper: InferenceEndpointMapper;
+    private readonly client;
+    constructor(endpointMapper?: InferenceEndpointMapper, config?: AnthropicConnectionConfig);
+    infer(inferenceInput: InferenceInput): Promise<InferenceOutput>;
+    stream(inferenceInput: InferenceInput): AsyncIterable<SemanticEvent>;
+}
+
+interface GeminiConnectionConfig {
+    baseURL?: string;
+    apiKey?: string;
+}
+/**
+ * Native Gemini adapter on the `@google/genai` SDK (`generateContent` /
+ * `generateContentStream`). Unlike an OpenAI-compat shim this maps our
+ * domain context to Gemini's native `contents`/`parts` shape, system
+ * instruction, `functionDeclarations`, and `thinkingConfig`, and reads
+ * thought parts + native usage metadata back out. Another `ModelRuntime`
+ * — no runner or port changes.
+ */
+declare class GeminiGenerateContent implements Endpoint {
+    endpointMapper: InferenceEndpointMapper;
+    private readonly client;
+    constructor(endpointMapper?: InferenceEndpointMapper, config?: GeminiConnectionConfig);
+    infer(inferenceInput: InferenceInput): Promise<InferenceOutput>;
+    stream(inferenceInput: InferenceInput): AsyncIterable<SemanticEvent>;
+}
+
+export { AnthropicMessages, ContextItem, DeveloperMessageItem, type Endpoint, FunctionCallItem, FunctionCallOutputItem, GeminiGenerateContent, type InferenceInput, type InferenceOutput, InputTokenDetails, McpClient, type McpServerConfig, McpToolRegistry, type McpToolSpec, ModelContext, type ModelContextRepository, ModelMessageItem, OpenAIChatCompletions, OpenAIResponses, OutputTokenDetails, Participant, ReasoningItem, RuntimeState, SemanticEvent, type StructuredOutputFormat, SystemMessageItem, TokenUsage, type Tool, UserMessageItem, createAgent, createHuman, defineRuntime, supportedModels };

package/dist/index.d.ts

@@ -422,4 +422,98 @@ declare function defineRuntime<TRuntimeState extends RuntimeState>(): {
     runLoop: (agentId: string, message: string, inferenceInput: InferenceInput) => void;
 };
 
-export { ContextItem, DeveloperMessageItem, type Endpoint, FunctionCallItem, FunctionCallOutputItem, type InferenceInput, type InferenceOutput, InputTokenDetails, McpClient, type McpServerConfig, McpToolRegistry, type McpToolSpec, ModelContext, type ModelContextRepository, ModelMessageItem, OutputTokenDetails, Participant, ReasoningItem, RuntimeState, SemanticEvent, type StructuredOutputFormat, SystemMessageItem, TokenUsage, type Tool, UserMessageItem, createAgent, createHuman, defineRuntime };
+declare const supportedModels: GenerativeModel[];
+
+declare class OpenAIResponses implements Endpoint {
+    endpointMapper: InferenceEndpointMapper;
+    private readonly client;
+    constructor(endpointMapper?: InferenceEndpointMapper);
+    infer(inferenceInput: InferenceInput): Promise<InferenceOutput>;
+    stream(inferenceInput: InferenceInput): AsyncIterable<SemanticEvent>;
+}
+
+/**
+ * Optional connection config. When omitted, the `openai` SDK reads
+ * `OPENAI_API_KEY` and `OPENAI_BASE_URL` from the environment, so the
+ * default-constructed runtime targets whatever `OPENAI_BASE_URL` points
+ * at (real OpenAI when unset). Provider presets (e.g. DeepSeek) pass an
+ * explicit base URL + credential.
+ */
+interface OpenAICompatibleConfig {
+    baseURL?: string;
+    apiKey?: string;
+    /**
+     * Extra request-body fields merged into every `chat.completions`
+     * call. This is how provider-specific quirks are handled **without
+     * a per-provider subclass in mozaik** — the consumer supplies the
+     * vendor-only fields (e.g. DeepSeek's `{ thinking: { type } }`,
+     * safety flags, routing hints). Standard fields the runtime already
+     * sets (`model`, `messages`, `tools`, `reasoning_effort`) take
+     * precedence and are not overwritten.
+     */
+    extraBody?: Record<string, unknown>;
+}
+/**
+ * Generic adapter for any **OpenAI Chat Completions**-compatible
+ * endpoint — real OpenAI, DeepSeek, Xiaomi MiMo, OpenRouter, vLLM,
+ * Ollama, etc. It speaks `/chat/completions` (not the Responses API),
+ * which is the dialect third-party OpenAI-compatible providers expose.
+ *
+ * The base URL and credential are configurable; everything else (the
+ * `ModelContext` ⇄ chat-message conversion, tool-call round-trip, token
+ * usage extraction) is provider-agnostic. Provider-specific request
+ * shaping (e.g. DeepSeek's `thinking` field) is supplied by the
+ * consumer via {@link OpenAICompatibleConfig.extraBody} — mozaik stays
+ * generic and gains no per-provider subclasses.
+ *
+ * Was `DeepSeekChatCompletions`; generalized so consumers can point it
+ * at any OpenAI-compatible endpoint.
+ */
+declare class OpenAIChatCompletions implements Endpoint {
+    endpointMapper: InferenceEndpointMapper;
+    private readonly client;
+    private readonly extraBody;
+    constructor(endpointMapper?: InferenceEndpointMapper, config?: OpenAICompatibleConfig);
+    private buildRequest;
+    infer(inferenceInput: InferenceInput): Promise<InferenceOutput>;
+    stream(inferenceInput: InferenceInput): AsyncIterable<SemanticEvent>;
+}
+
+interface AnthropicConnectionConfig {
+    baseURL?: string;
+    apiKey?: string;
+}
+/**
+ * Native Anthropic adapter on the `@anthropic-ai/sdk` (`messages.create`).
+ * Maps domain context to Anthropic's `messages`/`content` blocks shape,
+ * system prompt, tools, adaptive thinking, and structured output config.
+ */
+declare class AnthropicMessages implements Endpoint {
+    endpointMapper: InferenceEndpointMapper;
+    private readonly client;
+    constructor(endpointMapper?: InferenceEndpointMapper, config?: AnthropicConnectionConfig);
+    infer(inferenceInput: InferenceInput): Promise<InferenceOutput>;
+    stream(inferenceInput: InferenceInput): AsyncIterable<SemanticEvent>;
+}
+
+interface GeminiConnectionConfig {
+    baseURL?: string;
+    apiKey?: string;
+}
+/**
+ * Native Gemini adapter on the `@google/genai` SDK (`generateContent` /
+ * `generateContentStream`). Unlike an OpenAI-compat shim this maps our
+ * domain context to Gemini's native `contents`/`parts` shape, system
+ * instruction, `functionDeclarations`, and `thinkingConfig`, and reads
+ * thought parts + native usage metadata back out. Another `ModelRuntime`
+ * — no runner or port changes.
+ */
+declare class GeminiGenerateContent implements Endpoint {
+    endpointMapper: InferenceEndpointMapper;
+    private readonly client;
+    constructor(endpointMapper?: InferenceEndpointMapper, config?: GeminiConnectionConfig);
+    infer(inferenceInput: InferenceInput): Promise<InferenceOutput>;
+    stream(inferenceInput: InferenceInput): AsyncIterable<SemanticEvent>;
+}
+
+export { AnthropicMessages, ContextItem, DeveloperMessageItem, type Endpoint, FunctionCallItem, FunctionCallOutputItem, GeminiGenerateContent, type InferenceInput, type InferenceOutput, InputTokenDetails, McpClient, type McpServerConfig, McpToolRegistry, type McpToolSpec, ModelContext, type ModelContextRepository, ModelMessageItem, OpenAIChatCompletions, OpenAIResponses, OutputTokenDetails, Participant, ReasoningItem, RuntimeState, SemanticEvent, type StructuredOutputFormat, SystemMessageItem, TokenUsage, type Tool, UserMessageItem, createAgent, createHuman, defineRuntime, supportedModels };

package/dist/index.js

@@ -30,15 +30,19 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
+  AnthropicMessages: () => AnthropicMessages,
   ContextItem: () => ContextItem,
   DeveloperMessageItem: () => DeveloperMessageItem,
   FunctionCallItem: () => FunctionCallItem,
   FunctionCallOutputItem: () => FunctionCallOutputItem,
+  GeminiGenerateContent: () => GeminiGenerateContent,
   InputTokenDetails: () => InputTokenDetails,
   McpClient: () => McpClient,
   McpToolRegistry: () => McpToolRegistry,
   ModelContext: () => ModelContext,
   ModelMessageItem: () => ModelMessageItem,
+  OpenAIChatCompletions: () => OpenAIChatCompletions,
+  OpenAIResponses: () => OpenAIResponses,
   OutputTokenDetails: () => OutputTokenDetails,
   Participant: () => Participant,
   ReasoningItem: () => ReasoningItem,
@@ -49,7 +53,8 @@ __export(index_exports, {
   UserMessageItem: () => UserMessageItem,
   createAgent: () => createAgent,
   createHuman: () => createHuman,
-  defineRuntime: () => defineRuntime
+  defineRuntime: () => defineRuntime,
+  supportedModels: () => supportedModels
 });
 module.exports = __toCommonJS(index_exports);
 
@@ -2057,15 +2062,19 @@ function defineRuntime() {
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  AnthropicMessages,
   ContextItem,
   DeveloperMessageItem,
   FunctionCallItem,
   FunctionCallOutputItem,
+  GeminiGenerateContent,
   InputTokenDetails,
   McpClient,
   McpToolRegistry,
   ModelContext,
   ModelMessageItem,
+  OpenAIChatCompletions,
+  OpenAIResponses,
   OutputTokenDetails,
   Participant,
   ReasoningItem,
@@ -2076,6 +2085,7 @@ function defineRuntime() {
   UserMessageItem,
   createAgent,
   createHuman,
-  defineRuntime
+  defineRuntime,
+  supportedModels
 });
 //# sourceMappingURL=index.js.map