Create task_assigner_chain.py

AI_Agent/chains/task_assigner_chain.py

@@ -0,0 +1,17 @@
+# 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")}