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

ad-astra-video · web-flow · commit 5237d5c17b8c · 2024-12-26T14:42:34.000-06:00
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/checkout@v4.1.1
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

Original file line number	Diff line number	Diff line change
`@@ -61,6 +61,7 @@ var pipelineToImage = map[string]string{`
`61`	`61`	`"segment-anything-2": "livepeer/ai-runner:segment-anything-2",`
`62`	`62`	`"text-to-speech": "livepeer/ai-runner:text-to-speech",`
`63`	`63`	`"audio-to-text": "livepeer/ai-runner:audio-to-text",`
	`64`	`+ "llm": "livepeer/ai-runner:llm",`
`64`	`65`	`}`
`65`	`66`
`66`	`67`	`var livePipelineToImage = map[string]string{`