@auto-engineer/ai-gateway 0.1.1

package/dist/index.js

@@ -0,0 +1,351 @@
+import { generateText, streamText, generateObject, streamObject } from 'ai';
+import { openai } from '@ai-sdk/openai';
+import { anthropic } from '@ai-sdk/anthropic';
+import { google } from '@ai-sdk/google';
+import { xai } from '@ai-sdk/xai';
+import { configureAIProvider } from './config.js';
+import { z } from 'zod';
+import { getRegisteredToolsForAI } from './mcp-server.js';
+import { startServer } from './mcp-server.js';
+export var AIProvider;
+(function (AIProvider) {
+    AIProvider["OpenAI"] = "openai";
+    AIProvider["Anthropic"] = "anthropic";
+    AIProvider["Google"] = "google";
+    AIProvider["XAI"] = "xai";
+})(AIProvider || (AIProvider = {}));
+const defaultOptions = {
+    temperature: 0.7,
+    maxTokens: 1000,
+};
+function getDefaultModel(provider) {
+    switch (provider) {
+        case AIProvider.OpenAI:
+            return 'gpt-4o-mini';
+        case AIProvider.Anthropic:
+            return 'claude-sonnet-4-20250514';
+        case AIProvider.Google:
+            return 'gemini-2.5-pro';
+        case AIProvider.XAI:
+            return 'grok-3';
+        default:
+            // eslint-disable-next-line @typescript-eslint/restrict-template-expressions
+            throw new Error(`Unknown provider: ${provider}`);
+    }
+}
+function getModel(provider, model) {
+    const modelName = model ?? getDefaultModel(provider);
+    switch (provider) {
+        case AIProvider.OpenAI:
+            return openai(modelName);
+        case AIProvider.Anthropic:
+            return anthropic(modelName);
+        case AIProvider.Google:
+            return google(modelName);
+        case AIProvider.XAI:
+            return xai(modelName);
+        default:
+            throw new Error(`Unknown provider: ${provider}`);
+    }
+}
+export async function generateTextWithAI(prompt, provider, options = {}) {
+    const finalOptions = { ...defaultOptions, ...options };
+    const model = finalOptions.model ?? getDefaultModel(provider);
+    const modelInstance = getModel(provider, model);
+    if (finalOptions.includeTools === true) {
+        await startServer();
+        const result = await generateTextWithToolsAI(prompt, provider, options);
+        return result.text;
+    }
+    const result = await generateText({
+        model: modelInstance,
+        prompt,
+        temperature: finalOptions.temperature,
+        maxTokens: finalOptions.maxTokens,
+    });
+    return result.text;
+}
+export async function* streamTextWithAI(prompt, provider, options = {}) {
+    const finalOptions = { ...defaultOptions, ...options };
+    const model = getModel(provider, finalOptions.model);
+    const stream = await streamText({
+        model,
+        prompt,
+        temperature: finalOptions.temperature,
+        maxTokens: finalOptions.maxTokens,
+    });
+    for await (const chunk of stream.textStream) {
+        yield chunk;
+    }
+}
+/**
+ * Generates text using streaming and collects the full response.
+ * Optionally calls a stream callback for each token if provided.
+ * Always returns the complete collected response.
+ */
+export async function generateTextStreamingWithAI(prompt, provider, options = {}) {
+    const finalOptions = { ...defaultOptions, ...options };
+    let collectedResult = '';
+    const stream = streamTextWithAI(prompt, provider, finalOptions);
+    for await (const token of stream) {
+        // Collect all tokens for the final result
+        collectedResult += token;
+        // Call the stream callback if provided
+        if (finalOptions.streamCallback) {
+            finalOptions.streamCallback(token);
+        }
+    }
+    return collectedResult;
+}
+export async function generateTextWithToolsAI(prompt, provider, options = {}) {
+    const finalOptions = { ...defaultOptions, ...options };
+    const model = finalOptions.model ?? getDefaultModel(provider);
+    const modelInstance = getModel(provider, model);
+    const registeredTools = finalOptions.includeTools === true ? getRegisteredToolsForAI() : {};
+    console.log('registeredTools', registeredTools);
+    const hasTools = Object.keys(registeredTools).length > 0;
+    console.log('hasTools', hasTools);
+    // Build conversation messages
+    const messages = [{ role: 'user', content: prompt }];
+    let finalResult = '';
+    const allToolCalls = [];
+    let attempts = 0;
+    const maxAttempts = 5;
+    while (attempts < maxAttempts) {
+        attempts++;
+        const opts = {
+            model: modelInstance,
+            messages,
+            temperature: finalOptions.temperature,
+            maxTokens: finalOptions.maxTokens,
+            ...(hasTools && {
+                tools: registeredTools,
+                toolChoice: 'auto',
+            }),
+        };
+        console.log('opts', opts);
+        const result = await generateText(opts);
+        console.log('result', JSON.stringify(result, null, 2));
+        // Add assistant message to conversation
+        if (result.text) {
+            messages.push({ role: 'assistant', content: result.text });
+            finalResult = result.text;
+        }
+        // If there are tool calls, execute them and continue conversation
+        if (result.toolCalls !== undefined && result.toolCalls.length > 0) {
+            allToolCalls.push(...result.toolCalls);
+            // Execute tools and create a simple follow-up prompt
+            const toolResults = await executeToolCalls(result.toolCalls, registeredTools);
+            // Add the tool results as a user message and request a final response
+            messages.push({
+                role: 'user',
+                content: `${toolResults}Based on this product catalog data, please provide specific product recommendations for a soccer-loving daughter. Include product names, prices, and reasons why each item would be suitable.`,
+            });
+            // Continue the conversation to get AI's response to tool results
+            continue;
+        }
+        // If no tool calls, we're done
+        break;
+    }
+    return {
+        text: finalResult,
+        toolCalls: allToolCalls,
+    };
+}
+async function executeToolCalls(toolCalls, registeredTools) {
+    let toolResults = '';
+    for (const toolCall of toolCalls) {
+        try {
+            const toolCallObj = toolCall;
+            const tool = registeredTools[toolCallObj.toolName];
+            if (tool?.execute) {
+                const toolResult = await tool.execute(toolCallObj.args);
+                toolResults += `Tool ${toolCallObj.toolName} returned: ${String(toolResult)}\n\n`;
+            }
+            else {
+                toolResults += `Error: Tool ${toolCallObj.toolName} not found or missing execute function\n\n`;
+            }
+        }
+        catch (error) {
+            const toolCallObj = toolCall;
+            console.error(`Tool execution error for ${toolCallObj.toolName}:`, error);
+            toolResults += `Error executing tool ${toolCallObj.toolName}: ${String(error)}\n\n`;
+        }
+    }
+    return toolResults;
+}
+export async function generateTextWithImageAI(text, imageBase64, provider, options = {}) {
+    const finalOptions = { ...defaultOptions, ...options };
+    const model = finalOptions.model ?? getDefaultModel(provider);
+    const modelInstance = getModel(provider, model);
+    if (provider !== AIProvider.OpenAI && provider !== AIProvider.XAI) {
+        throw new Error(`Provider ${provider} does not support image inputs`);
+    }
+    const result = await generateText({
+        model: modelInstance,
+        messages: [
+            {
+                role: 'user',
+                content: [
+                    { type: 'text', text },
+                    { type: 'image', image: imageBase64 },
+                ],
+            },
+        ],
+        temperature: finalOptions.temperature,
+        maxTokens: finalOptions.maxTokens,
+    });
+    return result.text;
+}
+export function getAvailableProviders() {
+    const config = configureAIProvider();
+    const providers = [];
+    if (config.openai != null)
+        providers.push(AIProvider.OpenAI);
+    if (config.anthropic != null)
+        providers.push(AIProvider.Anthropic);
+    if (config.google != null)
+        providers.push(AIProvider.Google);
+    if (config.xai != null)
+        providers.push(AIProvider.XAI);
+    return providers;
+}
+function getEnhancedPrompt(prompt, lastError) {
+    const errorDetails = lastError.zodIssues
+        ? JSON.stringify(lastError.zodIssues, null, 2)
+        : lastError.validationDetails?.cause?.issues
+            ? JSON.stringify(lastError.validationDetails.cause.issues, null, 2)
+            : lastError.message;
+    return `${prompt}\\n\\n⚠️ IMPORTANT: Your previous response failed validation with the following errors:\\n${errorDetails}\\n\\nPlease fix these errors and ensure your response EXACTLY matches the required schema structure.`;
+}
+function isSchemaError(error) {
+    return (error.message.includes('response did not match schema') ||
+        error.message.includes('TypeValidationError') ||
+        error.name === 'AI_TypeValidationError');
+}
+function enhanceValidationError(error) {
+    const enhancedError = new Error(error.message);
+    Object.assign(enhancedError, error);
+    if (error.cause !== undefined || error.issues !== undefined || error.errors !== undefined) {
+        enhancedError.validationDetails = {
+            cause: error.cause,
+            issues: error.issues,
+            errors: error.errors,
+        };
+    }
+    if (error.message.includes('response did not match schema') &&
+        'cause' in error &&
+        typeof error.cause === 'object' &&
+        error.cause !== null &&
+        'issues' in error.cause) {
+        enhancedError.zodIssues = error.cause.issues;
+    }
+    return enhancedError;
+}
+function handleFailedRequest(lastError, maxRetries, attempt) {
+    if (!lastError || !isSchemaError(lastError) || attempt >= maxRetries - 1) {
+        return { shouldRetry: false, enhancedError: lastError };
+    }
+    console.log(`Schema validation failed on attempt ${attempt + 1}/${maxRetries}, retrying...`);
+    const enhancedError = enhanceValidationError(lastError);
+    return { shouldRetry: true, enhancedError };
+}
+export async function generateStructuredDataWithAI(prompt, provider, options) {
+    const maxSchemaRetries = 3;
+    let lastError;
+    if (options.includeTools === true)
+        await startServer();
+    const registeredTools = options.includeTools === true ? getRegisteredToolsForAI() : {};
+    console.log('registeredTools', registeredTools);
+    const hasTools = Object.keys(registeredTools).length > 0;
+    console.log('hasTools', hasTools);
+    for (let attempt = 0; attempt < maxSchemaRetries; attempt++) {
+        try {
+            const model = getModel(provider, options.model);
+            const enhancedPrompt = attempt > 0 && lastError ? getEnhancedPrompt(prompt, lastError) : prompt;
+            console.log('enhancedPrompt', enhancedPrompt);
+            const opts = {
+                model,
+                prompt: enhancedPrompt,
+                schema: options.schema,
+                schemaName: options.schemaName,
+                schemaDescription: options.schemaDescription,
+                temperature: options.temperature,
+                maxTokens: options.maxTokens,
+                ...(hasTools && {
+                    tools: registeredTools,
+                    toolChoice: 'auto',
+                }),
+            };
+            console.log('opts', opts);
+            const result = await generateObject(opts);
+            console.log('result', JSON.stringify(result, null, 2));
+            return result.object;
+        }
+        catch (error) {
+            lastError =
+                error instanceof Error
+                    ? error
+                    : new Error('An unknown error occurred');
+            const { shouldRetry, enhancedError } = handleFailedRequest(lastError, maxSchemaRetries, attempt);
+            lastError = enhancedError;
+            if (!shouldRetry) {
+                throw lastError;
+            }
+        }
+    }
+    // If we exhausted all retries, throw the last error
+    throw lastError;
+}
+export async function streamStructuredDataWithAI(prompt, provider, options) {
+    const maxSchemaRetries = 3;
+    let lastError;
+    for (let attempt = 0; attempt < maxSchemaRetries; attempt++) {
+        try {
+            const model = getModel(provider, options.model);
+            const enhancedPrompt = attempt > 0 && lastError ? getEnhancedPrompt(prompt, lastError) : prompt;
+            const result = streamObject({
+                model,
+                prompt: enhancedPrompt,
+                schema: options.schema,
+                schemaName: options.schemaName,
+                schemaDescription: options.schemaDescription,
+                temperature: options.temperature,
+                maxTokens: options.maxTokens,
+            });
+            // Stream partial objects if callback provided
+            if (options.onPartialObject) {
+                void (async () => {
+                    try {
+                        for await (const partialObject of result.partialObjectStream) {
+                            options.onPartialObject?.(partialObject);
+                        }
+                    }
+                    catch (streamError) {
+                        console.error('Error in partial object stream:', streamError);
+                    }
+                })();
+            }
+            // Return the final complete object
+            return await result.object;
+        }
+        catch (error) {
+            lastError =
+                error instanceof Error
+                    ? error
+                    : new Error('An unknown error occurred');
+            const { shouldRetry, enhancedError } = handleFailedRequest(lastError, maxSchemaRetries, attempt);
+            lastError = enhancedError;
+            if (!shouldRetry) {
+                throw lastError;
+            }
+        }
+    }
+    // If we exhausted all retries, throw the last error
+    throw lastError;
+}
+// Export zod for convenience
+export { z };
+// Export MCP server singleton functions
+export { server as mcpServer, registerTool, registerTools, startServer, isServerStarted, getRegisteredTools, getRegisteredToolsForAI, getToolHandler, getSchemaByName, executeRegisteredTool, } from './mcp-server.js';
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAC5E,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AACxC,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAC9C,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AACxC,OAAO,EAAE,GAAG,EAAE,MAAM,aAAa,CAAC;AAClC,OAAO,EAAE,mBAAmB,EAAE,MAAM,UAAU,CAAC;AAC/C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,uBAAuB,EAAE,MAAM,cAAc,CAAC;AACvD,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAqB3C,MAAM,CAAN,IAAY,UAKX;AALD,WAAY,UAAU;IACpB,+BAAiB,CAAA;IACjB,qCAAuB,CAAA;IACvB,+BAAiB,CAAA;IACjB,yBAAW,CAAA;AACb,CAAC,EALW,UAAU,KAAV,UAAU,QAKrB;AAoBD,MAAM,cAAc,GAAc;IAChC,WAAW,EAAE,GAAG;IAChB,SAAS,EAAE,IAAI;CAChB,CAAC;AAEF,SAAS,eAAe,CAAC,QAAoB;IAC3C,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,UAAU,CAAC,MAAM;YACpB,OAAO,aAAa,CAAC;QACvB,KAAK,UAAU,CAAC,SAAS;YACvB,OAAO,0BAA0B,CAAC;QACpC,KAAK,UAAU,CAAC,MAAM;YACpB,OAAO,gBAAgB,CAAC;QAC1B,KAAK,UAAU,CAAC,GAAG;YACjB,OAAO,QAAQ,CAAC;QAClB;YACE,4EAA4E;YAC5E,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,EAAE,CAAC,CAAC;IACrD,CAAC;AACH,CAAC;AAED,SAAS,QAAQ,CAAC,QAAoB,EAAE,KAAc;IACpD,MAAM,SAAS,GAAG,KAAK,IAAI,eAAe,CAAC,QAAQ,CAAC,CAAC;IAErD,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,UAAU,CAAC,MAAM;YACpB,OAAO,MAAM,CAAC,SAAS,CAAC,CAAC;QAC3B,KAAK,UAAU,CAAC,SAAS;YACvB,OAAO,SAAS,CAAC,SAAS,CAAC,CAAC;QAC9B,KAAK,UAAU,CAAC,MAAM;YACpB,OAAO,MAAM,CAAC,SAAS,CAAC,CAAC;QAC3B,KAAK,UAAU,CAAC,GAAG;YACjB,OAAO,GAAG,CAAC,SAAS,CAAC,CAAC;QACxB;YACE,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAkB,EAAE,CAAC,CAAC;IAC/D,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,MAAc,EACd,QAAoB,EACpB,UAAqB,EAAE;IAEvB,MAAM,YAAY,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,OAAO,EAAE,CAAC;IACvD,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,IAAI,eAAe,CAAC,QAAQ,CAAC,CAAC;IAC9D,MAAM,aAAa,GAAG,QAAQ,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IAEhD,IAAI,YAAY,CAAC,YAAY,KAAK,IAAI,EAAE,CAAC;QACvC,MAAM,WAAW,EAAE,CAAC;QACpB,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC,MAAM,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;QAExE,OAAO,MAAM,CAAC,IAAI,CAAC;IACrB,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;QAChC,KAAK,EAAE,aAAa;QACpB,MAAM;QACN,WAAW,EAAE,YAAY,CAAC,WAAW;QACrC,SAAS,EAAE,YAAY,CAAC,SAAS;KAClC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC,IAAI,CAAC;AACrB,CAAC;AAED,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC,gBAAgB,CACrC,MAAc,EACd,QAAoB,EACpB,UAAqB,EAAE;IAEvB,MAAM,YAAY,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,OAAO,EAAE,CAAC;IACvD,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,EAAE,YAAY,CAAC,KAAK,CAAC,CAAC;IAErD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC;QAC9B,KAAK;QACL,MAAM;QACN,WAAW,EAAE,YAAY,CAAC,WAAW;QACrC,SAAS,EAAE,YAAY,CAAC,SAAS;KAClC,CAAC,CAAC;IAEH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;QAC5C,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,2BAA2B,CAC/C,MAAc,EACd,QAAoB,EACpB,UAAqB,EAAE;IAEvB,MAAM,YAAY,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,OAAO,EAAE,CAAC;IACvD,IAAI,eAAe,GAAG,EAAE,CAAC;IAEzB,MAAM,MAAM,GAAG,gBAAgB,CAAC,MAAM,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC;IAEhE,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QACjC,0CAA0C;QAC1C,eAAe,IAAI,KAAK,CAAC;QAEzB,uCAAuC;QACvC,IAAI,YAAY,CAAC,cAAc,EAAE,CAAC;YAChC,YAAY,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IAED,OAAO,eAAe,CAAC;AACzB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC3C,MAAc,EACd,QAAoB,EACpB,UAAqB,EAAE;IAEvB,MAAM,YAAY,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,OAAO,EAAE,CAAC;IACvD,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,IAAI,eAAe,CAAC,QAAQ,CAAC,CAAC;IAC9D,MAAM,aAAa,GAAG,QAAQ,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IAEhD,MAAM,eAAe,GAAG,YAAY,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,CAAC,uBAAuB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAC5F,OAAO,CAAC,GAAG,CAAC,iBAAiB,EAAE,eAAe,CAAC,CAAC;IAChD,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;IACzD,OAAO,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;IAElC,8BAA8B;IAC9B,MAAM,QAAQ,GAA2D,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;IAE7G,IAAI,WAAW,GAAG,EAAE,CAAC;IACrB,MAAM,YAAY,GAAc,EAAE,CAAC;IACnC,IAAI,QAAQ,GAAG,CAAC,CAAC;IACjB,MAAM,WAAW,GAAG,CAAC,CAAC;IAEtB,OAAO,QAAQ,GAAG,WAAW,EAAE,CAAC;QAC9B,QAAQ,EAAE,CAAC;QAEX,MAAM,IAAI,GAAG;YACX,KAAK,EAAE,aAAa;YACpB,QAAQ;YACR,WAAW,EAAE,YAAY,CAAC,WAAW;YACrC,SAAS,EAAE,YAAY,CAAC,SAAS;YACjC,GAAG,CAAC,QAAQ,IAAI;gBACd,KAAK,EAAE,eAAe;gBACtB,UAAU,EAAE,MAAe;aAC5B,CAAC;SACH,CAAC;QACF,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAC1B,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,CAAC;QACxC,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QAEvD,wCAAwC;QACxC,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YAChB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;YAC3D,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC;QAC5B,CAAC;QAED,kEAAkE;QAClE,IAAI,MAAM,CAAC,SAAS,KAAK,SAAS,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAClE,YAAY,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC;YAEvC,qDAAqD;YACrD,MAAM,WAAW,GAAG,MAAM,gBAAgB,CAAC,MAAM,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;YAE9E,sEAAsE;YACtE,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,GAAG,WAAW,+LAA+L;aACvN,CAAC,CAAC;YAEH,iEAAiE;YACjE,SAAS;QACX,CAAC;QAED,+BAA+B;QAC/B,MAAM;IACR,CAAC;IAED,OAAO;QACL,IAAI,EAAE,WAAW;QACjB,SAAS,EAAE,YAAY;KACxB,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,gBAAgB,CAC7B,SAAoB,EACpB,eAGC;IAED,IAAI,WAAW,GAAG,EAAE,CAAC;IAErB,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,QAA+D,CAAC;YACpF,MAAM,IAAI,GAAG,eAAe,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;YACnD,IAAI,IAAI,EAAE,OAAO,EAAE,CAAC;gBAClB,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;gBACxD,WAAW,IAAI,QAAQ,WAAW,CAAC,QAAQ,cAAc,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC;YACpF,CAAC;iBAAM,CAAC;gBACN,WAAW,IAAI,eAAe,WAAW,CAAC,QAAQ,4CAA4C,CAAC;YACjG,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,WAAW,GAAG,QAAgC,CAAC;YACrD,OAAO,CAAC,KAAK,CAAC,4BAA4B,WAAW,CAAC,QAAQ,GAAG,EAAE,KAAK,CAAC,CAAC;YAC1E,WAAW,IAAI,wBAAwB,WAAW,CAAC,QAAQ,KAAK,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC;QACtF,CAAC;IACH,CAAC;IAED,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC3C,IAAY,EACZ,WAAmB,EACnB,QAAoB,EACpB,UAAqB,EAAE;IAEvB,MAAM,YAAY,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,OAAO,EAAE,CAAC;IACvD,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,IAAI,eAAe,CAAC,QAAQ,CAAC,CAAC;IAC9D,MAAM,aAAa,GAAG,QAAQ,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IAEhD,IAAI,QAAQ,KAAK,UAAU,CAAC,MAAM,IAAI,QAAQ,KAAK,UAAU,CAAC,GAAG,EAAE,CAAC;QAClE,MAAM,IAAI,KAAK,CAAC,YAAY,QAAQ,gCAAgC,CAAC,CAAC;IACxE,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;QAChC,KAAK,EAAE,aAAa;QACpB,QAAQ,EAAE;YACR;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE;oBACP,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;oBACtB,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE;iBACtC;aACF;SACF;QACD,WAAW,EAAE,YAAY,CAAC,WAAW;QACrC,SAAS,EAAE,YAAY,CAAC,SAAS;KAClC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC,IAAI,CAAC;AACrB,CAAC;AAED,MAAM,UAAU,qBAAqB;IACnC,MAAM,MAAM,GAAG,mBAAmB,EAAE,CAAC;IACrC,MAAM,SAAS,GAAiB,EAAE,CAAC;IACnC,IAAI,MAAM,CAAC,MAAM,IAAI,IAAI;QAAE,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;IAC7D,IAAI,MAAM,CAAC,SAAS,IAAI,IAAI;QAAE,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;IACnE,IAAI,MAAM,CAAC,MAAM,IAAI,IAAI;QAAE,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;IAC7D,IAAI,MAAM,CAAC,GAAG,IAAI,IAAI;QAAE,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;IACvD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,iBAAiB,CAAC,MAAc,EAAE,SAAgC;IACzE,MAAM,YAAY,GAAG,SAAS,CAAC,SAAS;QACtC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9C,CAAC,CAAC,SAAS,CAAC,iBAAiB,EAAE,KAAK,EAAE,MAAM;YAC1C,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,iBAAiB,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YACnE,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC;IAExB,OAAO,GAAG,MAAM,6FAA6F,YAAY,uGAAuG,CAAC;AACnO,CAAC;AAED,SAAS,aAAa,CAAC,KAAY;IACjC,OAAO,CACL,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,+BAA+B,CAAC;QACvD,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,qBAAqB,CAAC;QAC7C,KAAK,CAAC,IAAI,KAAK,wBAAwB,CACxC,CAAC;AACJ,CAAC;AAED,SAAS,sBAAsB,CAAC,KAA4B;IAC1D,MAAM,aAAa,GAAG,IAAI,KAAK,CAAC,KAAK,CAAC,OAAO,CAA0B,CAAC;IACxE,MAAM,CAAC,MAAM,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC;IAEpC,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;QAC1F,aAAa,CAAC,iBAAiB,GAAG;YAChC,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,MAAM,EAAE,KAAK,CAAC,MAAM;SACrB,CAAC;IACJ,CAAC;IAED,IACE,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,+BAA+B,CAAC;QACvD,OAAO,IAAI,KAAK;QAChB,OAAO,KAAK,CAAC,KAAK,KAAK,QAAQ;QAC/B,KAAK,CAAC,KAAK,KAAK,IAAI;QACpB,QAAQ,IAAI,KAAK,CAAC,KAAK,EACvB,CAAC;QACD,aAAa,CAAC,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,MAAmB,CAAC;IAC5D,CAAC;IACD,OAAO,aAAa,CAAC;AACvB,CAAC;AAED,SAAS,mBAAmB,CAC1B,SAA4C,EAC5C,UAAkB,EAClB,OAAe;IAEf,IAAI,CAAC,SAAS,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,IAAI,OAAO,IAAI,UAAU,GAAG,CAAC,EAAE,CAAC;QACzE,OAAO,EAAE,WAAW,EAAE,KAAK,EAAE,aAAa,EAAE,SAAS,EAAE,CAAC;IAC1D,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,uCAAuC,OAAO,GAAG,CAAC,IAAI,UAAU,eAAe,CAAC,CAAC;IAC7F,MAAM,aAAa,GAAG,sBAAsB,CAAC,SAAS,CAAC,CAAC;IAExD,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;AAC9C,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAChD,MAAc,EACd,QAAoB,EACpB,OAA+B;IAE/B,MAAM,gBAAgB,GAAG,CAAC,CAAC;IAC3B,IAAI,SAA4C,CAAC;IAEjD,IAAI,OAAO,CAAC,YAAY,KAAK,IAAI;QAAE,MAAM,WAAW,EAAE,CAAC;IACvD,MAAM,eAAe,GAAG,OAAO,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,CAAC,uBAAuB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IACvF,OAAO,CAAC,GAAG,CAAC,iBAAiB,EAAE,eAAe,CAAC,CAAC;IAChD,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;IACzD,OAAO,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;IAElC,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,gBAAgB,EAAE,OAAO,EAAE,EAAE,CAAC;QAC5D,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;YAEhD,MAAM,cAAc,GAAG,OAAO,GAAG,CAAC,IAAI,SAAS,CAAC,CAAC,CAAC,iBAAiB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;YAChG,OAAO,CAAC,GAAG,CAAC,gBAAgB,EAAE,cAAc,CAAC,CAAC;YAE9C,MAAM,IAAI,GAAG;gBACX,KAAK;gBACL,MAAM,EAAE,cAAc;gBACtB,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,iBAAiB,EAAE,OAAO,CAAC,iBAAiB;gBAC5C,WAAW,EAAE,OAAO,CAAC,WAAW;gBAChC,SAAS,EAAE,OAAO,CAAC,SAAS;gBAC5B,GAAG,CAAC,QAAQ,IAAI;oBACd,KAAK,EAAE,eAAe;oBACtB,UAAU,EAAE,MAAe;iBAC5B,CAAC;aACH,CAAC;YACF,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YAC1B,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,IAAI,CAAC,CAAC;YAC1C,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YACvD,OAAO,MAAM,CAAC,MAAM,CAAC;QACvB,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,SAAS;gBACP,KAAK,YAAY,KAAK;oBACpB,CAAC,CAAE,KAA+B;oBAClC,CAAC,CAAE,IAAI,KAAK,CAAC,2BAA2B,CAA2B,CAAC;YAExE,MAAM,EAAE,WAAW,EAAE,aAAa,EAAE,GAAG,mBAAmB,CAAC,SAAS,EAAE,gBAAgB,EAAE,OAAO,CAAC,CAAC;YACjG,SAAS,GAAG,aAAa,CAAC;YAE1B,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,MAAM,SAAS,CAAC;YAClB,CAAC;QACH,CAAC;IACH,CAAC;IAED,oDAAoD;IACpD,MAAM,SAAS,CAAC;AAClB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,MAAc,EACd,QAAoB,EACpB,OAAqC;IAErC,MAAM,gBAAgB,GAAG,CAAC,CAAC;IAC3B,IAAI,SAA4C,CAAC;IAEjD,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,gBAAgB,EAAE,OAAO,EAAE,EAAE,CAAC;QAC5D,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;YAEhD,MAAM,cAAc,GAAG,OAAO,GAAG,CAAC,IAAI,SAAS,CAAC,CAAC,CAAC,iBAAiB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;YAEhG,MAAM,MAAM,GAAG,YAAY,CAAC;gBAC1B,KAAK;gBACL,MAAM,EAAE,cAAc;gBACtB,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,iBAAiB,EAAE,OAAO,CAAC,iBAAiB;gBAC5C,WAAW,EAAE,OAAO,CAAC,WAAW;gBAChC,SAAS,EAAE,OAAO,CAAC,SAAS;aAC7B,CAAC,CAAC;YAEH,8CAA8C;YAC9C,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;gBAC5B,KAAK,CAAC,KAAK,IAAI,EAAE;oBACf,IAAI,CAAC;wBACH,IAAI,KAAK,EAAE,MAAM,aAAa,IAAI,MAAM,CAAC,mBAAmB,EAAE,CAAC;4BAC7D,OAAO,CAAC,eAAe,EAAE,CAAC,aAAa,CAAC,CAAC;wBAC3C,CAAC;oBACH,CAAC;oBAAC,OAAO,WAAW,EAAE,CAAC;wBACrB,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,WAAW,CAAC,CAAC;oBAChE,CAAC;gBACH,CAAC,CAAC,EAAE,CAAC;YACP,CAAC;YAED,mCAAmC;YACnC,OAAO,MAAM,MAAM,CAAC,MAAM,CAAC;QAC7B,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,SAAS;gBACP,KAAK,YAAY,KAAK;oBACpB,CAAC,CAAE,KAA+B;oBAClC,CAAC,CAAE,IAAI,KAAK,CAAC,2BAA2B,CAA2B,CAAC;YAExE,MAAM,EAAE,WAAW,EAAE,aAAa,EAAE,GAAG,mBAAmB,CAAC,SAAS,EAAE,gBAAgB,EAAE,OAAO,CAAC,CAAC;YACjG,SAAS,GAAG,aAAa,CAAC;YAE1B,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,MAAM,SAAS,CAAC;YAClB,CAAC;QACH,CAAC;IACH,CAAC;IAED,oDAAoD;IACpD,MAAM,SAAS,CAAC;AAClB,CAAC;AAED,6BAA6B;AAC7B,OAAO,EAAE,CAAC,EAAE,CAAC;AAEb,wCAAwC;AACxC,OAAO,EACL,MAAM,IAAI,SAAS,EACnB,YAAY,EACZ,aAAa,EACb,WAAW,EACX,eAAe,EACf,kBAAkB,EAClB,uBAAuB,EACvB,cAAc,EACd,eAAe,EACf,qBAAqB,GAGtB,MAAM,cAAc,CAAC"}

