@juspay/neurolink
Version:
Universal AI Development Platform with working MCP integration, multi-provider support, voice (TTS/STT/realtime), and professional CLI. 58+ external MCP servers discoverable, multimodal file processing, RAG pipelines. Build, test, and deploy AI applicatio
75 lines (74 loc) • 2.42 kB
TypeScript
/**
* Amazon SageMaker Provider Implementation (Simplified)
*
* This module provides a simplified SageMaker provider that extends BaseProvider
* and integrates with the NeuroLink ecosystem using existing patterns.
*/
import type { LanguageModel, Schema } from "ai";
import type { ZodType } from "zod";
import type { AIProviderName } from "../constants/enums.js";
import { BaseProvider } from "../core/baseProvider.js";
import type { NeuroLink } from "../neurolink.js";
import type { StreamOptions, StreamResult } from "../types/index.js";
/**
* Amazon SageMaker Provider extending BaseProvider
*/
export declare class AmazonSageMakerProvider extends BaseProvider {
private sagemakerModel;
private sagemakerConfig;
private modelConfig;
constructor(modelName?: string, endpointName?: string, region?: string, neurolink?: NeuroLink, credentials?: {
accessKeyId?: string;
secretAccessKey?: string;
sessionToken?: string;
region?: string;
endpoint?: string;
});
protected getProviderName(): AIProviderName;
protected getDefaultModel(): string;
protected getAISDKModel(): LanguageModel;
protected executeStream(_options: StreamOptions, _analysisSchema?: ZodType | Schema<unknown>): Promise<StreamResult>;
protected formatProviderError(error: unknown): Error;
/**
* Get SageMaker-specific provider information
*/
getSageMakerInfo(): {
endpointName: string;
modelType: string;
region: string;
configured: boolean;
};
/**
* Test basic configuration
*/
testConnection(): Promise<{
connected: boolean;
error?: string;
}>;
/**
* Public method to get the AI SDK model for CLI and external usage
*/
getModel(): Promise<LanguageModel>;
/**
* Test connectivity to the SageMaker endpoint
*/
testConnectivity(): Promise<{
success: boolean;
error?: string;
}>;
/**
* Get model capabilities and information
*/
getModelCapabilities(): {
capabilities: {
streaming: boolean;
toolCalling: boolean;
structuredOutput: boolean;
batchInference: boolean;
supportedResponseFormats: string[];
supportedToolTypes: string[];
maxBatchSize: number;
};
};
}
export default AmazonSageMakerProvider;