@cogitator-ai/swarms 0.1.0 → 0.3.1

package/dist/assessor/assessor.d.ts

@@ -0,0 +1,22 @@
+import type { SwarmConfig, AssessorConfig, AssessmentResult, TaskRequirements, ModelCandidate, Assessor } from '@cogitator-ai/types';
+export declare class SwarmAssessor implements Assessor {
+    private config;
+    private taskAnalyzer;
+    private modelDiscovery;
+    private modelScorer;
+    private roleMatcher;
+    private assessmentCache;
+    constructor(config?: AssessorConfig);
+    analyze(task: string, swarmConfig: SwarmConfig): Promise<AssessmentResult>;
+    assignModels(config: SwarmConfig, result: AssessmentResult): SwarmConfig;
+    suggestModels(requirements: TaskRequirements): Promise<ModelCandidate[]>;
+    private getCacheKey;
+    private hashString;
+    private getAgentNames;
+    private detectProvider;
+    private estimateTotalCost;
+    private estimateModelCost;
+    private optimizeForBudget;
+}
+export declare function createAssessor(config?: AssessorConfig): SwarmAssessor;
+//# sourceMappingURL=assessor.d.ts.map

package/dist/assessor/assessor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"assessor.d.ts","sourceRoot":"","sources":["../../src/assessor/assessor.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,WAAW,EACX,cAAc,EACd,gBAAgB,EAEhB,gBAAgB,EAChB,cAAc,EAGd,QAAQ,EAGT,MAAM,qBAAqB,CAAC;AA2B7B,qBAAa,aAAc,YAAW,QAAQ;IAC5C,OAAO,CAAC,MAAM,CAAyB;IACvC,OAAO,CAAC,YAAY,CAAe;IACnC,OAAO,CAAC,cAAc,CAAiB;IACvC,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,eAAe,CAAsE;gBAEjF,MAAM,GAAE,cAAmB;IAQjC,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,WAAW,GAAG,OAAO,CAAC,gBAAgB,CAAC;IA4GhF,YAAY,CAAC,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,gBAAgB,GAAG,WAAW;IAsDlE,aAAa,CAAC,YAAY,EAAE,gBAAgB,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;IAqB9E,OAAO,CAAC,WAAW;IAYnB,OAAO,CAAC,UAAU;IAUlB,OAAO,CAAC,aAAa;IAUrB,OAAO,CAAC,cAAc;IAStB,OAAO,CAAC,iBAAiB;IAkBzB,OAAO,CAAC,iBAAiB;IAWzB,OAAO,CAAC,iBAAiB;CAwC1B;AAED,wBAAgB,cAAc,CAAC,MAAM,CAAC,EAAE,cAAc,GAAG,aAAa,CAErE"}

package/dist/assessor/assessor.js