package/dist/index.specs.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=index.specs.d.ts.map

package/dist/index.specs.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.specs.d.ts","sourceRoot":"","sources":["../src/index.specs.ts"],"names":[],"mappings":""}

package/dist/index.specs.js

@@ -0,0 +1,14 @@
+import { describe, it, expect } from 'vitest';
+import { generateTextWithAI, streamTextWithAI, AIProvider } from './index.js';
+describe('AI Integration', () => {
+    process.env.OPENAI_API_KEY = 'test';
+    it('should export the correct types', () => {
+        expect(typeof generateTextWithAI).toBe('function');
+        expect(typeof streamTextWithAI).toBe('function');
+    });
+    it('should have valid AIProvider type', () => {
+        const providers = [AIProvider.OpenAI, AIProvider.Anthropic, AIProvider.Google, AIProvider.XAI];
+        expect(providers).toContain(AIProvider.OpenAI);
+    });
+});
+//# sourceMappingURL=index.specs.js.map

package/dist/index.specs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.specs.js","sourceRoot":"","sources":["../src/index.specs.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AAE3E,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,MAAM,CAAC;IAEpC,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,CAAC,OAAO,kBAAkB,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACnD,MAAM,CAAC,OAAO,gBAAgB,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,SAAS,GAAiB,CAAC,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,SAAS,EAAE,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC;QAC7G,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/mcp-server.d.ts

