@tanstack/ai
Version:
Core TanStack AI library - Open source AI SDK
75 lines (74 loc) • 2.99 kB
TypeScript
import { StreamChunk, SummarizationOptions, SummarizationResult } from '../../types.js';
/**
* Configuration for summarize adapter instances
*/
export interface SummarizeAdapterConfig {
apiKey?: string;
baseUrl?: string;
timeout?: number;
maxRetries?: number;
headers?: Record<string, string>;
}
/**
* Summarize adapter interface with pre-resolved generics.
*
* An adapter is created by a provider function: `provider('model')` → `adapter`
* All type resolution happens at the provider call site, not in this interface.
*
* Generic parameters:
* - TModel: The specific model name (e.g., 'gpt-4o')
* - TProviderOptions: Provider-specific options (already resolved)
*/
export interface SummarizeAdapter<TModel extends string = string, TProviderOptions extends object = Record<string, unknown>> {
/** Discriminator for adapter kind - used by generate() to determine API shape */
readonly kind: 'summarize';
/** Adapter name identifier */
readonly name: string;
/** The model this adapter is configured for */
readonly model: TModel;
/**
* @internal Type-only properties for inference. Not assigned at runtime.
*/
'~types': {
providerOptions: TProviderOptions;
};
/**
* Summarize the given text
*/
summarize: (options: SummarizationOptions) => Promise<SummarizationResult>;
/**
* Stream summarization of the given text.
* Optional - if not implemented, the activity layer will fall back to
* non-streaming summarize and yield the result as a single chunk.
*/
summarizeStream?: (options: SummarizationOptions) => AsyncIterable<StreamChunk>;
}
/**
* A SummarizeAdapter with any/unknown type parameters.
* Useful as a constraint in generic functions and interfaces.
*/
export type AnySummarizeAdapter = SummarizeAdapter<any, any>;
/**
* Abstract base class for summarize adapters.
* Extend this class to implement a summarize adapter for a specific provider.
*
* Generic parameters match SummarizeAdapter - all pre-resolved by the provider function.
*/
export declare abstract class BaseSummarizeAdapter<TModel extends string = string, TProviderOptions extends object = Record<string, unknown>> implements SummarizeAdapter<TModel, TProviderOptions> {
readonly kind: "summarize";
abstract readonly name: string;
readonly model: TModel;
'~types': {
providerOptions: TProviderOptions;
};
protected config: SummarizeAdapterConfig;
constructor(config: SummarizeAdapterConfig | undefined, model: TModel);
abstract summarize(options: SummarizationOptions): Promise<SummarizationResult>;
/**
* Stream summarization of the given text.
* Override this method in concrete implementations to enable streaming.
* If not overridden, the activity layer will fall back to non-streaming.
*/
summarizeStream?(options: SummarizationOptions): AsyncIterable<StreamChunk>;
protected generateId(): string;
}