---
license: apache-2.0
datasets:
- K8sAIOps/kubernetes_operator_3b_1.5k
- vicgalle/alpaca-gpt4
language:
- en
base_model:
- unsloth/Qwen2.5-3B-Instruct-bnb-4bit
---

# Kubernetes Feature Model

- **License:** Apache-2.0
- **Fine-tuned from model:** [unsloth/Qwen-2.5-3B-Instruct-bnb-4bit](https://huggingface.co/unsloth/Qwen-2.5-3B-Instruct-bnb-4bit)
- **Model type:** GGUF (compatible with Ollama)
- **Language:** English

This Qwen-based model was fine-tuned to assist users with Kubernetes commands and questions. It has three primary features:

1. **Generating accurate `kubectl` commands** based on user descriptions.
2. **Providing concise explanations about Kubernetes** for general queries.
3. **Politely requesting additional information** if the instruction is incomplete or ambiguous.

**Update:** Compared to the previous 1B model, the **3B model significantly reduces hallucinations** that sometimes occurred in the 1B model. Users can expect improved accuracy and reliability when interacting with this model.

The model was trained efficiently using [Unsloth](https://github.com/unslothai/unsloth) and Hugging Face's TRL library.

---

## How to Use the Model

This section provides instructions on how to run the model using Ollama and the provided Modelfile.

### Prerequisites

- Install [Ollama](https://github.com/jmorganca/ollama) on your system.
- Ensure you have access to the model hosted on Hugging Face: `hf.co/zhenyangzhao/Qwen2.5-3B-Instruct-Kubernetes`.

### Steps

1. **Create the Modelfile**

   Save the following content as a file named `Modelfile`:

   ```plaintext
   FROM hf.co/zhenyangzhao/Qwen2.5-3B-Instruct-Kubernetes
   PARAMETER temperature 0.3
   PARAMETER stop "<|endoftext|>"
   TEMPLATE """
   You are an AI assistant that helps users with Kubernetes commands and questions.
   **Your Behavior Guidelines:**
   1. **For clear and complete instructions:**
      - **Provide only** the exact `kubectl` command needed to fulfill the user's request.
      - Do not include extra explanations, placeholders, or context.
      - **Enclose the command within a code block** with `bash` syntax highlighting.
   2. **For incomplete or ambiguous instructions:**
      - **Politely ask** the user for the specific missing information.
      - Do **not** provide any commands or placeholders in your response.
      - Respond in plain text, clearly stating what information is needed.
   3. **For general Kubernetes questions:**
      - Provide a **concise and accurate explanation**.
      - Do **not** include any commands unless specifically requested.
      - Ensure that the explanation fully addresses the user's question.
   **Important Rules:**
   - **Do not generate CLI commands containing placeholders (e.g., <pod_name>, <resource_name>).**
   - Ensure all CLI commands are complete, valid, and executable as provided.
   - If user input is insufficient to form a complete command, ask for clarification instead of using placeholders.
   - Provide only the necessary CLI command output without any additional text.
   ### Instruction:
   {{ .Prompt }}
   ### Response:
   {{ .Response }}
   <|endoftext|>
   """
   ```

2. **Create the Model with Ollama**

   Open your terminal and run the following command to create the model:

   ```bash
   ollama create kubernetes_feature_model -f Modelfile
   ```

   This command tells Ollama to create a new model named `hybrid_kubernetes_feature_model` using the configuration specified in `Modelfile`.

3. **Run the Model**

   Start interacting with your model:

   ```bash
   ollama run kubernetes_feature_model
   ```

   This will initiate the model and prompt you for input based on the template provided.

   Alternatively, you can provide an instruction directly:

   ```bash
   ollama run kubernetes_feature_model -p "Create a namespace called 'production'."
   ```

   **Example Output:**

   ```bash
   kubectl create namespace production
   ```

---

## Model Details

### Purpose

The model assists users by:

- **Generating accurate `kubectl` commands** based on natural language descriptions.
- **Providing concise explanations about Kubernetes** for general queries.
- **Politely requesting additional information** if the instruction is incomplete or ambiguous.

### Intended Users

- Kubernetes administrators
- DevOps engineers
- Developers working with Kubernetes clusters

### Training Process

- **Base Model:** Unsloth's Qwen2.5-3B-Instruct
- **Fine-tuning:** Leveraged the Unsloth framework and Hugging Face's TRL library for efficient training.
- **Training Data:** Customized dataset focused on Kubernetes operations and features, including `kubectl` command usage and general Kubernetes concepts, containing approximately 1,500 entries.

---

## Model Features

### 1. Command Generation in Bash Format

When the model generates CLI commands, it provides them in `bash` format, enclosed within code blocks for easy execution and clarity. This allows users to copy and paste the commands directly into their terminal.

### 2. Handling Ambiguity with Polite Clarifications

If the instruction is incomplete or ambiguous, the model will politely ask for the specific missing information instead of making assumptions. This ensures accuracy and prevents the execution of incorrect commands.

### 3. Providing Concise Explanations

For general Kubernetes questions, the model offers concise and accurate explanations without unnecessary details, helping users understand concepts quickly.

### 4. Enhanced Accuracy with 3B Model

The transition to the 3B model has significantly **reduced hallucinations** that sometimes occurred in the 1B model. The 3B model provides more accurate and reliable responses, improving the overall user experience.

---

## Examples

### Example 1: Generating a Command

**Instruction:**

```
Create a namespace called 'development'.
```

**Response:**

```bash
kubectl create namespace development
```

---

### Example 2: Handling Ambiguity

**Instruction:**

```
Show me logs.
```

**Response:**

```
Which pod's logs would you like to retrieve?
```

---

### Example 3: Providing Explanations

**Instruction:**

```
What is a Deployment in Kubernetes?
```

**Response:**

```
A Deployment provides declarative updates for Pods and ReplicaSets, allowing you to manage your applications by defining the desired state and letting Kubernetes handle the rest.
```

---

## Limitations and Considerations

- **Accuracy:** While the 3B model improves accuracy, the model may occasionally produce incorrect or suboptimal commands. Always review the output before execution.
- **Security:** Be cautious when executing generated commands, especially in production environments.

---

## Feedback and Contributions

We welcome any comments or participation to improve the model and dataset. If you encounter issues or have suggestions for improvement:

- **GitHub:** [Unsloth Repository](https://github.com/unslothai/unsloth)
- **Contact:** Reach out to the developer, **dereklck**, for further assistance.

---

**Note:** This model provides assistance in generating Kubernetes commands and explanations based on user input. Always verify the generated commands in a safe environment before executing them in a production cluster.

---

If you have any further requests or need additional adjustments, please let me know!