@@ -0,0 +1,267 @@
+import { TaskAnalyzer } from './task-analyzer';
+import { ModelDiscovery } from './model-discovery';
+import { ModelScorer } from './scoring';
+import { RoleMatcher } from './role-matcher';
+const DEFAULT_CONFIG = {
+    mode: 'rules',
+    assessorModel: 'gpt-4o-mini',
+    preferLocal: true,
+    minCapabilityMatch: 0.3,
+    ollamaUrl: 'http://localhost:11434',
+    enabledProviders: ['ollama', 'openai', 'anthropic', 'google'],
+    cacheAssessments: true,
+    cacheTTL: 5 * 60 * 1000,
+};
+const TOKEN_ESTIMATES = {
+    simple: 500,
+    moderate: 1500,
+    complex: 4000,
+};
+export class SwarmAssessor {
+    config;
+    taskAnalyzer;
+    modelDiscovery;
+    modelScorer;
+    roleMatcher;
+    assessmentCache = new Map();
+    constructor(config = {}) {
+        this.config = { ...DEFAULT_CONFIG, ...config };
+        this.taskAnalyzer = new TaskAnalyzer();
+        this.modelDiscovery = new ModelDiscovery(this.config);
+        this.modelScorer = new ModelScorer();
+        this.roleMatcher = new RoleMatcher();
+    }
+    async analyze(task, swarmConfig) {
+        const cacheKey = this.getCacheKey(task, swarmConfig);
+        if (this.config.cacheAssessments) {
+            const cached = this.assessmentCache.get(cacheKey);
+            if (cached && Date.now() - cached.timestamp < this.config.cacheTTL) {
+                return cached.result;
+            }
+        }
+        const taskAnalysis = this.taskAnalyzer.analyze(task);
+        const discoveredModels = await this.modelDiscovery.discoverAll();
+        const agents = this.roleMatcher.extractAgentsFromConfig(swarmConfig);
+        const roleAnalyses = new Map();
+        const assignments = [];
+        const warnings = [];
+        for (const agent of agents) {
+            if (agent.metadata.locked) {
+                assignments.push({
+                    agentName: agent.agent.name,
+                    originalModel: agent.agent.model ?? 'unknown',
+                    assignedModel: agent.agent.model ?? 'unknown',
+                    provider: this.detectProvider(agent.agent.model ?? ''),
+                    score: 100,
+                    reasons: ['Model locked by configuration'],
+                    fallbackModels: [],
+                    locked: true,
+                });
+                continue;
+            }
+            const roleReqs = this.roleMatcher.analyzeRole(agent, taskAnalysis);
+            roleAnalyses.set(agent.agent.name, roleReqs);
+            const scoredModels = this.modelScorer.scoreAll(discoveredModels, roleReqs);
+            const minScore = this.config.minCapabilityMatch * 100;
+            const validModels = scoredModels.filter((s) => s.score >= minScore);
+            if (validModels.length === 0) {
+                const originalModel = agent.agent.model ?? 'gpt-4o-mini';
+                warnings.push(`No suitable model found for ${agent.agent.name}, keeping original: ${originalModel}`);
+                assignments.push({
+                    agentName: agent.agent.name,
+                    originalModel,
+                    assignedModel: originalModel,
+                    provider: this.detectProvider(originalModel),
+                    score: 50,
+                    reasons: ['No better model found, using original'],
+                    fallbackModels: [],
+                    locked: false,
+                });
+                continue;
+            }
+            let selectedModel = validModels[0];
+            if (this.config.preferLocal) {
+                const localModel = validModels.find((m) => m.model.isLocal);
+                if (localModel && localModel.score >= selectedModel.score * 0.8) {
+                    selectedModel = localModel;
+                }
+            }
+            assignments.push({
+                agentName: agent.agent.name,
+                originalModel: agent.agent.model ?? 'unknown',
+                assignedModel: selectedModel.model.id,
+                provider: selectedModel.model.provider,
+                score: selectedModel.score,
+                reasons: selectedModel.reasons,
+                fallbackModels: validModels.slice(1, 4).map((s) => s.model.id),
+                locked: false,
+            });
+        }
+        if (this.config.maxCostPerRun) {
+            this.optimizeForBudget(assignments, discoveredModels, this.config.maxCostPerRun, taskAnalysis.complexity);
+        }
+        const result = {
+            taskAnalysis,
+            roleAnalyses,
+            assignments,
+            totalEstimatedCost: this.estimateTotalCost(assignments, discoveredModels, taskAnalysis.complexity),
+            warnings,
+            discoveredModels,
+        };
+        if (this.config.cacheAssessments) {
+            this.assessmentCache.set(cacheKey, { result, timestamp: Date.now() });
+        }
+        return result;
+    }
+    assignModels(config, result) {
+        const updatedConfig = { ...config };
+        for (const assignment of result.assignments) {
+            if (assignment.locked)
+                continue;
+            const updateAgent = (agent) => {
+                if (!agent || agent.name !== assignment.agentName)
+                    return agent;
+                return { ...agent, model: assignment.assignedModel };
+            };
+            if (updatedConfig.supervisor) {
+                updatedConfig.supervisor = updateAgent(updatedConfig.supervisor);
+            }
+            if (updatedConfig.moderator) {
+                updatedConfig.moderator = updateAgent(updatedConfig.moderator);
+            }
+            if (updatedConfig.router) {
+                updatedConfig.router = updateAgent(updatedConfig.router);
+            }
+            if (updatedConfig.workers) {
+                updatedConfig.workers = updatedConfig.workers.map((w) => updateAgent(w) ?? w);
+            }
+            if (updatedConfig.agents) {
+                updatedConfig.agents = updatedConfig.agents.map((a) => updateAgent(a) ?? a);
+            }
+            if (updatedConfig.stages) {
+                updatedConfig.stages = updatedConfig.stages.map((stage) => ({
+                    ...stage,
+                    agent: updateAgent(stage.agent) ?? stage.agent,
+                }));
+            }
+        }
+        return updatedConfig;
+    }
+    async suggestModels(requirements) {
+        const discoveredModels = await this.modelDiscovery.discoverAll();
+        const roleReqs = {
+            ...requirements,
+            role: 'worker',
+            agentName: '_suggestion',
+        };
+        const scoredModels = this.modelScorer.scoreAll(discoveredModels, roleReqs);
+        return scoredModels.slice(0, 10).map((s) => ({
+            modelId: s.model.id,
+            provider: s.model.provider,
+            score: s.score,
+            reasons: s.reasons,
+            isLocal: s.model.isLocal,
+            estimatedCost: (s.model.pricing.input + s.model.pricing.output) / 2,
+            capabilities: s.model.capabilities,
+        }));
+    }
+    getCacheKey(task, config) {
+        const taskHash = this.hashString(task.slice(0, 500));
+        const configHash = this.hashString(JSON.stringify({
+            name: config.name,
+            strategy: config.strategy,
+            agents: this.getAgentNames(config),
+        }));
+        return `${taskHash}-${configHash}`;
+    }
+    hashString(str) {
+        let hash = 0;
+        for (let i = 0; i < str.length; i++) {
+            const char = str.charCodeAt(i);
+            hash = (hash << 5) - hash + char;
+            hash = hash & hash;
+        }
+        return Math.abs(hash).toString(36);
+    }
+    getAgentNames(config) {
+        const names = [];
+        if (config.supervisor)
+            names.push(config.supervisor.name);
+        if (config.workers)
+            names.push(...config.workers.map((w) => w.name));
+        if (config.agents)
+            names.push(...config.agents.map((a) => a.name));
+        if (config.moderator)
+            names.push(config.moderator.name);
+        if (config.router)
+            names.push(config.router.name);
+        return names;
+    }
+    detectProvider(modelId) {
+        const lower = modelId.toLowerCase();
+        if (lower.includes('gpt') || lower.includes('o1') || lower.includes('o3'))
+            return 'openai';
+        if (lower.includes('claude'))
+            return 'anthropic';
+        if (lower.includes('gemini'))
+            return 'google';
+        if (lower.includes('mistral') || lower.includes('mixtral'))
+            return 'mistral';
+        return 'ollama';
+    }
+    estimateTotalCost(assignments, discoveredModels, complexity = 'moderate') {
+        const estimatedTokens = TOKEN_ESTIMATES[complexity];
+        let total = 0;
+        for (const assignment of assignments) {
+            const model = discoveredModels.find((m) => m.id === assignment.assignedModel);
+            if (model && !model.isLocal) {
+                total +=
+                    (model.pricing.input * estimatedTokens + model.pricing.output * estimatedTokens) /
+                        1_000_000;
+            }
+        }
+        return total;
+    }
+    estimateModelCost(model, complexity = 'moderate') {
+        if (model.isLocal)
+            return 0;
+        const estimatedTokens = TOKEN_ESTIMATES[complexity];
+        return ((model.pricing.input * estimatedTokens + model.pricing.output * estimatedTokens) / 1_000_000);
+    }
+    optimizeForBudget(assignments, discoveredModels, budget, complexity = 'moderate') {
+        let currentCost = this.estimateTotalCost(assignments, discoveredModels, complexity);
+        if (currentCost <= budget)
+            return;
+        const byExpense = [...assignments]
+            .filter((a) => !a.locked)
+            .map((a) => {
+            const model = discoveredModels.find((m) => m.id === a.assignedModel);
+            return {
+                assignment: a,
+                model,
+                cost: model ? this.estimateModelCost(model, complexity) : 0,
+            };
+        })
+            .sort((a, b) => b.cost - a.cost);
+        for (const item of byExpense) {
+            if (currentCost <= budget)
+                break;
+            for (const fallbackId of item.assignment.fallbackModels) {
+                const fallbackModel = discoveredModels.find((m) => m.id === fallbackId);
+                if (fallbackModel?.isLocal) {
+                    const oldCost = item.model ? this.estimateModelCost(item.model, complexity) : 0;
+                    const newCost = this.estimateModelCost(fallbackModel, complexity);
+                    item.assignment.assignedModel = fallbackId;
+                    item.assignment.provider = fallbackModel.provider;
+                    item.assignment.reasons.push('Downgraded for cost optimization');
+                    currentCost -= oldCost - newCost;
+                    break;
+                }
+            }
+        }
+    }
+}
+export function createAssessor(config) {
+    return new SwarmAssessor(config);
+}
+//# sourceMappingURL=assessor.js.map

