@juspay/neurolink
Version:
Universal AI Development Platform with working MCP integration, multi-provider support, and professional CLI. Built-in tools operational, 58+ external MCP servers discoverable. Connect to filesystem, GitHub, database operations, and more. Build, test, and
68 lines (67 loc) • 2.21 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 { ZodType, ZodTypeDef } from "zod";
import type { Schema, LanguageModelV1 } from "ai";
import type { AIProviderName } from "../core/types.js";
import type { StreamOptions, StreamResult } from "../types/streamTypes.js";
import { BaseProvider } from "../core/baseProvider.js";
/**
* Amazon SageMaker Provider extending BaseProvider
*/
export declare class AmazonSageMakerProvider extends BaseProvider {
private sagemakerModel;
private sagemakerConfig;
private modelConfig;
constructor(modelName?: string, endpointName?: string);
protected getProviderName(): AIProviderName;
protected getDefaultModel(): string;
protected getAISDKModel(): LanguageModelV1;
protected executeStream(options: StreamOptions, analysisSchema?: ZodType<unknown, ZodTypeDef, unknown> | Schema<unknown>): Promise<StreamResult>;
protected handleProviderError(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<LanguageModelV1>;
/**
* 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;