Add Inference of SWE-bench for Data Interpreter

This commit is contained in:
mannaandpoem 2024-03-18 14:30:53 +08:00
parent e783e5b208
commit ef3be47f28
4 changed files with 180 additions and 1 deletions

View file

@ -182,3 +182,6 @@ class DataInterpreter(Role):
print(result)
data_info = DATA_INFO.format(info=result)
self.working_memory.add(Message(content=data_info, role="user", cause_by=CheckData))
def get_last_cell_source(self):
return self.execute_code.nb.cells[-1].source