kubePilot/
├── cmd/
│   ├── root.go          # 根命令，定义版本、全局Flag
│   ├── checkin.go       # 子命令：集群健康检查
│   └── briefing.go      # 子命令：集群资源简报
├── pkg/
│   └── utils/           # 工具库
└── main.go

var RootCmd = &cobra.Command{
    Use:   "kubePilot",
    Short: "K8s AIOps 工具链基座",
    Version: "v1.0.0",
}

func init() {
    // 挂载子命令
    RootCmd.AddCommand(checkinCmd)
    RootCmd.AddCommand(briefingCmd)

    // 定义全局Flag
    RootCmd.PersistentFlags().StringVar(&kubeconfig, "kubeconfig",
        "~/.kube/config", "K8s配置文件路径")
}

┌─────────────────────────────────────────┐
│           用户（自然语言输入）             │
└─────────────────────┬───────────────────┘
                      ▼
┌─────────────────────────────────────────┐
│           LLM（意图识别层）               │
│   Function Calling → 结构化JSON指令      │
└─────────────────────┬───────────────────┘
                      ▼
┌─────────────────────────────────────────┐
│        K8s Client（执行层）               │
│   client-go → 调用K8s API完成实际操作    │
└─────────────────────────────────────────┘

package utils

import (
    "fmt"
    "path/filepath"
    "strings"
    "k8s.io/client-go/discovery"
    "k8s.io/client-go/dynamic"
    "k8s.io/client-go/kubernetes"
    "k8s.io/client-go/tools/clientcmd"
    "k8s.io/client-go/util/homedir"
)

// ClientGo 封装三种客户端，实现全资源覆盖
type ClientGo struct {
    Clientset       *kubernetes.Clientset          // 标准客户端
    DynamicClient   dynamic.Interface              // 动态客户端
    DiscoveryClient discovery.DiscoveryInterface   // 发现客户端
}

func NewClientGo(kubeconfig string) (*ClientGo, error) {
    // 处理 ~ 路径
    if strings.HasPrefix(kubeconfig, "~") {
        homeDir := homedir.HomeDir()
        kubeconfig = filepath.Join(homeDir, kubeconfig[1:])
    }
    // 构建 Config
    config, err := clientcmd.BuildConfigFromFlags("", kubeconfig)
    if err != nil {
        return nil, fmt.Errorf("failed to build kubeconfig: %w", err)
    }
    // 初始化三类客户端
    clientset, err := kubernetes.NewForConfig(config)
    if err != nil {
        return nil, fmt.Errorf("failed to create clientset: %w", err)
    }
    dynamicClient, err := dynamic.NewForConfig(config)
    if err != nil {
        return nil, fmt.Errorf("failed to create dynamic client: %w", err)
    }
    discoveryClient, err := discovery.NewDiscoveryClientForConfig(config)
    if err != nil {
        return nil, fmt.Errorf("failed to create discovery client: %w", err)
    }
    return &ClientGo{
        Clientset:       clientset,
        DynamicClient:   dynamicClient,
        DiscoveryClient: discoveryClient,
    }, nil
}

package utils

import (
    "context"
    "errors"
    "os"
    "github.com/sashabaranov/go-openai"
)

type OpenAI struct {
    Client *openai.Client
    ctx    context.Context
}

func NewOpenAIClient() (*OpenAI, error) {
    apiKey := os.Getenv("OPENAI_API_KEY_rui")
    if apiKey == "" {
        return nil, errors.New("OPENAI_API_KEY environment variable is not set")
    }
    config := openai.DefaultConfig(apiKey)
    // 关键点：通过修改 BaseURL，实现多模型兼容
    config.BaseURL = "https://vip.apiyi.com/v1"
    client := openai.NewClientWithConfig(config)
    ctx := context.Background()
    return &OpenAI{Client: client, ctx: ctx}, nil
}

func (o *OpenAI) SendMessage(prompt string, content string) (string, error) {
    req := openai.ChatCompletionRequest{
        Model: openai.GPT4o,
        Messages: []openai.ChatCompletionMessage{
            {Role: "system", Content: prompt},
            {Role: "user", Content: content},
        },
    }
    resp, err := o.Client.CreateChatCompletion(o.ctx, req)
    if err != nil {
        return "", err
    }
    if len(resp.Choices) == 0 {
        return "", errors.New("no response from OpenAI")
    }
    return resp.Choices[0].Message.Content, nil
}

┌──────────┐    ┌──────────┐    ┌──────────┐    ┌──────────┐
│ 采集Event │ →  │ 关联Logs │ →  │ 构造Prompt│ →  │ AI分析输出│
└──────────┘    └──────────┘    └──────────┘    └──────────┘