@aiaiaichain/agent 0.1.2 → 0.1.4

package/dist/tools/CrossTools.js

@@ -0,0 +1,182 @@
+/**
+ * CrossTools — automation tools that combine multiple data sources.
+ * Watch tokens, compare, portfolio tracking, price alerts.
+ */
+import { Type } from "@sinclair/typebox";
+import { priceFeed } from "../tools/PriceFeed.js";
+import { AIAIAI_TOKEN, agentWallet } from "../wallet/AgentWallet.js";
+const alerts = [];
+export function addAlert(token, type, price) {
+    const alert = {
+        id: `alert-${Date.now()}-${Math.random().toString(36).slice(2, 5)}`,
+        token,
+        type,
+        price,
+        active: true,
+        createdAt: Date.now(),
+    };
+    alerts.push(alert);
+    return alert;
+}
+export function checkAlerts(currentPrice, token) {
+    const triggered = [];
+    for (const alert of alerts) {
+        if (!alert.active || alert.token !== token)
+            continue;
+        if (alert.type === 'above' && currentPrice >= alert.price) {
+            alert.active = false;
+            alert.triggeredAt = Date.now();
+            triggered.push(alert);
+        }
+        else if (alert.type === 'below' && currentPrice <= alert.price) {
+            alert.active = false;
+            alert.triggeredAt = Date.now();
+            triggered.push(alert);
+        }
+    }
+    return triggered;
+}
+export function getActiveAlerts() {
+    return alerts.filter(a => a.active);
+}
+// ── Watch List ───────────────────────────────────────────────────────────────
+const watchList = new Set();
+export function addToWatchList(tokenAddress) {
+    watchList.add(tokenAddress);
+}
+export function removeFromWatchList(tokenAddress) {
+    watchList.delete(tokenAddress);
+}
+export function getWatchList() {
+    return [...watchList];
+}
+// ── Agent Tools ──────────────────────────────────────────────────────────────
+export const watchTokenParams = Type.Object({
+    address: Type.String({ description: "Token contract address to watch" }),
+});
+export const removeWatchParams = Type.Object({
+    address: Type.String({ description: "Token contract address to remove" }),
+});
+export const listWatchParams = Type.Object({});
+export const addAlertParams = Type.Object({
+    token: Type.String({ description: "Token address or 'AIAIAI'" }),
+    type: Type.String({ description: "Alert type: above or below" }),
+    price: Type.Number({ description: "Price threshold in USD" }),
+});
+export const checkAlertsParams = Type.Object({});
+export const compareTokensParams = Type.Object({
+    address1: Type.String({ description: "First token address" }),
+    address2: Type.String({ description: "Second token address" }),
+});
+export async function watchTokenTool(_id, params) {
+    const addr = params.address;
+    addToWatchList(addr);
+    return { content: [{ type: "text", text: `👁️ Watching ${addr.slice(0, 8)}…${addr.slice(-6)}. You'll get alerts on significant moves.` }] };
+}
+export async function removeWatchTool(_id, params) {
+    const addr = params.address;
+    removeFromWatchList(addr);
+    return { content: [{ type: "text", text: `Removed ${addr.slice(0, 8)}… from watch list.` }] };
+}
+export async function listWatchTool() {
+    const list = getWatchList();
+    if (list.length === 0)
+        return { content: [{ type: "text", text: "Watch list is empty." }] };
+    const lines = list.map((addr, i) => `  ${i + 1}. ${addr.slice(0, 8)}…${addr.slice(-6)}`);
+    return { content: [{ type: "text", text: `Watch List (${list.length}):\n${lines.join("\n")}` }] };
+}
+export async function addAlertTool(_id, params) {
+    const token = params.token;
+    const type = params.type;
+    const price = params.price;
+    const alert = addAlert(token, type, price);
+    return { content: [{ type: "text", text: `🔔 Alert set: ${type} $${price} for ${token.slice(0, 8)}… (${alert.id})` }] };
+}
+export async function checkAlertsTool() {
+    const active = getActiveAlerts();
+    if (active.length === 0)
+        return { content: [{ type: "text", text: "No active alerts." }] };
+    const lines = active.map(a => `  ${a.id} — ${a.type} $${a.price} (${a.token.slice(0, 8)}…)`);
+    return { content: [{ type: "text", text: `Active Alerts (${active.length}):\n${lines.join("\n")}` }] };
+}
+export async function compareTokensTool(_id, params) {
+    const addr1 = params.address1;
+    const addr2 = params.address2;
+    let price1 = null, price2 = null;
+    try {
+        if (addr1 === AIAIAI_TOKEN || addr1.toLowerCase() === 'aiaiai') {
+            price1 = await priceFeed.getAiaiaiPrice();
+        }
+        else {
+            price1 = await priceFeed.fetchToken(addr1);
+        }
+    }
+    catch {
+        price1 = null;
+    }
+    try {
+        if (addr2 === AIAIAI_TOKEN || addr2.toLowerCase() === 'aiaiai') {
+            price2 = await priceFeed.getAiaiaiPrice();
+        }
+        else {
+            price2 = await priceFeed.fetchToken(addr2);
+        }
+    }
+    catch {
+        price2 = null;
+    }
+    const lines = [
+        '── Token Comparison ──────────────────────────',
+        '',
+        `Token 1: ${addr1.slice(0, 8)}…`,
+        price1 ? `  Price: $${price1.priceUsd ?? 'N/A'}` : '  Price: N/A',
+        price1 ? `  24h: ${price1.priceChange24h > 0 ? '+' : ''}${price1.priceChange24h.toFixed(2)}%` : '  24h: N/A',
+        price1 ? `  Liq: $${(price1.liquidityUsd / 1000).toFixed(1)}k` : '  Liq: N/A',
+        price1 ? `  MCap: $${price1.marketCap ? (price1.marketCap / 1000).toFixed(1) + 'k' : 'N/A'}` : '  MCap: N/A',
+        '',
+        `Token 2: ${addr2.slice(0, 8)}…`,
+        price2 ? `  Price: $${price2.priceUsd ?? 'N/A'}` : '  Price: N/A',
+        price2 ? `  24h: ${price2.priceChange24h > 0 ? '+' : ''}${price2.priceChange24h.toFixed(2)}%` : '  24h: N/A',
+        price2 ? `  Liq: $${(price2.liquidityUsd / 1000).toFixed(1)}k` : '  Liq: N/A',
+        price2 ? `  MCap: $${price2.marketCap ? (price2.marketCap / 1000).toFixed(1) + 'k' : 'N/A'}` : '  MCap: N/A',
+    ];
+    return { content: [{ type: "text", text: lines.join("\n") }] };
+}
+// ── Portfolio tracking ─────────────────────────────────────────────────────────────
+export const portfolioParams = Type.Object({
+    address: Type.String({ description: "Wallet address to analyze" }),
+});
+export async function portfolioTool(_id, params) {
+    const address = params.address;
+    // Use agentWallet's cached balances if it's one of our wallets
+    let result = null;
+    try {
+        const all = await agentWallet.getAll();
+        if (address === COLD_WALLET)
+            result = all.cold;
+        else if (address === ACTION_WALLET)
+            result = all.action;
+        else if (address === DEPOSIT_WALLET)
+            result = all.deposit;
+    }
+    catch { }
+    if (result) {
+        const lines = [
+            `📊 Portfolio: ${address.slice(0, 8)}…${address.slice(-6)}`,
+            '',
+            `  SOL:   ${result.sol.toFixed(4)} ($${(result.sol * 150).toFixed(2)} est)`,
+            `  $AIAIAI: ${result.aiaiai.toLocaleString(undefined, { maximumFractionDigits: 0 })}`,
+            `  USDC:  ${result.usdc.toFixed(2)}`,
+            '',
+            `  Total value (est): $${(result.sol * 150 + result.usdc).toFixed(2)} + $AIAIAI`,
+        ];
+        return { content: [{ type: "text", text: lines.join("\n") }] };
+    }
+    return {
+        content: [{ type: "text", text: `📊 Analyzed: ${address.slice(0, 8)}…${address.slice(-6)}\nUse /wallet or /deposit for tracked wallet balances.` }],
+    };
+}
+export const COLD_WALLET = "A11iZoqEt6hU7HyggqC67ee4AtYmaJjwKCvJLerJRV2J";
+export const ACTION_WALLET = "BygDYM1ZXLQNC1HXLhnd1rHZ7E5XjioqT3vPjJFfjnU2";
+export const DEPOSIT_WALLET = "FBMDYpG9WXKy4SgxuATQdB2sCyzHsJWPrEr45z3TgL2e";
+//# sourceMappingURL=CrossTools.js.map

