@push.rocks/smartai 0.2.0 → 0.3.2

package/dist_ts/00_commitinfo_data.js

@@ -3,7 +3,7 @@
  */
 export const commitinfo = {
     name: '@push.rocks/smartai',
-    version: '0.2.0',
+    version: '0.3.2',
     description: 'A TypeScript library for integrating and interacting with multiple AI models, offering capabilities for chat and potentially audio responses.'
 };
 //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiMDBfY29tbWl0aW5mb19kYXRhLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vdHMvMDBfY29tbWl0aW5mb19kYXRhLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBOztHQUVHO0FBQ0gsTUFBTSxDQUFDLE1BQU0sVUFBVSxHQUFHO0lBQ3hCLElBQUksRUFBRSxxQkFBcUI7SUFDM0IsT0FBTyxFQUFFLE9BQU87SUFDaEIsV0FBVyxFQUFFLCtJQUErSTtDQUM3SixDQUFBIn0=

package/dist_ts/provider.xai.d.ts

@@ -0,0 +1,43 @@
+import * as plugins from './plugins.js';
+import { MultiModalModel } from './abstract.classes.multimodal.js';
+export interface IXAIProviderOptions {
+    xaiToken: string;
+}
+export declare class XAIProvider extends MultiModalModel {
+    private options;
+    openAiApiClient: plugins.openai.default;
+    smartpdfInstance: plugins.smartpdf.SmartPdf;
+    constructor(optionsArg: IXAIProviderOptions);
+    start(): Promise<void>;
+    stop(): Promise<void>;
+    chatStream(input: ReadableStream<Uint8Array>): Promise<ReadableStream<string>>;
+    chat(optionsArg: {
+        systemMessage: string;
+        userMessage: string;
+        messageHistory: {
+            role: string;
+            content: string;
+        }[];
+    }): Promise<{
+        role: 'assistant';
+        message: string;
+    }>;
+    audio(optionsArg: {
+        message: string;
+    }): Promise<NodeJS.ReadableStream>;
+    vision(optionsArg: {
+        image: Buffer;
+        prompt: string;
+    }): Promise<string>;
+    document(optionsArg: {
+        systemMessage: string;
+        userMessage: string;
+        pdfDocuments: Uint8Array[];
+        messageHistory: {
+            role: string;
+            content: string;
+        }[];
+    }): Promise<{
+        message: any;
+    }>;
+}

package/dist_ts/provider.xai.js

