agent_cached_responses.py

import os
import json
import cryptocode

class AgentCachedResponses:
    DATABASE_ANSWERS_ENCRYPTED = "./database/answers_encrypted.json"

    def __init__(self):
        json_data_raw = None
        with open(AgentCachedResponses.DATABASE_ANSWERS_ENCRYPTED, "r") as f:
            json_data_raw = json.load(f)

        _json_answers_data_content = cryptocode.decrypt(
            json_data_raw["encoded_answers_data"],
            os.environ["ANSWERS_DATABASE_PASSWORD"]
        )

        self._json_answers_data = json.loads(_json_answers_data_content)
        
    def __call__(self, task_id: str, question: str, input_file: str) -> str:
        answer = self._json_answers_data.get(task_id)["answer"]
        return answer