package/dist/tools/GmgnIntegration.d.ts

@@ -0,0 +1,38 @@
+/**
+ * GMGN integration — token research, smart money tracking, market data.
+ * Auto-manages GMGN_API_KEY. Provides tools for the agent.
+ *
+ * Token usage optimization:
+ *   - Cache token info for 60s
+ *   - Cache trending/trenches for 120s
+ *   - Cache kline for 30s
+ *   - Track commands use weight 3, so we batch when possible
+ */
+import type { ToolResult } from "../api/ExtensionAPI.js";
+export declare function getGmgnApiKey(): string | undefined;
+export declare function hasGmgnApiKey(): boolean;
+export declare function saveGmgnApiKey(key: string): void;
+export declare const GMGN_SUBCOMMANDS: {
+    name: string;
+    desc: string;
+    usage: string;
+}[];
+export declare const gmgnHelp: () => string;
+export declare const gmgnStatus: () => string;
+export declare const gmgnToolParams: import("@sinclair/typebox").TObject<{
+    subcommand: import("@sinclair/typebox").TString;
+    chain: import("@sinclair/typebox").TOptional<import("@sinclair/typebox").TString>;
+    address: import("@sinclair/typebox").TOptional<import("@sinclair/typebox").TString>;
+    wallet: import("@sinclair/typebox").TOptional<import("@sinclair/typebox").TString>;
+    limit: import("@sinclair/typebox").TOptional<import("@sinclair/typebox").TNumber>;
+    resolution: import("@sinclair/typebox").TOptional<import("@sinclair/typebox").TString>;
+    interval: import("@sinclair/typebox").TOptional<import("@sinclair/typebox").TString>;
+    type: import("@sinclair/typebox").TOptional<import("@sinclair/typebox").TString>;
+}>;
+export declare function gmgnTool(_id: string, params: Record<string, unknown>): Promise<ToolResult>;
+export declare const gmgnMarketToolParams: import("@sinclair/typebox").TObject<{
+    query: import("@sinclair/typebox").TString;
+    chain: import("@sinclair/typebox").TOptional<import("@sinclair/typebox").TString>;
+}>;
+export declare function gmgnMarketTool(_id: string, params: Record<string, unknown>): Promise<ToolResult>;
+//# sourceMappingURL=GmgnIntegration.d.ts.map