@@ -0,0 +1,141 @@
+import * as plugins from './plugins.js';
+import * as paths from './paths.js';
+import { MultiModalModel } from './abstract.classes.multimodal.js';
+export class XAIProvider extends MultiModalModel {
+    constructor(optionsArg) {
+        super();
+        this.options = optionsArg;
+    }
+    async start() {
+        this.openAiApiClient = new plugins.openai.default({
+            apiKey: this.options.xaiToken,
+            baseURL: 'https://api.x.ai/v1',
+        });
+        this.smartpdfInstance = new plugins.smartpdf.SmartPdf();
+    }
+    async stop() { }
+    async chatStream(input) {
+        // Create a TextDecoder to handle incoming chunks
+        const decoder = new TextDecoder();
+        let buffer = '';
+        let currentMessage = null;
+        // Create a TransformStream to process the input
+        const transform = new TransformStream({
+            async transform(chunk, controller) {
+                buffer += decoder.decode(chunk, { stream: true });
+                // Try to parse complete JSON messages from the buffer
+                while (true) {
+                    const newlineIndex = buffer.indexOf('\n');
+                    if (newlineIndex === -1)
+                        break;
+                    const line = buffer.slice(0, newlineIndex);
+                    buffer = buffer.slice(newlineIndex + 1);
+                    if (line.trim()) {
+                        try {
+                            const message = JSON.parse(line);
+                            currentMessage = {
+                                role: message.role || 'user',
+                                content: message.content || '',
+                            };
+                        }
+                        catch (e) {
+                            console.error('Failed to parse message:', e);
+                        }
+                    }
+                }
+                // If we have a complete message, send it to X.AI
+                if (currentMessage) {
+                    const stream = await this.openAiApiClient.chat.completions.create({
+                        model: 'grok-2-latest',
+                        messages: [{ role: currentMessage.role, content: currentMessage.content }],
+                        stream: true,
+                    });
+                    // Process each chunk from X.AI
+                    for await (const chunk of stream) {
+                        const content = chunk.choices[0]?.delta?.content;
+                        if (content) {
+                            controller.enqueue(content);
+                        }
+                    }
+                    currentMessage = null;
+                }
+            },
+            flush(controller) {
+                if (buffer) {
+                    try {
+                        const message = JSON.parse(buffer);
+                        controller.enqueue(message.content || '');
+                    }
+                    catch (e) {
+                        console.error('Failed to parse remaining buffer:', e);
+                    }
+                }
+            }
+        });
+        // Connect the input to our transform stream
+        return input.pipeThrough(transform);
+    }
+    async chat(optionsArg) {
+        // Prepare messages array with system message, history, and user message
+        const messages = [
+            { role: 'system', content: optionsArg.systemMessage },
+            ...optionsArg.messageHistory.map(msg => ({
+                role: msg.role,
+                content: msg.content
+            })),
+            { role: 'user', content: optionsArg.userMessage }
+        ];
+        // Call X.AI's chat completion API
+        const completion = await this.openAiApiClient.chat.completions.create({
+            model: 'grok-2-latest',
+            messages: messages,
+            stream: false,
+        });
+        // Return the assistant's response
+        return {
+            role: 'assistant',
+            message: completion.choices[0]?.message?.content || ''
+        };
+    }
+    async audio(optionsArg) {
+        throw new Error('Audio generation is not supported by X.AI');
+    }
+    async vision(optionsArg) {
+        throw new Error('Vision tasks are not supported by X.AI');
+    }
+    async document(optionsArg) {
+        // First convert PDF documents to images
+        let pdfDocumentImageBytesArray = [];
+        for (const pdfDocument of optionsArg.pdfDocuments) {
+            const documentImageArray = await this.smartpdfInstance.convertPDFToPngBytes(pdfDocument);
+            pdfDocumentImageBytesArray = pdfDocumentImageBytesArray.concat(documentImageArray);
+        }
+        // Convert images to base64 for inclusion in the message
+        const imageBase64Array = pdfDocumentImageBytesArray.map(bytes => Buffer.from(bytes).toString('base64'));
+        // Combine document images into the user message
+        const enhancedUserMessage = `
+      ${optionsArg.userMessage}
+      
+      Document contents (as images):
+      ${imageBase64Array.map((img, i) => `Image ${i + 1}: <image data>`).join('\n')}
+    `;
+        // Use chat completion to analyze the documents
+        const messages = [
+            { role: 'system', content: optionsArg.systemMessage },
+            ...optionsArg.messageHistory.map(msg => ({
+                role: msg.role,
+                content: msg.content
+            })),
+            { role: 'user', content: enhancedUserMessage }
+        ];
+        const completion = await this.openAiApiClient.chat.completions.create({
+            model: 'grok-2-latest',
+            messages: messages,
+            stream: false,
+        });
+        return {
+            message: completion.choices[0]?.message?.content || ''
+        };
+    }
+}
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoicHJvdmlkZXIueGFpLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vdHMvcHJvdmlkZXIueGFpLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBLE9BQU8sS0FBSyxPQUFPLE1BQU0sY0FBYyxDQUFDO0FBQ3hDLE9BQU8sS0FBSyxLQUFLLE1BQU0sWUFBWSxDQUFDO0FBQ3BDLE9BQU8sRUFBRSxlQUFlLEVBQUUsTUFBTSxrQ0FBa0MsQ0FBQztBQVFuRSxNQUFNLE9BQU8sV0FBWSxTQUFRLGVBQWU7SUFLOUMsWUFBWSxVQUErQjtRQUN6QyxLQUFLLEVBQUUsQ0FBQztRQUNSLElBQUksQ0FBQyxPQUFPLEdBQUcsVUFBVSxDQUFDO0lBQzVCLENBQUM7SUFFTSxLQUFLLENBQUMsS0FBSztRQUNoQixJQUFJLENBQUMsZUFBZSxHQUFHLElBQUksT0FBTyxDQUFDLE1BQU0sQ0FBQyxPQUFPLENBQUM7WUFDaEQsTUFBTSxFQUFFLElBQUksQ0FBQyxPQUFPLENBQUMsUUFBUTtZQUM3QixPQUFPLEVBQUUscUJBQXFCO1NBQy9CLENBQUMsQ0FBQztRQUNILElBQUksQ0FBQyxnQkFBZ0IsR0FBRyxJQUFJLE9BQU8sQ0FBQyxRQUFRLENBQUMsUUFBUSxFQUFFLENBQUM7SUFDMUQsQ0FBQztJQUVNLEtBQUssQ0FBQyxJQUFJLEtBQUksQ0FBQztJQUVmLEtBQUssQ0FBQyxVQUFVLENBQUMsS0FBaUM7UUFDdkQsaURBQWlEO1FBQ2pELE1BQU0sT0FBTyxHQUFHLElBQUksV0FBVyxFQUFFLENBQUM7UUFDbEMsSUFBSSxNQUFNLEdBQUcsRUFBRSxDQUFDO1FBQ2hCLElBQUksY0FBYyxHQUE4QyxJQUFJLENBQUM7UUFFckUsZ0RBQWdEO1FBQ2hELE1BQU0sU0FBUyxHQUFHLElBQUksZUFBZSxDQUFxQjtZQUN4RCxLQUFLLENBQUMsU0FBUyxDQUFDLEtBQUssRUFBRSxVQUFVO2dCQUMvQixNQUFNLElBQUksT0FBTyxDQUFDLE1BQU0sQ0FBQyxLQUFLLEVBQUUsRUFBRSxNQUFNLEVBQUUsSUFBSSxFQUFFLENBQUMsQ0FBQztnQkFFbEQsc0RBQXNEO2dCQUN0RCxPQUFPLElBQUksRUFBRSxDQUFDO29CQUNaLE1BQU0sWUFBWSxHQUFHLE1BQU0sQ0FBQyxPQUFPLENBQUMsSUFBSSxDQUFDLENBQUM7b0JBQzFDLElBQUksWUFBWSxLQUFLLENBQUMsQ0FBQzt3QkFBRSxNQUFNO29CQUUvQixNQUFNLElBQUksR0FBRyxNQUFNLENBQUMsS0FBSyxDQUFDLENBQUMsRUFBRSxZQUFZLENBQUMsQ0FBQztvQkFDM0MsTUFBTSxHQUFHLE1BQU0sQ0FBQyxLQUFLLENBQUMsWUFBWSxHQUFHLENBQUMsQ0FBQyxDQUFDO29CQUV4QyxJQUFJLElBQUksQ0FBQyxJQUFJLEVBQUUsRUFBRSxDQUFDO3dCQUNoQixJQUFJLENBQUM7NEJBQ0gsTUFBTSxPQUFPLEdBQUcsSUFBSSxDQUFDLEtBQUssQ0FBQyxJQUFJLENBQUMsQ0FBQzs0QkFDakMsY0FBYyxHQUFHO2dDQUNmLElBQUksRUFBRSxPQUFPLENBQUMsSUFBSSxJQUFJLE1BQU07Z0NBQzVCLE9BQU8sRUFBRSxPQUFPLENBQUMsT0FBTyxJQUFJLEVBQUU7NkJBQy9CLENBQUM7d0JBQ0osQ0FBQzt3QkFBQyxPQUFPLENBQUMsRUFBRSxDQUFDOzRCQUNYLE9BQU8sQ0FBQyxLQUFLLENBQUMsMEJBQTBCLEVBQUUsQ0FBQyxDQUFDLENBQUM7d0JBQy9DLENBQUM7b0JBQ0gsQ0FBQztnQkFDSCxDQUFDO2dCQUVELGlEQUFpRDtnQkFDakQsSUFBSSxjQUFjLEVBQUUsQ0FBQztvQkFDbkIsTUFBTSxNQUFNLEdBQUcsTUFBTSxJQUFJLENBQUMsZUFBZSxDQUFDLElBQUksQ0FBQyxXQUFXLENBQUMsTUFBTSxDQUFDO3dCQUNoRSxLQUFLLEVBQUUsZUFBZTt3QkFDdEIsUUFBUSxFQUFFLENBQUMsRUFBRSxJQUFJLEVBQUUsY0FBYyxDQUFDLElBQUksRUFBRSxPQUFPLEVBQUUsY0FBYyxDQUFDLE9BQU8sRUFBRSxDQUFDO3dCQUMxRSxNQUFNLEVBQUUsSUFBSTtxQkFDYixDQUFDLENBQUM7b0JBRUgsK0JBQStCO29CQUMvQixJQUFJLEtBQUssRUFBRSxNQUFNLEtBQUssSUFBSSxNQUFNLEVBQUUsQ0FBQzt3QkFDakMsTUFBTSxPQUFPLEdBQUcsS0FBSyxDQUFDLE9BQU8sQ0FBQyxDQUFDLENBQUMsRUFBRSxLQUFLLEVBQUUsT0FBTyxDQUFDO3dCQUNqRCxJQUFJLE9BQU8sRUFBRSxDQUFDOzRCQUNaLFVBQVUsQ0FBQyxPQUFPLENBQUMsT0FBTyxDQUFDLENBQUM7d0JBQzlCLENBQUM7b0JBQ0gsQ0FBQztvQkFFRCxjQUFjLEdBQUcsSUFBSSxDQUFDO2dCQUN4QixDQUFDO1lBQ0gsQ0FBQztZQUVELEtBQUssQ0FBQyxVQUFVO2dCQUNkLElBQUksTUFBTSxFQUFFLENBQUM7b0JBQ1gsSUFBSSxDQUFDO3dCQUNILE1BQU0sT0FBTyxHQUFHLElBQUksQ0FBQyxLQUFLLENBQUMsTUFBTSxDQUFDLENBQUM7d0JBQ25DLFVBQVUsQ0FBQyxPQUFPLENBQUMsT0FBTyxDQUFDLE9BQU8sSUFBSSxFQUFFLENBQUMsQ0FBQztvQkFDNUMsQ0FBQztvQkFBQyxPQUFPLENBQUMsRUFBRSxDQUFDO3dCQUNYLE9BQU8sQ0FBQyxLQUFLLENBQUMsbUNBQW1DLEVBQUUsQ0FBQyxDQUFDLENBQUM7b0JBQ3hELENBQUM7Z0JBQ0gsQ0FBQztZQUNILENBQUM7U0FDRixDQUFDLENBQUM7UUFFSCw0Q0FBNEM7UUFDNUMsT0FBTyxLQUFLLENBQUMsV0FBVyxDQUFDLFNBQVMsQ0FBQyxDQUFDO0lBQ3RDLENBQUM7SUFFTSxLQUFLLENBQUMsSUFBSSxDQUFDLFVBSWpCO1FBQ0Msd0VBQXdFO1FBQ3hFLE1BQU0sUUFBUSxHQUFpQztZQUM3QyxFQUFFLElBQUksRUFBRSxRQUFRLEVBQUUsT0FBTyxFQUFFLFVBQVUsQ0FBQyxhQUFhLEVBQUU7WUFDckQsR0FBRyxVQUFVLENBQUMsY0FBYyxDQUFDLEdBQUcsQ0FBQyxHQUFHLENBQUMsRUFBRSxDQUFDLENBQUM7Z0JBQ3ZDLElBQUksRUFBRSxHQUFHLENBQUMsSUFBdUM7Z0JBQ2pELE9BQU8sRUFBRSxHQUFHLENBQUMsT0FBTzthQUNyQixDQUFDLENBQUM7WUFDSCxFQUFFLElBQUksRUFBRSxNQUFNLEVBQUUsT0FBTyxFQUFFLFVBQVUsQ0FBQyxXQUFXLEVBQUU7U0FDbEQsQ0FBQztRQUVGLGtDQUFrQztRQUNsQyxNQUFNLFVBQVUsR0FBRyxNQUFNLElBQUksQ0FBQyxlQUFlLENBQUMsSUFBSSxDQUFDLFdBQVcsQ0FBQyxNQUFNLENBQUM7WUFDcEUsS0FBSyxFQUFFLGVBQWU7WUFDdEIsUUFBUSxFQUFFLFFBQVE7WUFDbEIsTUFBTSxFQUFFLEtBQUs7U0FDZCxDQUFDLENBQUM7UUFFSCxrQ0FBa0M7UUFDbEMsT0FBTztZQUNMLElBQUksRUFBRSxXQUFXO1lBQ2pCLE9BQU8sRUFBRSxVQUFVLENBQUMsT0FBTyxDQUFDLENBQUMsQ0FBQyxFQUFFLE9BQU8sRUFBRSxPQUFPLElBQUksRUFBRTtTQUN2RCxDQUFDO0lBQ0osQ0FBQztJQUVNLEtBQUssQ0FBQyxLQUFLLENBQUMsVUFBK0I7UUFDaEQsTUFBTSxJQUFJLEtBQUssQ0FBQywyQ0FBMkMsQ0FBQyxDQUFDO0lBQy9ELENBQUM7SUFFTSxLQUFLLENBQUMsTUFBTSxDQUFDLFVBQTZDO1FBQy9ELE1BQU0sSUFBSSxLQUFLLENBQUMsd0NBQXdDLENBQUMsQ0FBQztJQUM1RCxDQUFDO0lBRU0sS0FBSyxDQUFDLFFBQVEsQ0FBQyxVQUtyQjtRQUNDLHdDQUF3QztRQUN4QyxJQUFJLDBCQUEwQixHQUFpQixFQUFFLENBQUM7UUFFbEQsS0FBSyxNQUFNLFdBQVcsSUFBSSxVQUFVLENBQUMsWUFBWSxFQUFFLENBQUM7WUFDbEQsTUFBTSxrQkFBa0IsR0FBRyxNQUFNLElBQUksQ0FBQyxnQkFBZ0IsQ0FBQyxvQkFBb0IsQ0FBQyxXQUFXLENBQUMsQ0FBQztZQUN6RiwwQkFBMEIsR0FBRywwQkFBMEIsQ0FBQyxNQUFNLENBQUMsa0JBQWtCLENBQUMsQ0FBQztRQUNyRixDQUFDO1FBRUQsd0RBQXdEO1FBQ3hELE1BQU0sZ0JBQWdCLEdBQUcsMEJBQTBCLENBQUMsR0FBRyxDQUFDLEtBQUssQ0FBQyxFQUFFLENBQzlELE1BQU0sQ0FBQyxJQUFJLENBQUMsS0FBSyxDQUFDLENBQUMsUUFBUSxDQUFDLFFBQVEsQ0FBQyxDQUN0QyxDQUFDO1FBRUYsZ0RBQWdEO1FBQ2hELE1BQU0sbUJBQW1CLEdBQUc7UUFDeEIsVUFBVSxDQUFDLFdBQVc7OztRQUd0QixnQkFBZ0IsQ0FBQyxHQUFHLENBQUMsQ0FBQyxHQUFHLEVBQUUsQ0FBQyxFQUFFLEVBQUUsQ0FBQyxTQUFTLENBQUMsR0FBRyxDQUFDLGdCQUFnQixDQUFDLENBQUMsSUFBSSxDQUFDLElBQUksQ0FBQztLQUM5RSxDQUFDO1FBRUYsK0NBQStDO1FBQy9DLE1BQU0sUUFBUSxHQUFpQztZQUM3QyxFQUFFLElBQUksRUFBRSxRQUFRLEVBQUUsT0FBTyxFQUFFLFVBQVUsQ0FBQyxhQUFhLEVBQUU7WUFDckQsR0FBRyxVQUFVLENBQUMsY0FBYyxDQUFDLEdBQUcsQ0FBQyxHQUFHLENBQUMsRUFBRSxDQUFDLENBQUM7Z0JBQ3ZDLElBQUksRUFBRSxHQUFHLENBQUMsSUFBdUM7Z0JBQ2pELE9BQU8sRUFBRSxHQUFHLENBQUMsT0FBTzthQUNyQixDQUFDLENBQUM7WUFDSCxFQUFFLElBQUksRUFBRSxNQUFNLEVBQUUsT0FBTyxFQUFFLG1CQUFtQixFQUFFO1NBQy9DLENBQUM7UUFFRixNQUFNLFVBQVUsR0FBRyxNQUFNLElBQUksQ0FBQyxlQUFlLENBQUMsSUFBSSxDQUFDLFdBQVcsQ0FBQyxNQUFNLENBQUM7WUFDcEUsS0FBSyxFQUFFLGVBQWU7WUFDdEIsUUFBUSxFQUFFLFFBQVE7WUFDbEIsTUFBTSxFQUFFLEtBQUs7U0FDZCxDQUFDLENBQUM7UUFFSCxPQUFPO1lBQ0wsT0FBTyxFQUFFLFVBQVUsQ0FBQyxPQUFPLENBQUMsQ0FBQyxDQUFDLEVBQUUsT0FBTyxFQUFFLE9BQU8sSUFBSSxFQUFFO1NBQ3ZELENBQUM7SUFDSixDQUFDO0NBQ0YifQ==