@@ -0,0 +1,50 @@
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { z } from 'zod';
+interface ToolResult {
+    content: Array<{
+        type: 'text';
+        text: string;
+    }>;
+    isError?: boolean;
+    [key: string]: unknown;
+}
+interface ToolDescription {
+    title: string;
+    description: string;
+    inputSchema: Record<string, z.ZodSchema>;
+    schema?: z.ZodSchema;
+    schemaName?: string;
+    schemaDescription?: string;
+}
+type ToolHandler<T extends Record<string, unknown> = Record<string, unknown>> = (params: T) => Promise<ToolResult>;
+interface RegisteredTool {
+    name: string;
+    description: ToolDescription;
+    handler: ToolHandler<Record<string, unknown>>;
+    aiSdkTool: {
+        parameters: z.ZodSchema;
+        description: string;
+        execute?: (args: Record<string, unknown>) => Promise<string>;
+    };
+}
+declare const server: McpServer;
+export { server };
+export declare function registerTool<T extends Record<string, unknown> = Record<string, unknown>>(name: string, description: ToolDescription, handler: ToolHandler<T>): void;
+export declare function registerTools<T extends Record<string, unknown> = Record<string, unknown>>(tools: Array<{
+    name: string;
+    description: ToolDescription;
+    handler: ToolHandler<T>;
+}>): void;
+export declare function startServer(): Promise<void>;
+export declare function isServerStarted(): boolean;
+export declare function getRegisteredTools(): RegisteredTool[];
+export declare function getRegisteredToolsForAI(): Record<string, {
+    parameters: z.ZodSchema;
+    description: string;
+    execute?: (args: Record<string, unknown>) => Promise<string>;
+}>;
+export declare function getToolHandler(name: string): ToolHandler | undefined;
+export declare function executeRegisteredTool(name: string, params: Record<string, unknown>): Promise<ToolResult>;
+export declare function getSchemaByName(schemaName: string): z.ZodSchema | undefined;
+export type { ToolResult, ToolDescription, ToolHandler, RegisteredTool };
+//# sourceMappingURL=mcp-server.d.ts.map

