wrapper-interpreter.py

from asyncio import timeout, TimeoutError, run
import sys
import json
sys.path.insert(0, "MetaGPT-DataExplainer/metagpt")

from metagpt.logs import logger
from metagpt.roles.de.data_explainer import DataExplainer
from metagpt.roles.di.data_interpreter import DataInterpreter
from metagpt.utils.recovery_util import save_history


async def _explain(request, save_dir):
    di = DataInterpreter()
    try:
        async with timeout(7200):
            rsp = await di.run(request)
    except TimeoutError as e:
        rsp = "========TIMEOUT======="
        print("========TIMEOUT=======")
    except Exception as e:
        rsp = "========ERROR======="
        print("========ERROR=======")
    logger.info(rsp)
    save_history(role=di, save_dir=save_dir)

with open("current-comp.json") as in_file:
    in_json = json.load(in_file)


run(_explain(in_json['request'], in_json['output_dir']))