Refactor to move routing logic to router instead of in the team.

.github/workflows/build-project.yml

@@ -0,0 +1,33 @@
+name: CI Build Flo-AI
+
+on:
+  pull_request:
+    branches:
+      - develop
+  workflow_dispatch:  # Allows manual triggering from the GitHub Actions UI
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+
+    steps:
+    - name: Checkout code
+      uses: actions/checkout@v3
+
+    - name: Set up Python
+      uses: actions/setup-python@v4
+      with:
+        python-version: '3.11'
+
+    - name: Install Poetry
+      run: |
+        curl -sSL https://install.python-poetry.org | python3
+
+    - name: Poetry lock
+      run: poetry lock
+
+    - name: Install dependencies
+      run: poetry install
+
+    - name: Run build
+      run: poetry build

.gitignore

@@ -10,4 +10,5 @@ dataset
 .deepeval-cache.json
 bin
 .DS_Store
-*.sql
+*.sql
+*.png

examples/email_reply_agent.ipynb

@@ -124,7 +124,9 @@
      "output_type": "stream",
      "text": [
       "/Users/vizsatiz/Documents/hub/flo/.venv/lib/python3.11/site-packages/sentence_transformers/cross_encoder/CrossEncoder.py:11: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html\n",
-      "  from tqdm.autonotebook import tqdm, trange\n"
+      "  from tqdm.autonotebook import tqdm, trange\n",
+      "/Users/vizsatiz/Documents/hub/flo/.venv/lib/python3.11/site-packages/transformers/tokenization_utils_base.py:1601: FutureWarning: `clean_up_tokenization_spaces` was not set. It will be set to `True` by default. This behavior will be depracted in transformers v4.45, and will be then set to `False` by default. For more details check this issue: https://github.com/huggingface/transformers/issues/31884\n",
+      "  warnings.warn(\n"
      ]
     }
    ],
@@ -160,7 +162,7 @@
     {
      "data": {
       "text/plain": [
-       "<flo_ai.state.flo_session.FloSession at 0x15d90ca90>"
+       "<flo_ai.state.flo_session.FloSession at 0x161db0510>"
       ]
      },
      "execution_count": 5,
@@ -200,7 +202,7 @@
     {
      "data": {
       "text/plain": [
-       "<flo_ai.state.flo_session.FloSession at 0x15d90ca90>"
+       "<flo_ai.state.flo_session.FloSession at 0x161db0510>"
       ]
      },
      "execution_count": 6,
@@ -227,7 +229,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 14,
+   "execution_count": 7,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -268,7 +270,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 21,
+   "execution_count": 8,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -308,7 +310,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 22,
+   "execution_count": 9,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -338,14 +340,28 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 23,
+   "execution_count": 10,
    "metadata": {},
    "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "huggingface/tokenizers: The current process just got forked, after parallelism has already been used. Disabling parallelism to avoid deadlocks...\n",
+      "To disable this warning, you can either:\n",
+      "\t- Avoid using `tokenizers` before the fork if possible\n",
+      "\t- Explicitly set the environment variable TOKENIZERS_PARALLELISM=(true | false)\n",
+      "huggingface/tokenizers: The current process just got forked, after parallelism has already been used. Disabling parallelism to avoid deadlocks...\n",
+      "To disable this warning, you can either:\n",
+      "\t- Avoid using `tokenizers` before the fork if possible\n",
+      "\t- Explicitly set the environment variable TOKENIZERS_PARALLELISM=(true | false)\n"
+     ]
+    },
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "{'SupportSupervisor-0XuY9': {'next': 'TransactionFetcher-7pEar'}}\n",
+      "{'SupportSupervisor-6y1Aw': {'next': 'TransactionFetcher-Rchvv'}}\n",
       "----\n",
       "\n",
       "\n",
@@ -358,45 +374,45 @@
       "\n",
       "Hi Vishnu,\n",
       "\n",
-      "Thank you for reaching out.\n",
+      "Thank you for reaching out regarding the failed transaction.\n",
       "\n",
-      "The transaction with ID 12123123432, which occurred on 23/07/2024 IST, failed due to insufficient balance in the account.\n",
+      "The transaction with ID 12123123432 occurred on 23/07/2024 IST and failed due to insufficient balance in the account.\n",
       "\n",
-      "If you have any further questions or need additional assistance, please feel free to ask.\n",
+      "If you have any further questions or need assistance with anything else, feel free to ask!\n",
       "\n",
       "Best regards,  \n",
       "Tom\u001b[0m\n",
       "\n",
       "\u001b[1m> Finished chain.\u001b[0m\n",
