From 3fd80fb26f70ca3debadab351fb35e1673e045d1 Mon Sep 17 00:00:00 2001
From: Neil Dwyer <neildwyer1991@gmail.com>
Date: Fri, 1 Dec 2023 16:27:30 -0800
Subject: [PATCH] Setup docs generation

---
 .github/workflows/publish-livekit-agents.yml  |  1 +
 .../publish-livekit-plugins-anthropic.yml     |  1 +
 .../publish-livekit-plugins-elevenlabs.yml    |  1 +
 .../publish-livekit-plugins-google.yml        |  1 +
 .../publish-livekit-plugins-openai.yml        |  1 +
 .../workflows/publish-livekit-plugins-vad.yml |  1 +
 .github/workflows/publish-package.yml         | 24 +++++++++++++++++++
 livekit-agents/livekit/agents/worker.py       |  3 +--
 8 files changed, 31 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/publish-livekit-agents.yml b/.github/workflows/publish-livekit-agents.yml
index 7c0702e58..916d9779c 100644
--- a/.github/workflows/publish-livekit-agents.yml
+++ b/.github/workflows/publish-livekit-agents.yml
@@ -8,4 +8,5 @@ jobs:
     uses: ./.github/workflows/publish-package.yml
     with:
       package_name: 'livekit-agents'
+      docs_entrypoint: 'livekit-plugins/livekit-agents/livekit/agents'
     secrets: inherit
diff --git a/.github/workflows/publish-livekit-plugins-anthropic.yml b/.github/workflows/publish-livekit-plugins-anthropic.yml
index cc2218491..f20e54013 100644
--- a/.github/workflows/publish-livekit-plugins-anthropic.yml
+++ b/.github/workflows/publish-livekit-plugins-anthropic.yml
@@ -8,4 +8,5 @@ jobs:
     uses: ./.github/workflows/publish-package.yml
     with:
       package_name: 'livekit-plugins/livekit-plugins-anthropic'
+      docs_entrypoint: 'livekit-plugins/livekit-plugins-anthropic/livekit/plugins/anthropic'
     secrets: inherit
diff --git a/.github/workflows/publish-livekit-plugins-elevenlabs.yml b/.github/workflows/publish-livekit-plugins-elevenlabs.yml
index bd8acb08d..0c1382aa1 100644
--- a/.github/workflows/publish-livekit-plugins-elevenlabs.yml
+++ b/.github/workflows/publish-livekit-plugins-elevenlabs.yml
@@ -8,4 +8,5 @@ jobs:
     uses: ./.github/workflows/publish-package.yml
     with:
       package_name: 'livekit-plugins/livekit-plugins-elevenlabs'
+      docs_entrypoint: 'livekit-plugins/livekit-plugins-elevenlabs/livekit/plugins/elevenlabs'
     secrets: inherit
diff --git a/.github/workflows/publish-livekit-plugins-google.yml b/.github/workflows/publish-livekit-plugins-google.yml
index 1f36a78cb..018822da2 100644
--- a/.github/workflows/publish-livekit-plugins-google.yml
+++ b/.github/workflows/publish-livekit-plugins-google.yml
@@ -8,4 +8,5 @@ jobs:
     uses: ./.github/workflows/publish-package.yml
     with:
       package_name: 'livekit-plugins/livekit-plugins-google'
+      docs_entrypoint: 'livekit-plugins/livekit-plugins-google/livekit/plugins/google'
     secrets: inherit
diff --git a/.github/workflows/publish-livekit-plugins-openai.yml b/.github/workflows/publish-livekit-plugins-openai.yml
index c90f62be8..b6deae42b 100644
--- a/.github/workflows/publish-livekit-plugins-openai.yml
+++ b/.github/workflows/publish-livekit-plugins-openai.yml
@@ -8,4 +8,5 @@ jobs:
     uses: ./.github/workflows/publish-package.yml
     with:
       package_name: 'livekit-plugins/livekit-plugins-openai'
+      docs_entrypoint: 'livekit-plugins/livekit-plugins-openai/livekit/plugins/openai'
     secrets: inherit
diff --git a/.github/workflows/publish-livekit-plugins-vad.yml b/.github/workflows/publish-livekit-plugins-vad.yml
index 23181fdd1..72f87c426 100644
--- a/.github/workflows/publish-livekit-plugins-vad.yml
+++ b/.github/workflows/publish-livekit-plugins-vad.yml
@@ -8,4 +8,5 @@ jobs:
     uses: ./.github/workflows/publish-package.yml
     with:
       package_name: 'livekit-plugins/livekit-plugins-vad'
+      docs_entrypoint: 'livekit-plugins/livekit-plugins-vad/livekit/plugins/vad'
     secrets: inherit
diff --git a/.github/workflows/publish-package.yml b/.github/workflows/publish-package.yml
index c884ce0b6..b6e49bc32 100644
--- a/.github/workflows/publish-package.yml
+++ b/.github/workflows/publish-package.yml
@@ -14,9 +14,16 @@ on:
       package_name:
         required: true
         type: string
+      docs_entrypoint:
+        required: true
+        type: string
     secrets:
       PYPI_API_TOKEN:
         required: true
+      DOCS_DEPLOY_AWS_ACCESS_KEY:
+        required: true
+      DOCS_DEPLOY_AWS_API_SECRET:
+        required: true
 
 permissions:
   contents: read
@@ -72,3 +79,20 @@ jobs:
       with:
         user: __token__
         password: ${{ secrets.PYPI_API_TOKEN }}
+
+  docs:
+    needs:
+    - publish
+
+    - name: Install dependencies
+      run: python -m pip install --upgrade pdoc  
+
+    - name: Build Docs
+        if: steps.changesets.outputs.published == 'true'
+        run: python -m pdoc ${{ inputs.docs_entrypoint }} --docformat=google --html --output-dir docs
+    - name: S3 Upload
+        run: aws s3 cp docs/ s3://livekit-docs/${{ inputs.package_name }} --recursive
+        env:
+          AWS_ACCESS_KEY_ID: ${{ secrets.DOCS_DEPLOY_AWS_ACCESS_KEY }}
+          AWS_SECRET_ACCESS_KEY: ${{ secrets.DOCS_DEPLOY_AWS_API_SECRET }}
+          AWS_DEFAULT_REGION: "us-east-1"
diff --git a/livekit-agents/livekit/agents/worker.py b/livekit-agents/livekit/agents/worker.py
index fd174c77c..9df40a3ac 100644
--- a/livekit-agents/livekit/agents/worker.py
+++ b/livekit-agents/livekit/agents/worker.py
@@ -79,8 +79,7 @@ def __init__(
         api_key: str = os.environ.get("LIVEKIT_API_KEY", ""),
         api_secret: str = os.environ.get("LIVEKIT_API_SECRET", ""),
     ) -> None:
-        """_summary_
-
+        """
         Args:
             job_request_cb (Callable[[JobRequest], Coroutine]): Callback that is triggered when a new Job is available.
             worker_type (JobType.ValueType): What kind of jobs this worker can handle.