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

fix: small fixes

This commit is contained in:
Shiven Mian 2024-02-17 23:11:23 -08:00
parent e156fec978
commit c1688fb7ce
1 changed files with 30 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -69,36 +69,36 @@ def generate_python_steps(function_name, steps):
def teach(): def teach():
root = Tk() root = Tk()
root.withdraw() root.withdraw()
skill_name = simpledialog.askstring("Skill Name", "Please enter the name for the skill:", parent=root)
skill_name = simpledialog.askstring("Skill Name", "Please enter the name for the skill:") if skill_name:
skill = Skill(skill_name) skill = Skill(skill_name)
while True: while True:
step = simpledialog.askstring("Next Step", "Enter the next step (or 'end' to finish): ") step = simpledialog.askstring("Next Step", "Enter the next step (or 'end' to finish): ", parent=root)
logger.info(f"Performing step: {step}") if step is None or step == "end":
if step == "end":
break
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):
if "format" in chunk and chunk["format"] == "execution":
content = chunk["content"]
language = content["format"]
code = content["content"]
chunk_code += code
interpreter.computer.run(code, language)
time.sleep(0.05)
accumulator.accumulate(chunk)
stepCheckDialog = StepCheckDialog(root)
stepCheckResult = stepCheckDialog.result
if stepCheckResult == "Yes" or stepCheckResult == "Task Complete":
skill.steps.append(step)
skill.code += chunk_code
if stepCheckResult == "Task Complete":
break break
elif step.strip() == "":
continue
logger.info(f"Performing step: {step}")
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):
if "format" in chunk and chunk["format"] == "execution":
content = chunk["content"]
language = content["format"]
code = content["content"]
chunk_code += code
interpreter.computer.run(code, language)
time.sleep(0.05)
accumulator.accumulate(chunk)
stepCheckDialog = StepCheckDialog(root)
stepCheckResult = stepCheckDialog.result
if stepCheckResult == "Yes" or stepCheckResult == "Task Complete":
skill.steps.append(step)
skill.code += chunk_code
if stepCheckResult == "Task Complete":
break
# Uncomment this incase you want steps instead of code # Uncomment this incase you want steps instead of code
#python_code = generate_python_steps(skill.skill_name, skill.steps) #python_code = generate_python_steps(skill.skill_name, skill.steps)
@ -106,5 +106,6 @@ def teach():
python_code = generate_python_code(skill.skill_name, skill.code) 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")
logger.info(f"Saving skill to: {filename}")
with open(filename, "w") as file: with open(filename, "w") as file:
file.write(python_code) file.write(python_code)