package com.example;

import com.google.adk.agents.BaseAgent;
import com.google.adk.agents.LlmAgent;
import com.google.adk.web.AgentLoader;
import com.google.common.collect.ImmutableList;
import java.util.NoSuchElementException;

public class MyAgentLoader implements AgentLoader {

    public static final MyAgentLoader INSTANCE = new MyAgentLoader();

    @Override
    public ImmutableList<String> listAgents() {
        return ImmutableList.of("chat_bot", "code_assistant");
    }

    @Override
    public BaseAgent loadAgent(String name) {
        switch (name) {
            case "chat_bot": return createChatBot();
            case "code_assistant": return createCodeAssistant();
            default: throw new NoSuchElementException("Agent not found: " + name);
        }
    }

    private BaseAgent createChatBot() {
        return LlmAgent.builder()
            .name("chat_bot")
            .description("A helpful chat bot")
            .model("gemini-2.0-flash")
            .instruction("You are a helpful assistant.")
            .build();
    }

    private BaseAgent createCodeAssistant() {
        return LlmAgent.builder()
            .name("code_assistant")
            .description("A code assistance agent")
            .model("gemini-2.0-flash")
            .instruction("You are a coding assistant. Help users with programming questions.")
            .build();
    }
}

public class SimpleAgentLoader implements AgentLoader {
    // No static field needed, uses default constructor
    @Override
    public ImmutableList<String> listAgents() {
        return ImmutableList.of("simple_agent");
    }

    @Override
    public BaseAgent loadAgent(String name) {
        if ("simple_agent".equals(name)) {
            return createSimpleAgent();
        }
        throw new NoSuchElementException("Agent not found: " + name);
    }

    private BaseAgent createSimpleAgent() {
        return LlmAgent.builder()
            .name("simple_agent")
            .description("A simple agent")
            .model("gemini-2.0-flash")
            .instruction("You are a helpful assistant.")
            .build();
    }
}

public class MultipleLoaders implements AgentLoader {
    public static final MultipleLoaders DEFAULT = new MultipleLoaders();
    public static final MultipleLoaders ADVANCED = new MultipleLoaders(true);

    private final boolean advanced;

    public MultipleLoaders() { this(false); }
    public MultipleLoaders(boolean advanced) { this.advanced = advanced; }

    @Override
    public ImmutableList<String> listAgents() {
        if (advanced) {
            return ImmutableList.of("advanced_agent");
        }
        return ImmutableList.of("basic_agent");
    }

    @Override
    public BaseAgent loadAgent(String name) {
        if (advanced && "advanced_agent".equals(name)) {
            return createAdvancedAgent();
        } else if (!advanced && "basic_agent".equals(name)) {
            return createBasicAgent();
        }
        throw new NoSuchElementException("Agent not found: " + name);
    }

    private BaseAgent createBasicAgent() {
        return LlmAgent.builder()
            .name("basic_agent")
            .description("A basic agent")
            .model("gemini-2.0-flash")
            .instruction("You are a basic helpful assistant.")
            .build();
    }

    private BaseAgent createAdvancedAgent() {
        return LlmAgent.builder()
            .name("advanced_agent")
            .description("An advanced agent with more capabilities")
            .model("gemini-2.0-flash")
            .instruction("You are an advanced assistant with enhanced capabilities.")
            .build();
    }
}