task-o-matic-core 0.1.0

package/dist/lib/ai-service/mcp-client.js

@@ -0,0 +1,48 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Context7Client = void 0;
+const mcp_1 = require("@ai-sdk/mcp");
+const ai_service_factory_1 = require("../../utils/ai-service-factory");
+class Context7Client {
+    mcpClient = null;
+    async initializeMCPClient() {
+        if (this.mcpClient)
+            return this.mcpClient;
+        try {
+            // Connect to Context7 MCP server using HTTP transport
+            this.mcpClient = await (0, mcp_1.experimental_createMCPClient)({
+                transport: {
+                    type: "http",
+                    url: "https://mcp.context7.com/mcp",
+                    headers: {
+                        CONTEXT7_API_KEY: process.env.CONTEXT7_API_KEY || "",
+                    },
+                },
+            });
+            return this.mcpClient;
+        }
+        catch (error) {
+            throw error;
+        }
+    }
+    async getMCPTools() {
+        const client = await this.initializeMCPClient();
+        if (!client) {
+            throw new Error("Failed to initialize MCP client");
+        }
+        // Return original tools without wrapping - tool wrapping doesn't work with MCP tools
+        // Context7 result saving will be handled in ai-operations.ts by processing toolResults
+        return await client.tools();
+    }
+    async closeMCPConnection() {
+        if (this.mcpClient) {
+            await this.mcpClient.close();
+            this.mcpClient = null;
+        }
+    }
+    async saveContext7Documentation(library, query, content) {
+        const storage = (0, ai_service_factory_1.getStorage)();
+        return await storage.saveContext7Documentation(library, query, content);
+    }
+}
+exports.Context7Client = Context7Client;

package/dist/lib/ai-service/model-provider.d.ts

@@ -0,0 +1,12 @@
+import type { LanguageModel } from "ai";
+import { AIConfig } from "../../types";
+export declare class ModelProvider {
+    getAIConfig(): AIConfig;
+    /**
+     * Get environment-based config using ConfigManager's getEnv callback.
+     * This ensures all env var access goes through a single source of truth.
+     */
+    private getEnvConfig;
+    getModel(aiConfig: AIConfig): LanguageModel;
+}
+//# sourceMappingURL=model-provider.d.ts.map

package/dist/lib/ai-service/model-provider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-provider.d.ts","sourceRoot":"","sources":["../../../src/lib/ai-service/model-provider.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAOvC,qBAAa,aAAa;IACjB,WAAW,IAAI,QAAQ;IAY9B;;;OAGG;IACH,OAAO,CAAC,YAAY;IAqCpB,QAAQ,CAAC,QAAQ,EAAE,QAAQ,GAAG,aAAa;CA2H5C"}

package/dist/lib/ai-service/model-provider.js

