Upload tool

code_generating.py

@@ -18,15 +18,15 @@ from transformers import LlamaForCausalLM, CodeLlamaTokenizer
 from transformers import PipelineTool
 
 
-QA_PROMPT = """Here is an example of how I want my code to be: '''{example}'''.
+QA_PROMPT = """Here is an example of how I want my code to be: '''{text}'''.
 
-Can you generate code for this prompt: '{prompt}'"""
+Can you generate code for this prompt: '{question}'"""
 
 
 class CodeGeneratingTool(PipelineTool):
     default_checkpoint = "codellama/CodeLlama-7b-Instruct-hf"
     description = (
-        "This is a tool that generates codes related to a prompt. It takes two arguments named `example`, which is a template on how the user wants their code to be generated, and `prompt`, which is the prompt of the code, and returns the code to the prompt."
+        "This is a tool that generates codes related to a prompt. It takes two arguments named `text`, which is a template on how the user wants their code to be generated, and `question`, which is the prompt of the code, and returns the code to the question."
     )
     name = "text_qa"
     pre_processor_class = CodeLlamaTokenizer