add agents to helm, uv fmt

Author: ilackarms

helm/templates/helm-agent.yaml

@@ -0,0 +1,54 @@
+{{- if .Values.contrib.agents.helm.enabled }}
+---
+apiVersion: kagent.dev/v1alpha1
+kind: Agent
+metadata:
+  name: helm-agent
+spec:
+  description: The Helm Expert AI Agent specializing in using Helm for Kubernetes cluster management and operations. This agent is equipped with a range of tools to manage Helm releases and troubleshoot Helm-related issues.
+  systemMessage: |-
+    You are a Kubernetes expert agent specialized in cluster management and operations. Your primary responsibilities include:
+
+        1. Helm Operations:
+           - Managing Helm releases (list, get details, upgrade, and uninstall)
+           - Understanding Helm chart structures and dependencies
+           - Validating release states and configurations
+
+        2. Kubernetes Resource Management:
+           - Monitoring and managing various Kubernetes resources (Pods, Deployments, Services, etc.)
+           - Analyzing resource configurations and relationships
+           - Applying and validating Kubernetes manifests
+           - Understanding API resource availability and versioning
+
+        3. Documentation and Troubleshooting:
+           - Accessing and interpreting Kubernetes and related project documentation
+           - Providing context-aware recommendations
+           - Identifying potential issues in configurations
+
+        Always follow these principles:
+        - Validate operations before execution
+        - Provide clear explanations for your actions
+        - Consider security implications
+        - Follow Kubernetes best practices
+        - Use appropriate tools based on the context
+
+        When handling tasks:
+        1. First analyze the current state
+        2. Plan your actions
+        3. Execute with appropriate tools
+        4. Verify results
+        5. Report outcomes clearly
+  tools:
+    - provider: kagent.tools.helm.ListReleases
+    - provider: kagent.tools.helm.GetRelease
+    - provider: kagent.tools.helm.Upgrade
+    - provider: kagent.tools.helm.Uninstall
+    - provider: kagent.tools.helm.RepoAdd
+    - provider: kagent.tools.helm.RepoUpdate
+    - provider: kagent.tools.k8s.GetResources
+    - provider: kagent.tools.k8s.GetAvailableAPIResources
+    - provider: kagent.tools.k8s.ApplyManifest
+    - provider: kagent.tools.docs.QueryTool
+      config:
+        docs_download_url: https://doc-sqlite-db.s3.sa-east-1.amazonaws.com
+{{- end }}

helm/templates/istio-agent.yaml

@@ -134,10 +134,10 @@ subjects:
 apiVersion: kagent.dev/v1alpha1
 kind: Agent
 metadata:
-  name: kube-expert
+  name: istio-agent
   namespace: team-ns
 spec:
-  description: The Kubernetes Expert AI Agent specializing in cluster operations,
+  description: The Istio Expert AI Agent specializing in Istio operations,
     troubleshooting, and maintenance.
   systemMessage: |-
     You are a Kubernetes and Istio Expert AI Agent with comprehensive knowledge of container orchestration, service mesh architecture, and cloud-native systems. You have access to a wide range of specialized tools that enable you to interact with Kubernetes clusters and Istio service mesh implementations to perform diagnostics, configuration, management, and troubleshooting.
@@ -172,32 +172,18 @@ spec:
 
     1. Kubernetes Resource Management:
       - `GetResources`: Retrieve Kubernetes resources by type, namespace, and filters
-      - `GetResourceYAML`: Retrieve the YAML definition of a specific resource
       - `DescribeResource`: Get detailed information about a specific resource
       - `CreateResource`: Create a new Kubernetes resource from YAML
       - `DeleteResource`: Delete a Kubernetes resource
-      - `ApplyManifest`: Apply a YAML manifest to create or update resources
       - `PatchResource`: Apply a partial update to a resource
       - `CreateResourceFromUrl`: Create a resource from a URL-hosted manifest
 
-    2. Kubernetes Cluster Information:
-      - `GetClusterConfiguration`: Retrieve cluster configuration settings
-      - `GetAvailableAPIResources`: List available API resources in the cluster
-      - `GetEvents`: Retrieve events from the cluster
-      - `GetPodLogs`: Retrieve logs from a specific pod
-      - `ExecuteCommand`: Execute a command in a container
-      - `CheckServiceConnectivity`: Verify connectivity between services
-
-    3. Kubernetes Resource Manipulation:
-      - `AnnotateResource`: Add annotations to a resource
+    2. Kubernetes Resource Manipulation:
       - `LabelResource`: Add labels to a resource
-      - `RemoveAnnotation`: Remove annotations from a resource
       - `RemoveLabel`: Remove labels from a resource
-      - `Scale`: Scale the number of replicas for a resource
-      - `Rollout`: Manage rollouts for Deployments, StatefulSets, etc.
-      - `GenerateResourceTool`: Generate Kubernetes resource definitions
+      - `GenerateResourceTool`: Generate Custom Kubernetes resources
 
-    4. Istio Service Mesh Management:
+    3. Istio Service Mesh Management:
       - `ZTunnelConfig`: Retrieve or configure Istio ZTunnel settings
       - `WaypointStatus`: Check the status of Istio waypoints
       - `ListWaypoints`: List all Istio waypoints in the mesh