package/dist/mcp-server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-server.d.ts","sourceRoot":"","sources":["../src/mcp-server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEpE,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,UAAU,UAAU;IAClB,OAAO,EAAE,KAAK,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;KACd,CAAC,CAAC;IACH,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AAED,UAAU,eAAe;IACvB,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC;IACzC,MAAM,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B;AAED,KAAK,WAAW,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,KAAK,OAAO,CAAC,UAAU,CAAC,CAAC;AAEnH,UAAU,cAAc;IACtB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,eAAe,CAAC;IAC7B,OAAO,EAAE,WAAW,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;IAC9C,SAAS,EAAE;QACT,UAAU,EAAE,CAAC,CAAC,SAAS,CAAC;QACxB,WAAW,EAAE,MAAM,CAAC;QACpB,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;KAC9D,CAAC;CACH;AAED,QAAA,MAAM,MAAM,WAGV,CAAC;AAoBH,OAAO,EAAE,MAAM,EAAE,CAAC;AAyClB,wBAAgB,YAAY,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACtF,IAAI,EAAE,MAAM,EACZ,WAAW,EAAE,eAAe,EAC5B,OAAO,EAAE,WAAW,CAAC,CAAC,CAAC,QAqBxB;AAED,wBAAgB,aAAa,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACvF,KAAK,EAAE,KAAK,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,eAAe,CAAC;IAC7B,OAAO,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC;CACzB,CAAC,QAQH;AAED,wBAAsB,WAAW,kBAKhC;AAED,wBAAgB,eAAe,YAE9B;AAED,wBAAgB,kBAAkB,IAAI,cAAc,EAAE,CAErD;AAED,wBAAgB,uBAAuB,IAAI,MAAM,CAC/C,MAAM,EACN;IACE,UAAU,EAAE,CAAC,CAAC,SAAS,CAAC;IACxB,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;CAC9D,CACF,CAgBA;AAED,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,SAAS,CAEpE;AAED,wBAAsB,qBAAqB,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,UAAU,CAAC,CAM9G;AAED,wBAAgB,eAAe,CAAC,UAAU,EAAE,MAAM,GAAG,CAAC,CAAC,SAAS,GAAG,SAAS,CAQ3E;AAED,YAAY,EAAE,UAAU,EAAE,eAAe,EAAE,WAAW,EAAE,cAAc,EAAE,CAAC"}

