AI_Agent/chains/task_assigner_chain.py

# AI_Agent/chains/task_assigner_chain.py
class TaskAssignerChain:
    """
    Assign each task to either Frontend or Backend.
    """
    def __init__(self, llm_adapter):
        self.llm = llm_adapter

    async def run(self, tasks_text: str):
        prompt = (
            "You are a project manager AI. Given a numbered list of technical tasks, "
            "assign each task to either the Frontend or Backend team. "
            "Return the results as a numbered list in the format: '1. Task description -> Frontend/Backend'.\n\n"
            f"Tasks:\n{tasks_text}"
        )
        out = await self.llm.generate(prompt, max_tokens=300)
        return {"assigned_tasks_text": out["text"], "assigned_tasks_raw": out.get("raw")}