package/dist/assessor/assessor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"assessor.js","sourceRoot":"","sources":["../../src/assessor/assessor.ts"],"names":[],"mappings":"AAaA,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAC/C,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACnD,OAAO,EAAE,WAAW,EAAE,MAAM,WAAW,CAAC;AACxC,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAM7C,MAAM,cAAc,GAA2B;IAC7C,IAAI,EAAE,OAAO;IACb,aAAa,EAAE,aAAa;IAC5B,WAAW,EAAE,IAAI;IACjB,kBAAkB,EAAE,GAAG;IACvB,SAAS,EAAE,wBAAwB;IACnC,gBAAgB,EAAE,CAAC,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,QAAQ,CAAC;IAC7D,gBAAgB,EAAE,IAAI;IACtB,QAAQ,EAAE,CAAC,GAAG,EAAE,GAAG,IAAI;CACxB,CAAC;AAEF,MAAM,eAAe,GAAmC;IACtD,MAAM,EAAE,GAAG;IACX,QAAQ,EAAE,IAAI;IACd,OAAO,EAAE,IAAI;CACd,CAAC;AAEF,MAAM,OAAO,aAAa;IAChB,MAAM,CAAyB;IAC/B,YAAY,CAAe;IAC3B,cAAc,CAAiB;IAC/B,WAAW,CAAc;IACzB,WAAW,CAAc;IACzB,eAAe,GAAG,IAAI,GAAG,EAA2D,CAAC;IAE7F,YAAY,SAAyB,EAAE;QACrC,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,MAAM,EAAE,CAAC;QAC/C,IAAI,CAAC,YAAY,GAAG,IAAI,YAAY,EAAE,CAAC;QACvC,IAAI,CAAC,cAAc,GAAG,IAAI,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACtD,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,EAAE,CAAC;QACrC,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,EAAE,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,IAAY,EAAE,WAAwB;QAClD,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;QACrD,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAC;YACjC,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAClD,IAAI,MAAM,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;gBACnE,OAAO,MAAM,CAAC,MAAM,CAAC;YACvB,CAAC;QACH,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAErD,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,WAAW,EAAE,CAAC;QAEjE,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,uBAAuB,CAAC,WAAW,CAAC,CAAC;QAErE,MAAM,YAAY,GAAG,IAAI,GAAG,EAA4B,CAAC;QACzD,MAAM,WAAW,GAAsB,EAAE,CAAC;QAC1C,MAAM,QAAQ,GAAa,EAAE,CAAC;QAE9B,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;gBAC1B,WAAW,CAAC,IAAI,CAAC;oBACf,SAAS,EAAE,KAAK,CAAC,KAAK,CAAC,IAAI;oBAC3B,aAAa,EAAG,KAAK,CAAC,KAA4B,CAAC,KAAK,IAAI,SAAS;oBACrE,aAAa,EAAG,KAAK,CAAC,KAA4B,CAAC,KAAK,IAAI,SAAS;oBACrE,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAE,KAAK,CAAC,KAA4B,CAAC,KAAK,IAAI,EAAE,CAAC;oBAC9E,KAAK,EAAE,GAAG;oBACV,OAAO,EAAE,CAAC,+BAA+B,CAAC;oBAC1C,cAAc,EAAE,EAAE;oBAClB,MAAM,EAAE,IAAI;iBACb,CAAC,CAAC;gBACH,SAAS;YACX,CAAC;YAED,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;YACnE,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;YAE7C,MAAM,YAAY,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,gBAAgB,EAAE,QAAQ,CAAC,CAAC;YAC3E,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,GAAG,GAAG,CAAC;YACtD,MAAM,WAAW,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,IAAI,QAAQ,CAAC,CAAC;YAEpE,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC7B,MAAM,aAAa,GAAI,KAAK,CAAC,KAA4B,CAAC,KAAK,IAAI,aAAa,CAAC;gBACjF,QAAQ,CAAC,IAAI,CACX,+BAA+B,KAAK,CAAC,KAAK,CAAC,IAAI,uBAAuB,aAAa,EAAE,CACtF,CAAC;gBACF,WAAW,CAAC,IAAI,CAAC;oBACf,SAAS,EAAE,KAAK,CAAC,KAAK,CAAC,IAAI;oBAC3B,aAAa;oBACb,aAAa,EAAE,aAAa;oBAC5B,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC;oBAC5C,KAAK,EAAE,EAAE;oBACT,OAAO,EAAE,CAAC,uCAAuC,CAAC;oBAClD,cAAc,EAAE,EAAE;oBAClB,MAAM,EAAE,KAAK;iBACd,CAAC,CAAC;gBACH,SAAS;YACX,CAAC;YAED,IAAI,aAAa,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;YACnC,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;gBAC5B,MAAM,UAAU,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBAC5D,IAAI,UAAU,IAAI,UAAU,CAAC,KAAK,IAAI,aAAa,CAAC,KAAK,GAAG,GAAG,EAAE,CAAC;oBAChE,aAAa,GAAG,UAAU,CAAC;gBAC7B,CAAC;YACH,CAAC;YAED,WAAW,CAAC,IAAI,CAAC;gBACf,SAAS,EAAE,KAAK,CAAC,KAAK,CAAC,IAAI;gBAC3B,aAAa,EAAG,KAAK,CAAC,KAA4B,CAAC,KAAK,IAAI,SAAS;gBACrE,aAAa,EAAE,aAAa,CAAC,KAAK,CAAC,EAAE;gBACrC,QAAQ,EAAE,aAAa,CAAC,KAAK,CAAC,QAAQ;gBACtC,KAAK,EAAE,aAAa,CAAC,KAAK;gBAC1B,OAAO,EAAE,aAAa,CAAC,OAAO;gBAC9B,cAAc,EAAE,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC9D,MAAM,EAAE,KAAK;aACd,CAAC,CAAC;QACL,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;YAC9B,IAAI,CAAC,iBAAiB,CACpB,WAAW,EACX,gBAAgB,EAChB,IAAI,CAAC,MAAM,CAAC,aAAa,EACzB,YAAY,CAAC,UAAU,CACxB,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAqB;YAC/B,YAAY;YACZ,YAAY;YACZ,WAAW;YACX,kBAAkB,EAAE,IAAI,CAAC,iBAAiB,CACxC,WAAW,EACX,gBAAgB,EAChB,YAAY,CAAC,UAAU,CACxB;YACD,QAAQ;YACR,gBAAgB;SACjB,CAAC;QAEF,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAC;YACjC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QACxE,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,YAAY,CAAC,MAAmB,EAAE,MAAwB;QACxD,MAAM,aAAa,GAAG,EAAE,GAAG,MAAM,EAAE,CAAC;QAEpC,KAAK,MAAM,UAAU,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;YAC5C,IAAI,UAAU,CAAC,MAAM;gBAAE,SAAS;YAEhC,MAAM,WAAW,GAAG,CAClB,KAAoB,EACL,EAAE;gBACjB,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,CAAC,SAAS;oBAAE,OAAO,KAAK,CAAC;gBAChE,OAAO,EAAE,GAAG,KAAK,EAAE,KAAK,EAAE,UAAU,CAAC,aAAa,EAAE,CAAC;YACvD,CAAC,CAAC;YAEF,IAAI,aAAa,CAAC,UAAU,EAAE,CAAC;gBAC7B,aAAa,CAAC,UAAU,GAAG,WAAW,CACpC,aAAa,CAAC,UAA8C,CAC1B,CAAC;YACvC,CAAC;YAED,IAAI,aAAa,CAAC,SAAS,EAAE,CAAC;gBAC5B,aAAa,CAAC,SAAS,GAAG,WAAW,CACnC,aAAa,CAAC,SAA6C,CAC1B,CAAC;YACtC,CAAC;YAED,IAAI,aAAa,CAAC,MAAM,EAAE,CAAC;gBACzB,aAAa,CAAC,MAAM,GAAG,WAAW,CAChC,aAAa,CAAC,MAA0C,CAC1B,CAAC;YACnC,CAAC;YAED,IAAI,aAAa,CAAC,OAAO,EAAE,CAAC;gBAC1B,aAAa,CAAC,OAAO,GAAG,aAAa,CAAC,OAAO,CAAC,GAAG,CAC/C,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,CAAqC,CAAC,IAAI,CAAC,CAC/B,CAAC;YACpC,CAAC;YAED,IAAI,aAAa,CAAC,MAAM,EAAE,CAAC;gBACzB,aAAa,CAAC,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC,GAAG,CAC7C,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,CAAqC,CAAC,IAAI,CAAC,CAChC,CAAC;YACnC,CAAC;YAED,IAAI,aAAa,CAAC,MAAM,EAAE,CAAC;gBACzB,aAAa,CAAC,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;oBAC1D,GAAG,KAAK;oBACR,KAAK,EAAE,WAAW,CAAC,KAAK,CAAC,KAAyC,CAAC,IAAI,KAAK,CAAC,KAAK;iBACnF,CAAC,CAAgC,CAAC;YACrC,CAAC;QACH,CAAC;QAED,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,YAA8B;QAChD,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,WAAW,EAAE,CAAC;QACjE,MAAM,QAAQ,GAAqB;YACjC,GAAG,YAAY;YACf,IAAI,EAAE,QAAQ;YACd,SAAS,EAAE,aAAa;SACzB,CAAC;QAEF,MAAM,YAAY,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,gBAAgB,EAAE,QAAQ,CAAC,CAAC;QAE3E,OAAO,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC3C,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE;YACnB,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,QAAQ;YAC1B,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,OAAO,EAAE,CAAC,CAAC,OAAO;YAClB,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO;YACxB,aAAa,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC;YACnE,YAAY,EAAE,CAAC,CAAC,KAAK,CAAC,YAAY;SACnC,CAAC,CAAC,CAAC;IACN,CAAC;IAEO,WAAW,CAAC,IAAY,EAAE,MAAmB;QACnD,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;QACrD,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAChC,IAAI,CAAC,SAAS,CAAC;YACb,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,MAAM,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC;SACnC,CAAC,CACH,CAAC;QACF,OAAO,GAAG,QAAQ,IAAI,UAAU,EAAE,CAAC;IACrC,CAAC;IAEO,UAAU,CAAC,GAAW;QAC5B,IAAI,IAAI,GAAG,CAAC,CAAC;QACb,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACpC,MAAM,IAAI,GAAG,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YAC/B,IAAI,GAAG,CAAC,IAAI,IAAI,CAAC,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC;YACjC,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC;QACrB,CAAC;QACD,OAAO,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACrC,CAAC;IAEO,aAAa,CAAC,MAAmB;QACvC,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,IAAI,MAAM,CAAC,UAAU;YAAE,KAAK,CAAC,IAAI,CAAE,MAAM,CAAC,UAA+B,CAAC,IAAI,CAAC,CAAC;QAChF,IAAI,MAAM,CAAC,OAAO;YAAE,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,CAAsB,CAAC,IAAI,CAAC,CAAC,CAAC;QAC3F,IAAI,MAAM,CAAC,MAAM;YAAE,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,CAAsB,CAAC,IAAI,CAAC,CAAC,CAAC;QACzF,IAAI,MAAM,CAAC,SAAS;YAAE,KAAK,CAAC,IAAI,CAAE,MAAM,CAAC,SAA8B,CAAC,IAAI,CAAC,CAAC;QAC9E,IAAI,MAAM,CAAC,MAAM;YAAE,KAAK,CAAC,IAAI,CAAE,MAAM,CAAC,MAA2B,CAAC,IAAI,CAAC,CAAC;QACxE,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,cAAc,CAAC,OAAe;QACpC,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QACpC,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC;YAAE,OAAO,QAAQ,CAAC;QAC3F,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,WAAW,CAAC;QACjD,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,QAAQ,CAAC;QAC9C,IAAI,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC;YAAE,OAAO,SAAS,CAAC;QAC7E,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,iBAAiB,CACvB,WAA8B,EAC9B,gBAAmC,EACnC,aAA6B,UAAU;QAEvC,MAAM,eAAe,GAAG,eAAe,CAAC,UAAU,CAAC,CAAC;QACpD,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,MAAM,KAAK,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,UAAU,CAAC,aAAa,CAAC,CAAC;YAC9E,IAAI,KAAK,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;gBAC5B,KAAK;oBACH,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,GAAG,eAAe,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,eAAe,CAAC;wBAChF,SAAS,CAAC;YACd,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,iBAAiB,CACvB,KAAsB,EACtB,aAA6B,UAAU;QAEvC,IAAI,KAAK,CAAC,OAAO;YAAE,OAAO,CAAC,CAAC;QAC5B,MAAM,eAAe,GAAG,eAAe,CAAC,UAAU,CAAC,CAAC;QACpD,OAAO,CACL,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,GAAG,eAAe,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,eAAe,CAAC,GAAG,SAAS,CAC7F,CAAC;IACJ,CAAC;IAEO,iBAAiB,CACvB,WAA8B,EAC9B,gBAAmC,EACnC,MAAc,EACd,aAA6B,UAAU;QAEvC,IAAI,WAAW,GAAG,IAAI,CAAC,iBAAiB,CAAC,WAAW,EAAE,gBAAgB,EAAE,UAAU,CAAC,CAAC;QACpF,IAAI,WAAW,IAAI,MAAM;YAAE,OAAO;QAElC,MAAM,SAAS,GAAG,CAAC,GAAG,WAAW,CAAC;aAC/B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;aACxB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACT,MAAM,KAAK,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,aAAa,CAAC,CAAC;YACrE,OAAO;gBACL,UAAU,EAAE,CAAC;gBACb,KAAK;gBACL,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;aAC5D,CAAC;QACJ,CAAC,CAAC;aACD,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;QAEnC,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;YAC7B,IAAI,WAAW,IAAI,MAAM;gBAAE,MAAM;YAEjC,KAAK,MAAM,UAAU,IAAI,IAAI,CAAC,UAAU,CAAC,cAAc,EAAE,CAAC;gBACxD,MAAM,aAAa,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,UAAU,CAAC,CAAC;gBACxE,IAAI,aAAa,EAAE,OAAO,EAAE,CAAC;oBAC3B,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;oBAChF,MAAM,OAAO,GAAG,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,UAAU,CAAC,CAAC;oBAElE,IAAI,CAAC,UAAU,CAAC,aAAa,GAAG,UAAU,CAAC;oBAC3C,IAAI,CAAC,UAAU,CAAC,QAAQ,GAAG,aAAa,CAAC,QAAQ,CAAC;oBAClD,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC;oBAEjE,WAAW,IAAI,OAAO,GAAG,OAAO,CAAC;oBACjC,MAAM;gBACR,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;CACF;AAED,MAAM,UAAU,cAAc,CAAC,MAAuB;IACpD,OAAO,IAAI,aAAa,CAAC,MAAM,CAAC,CAAC;AACnC,CAAC"}