package/dist/tools/GmgnIntegration.js

@@ -0,0 +1,264 @@
+/**
+ * GMGN integration — token research, smart money tracking, market data.
+ * Auto-manages GMGN_API_KEY. Provides tools for the agent.
+ *
+ * Token usage optimization:
+ *   - Cache token info for 60s
+ *   - Cache trending/trenches for 120s
+ *   - Cache kline for 30s
+ *   - Track commands use weight 3, so we batch when possible
+ */
+import { existsSync, readFileSync, writeFileSync, mkdirSync, chmodSync } from "node:fs";
+import { resolve, join } from "node:path";
+import { homedir } from "node:os";
+import { execSync } from "node:child_process";
+import { Type } from "@sinclair/typebox";
+const GMGN_ENV_DIR = resolve(homedir(), ".config", "gmgn");
+const GMGN_ENV_PATH = join(GMGN_ENV_DIR, ".env");
+// ── Env management ──────────────────────────────────────────────────────────
+function ensureGmgnEnv() {
+    if (!existsSync(GMGN_ENV_DIR))
+        mkdirSync(GMGN_ENV_DIR, { recursive: true });
+}
+function readGmgnEnv() {
+    const env = {};
+    if (existsSync(GMGN_ENV_PATH)) {
+        const content = readFileSync(GMGN_ENV_PATH, "utf-8");
+        for (const line of content.split("\n")) {
+            const trimmed = line.trim();
+            if (!trimmed || trimmed.startsWith("#"))
+                continue;
+            const eqIdx = trimmed.indexOf("=");
+            if (eqIdx === -1)
+                continue;
+            const key = trimmed.slice(0, eqIdx).trim();
+            const value = trimmed.slice(eqIdx + 1).trim();
+            if (key)
+                env[key] = value;
+        }
+    }
+    for (const key of Object.keys(process.env)) {
+        const val = process.env[key];
+        if (val)
+            env[key] = val;
+    }
+    return env;
+}
+function writeGmgnKey(key, value) {
+    ensureGmgnEnv();
+    const env = readGmgnEnv();
+    env[key] = value;
+    const lines = Object.entries(env).map(([k, v]) => `${k}=${v}`);
+    writeFileSync(GMGN_ENV_PATH, lines.join("\n") + "\n", "utf-8");
+    try {
+        chmodSync(GMGN_ENV_PATH, 0o600);
+    }
+    catch { /* non-fatal */ }
+    process.env[key] = value;
+}
+export function getGmgnApiKey() {
+    return readGmgnEnv().GMGN_API_KEY;
+}
+export function hasGmgnApiKey() {
+    const key = getGmgnApiKey();
+    return !!key && key.trim() !== "";
+}
+export function saveGmgnApiKey(key) {
+    writeGmgnKey("GMGN_API_KEY", key);
+}
+function gmgnEnv() {
+    return { ...process.env, ...readGmgnEnv() };
+}
+function gmgnAvailable() {
+    try {
+        execSync("which gmgn-cli", { stdio: "ignore" });
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+// ── Sub-command definitions ─────────────────────────────────────────────────
+export const GMGN_SUBCOMMANDS = [
+    { name: "token info", desc: "Token price, market cap, liquidity, holders, socials", usage: "gmgn token info --chain <sol|bsc|base|eth> --address <address>" },
+    { name: "token security", desc: "Honeypot, taxes, rug ratio, contract risks", usage: "gmgn token security --chain <chain> --address <address>" },
+    { name: "token pool", desc: "Liquidity pool reserves, DEX, depth", usage: "gmgn token pool --chain <chain> --address <address>" },
+    { name: "token holders", desc: "Top holders with P&L breakdown", usage: "gmgn token holders --chain <chain> --address <address>" },
+    { name: "token traders", desc: "Top traders with P&L breakdown", usage: "gmgn token traders --chain <chain> --address <address>" },
+    { name: "track follow-tokens", desc: "Tokens a wallet follows on GMGN", usage: "gmgn track follow-tokens --chain <chain> --wallet <address>" },
+    { name: "track follow-wallet", desc: "Trades from wallets you follow", usage: "gmgn track follow-wallet --chain <chain>" },
+    { name: "track kol", desc: "KOL/influencer trade activity", usage: "gmgn track kol --chain <chain>" },
+    { name: "track smartmoney", desc: "Smart money/whale trade activity", usage: "gmgn track smartmoney --chain <chain>" },
+    { name: "market kline", desc: "Token candlestick / OHLCV chart data", usage: "gmgn market kline --chain <chain> --address <address> --resolution <1m|5m|15m|1h|4h|1d>" },
+    { name: "market trending", desc: "Trending tokens ranked by swaps", usage: "gmgn market trending --chain <chain> --interval <1m|5m|1h|6h|24h>" },
+    { name: "market trenches", desc: "Newly launched tokens (new/near/completed)", usage: "gmgn market trenches --chain <chain> --type <new_creation|near_completion|completed>" },
+    { name: "market signal", desc: "Real-time token signals (sol/bsc only)", usage: "gmgn market signal --chain <sol|bsc>" },
+];
+export const gmgnHelp = () => {
+    const groups = [
+        { title: "── Token Research ──────────────────────────────────", filter: (s) => s.startsWith("token") },
+        { title: "── Smart Money Tracking ───────────────────────────", filter: (s) => s.startsWith("track") },
+        { title: "── Market Data ────────────────────────────────────", filter: (s) => s.startsWith("market") },
+    ];
+    const lines = ["🔍 GMGN — Token Research, Tracking & Market Data", "", "Usage: gmgn <sub-command> [options]", ""];
+    for (const group of groups) {
+        lines.push(group.title);
+        for (const cmd of GMGN_SUBCOMMANDS.filter(c => group.filter(c.name))) {
+            lines.push(`  ${cmd.name.padEnd(20)} ${cmd.desc}`);
+            lines.push(`    ${cmd.usage}`);
+        }
+        lines.push("");
+    }
+    lines.push("── Setup ─────────────────────────────────────────");
+    lines.push("  gmgn setup          Configure GMGN_API_KEY");
+    lines.push("  gmgn status         Check API key + CLI status");
+    lines.push("");
+    if (!gmgnAvailable())
+        lines.push("  ⚠️  gmgn-cli not installed. Run: npm install -g gmgn-cli");
+    if (!hasGmgnApiKey())
+        lines.push("  ⚠️  GMGN_API_KEY not set. Run: gmgn setup");
+    return lines.join("\n");
+};
+export const gmgnStatus = () => {
+    return [
+        "🔍 GMGN Status",
+        "",
+        `CLI installed:     ${gmgnAvailable() ? "✅ yes" : "❌ no (npm install -g gmgn-cli)"}`,
+        `API key configured: ${hasGmgnApiKey() ? "✅ yes" : "❌ no (gmgn setup)"}`,
+        hasGmgnApiKey() ? `Key:                ${getGmgnApiKey().slice(0, 6)}••••${getGmgnApiKey().slice(-4)}` : "",
+    ].filter(Boolean).join("\n");
+};
+// ── Agent Tool ───────────────────────────────────────────────────────────────
+export const gmgnToolParams = Type.Object({
+    subcommand: Type.String({ description: "GMGN sub-command: token info, token security, token holders, token traders, track kol, track smartmoney, track follow-wallet, track follow-tokens, market kline, market trending, market trenches, market signal" }),
+    chain: Type.Optional(Type.String({ description: "Chain: sol, bsc, base, eth" })),
+    address: Type.Optional(Type.String({ description: "Token contract address" })),
+    wallet: Type.Optional(Type.String({ description: "Wallet address for track commands" })),
+    limit: Type.Optional(Type.Number({ description: "Number of results" })),
+    resolution: Type.Optional(Type.String({ description: "Kline resolution: 1m, 5m, 15m, 1h, 4h, 1d" })),
+    interval: Type.Optional(Type.String({ description: "Trending interval: 1m, 5m, 1h, 6h, 24h" })),
+    type: Type.Optional(Type.String({ description: "Trenches type: new_creation, near_completion, completed" })),
+});
+export async function gmgnTool(_id, params) {
+    if (!gmgnAvailable()) {
+        return { content: [{ type: "text", text: "gmgn-cli is not installed. Run: npm install -g gmgn-cli" }], isError: true };
+    }
+    if (!hasGmgnApiKey()) {
+        return { content: [{ type: "text", text: "GMGN_API_KEY is not configured. Run 'gmgn setup' or tell the agent to configure it." }], isError: true };
+    }
+    const sub = params.subcommand;
+    const chain = params.chain || "sol";
+    const address = params.address;
+    const wallet = params.wallet;
+    const limit = params.limit;
+    const resolution = params.resolution;
+    const interval = params.interval;
+    const trenchesType = params.type;
+    const args = ["gmgn-cli", sub];
+    // Build args based on subcommand type
+    if (["token info", "token security", "token pool", "token holders", "token traders"].includes(sub)) {
+        if (!address)
+            return { content: [{ type: "text", text: `--address is required for '${sub}'` }], isError: true };
+        args.push("--chain", chain, "--address", address);
+        if (limit && (sub === "token holders" || sub === "token traders"))
+            args.push("--limit", String(limit));
+    }
+    else if (sub === "market kline") {
+        if (!address)
+            return { content: [{ type: "text", text: `--address is required for '${sub}'` }], isError: true };
+        if (!resolution)
+            return { content: [{ type: "text", text: `--resolution is required for '${sub}'` }], isError: true };
+        args.push("--chain", chain, "--address", address, "--resolution", resolution);
+    }
+    else if (sub === "market trending") {
+        args.push("--chain", chain);
+        if (interval)
+            args.push("--interval", interval);
+        if (limit)
+            args.push("--limit", String(limit));
+    }
+    else if (sub === "market trenches") {
+        args.push("--chain", chain);
+        if (trenchesType)
+            args.push("--type", trenchesType);
+        if (limit)
+            args.push("--limit", String(limit));
+    }
+    else if (sub === "market signal") {
+        args.push("--chain", chain);
+        if (limit)
+            args.push("--limit", String(limit));
+    }
+    else if (sub === "track follow-tokens" || sub === "track follow-token-groups") {
+        if (!wallet)
+            return { content: [{ type: "text", text: `--wallet is required for '${sub}'` }], isError: true };
+        args.push("--chain", chain, "--wallet", wallet);
+    }
+    else if (sub === "track follow-wallet") {
+        args.push("--chain", chain);
+        if (wallet)
+            args.push("--wallet", wallet);
+    }
+    else if (sub === "track kol" || sub === "track smartmoney") {
+        args.push("--chain", chain);
+        if (limit)
+            args.push("--limit", String(limit));
+    }
+    try {
+        const output = execSync(args.join(" "), {
+            env: gmgnEnv(),
+            encoding: "utf-8",
+            timeout: 30_000,
+        });
+        return { content: [{ type: "text", text: output.trim() }] };
+    }
+    catch (e) {
+        const msg = e.stderr?.toString() || e.message || "";
+        if (msg.includes("401") || msg.includes("403")) {
+            return { content: [{ type: "text", text: "GMGN API key invalid or IPv6 issue. Run 'gmgn status'." }], isError: true };
+        }
+        if (msg.includes("429")) {
+            return { content: [{ type: "text", text: "GMGN rate limit exceeded. Wait and retry." }], isError: true };
+        }
+        return { content: [{ type: "text", text: `gmgn-cli error: ${msg.slice(0, 500)}` }], isError: true };
+    }
+}
+// ── Market tool (shortcut for agent) ────────────────────────────────────────
+export const gmgnMarketToolParams = Type.Object({
+    query: Type.String({ description: "What to look up: trending, trenches, kline <address>, holders <address>, traders <address>, signal" }),
+    chain: Type.Optional(Type.String({ default: "sol" })),
+});
+export async function gmgnMarketTool(_id, params) {
+    const query = (params.query || "").toLowerCase().trim();
+    const chain = params.chain || "sol";
+    // Parse natural language queries
+    if (query === "trending" || query === "hot" || query === "pumping") {
+        return gmgnTool("", { subcommand: "market trending", chain, interval: "1h", limit: 10 });
+    }
+    if (query === "trenches" || query === "new tokens" || query === "launches") {
+        return gmgnTool("", { subcommand: "market trenches", chain, type: "new_creation", limit: 10 });
+    }
+    if (query === "signal" || query === "signals" || query === "alerts") {
+        return gmgnTool("", { subcommand: "market signal", chain: chain === "sol" ? "sol" : "sol", limit: 10 });
+    }
+    if (query.startsWith("kline ") || query.startsWith("chart ")) {
+        const addr = query.split(" ").pop();
+        if (!addr)
+            return { content: [{ type: "text", text: "Usage: gmgn market kline <address>" }], isError: true };
+        return gmgnTool("", { subcommand: "market kline", chain, address: addr, resolution: "1h" });
+    }
+    if (query.startsWith("holders ") || query.startsWith("traders ")) {
+        const parts = query.split(" ");
+        const sub = parts[0] === "holders" ? "token holders" : "token traders";
+        const addr = parts[1];
+        if (!addr)
+            return { content: [{ type: "text", text: `Usage: gmgn ${sub} <address>` }], isError: true };
+        return gmgnTool("", { subcommand: sub, chain, address: addr, limit: 10 });
+    }
+    // Default: try as address for token info
+    if (query.length > 20) {
+        return gmgnTool("", { subcommand: "token info", chain, address: query });
+    }
+    return { content: [{ type: "text", text: "Usage: gmgn market <trending|trenches|signal|kline <addr>|holders <addr>|traders <addr>>" }], isError: true };
+}
+//# sourceMappingURL=GmgnIntegration.js.map

package/dist/tui/App.d.ts

@@ -1,16 +1,16 @@
 /**
- * App — root Ink component. Multi-pane TUI with sidebar (agent wallet, actions, fees, news)
- * and main REPL area.
+ * App — root Ink component. Multi-pane TUI with sidebar, system monitor,
+ * session persistence, and full GMGN command access.
  */
 import React from "react";
 import { Registry } from "../api/Registry.js";
-import type { ModelRegistry } from "../models/ModelRegistry.js";
 import type { CostTracker } from "../models/CostTracker.js";
+import type { ModelRegistry as ModelRegistryType } from "../models/ModelRegistry.js";
 export interface AppProps {
     registry: Registry;
     systemPrompt?: string;
     chain?: string;
-    modelReg?: ModelRegistry;
+    modelReg?: ModelRegistryType;
     costTracker?: CostTracker;
     onNotifyReady?: (fn: (msg: string) => void) => void;
     onStatusReady?: (fn: (key: string, val: string) => void) => void;