@@ -0,0 +1,146 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ModelProvider = void 0;
+const openai_1 = require("@ai-sdk/openai");
+const anthropic_1 = require("@ai-sdk/anthropic");
+const openai_compatible_1 = require("@ai-sdk/openai-compatible");
+const ai_sdk_provider_1 = require("@openrouter/ai-sdk-provider");
+const config_1 = require("../config");
+const task_o_matic_error_1 = require("../../utils/task-o-matic-error");
+class ModelProvider {
+    getAIConfig() {
+        const config = config_1.configManager.getAIConfig();
+        const envConfig = this.getEnvConfig(config.provider);
+        // Override with environment variables if available
+        return {
+            ...config,
+            apiKey: config.apiKey || envConfig.apiKey,
+            baseURL: config.baseURL || envConfig.baseURL,
+        };
+    }
+    /**
+     * Get environment-based config using ConfigManager's getEnv callback.
+     * This ensures all env var access goes through a single source of truth.
+     */
+    getEnvConfig(provider) {
+        // Use a helper to get env vars - if configManager has custom callbacks,
+        // this will use those; otherwise falls back to process.env
+        const getEnv = (key) => {
+            try {
+                // Access through config structure which was built with getEnv callbacks
+                // or fall back to process.env for backwards compatibility
+                return process.env[key];
+            }
+            catch {
+                return undefined;
+            }
+        };
+        const envConfigMap = {
+            openai: {
+                apiKey: getEnv("OPENAI_API_KEY"),
+            },
+            anthropic: {
+                apiKey: getEnv("ANTHROPIC_API_KEY"),
+            },
+            zai: {
+                apiKey: getEnv("ZAI_API_KEY"),
+            },
+            openrouter: {
+                apiKey: getEnv("OPENROUTER_API_KEY"),
+                baseURL: "https://openrouter.ai/api/v1",
+            },
+            custom: {
+                apiKey: getEnv("CUSTOM_API_KEY"),
+                baseURL: getEnv("CUSTOM_API_URL"),
+            },
+        };
+        return envConfigMap[provider] || {};
+    }
+    getModel(aiConfig) {
+        const { provider, model, apiKey, baseURL } = aiConfig;
+        switch (provider) {
+            case "openai":
+                if (!apiKey)
+                    throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.AI_CONFIGURATION_ERROR, "OpenAI API key is required", {
+                        suggestions: [
+                            "Set the OPENAI_API_KEY environment variable.",
+                            "Run `task-o-matic config set-ai-key <key>`",
+                        ],
+                    });
+                return (0, openai_1.openai)(model);
+            case "anthropic":
+                if (!apiKey)
+                    throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.AI_CONFIGURATION_ERROR, "Anthropic API key is required", {
+                        suggestions: [
+                            "Set the ANTHROPIC_API_KEY environment variable.",
+                            "Run `task-o-matic config set-ai-key <key>`",
+                        ],
+                    });
+                return (0, anthropic_1.anthropic)(model);
+            case "openrouter":
+                if (!apiKey)
+                    throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.AI_CONFIGURATION_ERROR, "OpenRouter API key is required", {
+                        suggestions: [
+                            "Set the OPENROUTER_API_KEY environment variable.",
+                            "Run `task-o-matic config set-ai-key <key>`",
+                        ],
+                    });
+                const openRouterProvider = (0, ai_sdk_provider_1.createOpenRouter)({
+                    apiKey,
+                    headers: {
+                        "HTTP-Referer": "https://task-o-matic.dev",
+                        "X-Title": "Task-O-Matic",
+                    },
+                });
+                return openRouterProvider(model);
+            case "custom":
+                if (!apiKey)
+                    throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.AI_CONFIGURATION_ERROR, "Custom API key is required for custom provider", {
+                        suggestions: [
+                            "Set the CUSTOM_API_KEY environment variable.",
+                            "Run `task-o-matic config set-ai-key <key>`",
+                        ],
+                    });
+                if (!baseURL)
+                    throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.AI_CONFIGURATION_ERROR, "Custom provider requires baseURL", {
+                        suggestions: [
+                            "Set the CUSTOM_API_URL environment variable.",
+                            "Run `task-o-matic config set-ai-provider-url <url>`",
+                        ],
+                    });
+                const customProvider = (0, openai_compatible_1.createOpenAICompatible)({
+                    name: "custom",
+                    apiKey,
+                    baseURL,
+                });
+                return customProvider(model);
+            // disable for now, import error in the app
+            // case "gemini":
+            //   // Use Gemini CLI provider with OAuth via Proxy to handle lazy loading
+            //   return new GeminiProviderProxy(model, {
+            //     authType: "oauth-personal",
+            //   });
+            case "zai":
+                if (!apiKey)
+                    throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.AI_CONFIGURATION_ERROR, "Z.AI Coding plan API key is required", {
+                        suggestions: [
+                            "Set the ZAI_API_KEY environment variable.",
+                            "Run `task-o-matic config set-ai-key <key>`",
+                        ],
+                    });
+                const zaiProvider = (0, anthropic_1.createAnthropic)({
+                    baseURL: "https://api.z.ai/api/anthropic/v1",
+                    apiKey,
+                });
+                return zaiProvider(model);
+            default:
+                throw (0, task_o_matic_error_1.createStandardError)(task_o_matic_error_1.TaskOMaticErrorCodes.AI_CONFIGURATION_ERROR, `Unsupported provider: ${provider}`, {
+                    suggestions: [
+                        "Use one of the supported providers: 'openai', 'anthropic', 'openrouter', 'custom', 'zai'.",
+                        "Run `task-o-matic config set-ai-provider <provider>`",
+                    ],
+                });
+        }
+    }
+}
+exports.ModelProvider = ModelProvider;

package/dist/lib/ai-service/prd-operations.d.ts