package/dist/mcp-server.js

@@ -0,0 +1,121 @@
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { z } from 'zod';
+const server = new McpServer({
+    name: 'frontend-implementation',
+    version: '0.1.0',
+});
+const transport = new StdioServerTransport();
+let isStarted = false;
+const toolRegistry = new Map();
+async function cleanup() {
+    console.log('Cleaning up...');
+    await transport.close();
+    process.exit(0);
+}
+process.on('SIGTERM', () => {
+    void cleanup();
+});
+process.on('SIGINT', () => {
+    void cleanup();
+});
+export { server };
+function createMcpHandler(handler) {
+    return (args, _extra) => {
+        return handler(args);
+    };
+}
+function createAiSdkTool(description, handler) {
+    const parameterSchema = Object.keys(description.inputSchema).length > 0 ? z.object(description.inputSchema) : z.object({}).passthrough();
+    return {
+        parameters: parameterSchema,
+        description: description.description,
+        execute: async (args) => {
+            const result = await handler(args);
+            const textOutput = result.content[0]?.text || '';
+            // If a schema is provided, parse and validate the JSON output
+            if (description.schema) {
+                try {
+                    const parsed = JSON.parse(textOutput);
+                    description.schema.parse(parsed);
+                    return textOutput; // Return original text output for consistency
+                }
+                catch (parseError) {
+                    console.warn(`Tool failed to parse/validate JSON output:`, parseError);
+                    return textOutput; // Fallback to raw text
+                }
+            }
+            return textOutput;
+        },
+        ...(description.schema && { schema: description.schema }),
+        ...(description.schemaName != null && { schemaName: description.schemaName }),
+        ...(description.schemaDescription != null && { schemaDescription: description.schemaDescription }),
+    };
+}
+export function registerTool(name, description, handler) {
+    console.log(`🔧 Registering MCP tool: ${name}`);
+    if (isStarted) {
+        throw new Error('Cannot register tools after server has started');
+    }
+    const mcpHandler = createMcpHandler(handler);
+    const aiSdkTool = createAiSdkTool(description, handler);
+    const registeredTool = {
+        name,
+        description,
+        handler: handler,
+        aiSdkTool,
+    };
+    toolRegistry.set(name, registeredTool);
+    server.registerTool(name, description, mcpHandler);
+    console.log(`✅ Tool ${name} registered successfully. Total tools: ${toolRegistry.size}`);
+}
+export function registerTools(tools) {
+    if (isStarted) {
+        throw new Error('Cannot register tools after server has started');
+    }
+    tools.forEach((tool) => {
+        registerTool(tool.name, tool.description, tool.handler);
+    });
+}
+export async function startServer() {
+    if (isStarted)
+        return;
+    await server.connect(transport);
+    isStarted = true;
+}
+export function isServerStarted() {
+    return isStarted;
+}
+export function getRegisteredTools() {
+    return Array.from(toolRegistry.values());
+}
+export function getRegisteredToolsForAI() {
+    console.log(`📋 Getting registered tools for AI. Registry size: ${toolRegistry.size}`);
+    const tools = {};
+    for (const tool of toolRegistry.values()) {
+        tools[tool.name] = tool.aiSdkTool;
+        console.log(`  - Tool: ${tool.name}`);
+    }
+    console.log(`📊 Returning ${Object.keys(tools).length} tools for AI`);
+    return tools;
+}
+export function getToolHandler(name) {
+    return toolRegistry.get(name)?.handler;
+}
+export async function executeRegisteredTool(name, params) {
+    const tool = toolRegistry.get(name);
+    if (!tool) {
+        throw new Error(`Tool '${name}' not found`);
+    }
+    return await tool.handler(params);
+}
+export function getSchemaByName(schemaName) {
+    console.log();
+    for (const tool of toolRegistry.values()) {
+        if (tool.description?.schemaName === schemaName) {
+            return tool.description.schema;
+        }
+    }
+    return undefined;
+}
+//# sourceMappingURL=mcp-server.js.map

