File size: 8,474 Bytes
813c713 409b4f6 813c713 409b4f6 813c713 409b4f6 813c713 409b4f6 813c713 1eee9cd 264cabb 813c713 264cabb 63d819a 813c713 63d819a 813c713 63d819a 813c713 63d819a 813c713 63d819a 813c713 63d819a 813c713 63d819a 813c713 264cabb 813c713 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 |
import gradio as gr
import litellm
import os
import random
from tenacity import retry, stop_after_attempt, wait_fixed, retry_if_exception_type
from interpreter import interpreter
comments = [
"Generating function... π",
"Testing function... π§ͺ",
"Oops, something went wrong! π
",
"Function passed the test! π",
"Getting everything together... πͺ",
"Debugging in progress... π",
"Unleashing the power of LLMs! π§ ",
"Crafting the perfect function... π οΈ",
]
conversation_history = []
@retry(stop=stop_after_attempt(3), wait=wait_fixed(2), retry=retry_if_exception_type(litellm.exceptions.AuthenticationError))
def get_llm_response(prompt, model="gpt-4-turbo-preview"):
print(random.choice(comments))
try:
response = litellm.completion(
model=model,
messages=[{"role": "user", "content": prompt}],
temperature=0.7
)
return response.choices[0].message.content
except litellm.exceptions.AuthenticationError as e:
print(f"Authentication Error: {str(e)}")
raise e
def test_function(function_code):
try:
print("Executing the generated function... π")
interpreter.auto_run = True
output = interpreter.chat(function_code)
print(f"Function output: {output}")
print("Function passed the test! β
")
return True, None
except Exception as e:
print(f"Error occurred: {str(e)} β")
return False, str(e)
def generate_and_test_function(prompt, previous_code=None, previous_error=None, iteration=1):
print(f"Generating function for prompt (Iteration {iteration}): {prompt}")
# Append previous code and error to the prompt for context
if previous_code and previous_error:
prompt += f"\nPrevious code:\n{previous_code}\n\nPrevious error:\n{previous_error}\n\n"
prompt += "Please analyze the previous code and error, and provide suggestions and insights to fix the issue."
# Use GPT-3.5 for internal guidance
guidance_prompt = f"Provide guidance and suggestions for generating a function based on the following prompt and conversation history:\n{prompt}\n\nConversation History:\n{conversation_history}"
guidance_response = get_llm_response(guidance_prompt, model="gpt-3.5-turbo")
# Use GPT-4 for final guidance to Open Interpreter
generation_prompt = f"""
{prompt}
Guidance from super intelligent code bot:
{guidance_response}
Please generate a Python function that satisfies the prompt and follows the provided guidance, while adhering to these coding standards:
- Use descriptive and meaningful names for variables, functions, and classes.
- Follow the naming conventions: lowercase with underscores for functions and variables, CamelCase for classes.
- Keep functions small and focused, doing one thing well.
- Use 4 spaces for indentation, and avoid mixing spaces and tabs.
- Limit line length to 79 characters for better readability.
- Use docstrings to document functions, classes, and modules, describing their purpose, parameters, and return values.
- Use comments sparingly, and prefer descriptive names and clear code structure over comments.
- Handle exceptions appropriately and raise exceptions with clear error messages.
- Use blank lines to separate logical sections of code, but avoid excessive blank lines.
- Import modules in a specific order: standard library, third-party, and local imports, separated by blank lines.
- Use consistent quotes (single or double) for strings throughout the codebase.
- Follow the PEP 8 style guide for more detailed coding standards and best practices.
"""
generated_function = get_llm_response(generation_prompt, model="gpt-4")
print("Testing the generated function...")
success, error = test_function(generated_function)
# Append the generated function to the conversation history
conversation_history.append({"role": "assistant", "content": generated_function})
return success, error, generated_function
def save_function_to_file(generated_function, file_name):
with open(file_name, "w") as file:
file.write(generated_function)
print(f"Function saved to {file_name}")
# Example adjustment for the option handling part
def handle_post_success_actions(generated_function):
valid_option = False
while not valid_option:
print("\nOptions:")
# Options list here
option = input("Enter your choice (1-3): ")
if option in ["1", "2", "3"]:
valid_option = True
# Handle each option here
else:
print("Invalid choice. Please try again.")
def main(initial_prompt, run_mode, num_runs, console_output, command_input):
console_output = "Enter the initial prompt for the development process: " + initial_prompt + "\n"
yield console_output, gr.update(value="") # Clear the command input
while True:
console_output += "\nMenu:\n1. Generate and test a function π¨\n2. Exit π\n"
yield console_output, gr.update(interactive=True) # Wait for user input
choice = command_input
command_input = ""
yield console_output, gr.update(value="") # Clear the command input
if choice == "1":
if run_mode == "1":
success, error, generated_function = generate_and_test_function(initial_prompt)
if success:
generated_function = handle_post_success_actions(generated_function)
initial_prompt = f"Continue developing the function:\n{generated_function}"
else:
console_output += "Function test failed. π\n"
yield console_output, gr.update(interactive=True)
elif run_mode == "2":
for i in range(int(num_runs)):
console_output += f"\nRun {i+1}:\n"
yield console_output, gr.update(interactive=True)
success, error, generated_function = generate_and_test_function(initial_prompt)
if success:
generated_function = handle_post_success_actions(generated_function)
initial_prompt = f"Continue developing the function:\n{generated_function}"
else:
console_output += "Function test failed. π\n"
yield console_output, gr.update(interactive=True)
elif run_mode == "3":
while True:
success, error, generated_function = generate_and_test_function(initial_prompt)
if success:
generated_function = handle_post_success_actions(generated_function)
initial_prompt = f"Continue developing the function:\n{generated_function}"
else:
console_output += "Function test failed. Retrying...\n"
yield console_output, gr.update(interactive=True)
elif choice == "2":
console_output += "Exiting... Goodbye! π\n"
yield console_output, gr.update(interactive=False)
break
else:
console_output += "Invalid choice. Please try again. π
\n"
yield console_output, gr.update(interactive=True)
with gr.Blocks() as demo:
gr.Markdown("# LLM-Powered Function Generator")
with gr.Row():
with gr.Column():
initial_prompt = gr.Textbox(label="Initial Prompt")
run_mode = gr.Radio(["1", "2", "3"], label="Run Mode", info="1: Single Run, 2: Multiple Runs, 3: Continuous Mode")
num_runs = gr.Number(label="Number of Runs", visible=False, interactive=True)
start_button = gr.Button("Start")
with gr.Column():
console_output = gr.Textbox(label="Console Output", lines=20)
command_input = gr.Textbox(label="Command Input", lines=1)
run_mode.change(lambda x: gr.update(visible=x=="2"), inputs=run_mode, outputs=num_runs)
start_button.click(main, inputs=[initial_prompt, run_mode, num_runs, console_output, command_input], outputs=console_output)
command_input.submit(main, inputs=[initial_prompt, run_mode, num_runs, console_output, command_input], outputs=console_output)
demo.queue().launch()
|