Update logs

app.py

@@ -268,10 +268,12 @@ def text_to_text_generation(verbose, prompt):
         match response_num:
             case 0:
                 prompt = f"[INST] {prompt}\n Lets think step by step. [/INST] {start_template}"
+                print('Kubectl command prompt:')
                 print(prompt)
             case 1:
                 if retriever == "semantic_search":
                     question = prompt
+                    print('Semantic search prompt:')
                     print(
                         (
                         f"You are a helpful kubernetes professional. [INST] Use the following documentation, if it is relevant to answer the question below. [/INST]\nDocumentation: [RETRIEVED_RESULTS_FROM_BOOK] [INST] Answer the following question: {question} [/INST]\nAnswer:  \n")
@@ -292,7 +294,8 @@ def text_to_text_generation(verbose, prompt):
                     retrieved_results = google_search(prompt)
                     question = prompt
                     prompt = f"You are a helpful kubernetes professional. [INST] Use the following documentation, if it is relevant to answer the question below. [/INST]\nDocumentation: {retrieved_results} </s>\n<s> [INST] Answer the following question: {prompt} [/INST]\nAnswer: "
-
+                    
+                    print('Google search prompt:')
                     print(
                         (
                             f"You are a helpful kubernetes professional. [INST] Use the following documentation, if it is relevant to answer the question below. [/INST]\nDocumentation: [RETRIEVED_RESULTS_FROM_GOOGLE] [INST] Answer the following question: {question} [/INST]\nAnswer:  \n" 
@@ -309,10 +312,12 @@ def text_to_text_generation(verbose, prompt):
                     )
                 else:
                     prompt = f"[INST] Answer the following question: {prompt} [/INST]\nAnswer: "
+                    print('No retriever question prompt:')
                     print(prompt)
 
             case _:
                 prompt = f"[INST] {prompt} [/INST]"
+                print('Other question prompt:')
                 print(prompt)
 
         return prompt, md
@@ -343,7 +348,7 @@ def text_to_text_generation(verbose, prompt):
 
         return (prompt, answer[start:end].strip())
 
-    modes = ["Kubectl command", "Kubernetes related definition", "Other"]
+    modes = ["Kubectl command", "Kubernetes related", "Other"]
 
     print(f'{" Query Start ":-^40}')
     print("Classified as: " + modes[response_num])