Spaces:
Sleeping
Sleeping
Update gensim/sim_runner.py
Browse files- gensim/sim_runner.py +8 -7
gensim/sim_runner.py
CHANGED
@@ -79,13 +79,13 @@ class SimulationRunner:
|
|
79 |
|
80 |
self.generated_task = {'task-name': 'TASK_NAME_TEMPLATE', 'task-description': 'TASK_STRING_TEMPLATE', 'assets-used': ['ASSET_1', 'ASSET_2', Ellipsis]}
|
81 |
print("generated_task\n", self.generated_task)
|
82 |
-
yield "Task Generated ==>", None, None
|
83 |
self.generated_asset = self.agent.propose_assets()
|
84 |
# self.generated_asset = {}
|
85 |
print("generated_asset\n", self.generated_asset)
|
86 |
-
yield "Task Generated ==> Asset Generated ==> ", None, None
|
87 |
-
yield "Task Generated ==> Asset Generated ==> API Reviewed ==> ", None, None
|
88 |
-
yield "Task Generated ==> Asset Generated ==> API Reviewed ==> Error Reviewed ==> ", None, None
|
89 |
|
90 |
online_code_buffer = {}
|
91 |
for task_file in json.load(open(os.path.join('prompts/data', "generated_task_codes.json"))):
|
@@ -101,7 +101,7 @@ class SimulationRunner:
|
|
101 |
self.generated_code = online_code_buffer[random_task_file]
|
102 |
print("generated_code\n", self.generated_code)
|
103 |
print("curr_task_name\n", self.curr_task_name)
|
104 |
-
yield "Task Generated ==> Asset Generated ==> API Reviewed ==> Error Reviewed ==> Code Generated ==> ", self.generated_code, None
|
105 |
|
106 |
self.generated_tasks.append(self.generated_task)
|
107 |
self.generated_task_assets.append(self.generated_asset)
|
@@ -110,9 +110,10 @@ class SimulationRunner:
|
|
110 |
except:
|
111 |
to_print = highlight(f"{str(traceback.format_exc())}", PythonLexer(), HtmlFormatter())
|
112 |
print("Task Creation Exception:", highlight(f"{str(traceback.format_exc())}", PythonLexer(), TerminalFormatter()))
|
113 |
-
|
114 |
-
|
115 |
self.task_creation_pass = False
|
|
|
116 |
|
117 |
# self.curr_task_name = self.generated_task['task-name']
|
118 |
print("task creation time {:.3f}".format(time.time() - start_time))
|
|
|
79 |
|
80 |
self.generated_task = {'task-name': 'TASK_NAME_TEMPLATE', 'task-description': 'TASK_STRING_TEMPLATE', 'assets-used': ['ASSET_1', 'ASSET_2', Ellipsis]}
|
81 |
print("generated_task\n", self.generated_task)
|
82 |
+
yield "Task Generated ==>", "", None, None
|
83 |
self.generated_asset = self.agent.propose_assets()
|
84 |
# self.generated_asset = {}
|
85 |
print("generated_asset\n", self.generated_asset)
|
86 |
+
yield "Task Generated ==> Asset Generated ==> ", "", None, None
|
87 |
+
yield "Task Generated ==> Asset Generated ==> API Reviewed ==> ", "", None, None
|
88 |
+
yield "Task Generated ==> Asset Generated ==> API Reviewed ==> Error Reviewed ==> ", "", None, None
|
89 |
|
90 |
online_code_buffer = {}
|
91 |
for task_file in json.load(open(os.path.join('prompts/data', "generated_task_codes.json"))):
|
|
|
101 |
self.generated_code = online_code_buffer[random_task_file]
|
102 |
print("generated_code\n", self.generated_code)
|
103 |
print("curr_task_name\n", self.curr_task_name)
|
104 |
+
yield "Task Generated ==> Asset Generated ==> API Reviewed ==> Error Reviewed ==> Code Generated ==> ", "", self.generated_code, None
|
105 |
|
106 |
self.generated_tasks.append(self.generated_task)
|
107 |
self.generated_task_assets.append(self.generated_asset)
|
|
|
110 |
except:
|
111 |
to_print = highlight(f"{str(traceback.format_exc())}", PythonLexer(), HtmlFormatter())
|
112 |
print("Task Creation Exception:", highlight(f"{str(traceback.format_exc())}", PythonLexer(), TerminalFormatter()))
|
113 |
+
self.log = to_print
|
114 |
+
yield "Task Generated ==> Asset Generated ==> API Reviewed ==> Error Reviewed ==> Code Generation Failed", self.log, "", None
|
115 |
self.task_creation_pass = False
|
116 |
+
return
|
117 |
|
118 |
# self.curr_task_name = self.generated_task['task-name']
|
119 |
print("task creation time {:.3f}".format(time.time() - start_time))
|