@@ -0,0 +1,25 @@
+import { AIConfig, AIPRDParseResult, StreamingOptions, RetryConfig, BTSConfig } from "../../types";
+import { BaseOperations } from "./base-operations";
+export declare class PRDOperations extends BaseOperations {
+    parsePRD(prdContent: string, config?: Partial<AIConfig>, promptOverride?: string, userMessage?: string, streamingOptions?: StreamingOptions, retryConfig?: Partial<RetryConfig>, workingDirectory?: string, enableFilesystemTools?: boolean): Promise<AIPRDParseResult>;
+    reworkPRD(prdContent: string, feedback: string, config?: Partial<AIConfig>, promptOverride?: string, userMessage?: string, streamingOptions?: StreamingOptions, retryConfig?: Partial<RetryConfig>, workingDirectory?: string, enableFilesystemTools?: boolean): Promise<string>;
+    generatePRDQuestions(prdContent: string, config?: Partial<AIConfig>, promptOverride?: string, userMessage?: string, streamingOptions?: StreamingOptions, retryConfig?: Partial<RetryConfig>, workingDirectory?: string, enableFilesystemTools?: boolean): Promise<string[]>;
+    answerPRDQuestions(prdContent: string, questions: string[], config?: Partial<AIConfig>, contextInfo?: {
+        stackInfo?: string;
+        projectDescription?: string;
+    }, streamingOptions?: StreamingOptions, retryConfig?: Partial<RetryConfig>): Promise<Record<string, string>>;
+    generatePRD(description: string, config?: Partial<AIConfig>, promptOverride?: string, userMessage?: string, streamingOptions?: StreamingOptions, retryConfig?: Partial<RetryConfig>): Promise<string>;
+    combinePRDs(prds: string[], originalDescription: string, config?: Partial<AIConfig>, promptOverride?: string, userMessage?: string, streamingOptions?: StreamingOptions, retryConfig?: Partial<RetryConfig>): Promise<string>;
+    /**
+     * Suggest a technology stack based on PRD analysis.
+     */
+    suggestStack(prdContent: string, projectName?: string, config?: Partial<AIConfig>, promptOverride?: string, userMessage?: string, streamingOptions?: StreamingOptions, retryConfig?: Partial<RetryConfig>, workingDirectory?: string, enableFilesystemTools?: boolean): Promise<{
+        config: BTSConfig;
+        reasoning: string;
+    }>;
+    /**
+     * Validate and apply defaults to a partial BTSConfig.
+     */
+    private validateStackConfig;
+}
+//# sourceMappingURL=prd-operations.d.ts.map

package/dist/lib/ai-service/prd-operations.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"prd-operations.d.ts","sourceRoot":"","sources":["../../../src/lib/ai-service/prd-operations.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,QAAQ,EAER,gBAAgB,EAChB,gBAAgB,EAChB,WAAW,EAIX,SAAS,EACV,MAAM,aAAa,CAAC;AAcrB,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAMnD,qBAAa,aAAc,SAAQ,cAAc;IACzC,QAAQ,CACZ,UAAU,EAAE,MAAM,EAClB,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,EAClC,gBAAgB,CAAC,EAAE,MAAM,EACzB,qBAAqB,CAAC,EAAE,OAAO,GAC9B,OAAO,CAAC,gBAAgB,CAAC;IA6MtB,SAAS,CACb,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,MAAM,EAChB,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,EAClC,gBAAgB,CAAC,EAAE,MAAM,EACzB,qBAAqB,CAAC,EAAE,OAAO,GAC9B,OAAO,CAAC,MAAM,CAAC;IA+GZ,oBAAoB,CACxB,UAAU,EAAE,MAAM,EAClB,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,EAClC,gBAAgB,CAAC,EAAE,MAAM,EACzB,qBAAqB,CAAC,EAAE,OAAO,GAC9B,OAAO,CAAC,MAAM,EAAE,CAAC;IA6Hd,kBAAkB,CACtB,UAAU,EAAE,MAAM,EAClB,SAAS,EAAE,MAAM,EAAE,EACnB,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,WAAW,CAAC,EAAE;QACZ,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,kBAAkB,CAAC,EAAE,MAAM,CAAC;KAC7B,EACD,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GACjC,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IA8F5B,WAAW,CACf,WAAW,EAAE,MAAM,EACnB,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GACjC,OAAO,CAAC,MAAM,CAAC;IAqBZ,WAAW,CACf,IAAI,EAAE,MAAM,EAAE,EACd,mBAAmB,EAAE,MAAM,EAC3B,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GACjC,OAAO,CAAC,MAAM,CAAC;IA6BlB;;OAEG;IACG,YAAY,CAChB,UAAU,EAAE,MAAM,EAClB,WAAW,CAAC,EAAE,MAAM,EACpB,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,cAAc,CAAC,EAAE,MAAM,EACvB,WAAW,CAAC,EAAE,MAAM,EACpB,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,EAClC,gBAAgB,CAAC,EAAE,MAAM,EACzB,qBAAqB,CAAC,EAAE,OAAO,GAC9B,OAAO,CAAC;QAAE,MAAM,EAAE,SAAS,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC;IAiJpD;;OAEG;IACH,OAAO,CAAC,mBAAmB;CAqB5B"}