package/license

@@ -0,0 +1,19 @@
+Copyright (c) 2024 Task Venture Capital GmbH (hello@task.vc)
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@push.rocks/smartai",
-  "version": "0.2.0",
+  "version": "0.3.2",
   "private": false,
   "description": "A TypeScript library for integrating and interacting with multiple AI models, offering capabilities for chat and potentially audio responses.",
   "main": "dist_ts/index.js",

package/readme.md

@@ -1,80 +1,125 @@
 # @push.rocks/smartai
 
-Provides a standardized interface for integrating and conversing with multiple AI models, supporting operations like chat, streaming interactions, and audio responses.
-
-## Install
-
-To add @push.rocks/smartai to your project, run the following command in your terminal:
+[![npm version](https://badge.fury.io/js/%40push.rocks%2Fsmartai.svg)](https://www.npmjs.com/package/@push.rocks/smartai)
+
+SmartAi is a comprehensive TypeScript library that provides a standardized interface for integrating and interacting with multiple AI models. It supports a range of operations from synchronous and streaming chat to audio generation, document processing, and vision tasks.
+
+## Table of Contents
+
+- [Features](#features)
+- [Installation](#installation)
+- [Supported AI Providers](#supported-ai-providers)
+- [Quick Start](#quick-start)
+- [Usage Examples](#usage-examples)
+  - [Chat Interactions](#chat-interactions)
+  - [Streaming Chat](#streaming-chat)
+  - [Audio Generation](#audio-generation)
+  - [Document Processing](#document-processing)
+  - [Vision Processing](#vision-processing)
+- [Error Handling](#error-handling)
+- [Development](#development)
+  - [Running Tests](#running-tests)
+  - [Building the Project](#building-the-project)
+- [Contributing](#contributing)
+- [License](#license)
+- [Legal Information](#legal-information)
+
+## Features
+
+- **Unified API:** Seamlessly integrate multiple AI providers with a consistent interface.
+- **Chat & Streaming:** Support for both synchronous and real-time streaming chat interactions.
+- **Audio & Vision:** Generate audio responses and perform detailed image analysis.
+- **Document Processing:** Analyze PDFs and other documents using vision models.
+- **Extensible:** Easily extend the library to support additional AI providers.
+
+## Installation
+
+To install SmartAi, run the following command:
 
 ```bash
 npm install @push.rocks/smartai
 ```
 
-This command installs the package and adds it to your project's dependencies.
+This will add the package to your project’s dependencies.
 
 ## Supported AI Providers
 
-@push.rocks/smartai supports multiple AI providers, each with its own unique capabilities:
+SmartAi supports multiple AI providers. Configure each provider with its corresponding token or settings:
 
 ### OpenAI
-- Models: GPT-4, GPT-3.5-turbo, GPT-4-vision-preview
-- Features: Chat, Streaming, Audio Generation, Vision, Document Processing
-- Configuration:
+
+- **Models:** GPT-4, GPT-3.5-turbo, GPT-4-vision-preview
+- **Features:** Chat, Streaming, Audio Generation, Vision, Document Processing
+- **Configuration Example:**
+  
   ```typescript
   openaiToken: 'your-openai-token'
   ```
 
+### X.AI
+
+- **Models:** Grok-2-latest
+- **Features:** Chat, Streaming, Document Processing
+- **Configuration Example:**
+  
+  ```typescript
+  xaiToken: 'your-xai-token'
+  ```
+
 ### Anthropic
-- Models: Claude-3-opus-20240229
-- Features: Chat, Streaming, Vision, Document Processing
-- Configuration:
+
+- **Models:** Claude-3-opus-20240229
+- **Features:** Chat, Streaming, Vision, Document Processing
+- **Configuration Example:**
+  
   ```typescript
   anthropicToken: 'your-anthropic-token'
   ```
 
 ### Perplexity
-- Models: Mixtral-8x7b-instruct
-- Features: Chat, Streaming
-- Configuration:
+
+- **Models:** Mixtral-8x7b-instruct
+- **Features:** Chat, Streaming
+- **Configuration Example:**
+  
   ```typescript
   perplexityToken: 'your-perplexity-token'
   ```
 
 ### Groq
-- Models: Llama-3.3-70b-versatile
-- Features: Chat, Streaming
-- Configuration:
+
+- **Models:** Llama-3.3-70b-versatile
+- **Features:** Chat, Streaming
+- **Configuration Example:**
+  
   ```typescript
   groqToken: 'your-groq-token'
   ```
 
 ### Ollama
-- Models: Configurable (default: llama2, llava for vision/documents)
-- Features: Chat, Streaming, Vision, Document Processing
-- Configuration:
+
+- **Models:** Configurable (default: llama2; use llava for vision/document tasks)
+- **Features:** Chat, Streaming, Vision, Document Processing
+- **Configuration Example:**
+  
   ```typescript
-  baseUrl: 'http://localhost:11434' // Optional
-  model: 'llama2' // Optional
-  visionModel: 'llava' // Optional, for vision and document tasks
+  ollama: {
+    baseUrl: 'http://localhost:11434', // Optional
+    model: 'llama2',                  // Optional
+    visionModel: 'llava'               // Optional for vision and document tasks
+  }
   ```
 
-## Usage
-
-The `@push.rocks/smartai` package is a comprehensive solution for integrating and interacting with various AI models, designed to support operations ranging from chat interactions to audio responses. This documentation will guide you through the process of utilizing `@push.rocks/smartai` in your applications.
-
-### Getting Started
-
-Before you begin, ensure you have installed the package as described in the **Install** section above. Once installed, you can start integrating AI functionalities into your application.
-
-### Initializing SmartAi
+## Quick Start
 
-The first step is to import and initialize the `SmartAi` class with appropriate options for the AI services you plan to use:
+Initialize SmartAi with the provider configurations you plan to use:
 
 ```typescript
 import { SmartAi } from '@push.rocks/smartai';
 
 const smartAi = new SmartAi({
   openaiToken: 'your-openai-token',
+  xaiToken: 'your-xai-token',
   anthropicToken: 'your-anthropic-token',
   perplexityToken: 'your-perplexity-token',
   groqToken: 'your-groq-token',
@@ -87,35 +132,34 @@ const smartAi = new SmartAi({
 await smartAi.start();
 ```
 
-### Chat Interactions
+## Usage Examples
 
-#### Synchronous Chat
+### Chat Interactions
 
-For simple question-answer interactions:
+**Synchronous Chat:**
 
 ```typescript
 const response = await smartAi.openaiProvider.chat({
   systemMessage: 'You are a helpful assistant.',
   userMessage: 'What is the capital of France?',
-  messageHistory: [] // Previous messages in the conversation
+  messageHistory: [] // Include previous conversation messages if applicable
 });
 
 console.log(response.message);
 ```
 
-#### Streaming Chat
+### Streaming Chat
 
-For real-time, streaming interactions:
+**Real-Time Streaming:**
 
 ```typescript
 const textEncoder = new TextEncoder();
 const textDecoder = new TextDecoder();
 
-// Create input and output streams
+// Create a transform stream for sending and receiving data
 const { writable, readable } = new TransformStream();
 const writer = writable.getWriter();
 
-// Send a message
 const message = {
   role: 'user',
   content: 'Tell me a story about a brave knight'
@@ -123,91 +167,92 @@ const message = {
 
 writer.write(textEncoder.encode(JSON.stringify(message) + '\n'));
 
-// Process the response stream
+// Start streaming the response
 const stream = await smartAi.openaiProvider.chatStream(readable);
 const reader = stream.getReader();
 
 while (true) {
   const { done, value } = await reader.read();
   if (done) break;
-  console.log('AI:', value); // Process each chunk of the response
+  console.log('AI:', value);
 }
 ```
 
 ### Audio Generation
 
-For providers that support audio generation (currently OpenAI):
+Generate audio (supported by providers like OpenAI):
 
 ```typescript
 const audioStream = await smartAi.openaiProvider.audio({
   message: 'Hello, this is a test of text-to-speech'
 });
 
-// Handle the audio stream (e.g., save to file or play)
+// Process the audio stream, for example, play it or save to a file.
 ```
 
 ### Document Processing
 
-For providers that support document processing (OpenAI, Ollama, and Anthropic):
+Analyze and extract key information from documents:
 
 ```typescript
-// Using OpenAI
-const result = await smartAi.openaiProvider.document({
+// Example using OpenAI
+const documentResult = await smartAi.openaiProvider.document({
   systemMessage: 'Classify the document type',
   userMessage: 'What type of document is this?',
   messageHistory: [],
-  pdfDocuments: [pdfBuffer] // Uint8Array of PDF content
+  pdfDocuments: [pdfBuffer] // Uint8Array containing the PDF content
 });
+```
 
-// Using Ollama with llava
-const analysis = await smartAi.ollamaProvider.document({
+Other providers (e.g., Ollama and Anthropic) follow a similar pattern:
+
+```typescript
+// Using Ollama for document processing
+const ollamaResult = await smartAi.ollamaProvider.document({
   systemMessage: 'You are a document analysis assistant',
-  userMessage: 'Extract the key information from this document',
+  userMessage: 'Extract key information from this document',
   messageHistory: [],
-  pdfDocuments: [pdfBuffer] // Uint8Array of PDF content
+  pdfDocuments: [pdfBuffer]
 });
+```
 
-// Using Anthropic with Claude 3
-const anthropicAnalysis = await smartAi.anthropicProvider.document({
-  systemMessage: 'You are a document analysis assistant',
-  userMessage: 'Please analyze this document and extract key information',
+```typescript
+// Using Anthropic for document processing
+const anthropicResult = await smartAi.anthropicProvider.document({
+  systemMessage: 'Analyze the document',
+  userMessage: 'Please extract the main points',
   messageHistory: [],
-  pdfDocuments: [pdfBuffer] // Uint8Array of PDF content
+  pdfDocuments: [pdfBuffer]
 });
 ```
 
-Both providers will:
-1. Convert PDF documents to images
-2. Process each page using their vision models
-3. Return a comprehensive analysis based on the system message and user query
-
 ### Vision Processing
 
-For providers that support vision tasks (OpenAI, Ollama, and Anthropic):
+Analyze images with vision capabilities:
 
 ```typescript
-// Using OpenAI's GPT-4 Vision
-const description = await smartAi.openaiProvider.vision({
-  image: imageBuffer, // Buffer containing the image data
+// Using OpenAI GPT-4 Vision
+const imageDescription = await smartAi.openaiProvider.vision({
+  image: imageBuffer, // Uint8Array containing image data
   prompt: 'What do you see in this image?'
 });
 
-// Using Ollama's Llava model
-const analysis = await smartAi.ollamaProvider.vision({
+// Using Ollama for vision tasks
+const ollamaImageAnalysis = await smartAi.ollamaProvider.vision({
   image: imageBuffer,
   prompt: 'Analyze this image in detail'
 });
 
-// Using Anthropic's Claude 3
-const anthropicAnalysis = await smartAi.anthropicProvider.vision({
+// Using Anthropic for vision analysis
+const anthropicImageAnalysis = await smartAi.anthropicProvider.vision({
   image: imageBuffer,
-  prompt: 'Please analyze this image and describe what you see'
+  prompt: 'Describe the contents of this image'
 });
 ```
 
 ## Error Handling
 
-All providers implement proper error handling. It's recommended to wrap API calls in try-catch blocks:
+Always wrap API calls in try-catch blocks to manage errors effectively:
 
 ```typescript
 try {
@@ -216,26 +261,71 @@ try {
     userMessage: 'Hello!',
     messageHistory: []
   });
-} catch (error) {
+  console.log(response.message);
+} catch (error: any) {
   console.error('AI provider error:', error.message);
 }
 ```
 
-## License and Legal Information
+## Development
 
-This repository contains open-source code that is licensed under the MIT License. A copy of the MIT License can be found in the [license](license) file within this repository. 
+### Running Tests
 
-**Please note:** The MIT License does not grant permission to use the trade names, trademarks, service marks, or product names of the project, except as required for reasonable and customary use in describing the origin of the work and reproducing the content of the NOTICE file.
+To run the test suite, use the following command:
+
+```bash
+npm run test
+```
+
+Ensure your environment is configured with the appropriate tokens and settings for the providers you are testing.
+
+### Building the Project
+
+Compile the TypeScript code and build the package using:
+
+```bash
+npm run build
+```
+
+This command prepares the library for distribution.
+
+## Contributing
+
+Contributions are welcome! Please follow these steps:
+
+1. Fork the repository.
+2. Create a feature branch:  
+   ```bash
+   git checkout -b feature/my-feature
+   ```
+3. Commit your changes with clear messages:  
+   ```bash
+   git commit -m 'Add new feature'
+   ```
+4. Push your branch to your fork:  
+   ```bash
+   git push origin feature/my-feature
+   ```
+5. Open a Pull Request with a detailed description of your changes.
+
+## License
+
+This project is licensed under the [MIT License](LICENSE).
+
+## Legal Information
 
 ### Trademarks
 
-This project is owned and maintained by Task Venture Capital GmbH. The names and logos associated with Task Venture Capital GmbH and any related products or services are trademarks of Task Venture Capital GmbH and are not included within the scope of the MIT license granted herein. Use of these trademarks must comply with Task Venture Capital GmbH's Trademark Guidelines, and any usage must be approved in writing by Task Venture Capital GmbH.
+This project is owned and maintained by Task Venture Capital GmbH. The names and logos associated with Task Venture Capital GmbH and its related products or services are trademarks of Task Venture Capital GmbH and are not covered by the MIT License. Use of these trademarks must comply with Task Venture Capital GmbH's Trademark Guidelines.
 
 ### Company Information
 
 Task Venture Capital GmbH  
-Registered at District court Bremen HRB 35230 HB, Germany
+Registered at District Court Bremen HRB 35230 HB, Germany  
+Contact: hello@task.vc
+
+By using this repository, you agree to the terms outlined in this section.
 
-For any legal inquiries or if you require further information, please contact us via email at hello@task.vc.
+---
 
-By using this repository, you acknowledge that you have read this section, agree to comply with its terms, and understand that the licensing of the code does not imply endorsement by Task Venture Capital GmbH of any derivative works.
+Happy coding with SmartAi!

package/ts/00_commitinfo_data.ts

@@ -3,6 +3,6 @@
  */
 export const commitinfo = {
   name: '@push.rocks/smartai',
-  version: '0.2.0',
+  version: '0.3.2',
   description: 'A TypeScript library for integrating and interacting with multiple AI models, offering capabilities for chat and potentially audio responses.'
 }

package/ts/provider.xai.ts

@@ -0,0 +1,183 @@
+import * as plugins from './plugins.js';
+import * as paths from './paths.js';
+import { MultiModalModel } from './abstract.classes.multimodal.js';
+import type { ChatOptions, ChatResponse, ChatMessage } from './abstract.classes.multimodal.js';
+import type { ChatCompletionMessageParam } from 'openai/resources/chat/completions';
+
+export interface IXAIProviderOptions {
+  xaiToken: string;
+}
+
+export class XAIProvider extends MultiModalModel {
+  private options: IXAIProviderOptions;
+  public openAiApiClient: plugins.openai.default;
+  public smartpdfInstance: plugins.smartpdf.SmartPdf;
+
+  constructor(optionsArg: IXAIProviderOptions) {
+    super();
+    this.options = optionsArg;
+  }
+
+  public async start() {
+    this.openAiApiClient = new plugins.openai.default({
+      apiKey: this.options.xaiToken,
+      baseURL: 'https://api.x.ai/v1',
+    });
+    this.smartpdfInstance = new plugins.smartpdf.SmartPdf();
+  }
+
+  public async stop() {}
+
+  public async chatStream(input: ReadableStream<Uint8Array>): Promise<ReadableStream<string>> {
+    // Create a TextDecoder to handle incoming chunks
+    const decoder = new TextDecoder();
+    let buffer = '';
+    let currentMessage: { role: string; content: string; } | null = null;
+
+    // Create a TransformStream to process the input
+    const transform = new TransformStream<Uint8Array, string>({
+      async transform(chunk, controller) {
+        buffer += decoder.decode(chunk, { stream: true });
+
+        // Try to parse complete JSON messages from the buffer
+        while (true) {
+          const newlineIndex = buffer.indexOf('\n');
+          if (newlineIndex === -1) break;
+
+          const line = buffer.slice(0, newlineIndex);
+          buffer = buffer.slice(newlineIndex + 1);
+
+          if (line.trim()) {
+            try {
+              const message = JSON.parse(line);
+              currentMessage = {
+                role: message.role || 'user',
+                content: message.content || '',
+              };
+            } catch (e) {
+              console.error('Failed to parse message:', e);
+            }
+          }
+        }
+
+        // If we have a complete message, send it to X.AI
+        if (currentMessage) {
+          const stream = await this.openAiApiClient.chat.completions.create({
+            model: 'grok-2-latest',
+            messages: [{ role: currentMessage.role, content: currentMessage.content }],
+            stream: true,
+          });
+
+          // Process each chunk from X.AI
+          for await (const chunk of stream) {
+            const content = chunk.choices[0]?.delta?.content;
+            if (content) {
+              controller.enqueue(content);
+            }
+          }
+
+          currentMessage = null;
+        }
+      },
+
+      flush(controller) {
+        if (buffer) {
+          try {
+            const message = JSON.parse(buffer);
+            controller.enqueue(message.content || '');
+          } catch (e) {
+            console.error('Failed to parse remaining buffer:', e);
+          }
+        }
+      }
+    });
+
+    // Connect the input to our transform stream
+    return input.pipeThrough(transform);
+  }
+
+  public async chat(optionsArg: {
+    systemMessage: string;
+    userMessage: string;
+    messageHistory: { role: string; content: string; }[];
+  }): Promise<{ role: 'assistant'; message: string; }> {
+    // Prepare messages array with system message, history, and user message
+    const messages: ChatCompletionMessageParam[] = [
+      { role: 'system', content: optionsArg.systemMessage },
+      ...optionsArg.messageHistory.map(msg => ({
+        role: msg.role as 'system' | 'user' | 'assistant',
+        content: msg.content
+      })),
+      { role: 'user', content: optionsArg.userMessage }
+    ];
+
+    // Call X.AI's chat completion API
+    const completion = await this.openAiApiClient.chat.completions.create({
+      model: 'grok-2-latest',
+      messages: messages,
+      stream: false,
+    });
+
+    // Return the assistant's response
+    return {
+      role: 'assistant',
+      message: completion.choices[0]?.message?.content || ''
+    };
+  }
+
+  public async audio(optionsArg: { message: string }): Promise<NodeJS.ReadableStream> {
+    throw new Error('Audio generation is not supported by X.AI');
+  }
+
+  public async vision(optionsArg: { image: Buffer; prompt: string }): Promise<string> {
+    throw new Error('Vision tasks are not supported by X.AI');
+  }
+
+  public async document(optionsArg: {
+    systemMessage: string;
+    userMessage: string;
+    pdfDocuments: Uint8Array[];
+    messageHistory: { role: string; content: string; }[];
+  }): Promise<{ message: any }> {
+    // First convert PDF documents to images
+    let pdfDocumentImageBytesArray: Uint8Array[] = [];
+    
+    for (const pdfDocument of optionsArg.pdfDocuments) {
+      const documentImageArray = await this.smartpdfInstance.convertPDFToPngBytes(pdfDocument);
+      pdfDocumentImageBytesArray = pdfDocumentImageBytesArray.concat(documentImageArray);
+    }
+
+    // Convert images to base64 for inclusion in the message
+    const imageBase64Array = pdfDocumentImageBytesArray.map(bytes => 
+      Buffer.from(bytes).toString('base64')
+    );
+
+    // Combine document images into the user message
+    const enhancedUserMessage = `
+      ${optionsArg.userMessage}
+      
+      Document contents (as images):
+      ${imageBase64Array.map((img, i) => `Image ${i + 1}: <image data>`).join('\n')}
+    `;
+
+    // Use chat completion to analyze the documents
+    const messages: ChatCompletionMessageParam[] = [
+      { role: 'system', content: optionsArg.systemMessage },
+      ...optionsArg.messageHistory.map(msg => ({
+        role: msg.role as 'system' | 'user' | 'assistant',
+        content: msg.content
+      })),
+      { role: 'user', content: enhancedUserMessage }
+    ];
+
+    const completion = await this.openAiApiClient.chat.completions.create({
+      model: 'grok-2-latest',
+      messages: messages,
+      stream: false,
+    });
+
+    return {
+      message: completion.choices[0]?.message?.content || ''
+    };
+  }
+}