diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 1cceaa2..bff4aae 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -14,12 +14,26 @@ jobs:
         with:
           repository: 'specklesystems/speckle-automate-github-action'
           path: 'github-action'
+      - name: Install and configure Poetry
+        uses: snok/install-poetry@v1
+        with:
+          version: 1.3.2
+          virtualenvs-create: false
+          virtualenvs-in-project: false
+          installer-parallel: true
+      - name: Restore dependencies
+        run: poetry install
+      - name: Extract functionInputSchema
+        run: |
+          echo "function_input_schema=$(python schema_generation.py)" >> "$GITHUB_ENV"
       - uses: ./github-action
         with:
           speckle_server_url: 'https://automate.speckle.dev'
           speckle_token: ${{ secrets.SPECKLE_AUTOMATE_FUNCTION_PUBLISH_TOKEN }}
           speckle_function_id: ${{ secrets.SPECKLE_AUTOMATE_FUNCTION_ID }}
-          speckle_function_path: 'tasks/getting-started/typescript'
+          speckle_function_input_schema: ${{ env.function_input_schema }}
+          speckle_function_command: 'python main.py'
+          speckle_function_path: '.'
       - name: Log in to Speckle Automate Docker registry
         uses: docker/login-action@f4ef78c080cd8ba55a85445d5b36e214a81df20a
         with: