|
|
|
@ -7,18 +7,15 @@ from fastapi import FastAPI
|
|
|
|
|
from fastapi.staticfiles import StaticFiles
|
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
|
|
|
|
from ansi import ANSI, Color, Style, dim_multiline
|
|
|
|
|
from core.agents.manager import AgentManager
|
|
|
|
|
from core.handlers.base import BaseHandler, FileHandler, FileType
|
|
|
|
|
from core.handlers.dataframe import CsvToDataframe
|
|
|
|
|
from core.prompts.error import ERROR_PROMPT
|
|
|
|
|
from core.tools.base import BaseToolSet
|
|
|
|
|
from core.tools.cpu import ExitConversation, RequestsGet
|
|
|
|
|
from core.tools.editor import CodeEditor
|
|
|
|
|
from core.tools.terminal import Terminal
|
|
|
|
|
from core.upload import StaticUploader
|
|
|
|
|
from env import settings
|
|
|
|
|
from logger import logger
|
|
|
|
|
|
|
|
|
|
app = FastAPI()
|
|
|
|
|
|
|
|
|
@ -92,7 +89,7 @@ async def command(request: Request) -> Response:
|
|
|
|
|
except Exception as e:
|
|
|
|
|
return {"response": str(e), "files": []}
|
|
|
|
|
|
|
|
|
|
files = re.findall("image/\S*png|dataframe/\S*csv", res["output"])
|
|
|
|
|
files = re.findall("[file/\S*]", res["output"])
|
|
|
|
|
|
|
|
|
|
return {
|
|
|
|
|
"response": res["output"],
|
|
|
|
|