@zhive/cli 0.5.4 → 0.5.5

package/dist/shared/agent/analysis.js

@@ -1,14 +1,14 @@
 import { getMemoryLineCount, loadMemory, MEMORY_SOFT_LIMIT, saveMemory, } from '@zhive/sdk';
 import * as ai from 'ai';
 import { z } from 'zod';
-import { getModel } from '../config/ai-providers.js';
+import { getModel, getScreenModel } from '../config/ai-providers.js';
 import { buildMemoryExtractionPrompt } from './prompts/memory-prompt.js';
 import { extractErrorMessage, stripCodeFences } from './utils.js';
 import { wrapAISDK } from 'langsmith/experimental/vercel';
 import { buildScreenPrompt } from './prompts/prompt.js';
 import { buildMegathreadInputPrompt, buildMegathreadSystemPrompt, } from './prompts/megathread.js';
 import { clearSubagentUsage, getSubagentUsage, } from './tools/index.js';
-const { ToolLoopAgent, generateText, Output } = wrapAISDK(ai);
+const { ToolLoopAgent, generateText, generateObject, Output } = wrapAISDK(ai);
 // ─── Cache Helpers ─────────────────────────────────
 function cacheableSystem(content) {
     const message = {
@@ -63,11 +63,11 @@ export async function screenMegathreadRound(projectId, strategyContent) {
         const { system, prompt } = buildScreenPrompt(strategyContent, {
             projectId,
         });
-        const model = await getModel();
-        const res = await generateText({
+        const model = await getScreenModel();
+        const res = await generateObject({
             model,
             messages: [cacheableSystem(system), { role: 'user', content: prompt }],
-            experimental_output: Output.object({ schema: screenSchema }),
+            schema: screenSchema,
         });
         const usage = {
             inputTokens: res.usage?.inputTokens ?? 0,
@@ -78,11 +78,7 @@ export async function screenMegathreadRound(projectId, strategyContent) {
             toolNames: [],
             toolResults: [],
         };
-        const output = res.experimental_output;
-        if (!output) {
-            return { engage: true, usage };
-        }
-        return { engage: output.engage, usage };
+        return { engage: res.object.engage, usage };
     }
     catch (err) {
         const emptyUsage = {
@@ -136,9 +132,10 @@ export async function processMegathreadRound({ projectId, durationMs, recentComm
         return { skip: true, summary: '', conviction: 0, usage };
     }
     const prediction = output;
-    const summary = prediction.summary ?? '';
-    const conviction = prediction.conviction ?? 0;
-    return { skip: false, summary, conviction, usage };
+    if (prediction.summary === null || prediction.conviction === null) {
+        return { skip: true, summary: '', conviction: 0, usage };
+    }
+    return { skip: false, summary: prediction.summary, conviction: prediction.conviction, usage };
 }
 // ─── Memory Extraction ──────────────────────────────
 export async function extractAndSaveMemory(sessionMessages) {

package/dist/shared/agent/prompts/prompt.js

@@ -8,7 +8,7 @@ Your trading strategy:
 ${strategyContent}
 ---
 
-Only engage with projects that match the agent's sectors and expertise as defined in the trading strategy above.
+Only engage with projects that match the agent's sectors and expertise as defined in the trading strategy above. If the strategy's sectors include "all" or cover all sectors, always engage — the agent predicts on everything.
 
 Answer with only "yes" or "no".`;
     const prompt = `Project: ${projectId}

package/dist/shared/config/ai-providers.js

@@ -127,30 +127,36 @@ export function resolveModelInfo() {
     }
     return { provider: 'unknown', modelId: 'unknown', source: 'unknown' };
 }
+async function _loadModelForTier(tier) {
+    const agentKeys = getAgentProviderKeys();
+    const sortedProviders = [
+        ...AI_PROVIDERS.filter((p) => agentKeys.has(p.envVar)),
+        ...AI_PROVIDERS.filter((p) => !agentKeys.has(p.envVar)),
+    ];
+    for (const provider of sortedProviders) {
+        const keyValue = process.env[provider.envVar];
+        if (keyValue && keyValue.trim().length > 0) {
+            const overrideModel = process.env.HIVE_MODEL;
+            const modelId = tier === 'runtime' && overrideModel ? overrideModel : provider.models[tier];
+            const model = await provider.load(modelId);
+            return model;
+        }
+    }
+    throw new Error('No AI provider API key found in environment. ' +
+        'Set one of: ANTHROPIC_API_KEY, OPENAI_API_KEY, GOOGLE_GENERATIVE_AI_API_KEY, XAI_API_KEY, OPENROUTER_API_KEY');
+}
+let _screenModelPromise = null;
+export function getScreenModel() {
+    if (_screenModelPromise) {
+        return _screenModelPromise;
+    }
+    _screenModelPromise = _loadModelForTier('validation');
+    return _screenModelPromise;
+}
 export function getModel() {
     if (_modelPromise) {
         return _modelPromise;
     }
-    _modelPromise = (async () => {
-        const info = resolveModelInfo();
-        if (info.provider === 'unknown') {
-            throw new Error('No AI provider API key found in environment. ' +
-                'Set one of: ANTHROPIC_API_KEY, OPENAI_API_KEY, GOOGLE_GENERATIVE_AI_API_KEY, XAI_API_KEY, OPENROUTER_API_KEY');
-        }
-        const agentKeys = getAgentProviderKeys();
-        const sortedProviders = [
-            ...AI_PROVIDERS.filter((p) => agentKeys.has(p.envVar)),
-            ...AI_PROVIDERS.filter((p) => !agentKeys.has(p.envVar)),
-        ];
-        for (const provider of sortedProviders) {
-            const keyValue = process.env[provider.envVar];
-            if (keyValue && keyValue.trim().length > 0) {
-                const modelId = info.modelId;
-                const model = await provider.load(modelId);
-                return model;
-            }
-        }
-        throw new Error('Unreachable: resolveModelInfo succeeded but no provider found');
-    })();
+    _modelPromise = _loadModelForTier('runtime');
     return _modelPromise;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zhive/cli",
-  "version": "0.5.4",
+  "version": "0.5.5",
   "description": "CLI for bootstrapping zHive AI Agents",
   "type": "module",
   "main": "dist/index.js",