From c60844fb051553476886543accb37aeabd18d69a Mon Sep 17 00:00:00 2001
From: willtai <william.tai@neo4j.com>
Date: Thu, 12 Sep 2024 16:37:59 +0100
Subject: [PATCH] Add job to publish to pypi and test-pypi for neo4j-graphrag
 (#126)

---
 .github/workflows/publish.yaml | 26 +++++++++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/publish.yaml b/.github/workflows/publish.yaml
index e6257ecf..684f0a04 100644
--- a/.github/workflows/publish.yaml
+++ b/.github/workflows/publish.yaml
@@ -35,7 +35,7 @@ jobs:
     runs-on: ubuntu-latest
     environment:
       name: pypi
-      url: https://pypi.org/p/neo4j-genai
+      url: https://pypi.org/p/neo4j-graphrag
     permissions:
       id-token: write
     steps:
@@ -46,3 +46,27 @@ jobs:
           path: dist/
       - name: Publish distribution 📦 to PyPI
         uses: pypa/gh-action-pypi-publish@release/v1
+
+  publish-to-test-pypi:
+    name: TestPyPI
+    needs:
+      - build
+    runs-on: ubuntu-latest
+
+    environment:
+      name: testpypi
+      url: https://test.pypi.org/p/neo4j-graphrag
+
+    permissions:
+      id-token: write
+
+    steps:
+      - name: Download all the dists
+        uses: actions/download-artifact@v4
+        with:
+          name: python-package-distributions
+          path: dist/
+      - name: Publish distribution 📦 to TestPyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          repository-url: https://test.pypi.org/legacy/