ohlcv-ai 1.0.4 → 1.0.7

package/README.md

@@ -1,3 +1,12 @@
 # Ohlcv AI
 
 🧠AI agents focused on processing ohlcv data structures.
+
+# Install
+
+```bash
+npm i ohlcv-ai
+```
+```bash
+yarn add ohlcv-ai
+```

package/dist/index.d.ts

@@ -0,0 +1,615 @@
+export declare const ALIYUN_MODELS: Map<AliYunModelType, Model>;
+
+export declare class AliyunAI {
+    private apiKey;
+    private modelType;
+    private timeout;
+    /**
+     * Constructor - Minimal configuration
+     * @param config.apiKey - API key (required)
+     * @param config.modelType - Model type, default qwen-turbo
+     * @param config.timeout - Timeout, default 30 seconds
+     */
+    constructor(config: AliyunConfig);
+    /**
+     * Simplest method: single conversation
+     * @param message - User message
+     * @param options - Chat options
+     * @returns AI response
+     */
+    chat(message: string, options?: AliYunChatOptions): Promise<string>;
+    /**
+     * Multi-turn conversation
+     * @param messages - Message history
+     * @param options - Chat options
+     * @returns Complete API response
+     */
+    chatCompletion(messages: ChatMessage[], options?: {
+        temperature?: number;
+        maxTokens?: number;
+        stream?: boolean;
+        modelType?: AliYunModelType;
+    }): Promise<any>;
+    /**
+     * Streaming conversation (only supports OpenAI format)
+     * @param messages - Message history
+     * @param callback - Streaming callback function
+     * @param options - Chat options
+     */
+    chatStream(messages: ChatMessage[], callback: AliYunStreamCallback, options?: {
+        temperature?: number;
+        maxTokens?: number;
+        modelType?: AliYunModelType;
+    }): Promise<void>;
+    /**
+     * Switch model
+     * @param modelType - New model type
+     */
+    setModel(modelType: AliYunModelType): void;
+    /**
+     * Get current model configuration
+     */
+    getCurrentModel(): {
+        name: string;
+        displayName: string;
+        description?: string;
+    };
+    /**
+     * Test connection
+     * @returns Connection test result
+     */
+    testConnection(): Promise<{
+        success: boolean;
+        model: string;
+        response?: string;
+        error?: string;
+    }>;
+    private buildOpenAIRequest;
+    private buildDashScopeRequest;
+    private makeRequest;
+    private makeStreamRequest;
+    private extractContent;
+    /**
+     * Specialized method for processing OHLCV arrays
+     * @param ohlcvArray - OHLCV data array
+     * @param instructions - Processing instructions, supports Chinese and English (optional, default: "Based on these OHLCV data, predict the next period")
+     * @param count - Number of OHLCV data items to return (optional, default: 1)
+     * @param options - Chat options
+     * @returns Predicted OHLCV array
+     */
+    predictingOHLCV(ohlcvArray: OHLCV[], instructions?: string, count?: number, options?: AliYunChatOptions): Promise<OHLCV[]>;
+    /**
+     * Parse AI returned OHLCV response
+     * @private
+     */
+    private parseOHLCVResponse;
+}
+
+export declare interface AliYunChatOptions {
+    temperature?: number;
+    maxTokens?: number;
+    stream?: boolean;
+    systemPrompt?: string;
+    modelType?: AliYunModelType;
+}
+
+export declare interface AliyunConfig {
+    apiKey: string;
+    modelType?: AliYunModelType;
+    timeout?: number;
+}
+
+export declare enum AliYunModelType {
+    QWEN_TURBO = "qwen-turbo",
+    QWEN_PLUS = "qwen-plus",
+    QWEN_MAX = "qwen-max",
+    QWEN_MAX_LONGCONTEXT = "qwen-max-longcontext",
+    QWEN_2_5B = "qwen2.5-0.5b",
+    QWEN_2_5B_INSTRUCT = "qwen2.5-0.5b-instruct",
+    QWEN_2_5B_7B = "qwen2.5-7b",
+    QWEN_2_5B_7B_INSTRUCT = "qwen2.5-7b-instruct",
+    QWEN_2_5B_14B = "qwen2.5-14b",
+    QWEN_2_5B_14B_INSTRUCT = "qwen2.5-14b-instruct",
+    QWEN_2_5B_32B = "qwen2.5-32b",
+    QWEN_2_5B_32B_INSTRUCT = "qwen2.5-32b-instruct",
+    QWEN_2_5B_72B = "qwen2.5-72b",
+    QWEN_2_5B_72B_INSTRUCT = "qwen2.5-72b-instruct",
+    QWEN_2_5B_CODER = "qwen2.5-coder",
+    QWEN_2_5B_CODER_7B = "qwen2.5-coder-7b",
+    QWEN_2_5B_CODER_14B = "qwen2.5-coder-14b",
+    QWEN_2_5B_CODER_32B = "qwen2.5-coder-32b",
+    QWEN_VL_LITE = "qwen-vl-lite",
+    QWEN_VL_PLUS = "qwen-vl-plus",
+    QWEN_VL_MAX = "qwen-vl-max",
+    QWEN_AUDIO_TURBO = "qwen-audio-turbo",
+    QWEN_AUDIO_CHAT = "qwen-audio-chat",
+    QWEN_MATH_7B = "qwen-math-7b",
+    LLAMA2_7B_CHAT_V2 = "llama2-7b-chat-v2",
+    BAICHUAN2_7B_CHAT_V1 = "baichuan2-7b-chat-v1",
+    QWEN_FINANCIAL = "qwen-financial",
+    QWEN_FINANCIAL_14B = "qwen-financial-14b",
+    QWEN_FINANCIAL_32B = "qwen-financial-32b",
+    QWEN_MEDICAL = "qwen-medical",
+    QWEN_MEDICAL_14B = "qwen-medical-14b",
+    QWEN_MEDICAL_32B = "qwen-medical-32b",
+    QWEN_OMNI = "qwen-omni",
+    QWEN_OMNI_PRO = "qwen-omni-pro"
+}
+
+export declare type AliYunStreamCallback = (chunk: string, isEnd: boolean) => void;
+
+export declare interface ChatMessage {
+    role: 'system' | 'user' | 'assistant';
+    content: string;
+}
+
+export declare interface CostEstimate {
+    inputTokens: number;
+    outputTokens: number;
+    inputCost: number;
+    outputCost: number;
+    totalCost: number;
+}
+
+/**
+ * Factory function for quick instance creation
+ */
+export declare function createAliyunAI(apiKey: string, modelType?: AliYunModelType): AliyunAI;
+
+/**
+ * Factory function for quick instance creation
+ */
+export declare function createDeepSeekAI(apiKey: string, modelType?: DeepSeekModelType): DeepSeekAI;
+
+/**
+ * Factory function for quick instance creation
+ */
+export declare function createOpenAI(apiKey: string, modelType?: OpenAIModelType): OpenAI;
+
+export declare const DEEPSEEK_MODELS: Map<DeepSeekModelType, DeepSeekModel>;
+
+export declare class DeepSeekAI {
+    private apiKey;
+    private modelType;
+    private timeout;
+    private baseURL;
+    /**
+     * Constructor - Minimal configuration
+     * @param config.apiKey - API key (required)
+     * @param config.modelType - Model type, default deepseek-chat
+     * @param config.timeout - Timeout, default 30 seconds
+     * @param config.baseURL - Base URL for API, default official endpoint
+     */
+    constructor(config: DeepSeekConfig);
+    /**
+     * Simplest method: single conversation
+     * @param message - User message
+     * @param options - Chat options
+     * @returns AI response
+     */
+    chat(message: string, options?: DeepSeekChatOptions): Promise<string>;
+    /**
+     * Multi-turn conversation
+     * @param messages - Message history
+     * @param options - Chat options
+     * @returns Complete API response
+     */
+    chatCompletion(messages: DeepSeekChatMessage[], options?: DeepSeekChatOptions): Promise<any>;
+    /**
+     * Streaming conversation
+     * @param messages - Message history
+     * @param callback - Streaming callback function
+     * @param options - Chat options
+     */
+    chatStream(messages: DeepSeekChatMessage[], callback: DeepSeekStreamCallback, options?: DeepSeekChatOptions): Promise<void>;
+    /**
+     * Specialized method for processing OHLCV arrays
+     * @param ohlcvArray - OHLCV data array
+     * @param instructions - Processing instructions (optional)
+     * @param count - Number of OHLCV data items to return (optional, default: 1)
+     * @param options - Chat options
+     * @returns Predicted OHLCV array
+     */
+    predictingOHLCV(ohlcvArray: OHLCV[], instructions?: string, count?: number, options?: DeepSeekChatOptions): Promise<OHLCV[]>;
+    /**
+     * Switch model
+     * @param modelType - New model type
+     */
+    setModel(modelType: DeepSeekModelType): void;
+    /**
+     * Get current model configuration
+     */
+    getCurrentModel(): {
+        name: string;
+        displayName: string;
+        description?: string;
+    };
+    /**
+     * Test connection
+     * @returns Connection test result
+     */
+    testConnection(): Promise<{
+        success: boolean;
+        model: string;
+        response?: string;
+        error?: string;
+    }>;
+    private buildOpenAIRequest;
+    private makeRequest;
+    private makeStreamRequest;
+    private extractContent;
+    private parseOHLCVResponse;
+}
+
+export declare interface DeepSeekChatMessage {
+    role: 'system' | 'user' | 'assistant' | 'tool';
+    content: string;
+    name?: string;
+    tool_call_id?: string;
+}
+
+export declare interface DeepSeekChatOptions {
+    temperature?: number;
+    maxTokens?: number;
+    stream?: boolean;
+    systemPrompt?: string;
+    modelType?: DeepSeekModelType;
+    topP?: number;
+    frequencyPenalty?: number;
+    presencePenalty?: number;
+    stop?: string[];
+    tools?: any[];
+    toolChoice?: string | object;
+}
+
+export declare interface DeepSeekConfig {
+    apiKey: string;
+    modelType?: DeepSeekModelType;
+    timeout?: number;
+    baseURL?: string;
+}
+
+declare interface DeepSeekModel {
+    name: string;
+    displayName: string;
+    endpoint: string;
+    endpoints?: string[];
+    format: 'openai' | 'deepseek';
+    description?: string;
+    maxTokens?: number;
+    contextLength?: number;
+    capabilities?: string[];
+    version?: string;
+}
+
+export declare enum DeepSeekModelType {
+    DEEPSEEK_CHAT = "deepseek-chat",
+    DEEPSEEK_CHAT_LITE = "deepseek-chat-lite",
+    DEEPSEEK_CHAT_PRO = "deepseek-chat-pro",
+    DEEPSEEK_CHAT_MAX = "deepseek-chat-max",
+    DEEPSEEK_CODER = "deepseek-coder",
+    DEEPSEEK_CODER_LITE = "deepseek-coder-lite",
+    DEEPSEEK_CODER_PRO = "deepseek-coder-pro",
+    DEEPSEEK_MATH = "deepseek-math",
+    DEEPSEEK_MATH_PRO = "deepseek-math-pro",
+    DEEPSEEK_REASONER = "deepseek-reasoner",
+    DEEPSEEK_REASONER_PRO = "deepseek-reasoner-pro",
+    DEEPSEEK_VISION = "deepseek-vision",
+    DEEPSEEK_VISION_PRO = "deepseek-vision-pro",
+    DEEPSEEK_FINANCE = "deepseek-finance",
+    DEEPSEEK_LAW = "deepseek-law",
+    DEEPSEEK_MEDICAL = "deepseek-medical",
+    DEEPSEEK_RESEARCH = "deepseek-research",
+    DEEPSEEK_OMNI = "deepseek-omni",
+    DEEPSEEK_OMNI_PRO = "deepseek-omni-pro",
+    DEEPSEEK_LLM = "deepseek-llm",
+    DEEPSEEK_LLM_67B = "deepseek-llm-67b",
+    DEEPSEEK_LLM_131B = "deepseek-llm-131b"
+}
+
+export declare type DeepSeekStreamCallback = (chunk: string, isEnd: boolean) => void;
+
+export declare function estimateCost(model: OpenAIModel, inputTokens: number, outputTokens?: number): CostEstimate;
+
+export declare function getAllOpenAIModels(): OpenAIModel[];
+
+export declare function getAudioModelsOpenAI(): OpenAIModel[];
+
+export declare function getAvailableOpenAIModelTypes(): OpenAIModelType[];
+
+export declare function getChatModels(): OpenAIModel[];
+
+export declare function getCompletionModels(): OpenAIModel[];
+
+export declare function getCostEfficientModels(): OpenAIModel[];
+
+export declare function getEmbeddingModels(): OpenAIModel[];
+
+export declare function getHighContextModels(): OpenAIModel[];
+
+export declare function getLatestModels(): OpenAIModel[];
+
+export declare function getMultimodalModelsOpenAI(): OpenAIModel[];
+
+export declare function getOpenAIModel(type: OpenAIModelType): OpenAIModel | undefined;
+
+export declare function getOpenAIModelByName(name: string): OpenAIModel | undefined;
+
+export declare function getVisionModelsOpenAI(): OpenAIModel[];
+
+declare interface Model {
+    name: string;
+    displayName: string;
+    endpoint: string;
+    endpoints?: string[];
+    format: 'openai' | 'dashscope';
+    description?: string;
+    maxTokens?: number;
+    contextLength?: number;
+    capabilities?: string[];
+}
+
+export declare interface OHLCV {
+    open: number;
+    high: number;
+    low: number;
+    close: number;
+    volume: number;
+}
+
+export declare class OpenAI {
+    private apiKey;
+    private modelType;
+    private timeout;
+    private organization?;
+    private baseURL;
+    /**
+     * Constructor - Minimal configuration
+     * @param config.apiKey - API key (required)
+     * @param config.modelType - Model type, default gpt-3.5-turbo
+     * @param config.timeout - Timeout, default 30 seconds
+     * @param config.organization - Organization ID (optional)
+     * @param config.baseURL - Custom base URL (optional)
+     */
+    constructor(config: OpenAIConfig);
+    /**
+     * Simplest method: single conversation
+     * @param message - User message
+     * @param options - Chat options
+     * @returns AI response
+     */
+    chat(message: string, options?: OpenAIChatOptions): Promise<string>;
+    /**
+     * Multi-turn conversation
+     * @param messages - Message history
+     * @param options - Chat options
+     * @returns Complete API response
+     */
+    chatCompletion(messages: ChatMessage[], options?: {
+        temperature?: number;
+        maxTokens?: number;
+        stream?: boolean;
+        modelType?: OpenAIModelType;
+        topP?: number;
+        frequencyPenalty?: number;
+        presencePenalty?: number;
+        stop?: string[];
+    }): Promise<any>;
+    /**
+     * Streaming conversation
+     * @param messages - Message history
+     * @param callback - Streaming callback function
+     * @param options - Chat options
+     */
+    chatStream(messages: ChatMessage[], callback: OpenAIStreamCallback, options?: {
+        temperature?: number;
+        maxTokens?: number;
+        modelType?: OpenAIModelType;
+        topP?: number;
+        frequencyPenalty?: number;
+        presencePenalty?: number;
+        stop?: string[];
+    }): Promise<void>;
+    /**
+     * Generate images using DALL-E
+     * @param prompt - Image generation prompt
+     * @param options - Image generation options
+     * @returns Generated image URLs
+     */
+    generateImage(prompt: string, options?: {
+        modelType?: OpenAIModelType;
+        n?: number;
+        size?: '256x256' | '512x512' | '1024x1024' | '1792x1024' | '1024x1792';
+        quality?: 'standard' | 'hd';
+        style?: 'vivid' | 'natural';
+        responseFormat?: 'url' | 'b64_json';
+    }): Promise<string[]>;
+    /**
+     * Create text embeddings
+     * @param input - Text or array of texts to embed
+     * @param options - Embedding options
+     * @returns Embedding vectors
+     */
+    createEmbeddings(input: string | string[], options?: {
+        modelType?: OpenAIModelType;
+        dimensions?: number;
+    }): Promise<number[][]>;
+    /**
+     * Transcribe audio using Whisper
+     * @param audioData - Base64 encoded audio data or file path
+     * @param options - Transcription options
+     * @returns Transcribed text
+     */
+    transcribeAudio(audioData: string | File, options?: {
+        modelType?: OpenAIModelType;
+        language?: string;
+        prompt?: string;
+        responseFormat?: 'json' | 'text' | 'srt' | 'verbose_json' | 'vtt';
+        temperature?: number;
+    }): Promise<string>;
+    /**
+     * Text-to-speech conversion
+     * @param text - Text to convert to speech
+     * @param options - TTS options
+     * @returns Audio data (base64 or blob)
+     */
+    textToSpeech(text: string, options?: {
+        modelType?: OpenAIModelType;
+        voice?: 'alloy' | 'echo' | 'fable' | 'onyx' | 'nova' | 'shimmer';
+        responseFormat?: 'mp3' | 'opus' | 'aac' | 'flac' | 'wav' | 'pcm';
+        speed?: number;
+    }): Promise<ArrayBuffer>;
+    /**
+     * Content moderation
+     * @param input - Text to moderate
+     * @param options - Moderation options
+     * @returns Moderation results
+     */
+    moderateContent(input: string, options?: {
+        modelType?: OpenAIModelType;
+    }): Promise<any>;
+    /**
+     * Switch model
+     * @param modelType - New model type
+     */
+    setModel(modelType: OpenAIModelType): void;
+    /**
+     * Get current model configuration
+     */
+    getCurrentModel(): {
+        name: string;
+        displayName: string;
+        description?: string;
+    };
+    /**
+     * Test connection
+     * @returns Connection test result
+     */
+    testConnection(): Promise<{
+        success: boolean;
+        model: string;
+        response?: string;
+        error?: string;
+    }>;
+    /**
+     * Estimate cost for a request
+     * @param inputTokens - Number of input tokens
+     * @param outputTokens - Number of output tokens
+     * @param modelType - Model type (optional, uses current if not provided)
+     * @returns Cost estimate
+     */
+    estimateCost(inputTokens: number, outputTokens?: number, modelType?: OpenAIModelType): {
+        inputCost: number;
+        outputCost: number;
+        totalCost: number;
+    };
+    private buildOpenAIRequest;
+    private makeRequest;
+    private makeFormDataRequest;
+    private makeStreamRequest;
+    private extractContent;
+    /**
+     * Specialized method for processing OHLCV arrays
+     * @param ohlcvArray - OHLCV data array
+     * @param instructions - Processing instructions (optional, default: "Based on these OHLCV data, predict the next period")
+     * @param count - Number of OHLCV data items to return (optional, default: 1)
+     * @param options - Chat options
+     * @returns Predicted OHLCV array
+     */
+    analyzeOHLCV(ohlcvArray: OHLCV[], instructions?: string, count?: number, options?: OpenAIChatOptions): Promise<OHLCV[]>;
+    /**
+     * Parse AI returned OHLCV response
+     * @private
+     */
+    private parseOHLCVResponse;
+}
+
+export declare const OPENAI_MODELS: Map<OpenAIModelType, OpenAIModel>;
+
+export declare interface OpenAIChatOptions {
+    temperature?: number;
+    maxTokens?: number;
+    stream?: boolean;
+    systemPrompt?: string;
+    modelType?: OpenAIModelType;
+    topP?: number;
+    frequencyPenalty?: number;
+    presencePenalty?: number;
+    stop?: string[];
+}
+
+export declare interface OpenAIConfig {
+    apiKey: string;
+    modelType?: OpenAIModelType;
+    timeout?: number;
+    organization?: string;
+    baseURL?: string;
+}
+
+export declare interface OpenAIModel {
+    name: string;
+    displayName: string;
+    endpoint: string;
+    format: 'openai';
+    description: string;
+    maxTokens?: number;
+    contextLength?: number;
+    capabilities: string[];
+    inputCostPer1KTokens?: number;
+    outputCostPer1KTokens?: number;
+    supportedFeatures?: string[];
+}
+
+export declare enum OpenAIModelType {
+    GPT4 = "gpt-4",
+    GPT4_0314 = "gpt-4-0314",
+    GPT4_0613 = "gpt-4-0613",
+    GPT4_32K = "gpt-4-32k",
+    GPT4_32K_0314 = "gpt-4-32k-0314",
+    GPT4_32K_0613 = "gpt-4-32k-0613",
+    GPT4_TURBO = "gpt-4-turbo",
+    GPT4_TURBO_PREVIEW = "gpt-4-turbo-preview",
+    GPT4_TURBO_2024_04_09 = "gpt-4-turbo-2024-04-09",
+    GPT4_OMNI = "gpt-4o",
+    GPT4_OMNI_2024_05_13 = "gpt-4o-2024-05-13",
+    GPT4_OMNI_MINI = "gpt-4o-mini",
+    GPT4_OMNI_MINI_2024_07_18 = "gpt-4o-mini-2024-07-18",
+    GPT3_5_TURBO = "gpt-3.5-turbo",
+    GPT3_5_TURBO_0125 = "gpt-3.5-turbo-0125",
+    GPT3_5_TURBO_1106 = "gpt-3.5-turbo-1106",
+    GPT3_5_TURBO_INSTRUCT = "gpt-3.5-turbo-instruct",
+    GPT3_5_TURBO_16K = "gpt-3.5-turbo-16k",
+    GPT3_5_TURBO_16K_0613 = "gpt-3.5-turbo-16k-0613",
+    DAVINCI_002 = "davinci-002",
+    BABBAGE_002 = "babbage-002",
+    TEXT_DAVINCI_003 = "text-davinci-003",
+    TEXT_DAVINCI_002 = "text-davinci-002",
+    TEXT_DAVINCI_001 = "text-davinci-001",
+    TEXT_CURIE_001 = "text-curie-001",
+    TEXT_BABBAGE_001 = "text-babbage-001",
+    TEXT_ADA_001 = "text-ada-001",
+    TEXT_EMBEDDING_ADA_002 = "text-embedding-ada-002",
+    TEXT_EMBEDDING_3_SMALL = "text-embedding-3-small",
+    TEXT_EMBEDDING_3_LARGE = "text-embedding-3-large",
+    DALL_E_2 = "dall-e-2",
+    DALL_E_3 = "dall-e-3",
+    WHISPER_1 = "whisper-1",
+    TTS_1 = "tts-1",
+    TTS_1_HD = "tts-1-hd",
+    MODERATION_LATEST = "text-moderation-latest",
+    MODERATION_STABLE = "text-moderation-stable",
+    GPT3_5_TURBO_FINETUNED = "ft:gpt-3.5-turbo-0125:personal:",
+    GPT4_FINETUNED = "ft:gpt-4-0125-preview:personal:",
+    GPT4_VISION_PREVIEW = "gpt-4-vision-preview"
+}
+
+export declare type OpenAIStreamCallback = (chunk: string, isEnd: boolean) => void;
+
+export declare function suggestModel(requirements: {
+    taskType: 'chat' | 'completion' | 'embedding' | 'image' | 'audio';
+    budget?: number;
+    contextLength?: number;
+    features?: string[];
+}): OpenAIModel[];
+
+export { }