routers/prompt.py

from model.llm_agent import llm_agent


class LLMPrompt(object):
    def __init__(self):
        pass

    def prompt(self, req: dict):
        text = req.get("text")
        result = llm_agent.prompt(text)

        return {"result": result}

llm_prompt = LLMPrompt()