package/dist/assessor/index.d.ts

@@ -0,0 +1,6 @@
+export { SwarmAssessor, createAssessor } from './assessor';
+export { TaskAnalyzer } from './task-analyzer';
+export { ModelDiscovery } from './model-discovery';
+export { ModelScorer, type ScoredModel } from './scoring';
+export { RoleMatcher } from './role-matcher';
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/assessor/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAC3D,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAC/C,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACnD,OAAO,EAAE,WAAW,EAAE,KAAK,WAAW,EAAE,MAAM,WAAW,CAAC;AAC1D,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC"}

package/dist/assessor/index.js

@@ -0,0 +1,6 @@
+export { SwarmAssessor, createAssessor } from './assessor';
+export { TaskAnalyzer } from './task-analyzer';
+export { ModelDiscovery } from './model-discovery';
+export { ModelScorer } from './scoring';
+export { RoleMatcher } from './role-matcher';
+//# sourceMappingURL=index.js.map

package/dist/assessor/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/assessor/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAC3D,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAC/C,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACnD,OAAO,EAAE,WAAW,EAAoB,MAAM,WAAW,CAAC;AAC1D,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC"}

package/dist/assessor/model-discovery.d.ts

@@ -0,0 +1,17 @@
+import type { DiscoveredModel, ModelProvider, AssessorConfig } from '@cogitator-ai/types';
+export declare class ModelDiscovery {
+    private ollamaUrl;
+    private enabledProviders;
+    private cache;
+    private cacheTTL;
+    constructor(config: AssessorConfig);
+    discoverAll(): Promise<DiscoveredModel[]>;
+    discoverOllama(): Promise<DiscoveredModel[]>;
+    private transformOllamaModel;
+    private inferOllamaCapabilities;
+    private inferContextWindow;
+    private formatModelName;
+    getCloudModels(filterProviders?: ModelProvider[]): DiscoveredModel[];
+    checkOllamaAvailability(): Promise<boolean>;
+}
+//# sourceMappingURL=model-discovery.d.ts.map

