|
|
@ -68,7 +68,7 @@ async def execute(request: ExecuteRequest) -> ExecuteResponse:
|
|
|
|
return {"answer": str(e), "files": []}
|
|
|
|
return {"answer": str(e), "files": []}
|
|
|
|
|
|
|
|
|
|
|
|
files = re.findall(r"\[file://\S*\]", res["output"])
|
|
|
|
files = re.findall(r"\[file://\S*\]", res["output"])
|
|
|
|
files = [file[1:-1] for file in files]
|
|
|
|
files = [file[1:-1].split("file://")[1] for file in files]
|
|
|
|
|
|
|
|
|
|
|
|
return {
|
|
|
|
return {
|
|
|
|
"answer": res["output"],
|
|
|
|
"answer": res["output"],
|
|
|
@ -97,7 +97,7 @@ async def execute_async(execution_id: str):
|
|
|
|
if execution.status == "SUCCESS" and execution.result:
|
|
|
|
if execution.status == "SUCCESS" and execution.result:
|
|
|
|
output = execution.result.get("output", "")
|
|
|
|
output = execution.result.get("output", "")
|
|
|
|
files = re.findall(r"\[file://\S*\]", output)
|
|
|
|
files = re.findall(r"\[file://\S*\]", output)
|
|
|
|
files = [file[1:-1] for file in files]
|
|
|
|
files = [file[1:-1].split("file://")[1] for file in files]
|
|
|
|
result = {
|
|
|
|
result = {
|
|
|
|
"answer": output,
|
|
|
|
"answer": output,
|
|
|
|
"files": [uploader.upload(file) for file in files],
|
|
|
|
"files": [uploader.upload(file) for file in files],
|
|
|
|