@inferencesh/sdk 0.4.4 → 0.4.8

package/dist/agent.js

@@ -1,188 +0,0 @@
-"use strict";
-/**
- * Headless Agent SDK
- *
- * Chat with AI agents without UI dependencies.
- */
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Agent = void 0;
-const stream_1 = require("./stream");
-const eventsource_1 = require("eventsource");
-const types_1 = require("./types");
-// =============================================================================
-// Agent Class
-// =============================================================================
-class Agent {
-    constructor(config, options) {
-        this.chatId = null;
-        // Unified stream for both Chat and ChatMessage events (uses TypedEvents)
-        this.stream = null;
-        this.apiKey = config.apiKey;
-        this.baseUrl = config.baseUrl || 'https://api.inference.sh';
-        this.options = options;
-    }
-    /** Get current chat ID */
-    get currentChatId() {
-        return this.chatId;
-    }
-    /** Send a message to the agent */
-    async sendMessage(text, options = {}) {
-        const isAdHoc = 'core_app_ref' in this.options;
-        // Upload files if provided
-        let imageUri;
-        let fileUris;
-        if (options.files && options.files.length > 0) {
-            const uploadedFiles = await Promise.all(options.files.map(f => this.uploadFile(f)));
-            // Separate images from other files
-            const images = uploadedFiles.filter(f => f.content_type?.startsWith('image/'));
-            const others = uploadedFiles.filter(f => !f.content_type?.startsWith('image/'));
-            if (images.length > 0)
-                imageUri = images[0].uri;
-            if (others.length > 0)
-                fileUris = others.map(f => f.uri);
-        }
-        // Both template and ad-hoc use /agents/run
-        const input = { text, image: imageUri, files: fileUris, role: 'user', context: [], system_prompt: '', context_size: 0 };
-        const body = isAdHoc
-            ? { chat_id: this.chatId, agent_config: this.options, input }
-            : { chat_id: this.chatId, agent: this.options.agent, input };
-        const response = await this.request('post', '/agents/run', { data: body });
-        // Update chat ID if new
-        if (!this.chatId && response.assistant_message.chat_id) {
-            this.chatId = response.assistant_message.chat_id;
-            this.startStreaming(options);
-        }
-        return response.assistant_message;
-    }
-    /** Get chat by ID */
-    async getChat(chatId) {
-        const id = chatId || this.chatId;
-        if (!id)
-            return null;
-        return this.request('get', `/chats/${id}`);
-    }
-    /** Stop the current chat generation */
-    async stopChat() {
-        if (!this.chatId)
-            return;
-        await this.request('post', `/chats/${this.chatId}/stop`);
-    }
-    /**
-     * Submit a tool result
-     * @param toolInvocationId - The tool invocation ID
-     * @param resultOrAction - Either a raw result string, or an object with action and optional form_data (will be JSON-serialized)
-     */
-    async submitToolResult(toolInvocationId, resultOrAction) {
-        // Serialize widget actions to JSON string
-        const result = typeof resultOrAction === 'string'
-            ? resultOrAction
-            : JSON.stringify(resultOrAction);
-        await this.request('post', `/tools/${toolInvocationId}`, { data: { result } });
-    }
-    /** Stop streaming and cleanup */
-    disconnect() {
-        this.stream?.stop();
-        this.stream = null;
-    }
-    /** Reset the agent (start fresh chat) */
-    reset() {
-        this.disconnect();
-        this.chatId = null;
-    }
-    /** Upload a file and return the file object */
-    async uploadFile(data) {
-        // Create file record
-        let contentType = 'application/octet-stream';
-        let size;
-        if (data instanceof Blob) {
-            contentType = data.type || 'application/octet-stream';
-            size = data.size;
-        }
-        const files = await this.request('post', '/files', { data: { files: [{ uri: '', content_type: contentType, size }] } });
-        const file = files[0];
-        if (!file.upload_url)
-            throw new Error('No upload URL');
-        // Convert to blob if needed
-        let blob;
-        if (data instanceof Blob) {
-            blob = data;
-        }
-        else if (data.startsWith('data:')) {
-            const matches = data.match(/^data:([^;]+);base64,(.+)$/);
-            if (!matches)
-                throw new Error('Invalid data URI');
-            const bytes = Uint8Array.from(atob(matches[2]), c => c.charCodeAt(0));
-            blob = new Blob([bytes], { type: matches[1] });
-        }
-        else {
-            const bytes = Uint8Array.from(atob(data), c => c.charCodeAt(0));
-            blob = new Blob([bytes], { type: contentType });
-        }
-        // Upload to signed URL
-        const uploadResp = await fetch(file.upload_url, {
-            method: 'PUT',
-            body: blob,
-            headers: { 'Content-Type': blob.type },
-        });
-        if (!uploadResp.ok)
-            throw new Error('Upload failed');
-        return { uri: file.uri, content_type: file.content_type };
-    }
-    // =============================================================================
-    // Private Methods
-    // =============================================================================
-    startStreaming(options) {
-        if (!this.chatId)
-            return;
-        // Unified stream with TypedEvents (single SSE connection for both Chat and ChatMessage)
-        this.stream = new stream_1.StreamManager({
-            createEventSource: async () => this.createEventSource(`/chats/${this.chatId}/stream`),
-            autoReconnect: true,
-        });
-        // Listen for Chat object updates (status changes)
-        this.stream.addEventListener('chats', (chat) => {
-            options.onChat?.(chat);
-        });
-        // Listen for ChatMessage updates
-        this.stream.addEventListener('chat_messages', (message) => {
-            options.onMessage?.(message);
-            // Check for client tool invocations
-            if (message.tool_invocations && options.onToolCall) {
-                for (const inv of message.tool_invocations) {
-                    if (inv.type === types_1.ToolTypeClient && inv.status === types_1.ToolInvocationStatusAwaitingInput) {
-                        options.onToolCall({
-                            id: inv.id,
-                            name: inv.function?.name || '',
-                            args: inv.function?.arguments || {},
-                        });
-                    }
-                }
-            }
-        });
-        this.stream.connect();
-    }
-    createEventSource(endpoint) {
-        return new eventsource_1.EventSource(`${this.baseUrl}${endpoint}`, {
-            fetch: (input, init) => fetch(input, {
-                ...init,
-                headers: { ...init.headers, Authorization: `Bearer ${this.apiKey}` },
-            }),
-        });
-    }
-    async request(method, endpoint, options = {}) {
-        const response = await fetch(`${this.baseUrl}${endpoint}`, {
-            method: method.toUpperCase(),
-            headers: {
-                'Content-Type': 'application/json',
-                Authorization: `Bearer ${this.apiKey}`,
-            },
-            body: options.data ? JSON.stringify(options.data) : undefined,
-        });
-        const json = await response.json();
-        if (!response.ok || !json.success) {
-            throw new Error(json.error?.message || 'Request failed');
-        }
-        return json.data;
-    }
-}
-exports.Agent = Agent;