package/dist/assessor/model-discovery.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-discovery.d.ts","sourceRoot":"","sources":["../../src/assessor/model-discovery.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,eAAe,EACf,aAAa,EAEb,cAAc,EACf,MAAM,qBAAqB,CAAC;AA8M7B,qBAAa,cAAc;IACzB,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,gBAAgB,CAAqB;IAC7C,OAAO,CAAC,KAAK,CAAuE;IACpF,OAAO,CAAC,QAAQ,CAAS;gBAEb,MAAM,EAAE,cAAc;IAQ5B,WAAW,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;IAqBzC,cAAc,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;IAkBlD,OAAO,CAAC,oBAAoB;IAoB5B,OAAO,CAAC,uBAAuB;IAS/B,OAAO,CAAC,kBAAkB;IAS1B,OAAO,CAAC,eAAe;IASvB,cAAc,CAAC,eAAe,CAAC,EAAE,aAAa,EAAE,GAAG,eAAe,EAAE;IAO9D,uBAAuB,IAAI,OAAO,CAAC,OAAO,CAAC;CAUlD"}

package/dist/assessor/model-discovery.js

@@ -0,0 +1,286 @@
+const KNOWN_MODEL_CAPABILITIES = {
+    llama3: { supportsTools: true, supportsJson: true },
+    'llama3.1': { supportsTools: true, supportsJson: true },
+    'llama3.2': { supportsTools: true, supportsJson: true },
+    'llama3.3': { supportsTools: true, supportsJson: true },
+    qwen: { supportsTools: true, supportsJson: true },
+    qwen2: { supportsTools: true, supportsJson: true },
+    'qwen2.5': { supportsTools: true, supportsJson: true },
+    mistral: { supportsTools: true, supportsJson: true },
+    mixtral: { supportsTools: true, supportsJson: true },
+    llava: { supportsVision: true, supportsJson: true },
+    bakllava: { supportsVision: true, supportsJson: true },
+    'llama3.2-vision': { supportsVision: true, supportsTools: true, supportsJson: true },
+    'minicpm-v': { supportsVision: true, supportsJson: true },
+    moondream: { supportsVision: true },
+    deepseek: { supportsTools: true, supportsJson: true },
+    phi3: { supportsJson: true },
+    phi4: { supportsTools: true, supportsJson: true },
+    gemma: { supportsJson: true },
+    gemma2: { supportsJson: true },
+    codellama: { supportsJson: true },
+    starcoder: { supportsJson: true },
+};
+const KNOWN_CONTEXT_WINDOWS = {
+    'llama3.2:1b': 128000,
+    'llama3.2:3b': 128000,
+    'llama3.1': 128000,
+    llama3: 8192,
+    'qwen2.5': 128000,
+    qwen2: 32768,
+    mistral: 32768,
+    mixtral: 32768,
+    phi3: 128000,
+    phi4: 16384,
+    gemma2: 8192,
+    deepseek: 64000,
+};
+const CLOUD_MODELS = [
+    {
+        id: 'gpt-4o',
+        provider: 'openai',
+        displayName: 'GPT-4o',
+        capabilities: {
+            supportsVision: true,
+            supportsTools: true,
+            supportsJson: true,
+            supportsStreaming: true,
+        },
+        pricing: { input: 2.5, output: 10 },
+        contextWindow: 128000,
+        isLocal: false,
+        isAvailable: true,
+    },
+    {
+        id: 'gpt-4o-mini',
+        provider: 'openai',
+        displayName: 'GPT-4o Mini',
+        capabilities: {
+            supportsVision: true,
+            supportsTools: true,
+            supportsJson: true,
+            supportsStreaming: true,
+        },
+        pricing: { input: 0.15, output: 0.6 },
+        contextWindow: 128000,
+        isLocal: false,
+        isAvailable: true,
+    },
+    {
+        id: 'gpt-4-turbo',
+        provider: 'openai',
+        displayName: 'GPT-4 Turbo',
+        capabilities: {
+            supportsVision: true,
+            supportsTools: true,
+            supportsJson: true,
+            supportsStreaming: true,
+        },
+        pricing: { input: 10, output: 30 },
+        contextWindow: 128000,
+        isLocal: false,
+        isAvailable: true,
+    },
+    {
+        id: 'claude-sonnet-4-20250514',
+        provider: 'anthropic',
+        displayName: 'Claude Sonnet 4',
+        capabilities: {
+            supportsVision: true,
+            supportsTools: true,
+            supportsJson: true,
+            supportsStreaming: true,
+        },
+        pricing: { input: 3, output: 15 },
+        contextWindow: 200000,
+        isLocal: false,
+        isAvailable: true,
+    },
+    {
+        id: 'claude-3-5-sonnet-20241022',
+        provider: 'anthropic',
+        displayName: 'Claude 3.5 Sonnet',
+        capabilities: {
+            supportsVision: true,
+            supportsTools: true,
+            supportsJson: true,
+            supportsStreaming: true,
+        },
+        pricing: { input: 3, output: 15 },
+        contextWindow: 200000,
+        isLocal: false,
+        isAvailable: true,
+    },
+    {
+        id: 'claude-3-5-haiku-20241022',
+        provider: 'anthropic',
+        displayName: 'Claude 3.5 Haiku',
+        capabilities: { supportsTools: true, supportsJson: true, supportsStreaming: true },
+        pricing: { input: 0.8, output: 4 },
+        contextWindow: 200000,
+        isLocal: false,
+        isAvailable: true,
+    },
+    {
+        id: 'claude-3-opus-20240229',
+        provider: 'anthropic',
+        displayName: 'Claude 3 Opus',
+        capabilities: {
+            supportsVision: true,
+            supportsTools: true,
+            supportsJson: true,
+            supportsStreaming: true,
+        },
+        pricing: { input: 15, output: 75 },
+        contextWindow: 200000,
+        isLocal: false,
+        isAvailable: true,
+    },
+    {
+        id: 'gemini-1.5-pro',
+        provider: 'google',
+        displayName: 'Gemini 1.5 Pro',
+        capabilities: {
+            supportsVision: true,
+            supportsTools: true,
+            supportsJson: true,
+            supportsStreaming: true,
+        },
+        pricing: { input: 1.25, output: 5 },
+        contextWindow: 2000000,
+        isLocal: false,
+        isAvailable: true,
+    },
+    {
+        id: 'gemini-1.5-flash',
+        provider: 'google',
+        displayName: 'Gemini 1.5 Flash',
+        capabilities: {
+            supportsVision: true,
+            supportsTools: true,
+            supportsJson: true,
+            supportsStreaming: true,
+        },
+        pricing: { input: 0.075, output: 0.3 },
+        contextWindow: 1000000,
+        isLocal: false,
+        isAvailable: true,
+    },
+    {
+        id: 'gemini-2.0-flash',
+        provider: 'google',
+        displayName: 'Gemini 2.0 Flash',
+        capabilities: {
+            supportsVision: true,
+            supportsTools: true,
+            supportsJson: true,
+            supportsStreaming: true,
+        },
+        pricing: { input: 0.1, output: 0.4 },
+        contextWindow: 1000000,
+        isLocal: false,
+        isAvailable: true,
+    },
+];
+export class ModelDiscovery {
+    ollamaUrl;
+    enabledProviders;
+    cache = new Map();
+    cacheTTL;
+    constructor(config) {
+        this.ollamaUrl = config.ollamaUrl ?? 'http://localhost:11434';
+        this.enabledProviders = new Set(config.enabledProviders ?? ['ollama', 'openai', 'anthropic', 'google']);
+        this.cacheTTL = config.cacheTTL ?? 5 * 60 * 1000;
+    }
+    async discoverAll() {
+        const cacheKey = 'all';
+        const cached = this.cache.get(cacheKey);
+        if (cached && Date.now() - cached.timestamp < this.cacheTTL) {
+            return cached.models;
+        }
+        const models = [];
+        if (this.enabledProviders.has('ollama')) {
+            const ollamaModels = await this.discoverOllama();
+            models.push(...ollamaModels);
+        }
+        const cloudModels = this.getCloudModels();
+        models.push(...cloudModels);
+        this.cache.set(cacheKey, { models, timestamp: Date.now() });
+        return models;
+    }
+    async discoverOllama() {
+        try {
+            const response = await fetch(`${this.ollamaUrl}/api/tags`, {
+                signal: AbortSignal.timeout(5000),
+            });
+            if (!response.ok) {
+                return [];
+            }
+            const data = (await response.json());
+            return data.models.map((model) => this.transformOllamaModel(model));
+        }
+        catch {
+            return [];
+        }
+    }
+    transformOllamaModel(model) {
+        const name = model.name.toLowerCase();
+        const capabilities = this.inferOllamaCapabilities(name);
+        const contextWindow = this.inferContextWindow(name);
+        return {
+            id: model.name,
+            provider: 'ollama',
+            displayName: this.formatModelName(model.name),
+            capabilities: {
+                ...capabilities,
+                supportsStreaming: true,
+            },
+            pricing: { input: 0, output: 0 },
+            contextWindow,
+            isLocal: true,
+            isAvailable: true,
+        };
+    }
+    inferOllamaCapabilities(name) {
+        for (const [pattern, caps] of Object.entries(KNOWN_MODEL_CAPABILITIES)) {
+            if (name.includes(pattern.toLowerCase())) {
+                return { ...caps };
+            }
+        }
+        return { supportsStreaming: true };
+    }
+    inferContextWindow(name) {
+        for (const [pattern, size] of Object.entries(KNOWN_CONTEXT_WINDOWS)) {
+            if (name.includes(pattern.toLowerCase())) {
+                return size;
+            }
+        }
+        return 4096;
+    }
+    formatModelName(name) {
+        const [base, tag] = name.split(':');
+        const formatted = base
+            .split(/[-_]/)
+            .map((part) => part.charAt(0).toUpperCase() + part.slice(1))
+            .join(' ');
+        return tag && tag !== 'latest' ? `${formatted} (${tag})` : formatted;
+    }
+    getCloudModels(filterProviders) {
+        const providers = filterProviders
+            ? new Set(filterProviders.filter((p) => this.enabledProviders.has(p)))
+            : this.enabledProviders;
+        return CLOUD_MODELS.filter((model) => providers.has(model.provider));
+    }
+    async checkOllamaAvailability() {
+        try {
+            const response = await fetch(`${this.ollamaUrl}/api/tags`, {
+                signal: AbortSignal.timeout(2000),
+            });
+            return response.ok;
+        }
+        catch {
+            return false;
+        }
+    }
+}
+//# sourceMappingURL=model-discovery.js.map

