import { FunctionTool, LlmAgent } from "@google/adk";
import { OllamaModel } from "@yagolopez/adk-utils";
import { z } from "zod";

// Define a custom tool
const getCurrentTime = new FunctionTool({
  name: "get_current_time",
  description: "Returns the current time in a specified city.",
  parameters: z.object({
    city: z.string().describe("The name of the city"),
  }),
  execute: ({ city }) => ({
    status: "success",
    report: `The current time in ${city} is ${new Date().toLocaleTimeString()}`,
  }),
});

// Create an agent with Ollama model
export const rootAgent = new LlmAgent({
  name: "agent1",
  model: new OllamaModel("gpt-oss:120b-cloud", "https://ollama.com"),
  description: "Agent with custom tools for time and diagrams",
  instruction: "You are a helpful assistant.",
  tools: [getCurrentTime],
});