MirrorMirror
/
01
mirror of https://github.com/OpenInterpreter/01.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat: add function to add steps instead, force task completion

This commit is contained in:
Shiven Mian 2024-02-17 14:01:52 -08:00
parent e366f3825b
commit 6e3993d0c2
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
01OS/01OS/server/teach.py
View File

@ -22,13 +22,19 @@ def to_camel_case(text):
camel_case_string = words[0].lower() + ''.join(word.title() for word in words[1:]) camel_case_string = words[0].lower() + ''.join(word.title() for word in words[1:])
return camel_case_string return camel_case_string
def generate_python_code(function_name, steps, code): def generate_python_code(function_name, code):
code_string = f'def {to_camel_case(function_name)}():\n' code_string = f'def {to_camel_case(function_name)}():\n'
code_string += f' """{function_name}"""\n' code_string += f' """{function_name}"""\n'
indented_code = textwrap.indent(code, ' ') indented_code = textwrap.indent(code, ' ')
code_string += indented_code + '\n' code_string += indented_code + '\n'
return code_string return code_string
def generate_python_steps(function_name, steps):
code_string = f'def {to_camel_case(function_name)}():\n'
code_string += f' """{function_name}"""\n'
code_string += f' print({steps})\n'
return code_string
def teach(): def teach():
root = tk.Tk() root = tk.Tk()
root.withdraw() root.withdraw()
@ -42,6 +48,8 @@ def teach():
break break
chunk_code = "" chunk_code = ""
interpreter.computer.languages = [l for l in interpreter.computer.languages if l.name.lower() == "python"]
interpreter.force_task_completion = True
for chunk in interpreter.chat(step, stream=True, display=False): for chunk in interpreter.chat(step, stream=True, display=False):
if "format" in chunk and chunk["format"] == "execution": if "format" in chunk and chunk["format"] == "execution":
content = chunk["content"] content = chunk["content"]
@ -57,7 +65,10 @@ def teach():
skill.steps.append(step) skill.steps.append(step)
skill.code += chunk_code skill.code += chunk_code
python_code = generate_python_code(skill.skill_name, skill.steps, skill.code) # Uncomment this incase you want steps instead of code
#python_code = generate_python_steps(skill.skill_name, skill.steps)
python_code = generate_python_code(skill.skill_name, skill.code)
SKILLS_DIR = os.path.dirname(__file__) + "/skills" SKILLS_DIR = os.path.dirname(__file__) + "/skills"
filename = os.path.join(SKILLS_DIR, f"{skill.skill_name.replace(' ', '_')}.py") filename = os.path.join(SKILLS_DIR, f"{skill.skill_name.replace(' ', '_')}.py")
with open(filename, "w") as file: with open(filename, "w") as file: