ask-ai-mcp/dist/ai-service.js

A Model Context Protocol server enabling AI-to-AI collaboration across multiple providers.

import { createAnthropic } from "@ai-sdk/anthropic";
import { createGoogleGenerativeAI } from "@ai-sdk/google";
import { createOpenAI } from "@ai-sdk/openai";
import { createPerplexity } from "@ai-sdk/perplexity";
import { generateText } from "ai";
export class AIService {
    config;
    constructor(config) {
        this.config = config;
        this.validateConfig();
    }
    validateConfig() {
        if (!this.config.model) {
            throw new Error("Model is required");
        }
        if (this.config.maxTokens !== undefined && this.config.maxTokens <= 0) {
            throw new Error("Max tokens must be greater than 0");
        }
    }
    getModel() {
        switch (this.config.provider) {
            case "openai":
            case "openai-compatible": {
                const settings = {};
                if (this.config.baseUrl) {
                    settings.baseURL = this.config.baseUrl;
                }
                else if (this.config.provider === "openai-compatible") {
                    throw new Error("Base URL is required for OpenAI-compatible providers");
                }
                if (this.config.apiKey)
                    settings.apiKey = this.config.apiKey;
                const openai = createOpenAI(settings);
                return openai(this.config.model);
            }
            case "anthropic": {
                const settings = {};
                if (this.config.apiKey)
                    settings.apiKey = this.config.apiKey;
                if (this.config.baseUrl)
                    settings.baseURL = this.config.baseUrl;
                const anthropic = createAnthropic(settings);
                return anthropic(this.config.model);
            }
            case "google": {
                const settings = {};
                if (this.config.apiKey)
                    settings.apiKey = this.config.apiKey;
                if (this.config.baseUrl)
                    settings.baseURL = this.config.baseUrl;
                const google = createGoogleGenerativeAI(settings);
                return google(this.config.model);
            }
            case "perplexity": {
                const settings = {};
                if (this.config.apiKey)
                    settings.apiKey = this.config.apiKey;
                if (this.config.baseUrl)
                    settings.baseURL = this.config.baseUrl;
                const perplexity = createPerplexity(settings);
                return perplexity(this.config.model);
            }
            default:
                throw new Error(`Unsupported provider: ${this.config.provider}`);
        }
    }
    async askAI(request) {
        try {
            const model = this.getModel();
            // Construct the prompt with context if provided
            let prompt = `<question>\n${request.question}\n</question>`;
            if (request.context && request.context.trim()) {
                prompt = `<context>
${request.context}
</context>

${prompt}`;
            }
            const generateOptions = {
                model,
                prompt,
                system: `You are a large language model acting as a collaborative assistant.

You are being queried by another AI model that is using a tool to ask for your help.

Your role is to provide a helpful, peer-level response to assist the calling AI. It may be seeking a second opinion, a different perspective, or help with a problem it cannot solve on its own.

Key considerations:
- The request is from another AI. Your response will be programmatically processed and used to help it complete its task.
- Frame your response as a suggestion or an alternative viewpoint, not as a definitive command or factual statement.
- The calling AI has provided the context it has available. If the context seems insufficient, you can note what additional information would be helpful, but understand that the calling AI may not be able to provide it.
- The questioner's assumptions, viewpoint, or chosen direction might be incorrect. Step back, adopt a broader perspective, and think carefully before answering.
- Be aware of your own limitations and knowledge cutoff. If you are uncertain or a request is beyond your capabilities, state that clearly rather than give a misleading or evasive reply.

Your goal is to help your fellow AI overcome its challenge. Provide clear, structured, and actionable information.`,
            };
            // Add optional parameters only if configured
            if (this.config.temperature !== undefined) {
                generateOptions.temperature = this.config.temperature;
            }
            if (this.config.maxTokens !== undefined) {
                generateOptions.maxTokens = this.config.maxTokens;
            }
            if (this.config.reasoningEffort !== undefined) {
                generateOptions.providerOptions = {
                    openai: { reasoningEffort: this.config.reasoningEffort },
                };
            }
            const result = await generateText(generateOptions);
            return {
                response: result.text,
                provider: this.config.provider,
                model: result.response.modelId,
            };
        }
        catch (error) {
            const errorMessage = error instanceof Error ? error.message : "Unknown error occurred";
            throw new Error(`AI service error: ${errorMessage}`);
        }
    }
}
//# sourceMappingURL=ai-service.js.map