diff --git a/.github/workflows/python-publish-test.yml b/.github/workflows/python-publish-test.yml
new file mode 100644
index 000000000..64679d582
--- /dev/null
+++ b/.github/workflows/python-publish-test.yml
@@ -0,0 +1,79 @@
+name: Publish Bailo Distribution to TestPyPI
+
+on:
+  workflow_dispatch:
+    inputs:
+      tags:
+        description: 'Publish to TestPyPI.'
+
+jobs:
+  build:
+    name: Build distribution
+    runs-on: ubuntu-latest
+    outputs:
+      files-changed-output: ${{ steps.files-changed.outputs.any_changed }}
+
+    steps:
+      - uses: actions/checkout@v3
+        with:
+          fetch-depth: 0
+
+      # Check for changed init file (for new version)
+      - name: Get changed init files
+        id: files-changed
+        uses: tj-actions/changed-files@v40
+        with:
+          files: |
+            lib/python-beta/src/bailo/__init__.py
+
+      # Setup Python
+      - name: Set up Python
+        uses: actions/setup-python@v4
+        with:
+          python-version: "3.11"
+
+      # Install build package
+      - name: Install pypa/build
+        run: python3 -m pip install build --user
+
+      # Build Bailo distribution
+      - name: Build a binary wheel and a source tarball
+        run: |
+          cd lib/python-beta
+          python3 -m build
+
+      # Upload Bailo artifact
+      - name: Store the distribution packages
+        uses: actions/upload-artifact@v3
+        with:
+          name: python-package-distributions
+          path: lib/python-beta/dist/
+
+  publish-to-testpypi:
+    name: Publish Bailo distribution to TestPyPI
+    needs:
+    - build
+    runs-on: ubuntu-latest
+
+    environment:
+      name: testpypi
+      url: https://test.pypi.org/p/bailo
+
+    permissions:
+      id-token: write
+
+    steps:
+      #Retrieve Bailo artifact
+      - name: Download all the dists
+        if: needs.build.outputs.files-changed-output == 'true'
+        uses: actions/download-artifact@v3
+        with:
+          name: python-package-distributions
+          path: dist/
+
+      # Publish artifact to TestPyPI
+      - name: Publish distribution to TestPyPI
+        if: needs.build.outputs.files-changed-output == 'true'
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          repository-url: https://test.pypi.org/legacy/
\ No newline at end of file
diff --git a/.github/workflows/python-publish.yml b/.github/workflows/python-publish.yml
new file mode 100644
index 000000000..901613f4d
--- /dev/null
+++ b/.github/workflows/python-publish.yml
@@ -0,0 +1,75 @@
+name: Publish Bailo Distribution to PyPI
+
+on:
+  push:
+    branches: [main]
+
+jobs:
+  build:
+    name: Build distribution
+    runs-on: ubuntu-latest
+    outputs:
+      files-changed-output: ${{ steps.files-changed.outputs.any_changed }}
+
+    steps:
+      - uses: actions/checkout@v3
+        with:
+          fetch-depth: 0
+
+      # Check for changed init file (for new version)
+      - name: Get changed init files
+        id: files-changed
+        uses: tj-actions/changed-files@v40
+        with:
+          files: |
+            lib/python-beta/src/bailo/__init__.py
+
+      # Setup Python
+      - name: Set up Python
+        uses: actions/setup-python@v4
+        with:
+          python-version: "3.11"
+
+      # Install build package
+      - name: Install pypa/build
+        run: python3 -m pip install build --user
+
+      # Build Bailo distribution
+      - name: Build a binary wheel and a source tarball
+        run: |
+          cd lib/python-beta
+          python3 -m build
+
+      # Upload Bailo artifact
+      - name: Store the distribution packages
+        uses: actions/upload-artifact@v3
+        with:
+          name: python-package-distributions
+          path: lib/python-beta/dist/
+
+  publish-to-pypi:
+    name: Publish Bailo distribution to PyPI
+    needs:
+    - build
+    runs-on: ubuntu-latest
+
+    environment:
+      name: pypi
+      url: https://pypi.org/p/bailo
+
+    permissions:
+      id-token: write
+
+    steps:
+      # Retrieve Bailo artifact
+      - name: Download all the dists
+        if: needs.build.outputs.files-changed-output == 'true'
+        uses: actions/download-artifact@v3
+        with:
+          name: python-package-distributions
+          path: lib/python-beta/dist/
+
+      # Publish artifact to PyPI
+      - name: Publish distribution to PyPI
+        if: needs.build.outputs.files-changed-output == 'true'
+        uses: pypa/gh-action-pypi-publish@release/v1
\ No newline at end of file
diff --git a/lib/python-beta/src/bailo/__init__.py b/lib/python-beta/src/bailo/__init__.py
index 7e90b6cf0..6abc805d2 100644
--- a/lib/python-beta/src/bailo/__init__.py
+++ b/lib/python-beta/src/bailo/__init__.py
@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-__version__ = "0.2.0"
+__version__ = "0.2.1"
 
 from bailo.core.agent import Agent
 from bailo.core.client import Client