package/dist/mcp-server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-server.js","sourceRoot":"","sources":["../src/mcp-server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAiCxB,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC;IAC3B,IAAI,EAAE,yBAAyB;IAC/B,OAAO,EAAE,OAAO;CACjB,CAAC,CAAC;AAEH,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;AAE7C,IAAI,SAAS,GAAG,KAAK,CAAC;AACtB,MAAM,YAAY,GAAG,IAAI,GAAG,EAA0B,CAAC;AAEvD,KAAK,UAAU,OAAO;IACpB,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAC9B,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC;IACxB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC;AAED,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;IACzB,KAAK,OAAO,EAAE,CAAC;AACjB,CAAC,CAAC,CAAC;AACH,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE;IACxB,KAAK,OAAO,EAAE,CAAC;AACjB,CAAC,CAAC,CAAC;AAEH,OAAO,EAAE,MAAM,EAAE,CAAC;AAElB,SAAS,gBAAgB,CACvB,OAAuB;IAEvB,OAAO,CAAC,IAA6B,EAAE,MAAe,EAAE,EAAE;QACxD,OAAO,OAAO,CAAC,IAAS,CAAC,CAAC;IAC5B,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,WAA4B,EAAE,OAAoB;IACzE,MAAM,eAAe,GACnB,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;IAEnH,OAAO;QACL,UAAU,EAAE,eAAe;QAC3B,WAAW,EAAE,WAAW,CAAC,WAAW;QACpC,OAAO,EAAE,KAAK,EAAE,IAA6B,EAAE,EAAE;YAC/C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;YACnC,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;YAEjD,8DAA8D;YAC9D,IAAI,WAAW,CAAC,MAAM,EAAE,CAAC;gBACvB,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAY,CAAC;oBACjD,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;oBACjC,OAAO,UAAU,CAAC,CAAC,8CAA8C;gBACnE,CAAC;gBAAC,OAAO,UAAU,EAAE,CAAC;oBACpB,OAAO,CAAC,IAAI,CAAC,4CAA4C,EAAE,UAAU,CAAC,CAAC;oBACvE,OAAO,UAAU,CAAC,CAAC,uBAAuB;gBAC5C,CAAC;YACH,CAAC;YAED,OAAO,UAAU,CAAC;QACpB,CAAC;QACD,GAAG,CAAC,WAAW,CAAC,MAAM,IAAI,EAAE,MAAM,EAAE,WAAW,CAAC,MAAM,EAAE,CAAC;QACzD,GAAG,CAAC,WAAW,CAAC,UAAU,IAAI,IAAI,IAAI,EAAE,UAAU,EAAE,WAAW,CAAC,UAAU,EAAE,CAAC;QAC7E,GAAG,CAAC,WAAW,CAAC,iBAAiB,IAAI,IAAI,IAAI,EAAE,iBAAiB,EAAE,WAAW,CAAC,iBAAiB,EAAE,CAAC;KACnG,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,YAAY,CAC1B,IAAY,EACZ,WAA4B,EAC5B,OAAuB;IAEvB,OAAO,CAAC,GAAG,CAAC,4BAA4B,IAAI,EAAE,CAAC,CAAC;IAEhD,IAAI,SAAS,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;IACpE,CAAC;IAED,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAC7C,MAAM,SAAS,GAAG,eAAe,CAAC,WAAW,EAAE,OAA+C,CAAC,CAAC;IAEhG,MAAM,cAAc,GAAmB;QACrC,IAAI;QACJ,WAAW;QACX,OAAO,EAAE,OAA+C;QACxD,SAAS;KACV,CAAC;IAEF,YAAY,CAAC,GAAG,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;IACvC,MAAM,CAAC,YAAY,CAAC,IAAI,EAAE,WAAW,EAAE,UAAU,CAAC,CAAC;IACnD,OAAO,CAAC,GAAG,CAAC,UAAU,IAAI,0CAA0C,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC;AAC3F,CAAC;AAED,MAAM,UAAU,aAAa,CAC3B,KAIE;IAEF,IAAI,SAAS,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;IACpE,CAAC;IACD,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QACrB,YAAY,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW;IAC/B,IAAI,SAAS;QAAE,OAAO;IAEtB,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAChC,SAAS,GAAG,IAAI,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,eAAe;IAC7B,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,kBAAkB;IAChC,OAAO,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC;AAC3C,CAAC;AAED,MAAM,UAAU,uBAAuB;IAQrC,OAAO,CAAC,GAAG,CAAC,sDAAsD,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC;IACvF,MAAM,KAAK,GAOP,EAAE,CAAC;IACP,KAAK,MAAM,IAAI,IAAI,YAAY,CAAC,MAAM,EAAE,EAAE,CAAC;QACzC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC;QAClC,OAAO,CAAC,GAAG,CAAC,aAAa,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IACxC,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,gBAAgB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,eAAe,CAAC,CAAC;IACtE,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,IAAY;IACzC,OAAO,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC;AACzC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,IAAY,EAAE,MAA+B;IACvF,MAAM,IAAI,GAAG,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACpC,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,MAAM,IAAI,KAAK,CAAC,SAAS,IAAI,aAAa,CAAC,CAAC;IAC9C,CAAC;IACD,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;AACpC,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,UAAkB;IAChD,OAAO,CAAC,GAAG,EAAE,CAAC;IACd,KAAK,MAAM,IAAI,IAAI,YAAY,CAAC,MAAM,EAAE,EAAE,CAAC;QACzC,IAAI,IAAI,CAAC,WAAW,EAAE,UAAU,KAAK,UAAU,EAAE,CAAC;YAChD,OAAO,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;QACjC,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC"}

