runner: fix open api code gen, add llm to workflows and container list (

#384)
livepeer · Dec 26, 2024 · 5237d5c · 5237d5c
1 parent b81f898
commit 5237d5c
Show file tree

Hide file tree

Showing 5 changed files with 93 additions and 87 deletions.
diff --git a/.github/workflows/ai-runner-pipelines-docker.yaml b/.github/workflows/ai-runner-pipelines-docker.yaml
@@ -34,6 +34,7 @@ jobs:
           - docker/Dockerfile.segment_anything_2
           - docker/Dockerfile.text_to_speech
           - docker/Dockerfile.audio_to_text
+          - docker/Dockerfile.llm
     steps:
       - name: Check out code
         uses: actions/[email protected]

diff --git a/runner/gateway.openapi.yaml b/runner/gateway.openapi.yaml
@@ -525,7 +525,8 @@ components:
     AudioResponse:
       properties:
         audio:
-          $ref: '#/components/schemas/MediaURL'
+          allOf:
+          - $ref: '#/components/schemas/MediaURL'
           description: The generated audio.
       type: object
       required:
@@ -826,7 +827,8 @@ components:
     HTTPError:
       properties:
         detail:
-          $ref: '#/components/schemas/APIError'
+          allOf:
+          - $ref: '#/components/schemas/APIError'
           description: Detailed error information.
       type: object
       required:

diff --git a/runner/openapi.yaml b/runner/openapi.yaml
@@ -558,7 +558,8 @@ components:
     AudioResponse:
       properties:
         audio:
-          $ref: '#/components/schemas/MediaURL'
+          allOf:
+          - $ref: '#/components/schemas/MediaURL'
           description: The generated audio.
       type: object
       required:
@@ -918,7 +919,8 @@ components:
     HTTPError:
       properties:
         detail:
-          $ref: '#/components/schemas/APIError'
+          allOf:
+          - $ref: '#/components/schemas/APIError'
           description: Detailed error information.
       type: object
       required:

diff --git a/worker/docker.go b/worker/docker.go
@@ -61,6 +61,7 @@ var pipelineToImage = map[string]string{
 	"segment-anything-2": "livepeer/ai-runner:segment-anything-2",
 	"text-to-speech":     "livepeer/ai-runner:text-to-speech",
 	"audio-to-text":      "livepeer/ai-runner:audio-to-text",
+	"llm":                "livepeer/ai-runner:llm",
 }
 
 var livePipelineToImage = map[string]string{

diff --git a/worker/runner.gen.go b/worker/runner.gen.go