Add input for endpoint

Author: dsanders11

README.md

@@ -49,16 +49,17 @@ jobs:
 
 ### Configuration
 
-| Input                        | Description                                                                        | Default               | Required |
-| ---------------------------- | ---------------------------------------------------------------------------------- | --------------------- | -------- |
-| `token`                      | GitHub token with issues and pull-requests permissions                             | `${{ github.token }}` | No       |
-| `spam-label`                 | Label to add when generic spam is detected                                         | `spam`                | No       |
-| `ai-label`                   | Label to add when AI-generated content is detected                                 | `ai-generated`        | No       |
-| `minimize-detected-comments` | Whether to minimize comments detected as spam                                      | `true`                | No       |
-| `custom-prompt-path`         | Path to a custom YAML prompt file in your repository (relative to repository root) | (none)                | No       |
-| `enable-spam-detection`      | Enable built-in spam detection prompt                                              | `true`                | No       |
-| `enable-link-spam-detection` | Enable built-in link spam detection prompt                                         | `true`                | No       |
-| `enable-ai-detection`        | Enable built-in AI-generated content detection prompt                              | `true`                | No       |
+| Input                        | Description                                                                        | Default                              | Required |
+| ---------------------------- | ---------------------------------------------------------------------------------- | ------------------------------------ | -------- |
+| `token`                      | GitHub token with issues and pull-requests permissions                             | `${{ github.token }}`                | No       |
+| `spam-label`                 | Label to add when generic spam is detected                                         | `spam`                               | No       |
+| `ai-label`                   | Label to add when AI-generated content is detected                                 | `ai-generated`                       | No       |
+| `minimize-detected-comments` | Whether to minimize comments detected as spam                                      | `true`                               | No       |
+| `custom-prompt-path`         | Path to a custom YAML prompt file in your repository (relative to repository root) | (none)                               | No       |
+| `enable-spam-detection`      | Enable built-in spam detection prompt                                              | `true`                               | No       |
+| `enable-link-spam-detection` | Enable built-in link spam detection prompt                                         | `true`                               | No       |
+| `enable-ai-detection`        | Enable built-in AI-generated content detection prompt                              | `true`                               | No       |
+| `endpoint`                   | The endpoint to use for inference                                                  | `https://models.github.ai/inference` | No       |
 
 ## Inference
 

action.yml

@@ -43,6 +43,10 @@ inputs:
     description: 'Enable built-in AI-generated content detection prompt'
     required: false
     default: 'true'
+  endpoint:
+    description: The endpoint to use for inference
+    required: false
+    default: 'https://models.github.ai/inference'
 
 runs:
   using: 'node16'

badges/coverage.svg

@@ -17,6 +17,7 @@ async function run(): Promise<void> {
     const aiLabel = core.getInput('ai-label')
     const minimizeComments = core.getBooleanInput('minimize-detected-comments')
     const customPromptPath = core.getInput('custom-prompt-path')
+    const endpoint = core.getInput('endpoint')
 
     // Built-in prompt configuration
     const enableSpamDetection = core.getBooleanInput('enable-spam-detection')
@@ -27,7 +28,7 @@ async function run(): Promise<void> {
 
     const openai = new OpenAI({
       apiKey: token,
-      baseURL: 'https://models.github.ai/inference'
+      baseURL: endpoint
     })
     const octokit = github.getOctokit(token)