package/package.json

@@ -0,0 +1,38 @@
+{
+  "name": "@auto-engineer/ai-gateway",
+  "version": "0.1.1",
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "scripts": {
+    "build": "tsc && tsx ../../scripts/fix-esm-imports.ts",
+    "dev": "tsc --watch",
+    "test": "vitest run",
+    "lint": "eslint 'src/**/*.ts' --max-warnings 0 --config ../../eslint.config.ts",
+    "lint:fix": "eslint 'src/**/*.ts' --fix --config ../../eslint.config.ts",
+    "type-check": "tsc --noEmit",
+    "format": "prettier --write \"**/*.{js,ts,json,md,yml,yaml}\" --ignore-path ../../.prettierignore",
+    "format:fix": "prettier --write \"**/*.{js,ts,json,md,yml,yaml}\" --ignore-path ../../.prettierignore",
+    "link:dev": "pnpm build && pnpm link --global",
+    "unlink:dev": "pnpm unlink --global"
+  },
+  "dependencies": {
+    "@ai-sdk/anthropic": "^1.2.12",
+    "@ai-sdk/google": "^1.2.19",
+    "@ai-sdk/openai": "^1.3.22",
+    "@ai-sdk/xai": "^1.2.16",
+    "@modelcontextprotocol/sdk": "^1.3.0",
+    "ai": "^4.3.16",
+    "dotenv": "^16.4.5",
+    "zod": "^3.25.67"
+  },
+  "publishConfig": {
+    "access": "public"
+  }
+}

