genkit/src/beta.ts

Genkit AI framework

/**
 * @license
 *
 * Copyright 2025 Google LLC
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

/**
 * Beta features including interrupts, stream managers, and the experimental
 * {@link GenkitBeta} class.
 *
 * ```ts
 * import { genkit } from 'genkit/beta';
 * ```
 *
 * @module beta
 */

export {
  AgentInitSchema,
  AgentInputSchema,
  AgentOutputSchema,
  AgentStreamChunkSchema,
  type AgentInit,
  type AgentInput,
  type AgentOutput,
} from '@genkit-ai/ai';
export { Session, type Artifact } from '@genkit-ai/ai/session';
export {
  AgentError,
  remoteAgent,
  type AgentAPI,
  type AgentChat,
  type AgentChunk,
  type AgentInterrupt,
  type AgentResponse,
  type AgentTurn,
  type DetachedTask,
  type RemoteAgentOptions,
} from './client/agent.js';

export type { AgentFinishReason } from './client/types.js';

export {
  InMemoryStreamManager,
  StreamNotFoundError,
  type ActionStreamInput,
  type ActionStreamSubscriber,
  type StreamManager,
} from '@genkit-ai/core';
export { AsyncTaskQueue, lazy } from '@genkit-ai/core/async';
export * from './common.js';
export {
  FileSessionStore,
  GenkitBeta,
  InMemorySessionStore,
  SessionRunner,
  applyPatch,
  diff,
  genkit,
  type Agent,
  type AgentFn,
  type AgentStreamChunk,
  type ClientTransform,
  type GenkitBetaOptions,
  type JsonPatch,
  type JsonPatchOperation,
  type SessionSnapshot,
  type SessionSnapshotInput,
  type SessionState,
  type SessionStore,
  type SessionStoreOptions,
  type SnapshotMutator,
} from './genkit-beta.js';