RUN YOUR AGENT AS THE MAIN MODEL IN PI

import { Type } from "@earendil-works/pi-ai";
import { defineTool, type ExtensionAPI } from "@earendil-works/pi-coding-agent";

const API = "https://api.example-stocks.com";

// The model reads `content`. pi stores `details` in the session.
function result(text: string, details: Record<string, unknown>) {
	return { content: [{ type: "text" as const, text }], details };
}

const searchCompany = defineTool({
	name: "search_company",
	label: "Search Company",
	description: "Find a company's stock ticker symbol by company name. Call this before get_stock_price.",
	parameters: Type.Object({
		name: Type.String({ description: "Company name, e.g. 'Apple'." }),
	}),

	async execute(_toolCallId, params, signal) {
		const res = await fetch(`${API}/search?q=${encodeURIComponent(params.name)}`, { signal });
		if (res.status === 404) return result(`No company found matching "${params.name}".`, { found: false });
		if (!res.ok) throw new Error(`Search returned HTTP ${res.status}`);

		const { ticker, name } = (await res.json()) as { ticker: string; name: string };
		return result(`Found ${name} (ticker: ${ticker}).`, { found: true, ticker, name });
	},
});

const getStockPrice = defineTool({
	name: "get_stock_price",
	label: "Get Stock Price",
	description: "Get the latest stock price for a ticker symbol returned by search_company.",
	parameters: Type.Object({
		ticker: Type.String({ description: "Stock ticker symbol, e.g. 'AAPL'." }),
	}),

	async execute(_toolCallId, params, signal) {
		const res = await fetch(`${API}/quote/${encodeURIComponent(params.ticker)}`, { signal });
		if (res.status === 404) return result(`No price found for "${params.ticker}".`, { found: false });
		if (!res.ok) throw new Error(`Quote returned HTTP ${res.status}`);

		const { price, currency } = (await res.json()) as { price: number; currency: string };
		return result(`${params.ticker}: ${price} ${currency}.`, { found: true, price, currency });
	},
});

export default function (pi: ExtensionAPI) {
	pi.registerTool(searchCompany);
	pi.registerTool(getStockPrice);
}

{
  "providers": {
    "oumi": {
      "baseUrl": "https://inference.oumi.ai/v1",
      "api": "openai-completions",
      "apiKey": "!cat /Users/you/.config/oumi/api.key",
      "name": "Oumi",
      "models": [
        {
          "id": "projects/{project_id}/deployments/{deployment_id}",
          "name": "Oumi Stock Lookup",
          "input": ["text"],
          "contextWindow": 32768,
          "maxTokens": 4096,
          "cost": { "input": 0, "output": 0, "cacheRead": 0, "cacheWrite": 0 }
        }
      ]
    }
  }
}