package/src/config.ts

@@ -0,0 +1,39 @@
+import dotenv from 'dotenv';
+import { fileURLToPath } from 'url';
+import { dirname, resolve } from 'path';
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+dotenv.config({ path: resolve(__dirname, '../../../.env') });
+
+export interface AIConfig {
+  openai?: {
+    apiKey: string;
+  };
+  anthropic?: {
+    apiKey: string;
+  };
+  google?: {
+    apiKey: string;
+  };
+  xai?: {
+    apiKey: string;
+  };
+}
+
+export function configureAIProvider(): AIConfig {
+  const config = {
+    openai: process.env.OPENAI_API_KEY != null ? { apiKey: process.env.OPENAI_API_KEY } : undefined,
+    anthropic: process.env.ANTHROPIC_API_KEY != null ? { apiKey: process.env.ANTHROPIC_API_KEY } : undefined,
+    google:
+      process.env.GOOGLE_GENERATIVE_AI_API_KEY != null
+        ? { apiKey: process.env.GOOGLE_GENERATIVE_AI_API_KEY }
+        : undefined,
+    xai: process.env.XAI_API_KEY != null ? { apiKey: process.env.XAI_API_KEY } : undefined,
+  };
+  if (config.openai == null && config.anthropic == null && config.google == null && config.xai == null) {
+    throw new Error(
+      'At least one AI provider must be configured. Please set OPENAI_API_KEY, ANTHROPIC_API_KEY, GOOGLE_GENERATIVE_AI_API_KEY, or XAI_API_KEY environment variables.',
+    );
+  }
+  return config;
+}