-      "{'TransactionFetcher-7pEar': {'messages': [HumanMessage(content='Subject: Re: Failed Transaction Inquiry\\n\\nHi Vishnu,\\n\\nThank you for reaching out.\\n\\nThe transaction with ID 12123123432, which occurred on 23/07/2024 IST, failed due to insufficient balance in the account.\\n\\nIf you have any further questions or need additional assistance, please feel free to ask.\\n\\nBest regards,  \\nTom', name='TransactionFetcher-7pEar')]}}\n",
+      "{'TransactionFetcher-Rchvv': {'messages': [HumanMessage(content='Subject: Re: Failed Transaction Inquiry\\n\\nHi Vishnu,\\n\\nThank you for reaching out regarding the failed transaction.\\n\\nThe transaction with ID 12123123432 occurred on 23/07/2024 IST and failed due to insufficient balance in the account.\\n\\nIf you have any further questions or need assistance with anything else, feel free to ask!\\n\\nBest regards,  \\nTom', name='TransactionFetcher-Rchvv')]}}\n",
       "----\n",
-      "{'SupportSupervisor-0XuY9': {'next': 'EmailSender-YJNuR'}}\n",
+      "{'SupportSupervisor-6y1Aw': {'next': 'EmailSender-N1MZP'}}\n",
       "----\n",
       "\n",
       "\n",
       "\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n",
       "\u001b[32;1m\u001b[1;3m\n",
-      "Invoking: `email_triage` with `{'email': 'Subject: Re: Failed Transaction Inquiry\\n\\nHi Vishnu,\\n\\nThank you for reaching out.\\n\\nThe transaction with ID 12123123432, which occurred on 23/07/2024 IST, failed due to insufficient balance in the account.\\n\\nIf you have any further questions or need additional assistance, please feel free to ask.\\n\\nBest regards,\\nTom'}`\n",
+      "Invoking: `email_triage` with `{'email': 'Subject: Re: Failed Transaction Inquiry\\n\\nHi Vishnu,\\n\\nThank you for reaching out regarding the failed transaction.\\n\\nThe transaction with ID 12123123432 occurred on 23/07/2024 IST and failed due to insufficient balance in the account.\\n\\nIf you have any further questions or need assistance with anything else, feel free to ask!\\n\\nBest regards,\\nTom'}`\n",
       "\n",
       "\n",
       "\u001b[0mSubject: Re: Failed Transaction Inquiry\n",
       "\n",
       "Hi Vishnu,\n",
       "\n",
-      "Thank you for reaching out.\n",
+      "Thank you for reaching out regarding the failed transaction.\n",
       "\n",
-      "The transaction with ID 12123123432, which occurred on 23/07/2024 IST, failed due to insufficient balance in the account.\n",
+      "The transaction with ID 12123123432 occurred on 23/07/2024 IST and failed due to insufficient balance in the account.\n",
       "\n",
-      "If you have any further questions or need additional assistance, please feel free to ask.\n",
+      "If you have any further questions or need assistance with anything else, feel free to ask!\n",
       "\n",
       "Best regards,\n",
       "Tom\n",
-      "\u001b[36;1m\u001b[1;3mEmail sent successfully\u001b[0m\u001b[32;1m\u001b[1;3mI have sent the email to Vishnu with the details regarding the failed transaction. If there are any further questions or additional assistance needed, Vishnu can reach out again.\u001b[0m\n",
+      "\u001b[36;1m\u001b[1;3mEmail sent successfully\u001b[0m\u001b[32;1m\u001b[1;3mI have sent the response to Vishnu regarding the failed transaction. If you need any further assistance, feel free to ask!\u001b[0m\n",
       "\n",
       "\u001b[1m> Finished chain.\u001b[0m\n",
-      "{'EmailSender-YJNuR': {'messages': [HumanMessage(content='I have sent the email to Vishnu with the details regarding the failed transaction. If there are any further questions or additional assistance needed, Vishnu can reach out again.', name='EmailSender-YJNuR')]}}\n",
+      "{'EmailSender-N1MZP': {'messages': [HumanMessage(content='I have sent the response to Vishnu regarding the failed transaction. If you need any further assistance, feel free to ask!', name='EmailSender-N1MZP')]}}\n",
       "----\n",
-      "{'SupportSupervisor-0XuY9': {'next': 'FINISH'}}\n",
+      "{'SupportSupervisor-6y1Aw': {'next': 'FINISH'}}\n",
       "----\n"
      ]
     }