import { cohere } from "@agentium/core";

const model = cohere("command-r-plus");

import { Agent, cohere } from "@agentium/core";

const agent = new Agent({
  name: "cohere-agent",
  model: cohere("command-r-plus"),
  instructions: "You are a helpful assistant.",
});

const result = await agent.run("Summarize the key benefits of microservices architecture.");
console.log(result.text);

import { Agent, cohere, defineTool } from "@agentium/core";
import { z } from "zod";

const agent = new Agent({
  name: "rag-agent",
  model: cohere("command-r-plus"),
  instructions:
    "You are a knowledge-grounded assistant. Always use the search tool to find " +
    "relevant information before answering. Cite your sources.",
  tools: [
    defineTool({
      name: "searchDocs",
      description: "Search internal documentation",
      parameters: z.object({ query: z.string() }),
      execute: async ({ query }) =>
        JSON.stringify([
          { title: "Deployment Guide", content: `Guide for: ${query}...`, source: "docs/deploy.md" },
          { title: "API Reference", content: `API info for: ${query}...`, source: "docs/api.md" },
        ]),
    }),
  ],
  maxToolRoundtrips: 3,
});

const result = await agent.run("How do I deploy our service to production?");
console.log(result.text);

import { Agent, cohere, defineTool } from "@agentium/core";
import { z } from "zod";

const agent = new Agent({
  name: "tool-agent",
  model: cohere("command-r-plus"),
  tools: [
    defineTool({
      name: "createTicket",
      description: "Create a support ticket",
      parameters: z.object({
        title: z.string(),
        priority: z.enum(["low", "medium", "high"]),
        description: z.string(),
      }),
      execute: async ({ title, priority }) => `Ticket created: "${title}" [${priority}] — ID: TK-1234`,
    }),
  ],
});

const result = await agent.run("Create a high-priority ticket: 'API returning 500 errors on /users endpoint'");
console.log(result.text);

import { Agent, cohere, CostTracker, defineTool } from "@agentium/core";
import { z } from "zod";

const costTracker = new CostTracker({
  pricing: {
    "command-r-plus": { promptPer1k: 0.0025, completionPer1k: 0.01 },
    "command-r": { promptPer1k: 0.00015, completionPer1k: 0.0006 },
  },
});

const agent = new Agent({
  name: "enterprise-agent",
  model: cohere("command-r-plus"),
  instructions: "You are an enterprise assistant with document search capabilities.",
  tools: [
    defineTool({
      name: "search",
      description: "Search knowledge base",
      parameters: z.object({ query: z.string() }),
      execute: async ({ query }) => `Found 5 results for "${query}"`,
    }),
  ],
  costTracker,
  maxToolRoundtrips: 5,
});

const result = await agent.run("What is our company's vacation policy?");
console.log(result.text);
console.log(`Cost: $${costTracker.getSummary().totalCost.toFixed(4)}`);