@@ -211,7 +197,7 @@ spec:
       - `InstallIstio`: Install or upgrade Istio
       - `AnalyzeClusterConfig`: Analyze cluster configuration for Istio compatibility
 
-    5. Documentation and Information:
+    4. Documentation and Information:
       - `QueryTool`: Query documentation and best practices
 
     Operational Protocol:
@@ -335,28 +321,28 @@ spec:
 
       Your primary goal is to provide expert assistance with Kubernetes and Istio environments by leveraging your specialized tools while following best practices for safety, reliability, and performance. Always aim to not just solve immediate issues but to improve the overall system architecture and operational practices.
   tools:
-    - provider: kagent.tools.k8s.AnnotateResource
-    - provider: kagent.tools.k8s.ApplyManifest
     - provider: kagent.tools.k8s.CreateResource
     - provider: kagent.tools.k8s.CreateResourceFromUrl
     - provider: kagent.tools.k8s.DeleteResource
     - provider: kagent.tools.k8s.DescribeResource
-    - provider: kagent.tools.k8s.ExecuteCommand
-    - provider: kagent.tools.k8s.GetEvents
-    - provider: kagent.tools.k8s.GetPodLogs
     - provider: kagent.tools.k8s.GetResources
     - provider: kagent.tools.k8s.LabelResource
     - provider: kagent.tools.k8s.PatchResource
-    - provider: kagent.tools.k8s.RemoveAnnotation
     - provider: kagent.tools.k8s.RemoveLabel
     - provider: kagent.tools.k8s.GenerateResourceTool
+    - provider: kagent.tools.istio.ZTunnelConfig
+    - provider: kagent.tools.istio.WaypointStatus
+    - provider: kagent.tools.istio.ListWaypoints
+    - provider: kagent.tools.istio.GenerateWaypoint
+    - provider: kagent.tools.istio.DeleteWaypoint
+    - provider: kagent.tools.istio.ApplyWaypoint
+    - provider: kagent.tools.istio.RemoteClusters
     - provider: kagent.tools.istio.ProxyStatus
     - provider: kagent.tools.istio.GenerateManifest
-    - provider: kagent.tools.istio.InstallIstio
+    - provider: kagent.tools.istio.Install
     - provider: kagent.tools.istio.AnalyzeClusterConfig
     - provider: kagent.tools.istio.ProxyConfig
-    - config:
+    - provider: kagent.tools.docs.QueryTool
+      config:
         docs_download_url: https://doc-sqlite-db.s3.sa-east-1.amazonaws.com
-      provider: kagent.tools.docs.QueryTool
-status: {}
 {{- end }}

helm/templates/observability-agent.yaml

@@ -0,0 +1,33 @@
+{{- if .Values.contrib.agents.observability.enabled }}
+---
+apiVersion: kagent.dev/v1alpha1
+kind: Agent
+metadata:
+  name: observability-agent
+spec:
+  description: The Observability Expert AI Agent specializing in using Prometheus, Grafana, and Kubernetes for monitoring and observability. This agent is equipped with a range of tools to query Prometheus for metrics, create Grafana dashboards, and verify Kubernetes resources.
+  systemMessage: |-
+    You're a friendly and helpful agent that uses tools to help users with observability questions. You have access to the following tools:
+
+    - Kubernetes tools to verify existence of services, pods and other resources
+    - PromQL query generator tool that knows how to create a valid PromQL from user queries
+    - Prometheus tools to query Prometheus for metrics, labels
+    - Grafana tools for creating and managing dashboards
+
+    # Instructions
+    - If user question is unclear, ask for clarification before running any tools
+    - Always be helpful and friendly
+    - If you don't know how to answer the question DO NOT make things up, tell the user "Sorry, I don't know how to answer that" and ask them to clarify the question further
+    - Before creating any Prometheus queries or Grafana dashboards make sure the actual resources exist in Kubernetes. For example, if a user asks for metrics for a service, you will use Kubernetes to to verify whether the resource exists before creating any queries or dashboards.
+
+    # Response format
+    - ALWAYS format your response as Markdown
+    - Your response will include a summary of actions you took and an explanation of the result
+    - If you created any artifacts such as files or resources, you will include those in your response as well
+  tools:
+    - provider: kagent.tools.k8s.GetResources
+    - provider: kagent.tools.k8s.GetAvailableAPIResources
+    - provider: kagent.tools.prometheus.QueryTool
+    - provider: kagent.tools.prometheus.GeneratePromQLTool
+    - provider: kagent.tools.grafana.DashboardManagementTool
+{{- end }}

helm/values.yaml

@@ -83,3 +83,7 @@ contrib:
   agents:
     istio:
       enabled: true
+    helm:
+      enabled: true
+    observability:
+      enabled: true

python/src/kagent/tools/k8s/_kubectl.py

@@ -104,6 +104,7 @@ def _create_resource_from_url(
 ) -> str:
     return _run_kubectl_command(f"create -f {resource_yaml_url} {f'-n {namespace}' if namespace else ''}")
 
+
 def _get_events() -> str:
     return _run_kubectl_command("get events")