Extend examples for classifier

app.py

@@ -226,11 +226,20 @@ def text_to_text_generation(verbose, prompt):
 
     is_kubectl_prompt = (
         f"You are a helpful assistant who classifies prompts into three categories. [INST] Respond with 0 if it pertains to a 'kubectl' operation. This is an instruction that can be answered with a 'kubectl' action. Look for keywords like 'get', 'list', 'create', 'show', 'view', and other command-like words. This category is an instruction instead of a question. Respond with 1 only if the prompt is a question, and is about a definition related to Kubernetes, or non-action inquiries. Respond with 2 every other scenario, for example if the question is a general question, not related to Kubernetes or 'kubectl'.\n"
-        f"So for instance the following:\n"
-        f'text: "List all pods in Kubernetes"\n'
-        f"Would get a response:\n"
-        f"response (0/1/2): 0 [/INST] \n"
-        f'text: "{prompt}"'
+        f"Here are some examples:\n"
+        f"text: List all pods in Kubernetes\n"
+        f"response (0/1/2): 0 \n"
+        f"text: What is a headless service and how to create one?\n"
+        f"response (0/1/2): 1 \n"
+        f"text: What is the capital of Hungary?\n"
+        f"response (0/1/2): 2 \n"
+        f"text: Display detailed information about the pod 'web-app-pod-1'\n"
+        f"response (0/1/2): 0 \n"
+        f"text: What are some typical foods in Germany?\n"
+        f"response (0/1/2): 2 \n"
+        f"text: What is a LoadBalancer in Kubernetes?\n"
+        f"response (0/1/2): 1 \n"
+        f'Classify the following: [/INST] \ntext: "{prompt}\n"'
         f"response (0/1/2): "
     )