fix: files regex

1 year ago · c8ce069ddd
parent 0200369f68
commit c8ce069ddd
1 changed files with 2 additions and 1 deletions
--- a/api/main.py
+++ b/api/main.py
@ -118,7 +118,8 @@ async def execute(request: ExecuteRequest) -> ExecuteResponse:
    except Exception as e:
        return {"answer": str(e), "files": []}

-    files = re.findall("[file/\S*]", res["output"])
+    files = re.findall(r"\[file/\S*\]", res["output"])
+    files = [file[1:-1] for file in files]

    return {
        "answer": res["output"],