package/dist/assessor/model-discovery.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-discovery.js","sourceRoot":"","sources":["../../src/assessor/model-discovery.ts"],"names":[],"mappings":"AAwBA,MAAM,wBAAwB,GAAmD;IAC/E,MAAM,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE;IACnD,UAAU,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE;IACvD,UAAU,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE;IACvD,UAAU,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE;IACvD,IAAI,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE;IACjD,KAAK,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE;IAClD,SAAS,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE;IACtD,OAAO,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE;IACpD,OAAO,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE;IACpD,KAAK,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE;IACnD,QAAQ,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE;IACtD,iBAAiB,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE;IACpF,WAAW,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE;IACzD,SAAS,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE;IACnC,QAAQ,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE;IACrD,IAAI,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE;IAC5B,IAAI,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE;IACjD,KAAK,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE;IAC7B,MAAM,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE;IAC9B,SAAS,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE;IACjC,SAAS,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE;CAClC,CAAC;AAEF,MAAM,qBAAqB,GAA2B;IACpD,aAAa,EAAE,MAAM;IACrB,aAAa,EAAE,MAAM;IACrB,UAAU,EAAE,MAAM;IAClB,MAAM,EAAE,IAAI;IACZ,SAAS,EAAE,MAAM;IACjB,KAAK,EAAE,KAAK;IACZ,OAAO,EAAE,KAAK;IACd,OAAO,EAAE,KAAK;IACd,IAAI,EAAE,MAAM;IACZ,IAAI,EAAE,KAAK;IACX,MAAM,EAAE,IAAI;IACZ,QAAQ,EAAE,KAAK;CAChB,CAAC;AAEF,MAAM,YAAY,GAAsB;IACtC;QACE,EAAE,EAAE,QAAQ;QACZ,QAAQ,EAAE,QAAQ;QAClB,WAAW,EAAE,QAAQ;QACrB,YAAY,EAAE;YACZ,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;YACnB,YAAY,EAAE,IAAI;YAClB,iBAAiB,EAAE,IAAI;SACxB;QACD,OAAO,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE;QACnC,aAAa,EAAE,MAAM;QACrB,OAAO,EAAE,KAAK;QACd,WAAW,EAAE,IAAI;KAClB;IACD;QACE,EAAE,EAAE,aAAa;QACjB,QAAQ,EAAE,QAAQ;QAClB,WAAW,EAAE,aAAa;QAC1B,YAAY,EAAE;YACZ,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;YACnB,YAAY,EAAE,IAAI;YAClB,iBAAiB,EAAE,IAAI;SACxB;QACD,OAAO,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE;QACrC,aAAa,EAAE,MAAM;QACrB,OAAO,EAAE,KAAK;QACd,WAAW,EAAE,IAAI;KAClB;IACD;QACE,EAAE,EAAE,aAAa;QACjB,QAAQ,EAAE,QAAQ;QAClB,WAAW,EAAE,aAAa;QAC1B,YAAY,EAAE;YACZ,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;YACnB,YAAY,EAAE,IAAI;YAClB,iBAAiB,EAAE,IAAI;SACxB;QACD,OAAO,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;QAClC,aAAa,EAAE,MAAM;QACrB,OAAO,EAAE,KAAK;QACd,WAAW,EAAE,IAAI;KAClB;IACD;QACE,EAAE,EAAE,0BAA0B;QAC9B,QAAQ,EAAE,WAAW;QACrB,WAAW,EAAE,iBAAiB;QAC9B,YAAY,EAAE;YACZ,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;YACnB,YAAY,EAAE,IAAI;YAClB,iBAAiB,EAAE,IAAI;SACxB;QACD,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE;QACjC,aAAa,EAAE,MAAM;QACrB,OAAO,EAAE,KAAK;QACd,WAAW,EAAE,IAAI;KAClB;IACD;QACE,EAAE,EAAE,4BAA4B;QAChC,QAAQ,EAAE,WAAW;QACrB,WAAW,EAAE,mBAAmB;QAChC,YAAY,EAAE;YACZ,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;YACnB,YAAY,EAAE,IAAI;YAClB,iBAAiB,EAAE,IAAI;SACxB;QACD,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE;QACjC,aAAa,EAAE,MAAM;QACrB,OAAO,EAAE,KAAK;QACd,WAAW,EAAE,IAAI;KAClB;IACD;QACE,EAAE,EAAE,2BAA2B;QAC/B,QAAQ,EAAE,WAAW;QACrB,WAAW,EAAE,kBAAkB;QAC/B,YAAY,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,iBAAiB,EAAE,IAAI,EAAE;QAClF,OAAO,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,EAAE;QAClC,aAAa,EAAE,MAAM;QACrB,OAAO,EAAE,KAAK;QACd,WAAW,EAAE,IAAI;KAClB;IACD;QACE,EAAE,EAAE,wBAAwB;QAC5B,QAAQ,EAAE,WAAW;QACrB,WAAW,EAAE,eAAe;QAC5B,YAAY,EAAE;YACZ,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;YACnB,YAAY,EAAE,IAAI;YAClB,iBAAiB,EAAE,IAAI;SACxB;QACD,OAAO,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;QAClC,aAAa,EAAE,MAAM;QACrB,OAAO,EAAE,KAAK;QACd,WAAW,EAAE,IAAI;KAClB;IACD;QACE,EAAE,EAAE,gBAAgB;QACpB,QAAQ,EAAE,QAAQ;QAClB,WAAW,EAAE,gBAAgB;QAC7B,YAAY,EAAE;YACZ,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;YACnB,YAAY,EAAE,IAAI;YAClB,iBAAiB,EAAE,IAAI;SACxB;QACD,OAAO,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE;QACnC,aAAa,EAAE,OAAO;QACtB,OAAO,EAAE,KAAK;QACd,WAAW,EAAE,IAAI;KAClB;IACD;QACE,EAAE,EAAE,kBAAkB;QACtB,QAAQ,EAAE,QAAQ;QAClB,WAAW,EAAE,kBAAkB;QAC/B,YAAY,EAAE;YACZ,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;YACnB,YAAY,EAAE,IAAI;YAClB,iBAAiB,EAAE,IAAI;SACxB;QACD,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE;QACtC,aAAa,EAAE,OAAO;QACtB,OAAO,EAAE,KAAK;QACd,WAAW,EAAE,IAAI;KAClB;IACD;QACE,EAAE,EAAE,kBAAkB;QACtB,QAAQ,EAAE,QAAQ;QAClB,WAAW,EAAE,kBAAkB;QAC/B,YAAY,EAAE;YACZ,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;YACnB,YAAY,EAAE,IAAI;YAClB,iBAAiB,EAAE,IAAI;SACxB;QACD,OAAO,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;QACpC,aAAa,EAAE,OAAO;QACtB,OAAO,EAAE,KAAK;QACd,WAAW,EAAE,IAAI;KAClB;CACF,CAAC;AAEF,MAAM,OAAO,cAAc;IACjB,SAAS,CAAS;IAClB,gBAAgB,CAAqB;IACrC,KAAK,GAAG,IAAI,GAAG,EAA4D,CAAC;IAC5E,QAAQ,CAAS;IAEzB,YAAY,MAAsB;QAChC,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,wBAAwB,CAAC;QAC9D,IAAI,CAAC,gBAAgB,GAAG,IAAI,GAAG,CAC7B,MAAM,CAAC,gBAAgB,IAAI,CAAC,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,QAAQ,CAAC,CACvE,CAAC;QACF,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,WAAW;QACf,MAAM,QAAQ,GAAG,KAAK,CAAC;QACvB,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACxC,IAAI,MAAM,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC5D,OAAO,MAAM,CAAC,MAAM,CAAC;QACvB,CAAC;QAED,MAAM,MAAM,GAAsB,EAAE,CAAC;QAErC,IAAI,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YACxC,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;YACjD,MAAM,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,CAAC;QAC/B,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QAC1C,MAAM,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,CAAC;QAE5B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAC5D,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,cAAc;QAClB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,SAAS,WAAW,EAAE;gBACzD,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC;aAClC,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,OAAO,EAAE,CAAC;YACZ,CAAC;YAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAuB,CAAC;YAE3D,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC;QACtE,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAEO,oBAAoB,CAAC,KAAsB;QACjD,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;QACtC,MAAM,YAAY,GAAG,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,CAAC;QACxD,MAAM,aAAa,GAAG,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;QAEpD,OAAO;YACL,EAAE,EAAE,KAAK,CAAC,IAAI;YACd,QAAQ,EAAE,QAAQ;YAClB,WAAW,EAAE,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC;YAC7C,YAAY,EAAE;gBACZ,GAAG,YAAY;gBACf,iBAAiB,EAAE,IAAI;aACxB;YACD,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE;YAChC,aAAa;YACb,OAAO,EAAE,IAAI;YACb,WAAW,EAAE,IAAI;SAClB,CAAC;IACJ,CAAC;IAEO,uBAAuB,CAAC,IAAY;QAC1C,KAAK,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,wBAAwB,CAAC,EAAE,CAAC;YACvE,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC;gBACzC,OAAO,EAAE,GAAG,IAAI,EAAE,CAAC;YACrB,CAAC;QACH,CAAC;QACD,OAAO,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC;IACrC,CAAC;IAEO,kBAAkB,CAAC,IAAY;QACrC,KAAK,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,EAAE,CAAC;YACpE,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC;gBACzC,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,eAAe,CAAC,IAAY;QAClC,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACpC,MAAM,SAAS,GAAG,IAAI;aACnB,KAAK,CAAC,MAAM,CAAC;aACb,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;aAC3D,IAAI,CAAC,GAAG,CAAC,CAAC;QACb,OAAO,GAAG,IAAI,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,SAAS,KAAK,GAAG,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC;IACvE,CAAC;IAED,cAAc,CAAC,eAAiC;QAC9C,MAAM,SAAS,GAAG,eAAe;YAC/B,CAAC,CAAC,IAAI,GAAG,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACtE,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC;QAC1B,OAAO,YAAY,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;IACvE,CAAC;IAED,KAAK,CAAC,uBAAuB;QAC3B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,SAAS,WAAW,EAAE;gBACzD,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC;aAClC,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC,EAAE,CAAC;QACrB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;CACF"}