update

runner/app/main.py

@@ -50,10 +50,10 @@ def load_pipeline(pipeline: str, model_id: str) -> any:
             from app.pipelines.upscale import UpscalePipeline
 
             return UpscalePipeline(model_id)
-        case "sentiment-analysis":
-            from app.pipelines.sentiment_analysis import SentimentAnalysisPipeline
+        case "text-sentiment-analysis":
+            from app.pipelines.text_sentiment_analysis import TextSentimentAnalysisPipeline
 
-            return SentimentAnalysisPipeline(model_id)
+            return TextSentimentAnalysisPipeline(model_id)
         case "segment-anything-2":
             from app.pipelines.segment_anything_2 import SegmentAnything2Pipeline
 
@@ -88,10 +88,10 @@ def load_route(pipeline: str) -> any:
             from app.routes import upscale
 
             return upscale.router
-        case "sentiment-analysis":
-            from app.routes import sentiment_analysis
+        case "text-sentiment-analysis":
+            from app.routes import text_sentiment_analysis
 
-            return sentiment_analysis.router
+            return text_sentiment_analysis.router
         case "segment-anything-2":
             from app.routes import segment_anything_2
 

runner/app/pipelines/sentiment_analysis.py → runner/app/pipelines/text_sentiment_analysis.py

@@ -6,7 +6,7 @@
 logger = logging.getLogger(__name__)
 
 
-class SentimentAnalysisPipeline(Pipeline):
+class TextSentimentAnalysisPipeline(Pipeline):
     def __init__(self, model_id: str):
         self.model_id = model_id
         kwargs = {}
@@ -31,4 +31,4 @@ def __call__(self, text: str, **kwargs):
         return self.ldm(text, **kwargs)
 
     def __str__(self) -> str:
-        return f"SentimentAnalysisPipeline model_id={self.model_id}"
+        return f"TextSentimentAnalysisPipeline model_id={self.model_id}"

runner/app/routes/sentiment_analysis.py → runner/app/routes/text_sentiment_analysis.py

@@ -4,7 +4,7 @@
 
 from app.dependencies import get_pipeline
 from app.pipelines.base import Pipeline
-from app.routes.util import HTTPError, SentimentAnalysisResponse, http_error
+from app.routes.util import HTTPError, TextSentimentAnalysisResponse, http_error
 from fastapi import APIRouter, Depends, status
 from fastapi.responses import JSONResponse
 from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer
@@ -15,6 +15,15 @@
 logger = logging.getLogger(__name__)
 
 RESPONSES = {
+    status.HTTP_200_OK: {
+        "content": {
+            "application/json": {
+                "schema": {
+                    "x-speakeasy-name-override": "data",
+                }
+            }
+        },
+    },
     status.HTTP_400_BAD_REQUEST: {"model": HTTPError},
     status.HTTP_401_UNAUTHORIZED: {"model": HTTPError},
     status.HTTP_500_INTERNAL_SERVER_ERROR: {"model": HTTPError},
@@ -41,18 +50,22 @@ def handle_pipeline_error(e: Exception) -> JSONResponse:
 
 
 @router.post(
-    "/sentiment-analysis",
-    response_model=SentimentAnalysisResponse,
+    "/text-sentiment-analysis",
+    response_model=TextSentimentAnalysisResponse,
     responses=RESPONSES,
-    description="Analyze the sentiment of the provided text."
+    description="Analyze the sentiment of a given text inputs.",
+    operation_id="analyzeSentiment",
+    summary="Text Sentiment Analysis",
+    tags=["analysis"],
+    openapi_extra={"x-speakeasy-name-override": "textSentimentAnalysis"},
 )
 @router.post(
-    "/sentiment-analysis/",
-    response_model=SentimentAnalysisResponse,
+    "/text-sentiment-analysis/",
+    response_model=TextSentimentAnalysisResponse,
     responses=RESPONSES,
     include_in_schema=False,
 )
-async def sentiment_analysis(
+async def text_sentiment_analysis(
     model_id: Annotated[
         str,
         Field(
@@ -62,7 +75,7 @@ async def sentiment_analysis(
     ],
     text_input: Annotated[
         str,
-        Field(description="Text to classify. Separate multiple sentences by commas.")
+        Field(description="Text to analyze. For multiple sentences, separate them with commas.")
     ],
     pipeline: Pipeline = Depends(get_pipeline),
     token: HTTPAuthorizationCredentials = Depends(HTTPBearer(auto_error=False)),

runner/app/routes/util.py

@@ -63,7 +63,7 @@ class LabelScore(BaseModel):
     score: float
 
 
-class SentimentAnalysisResponse(BaseModel):
+class TextSentimentAnalysisResponse(BaseModel):
     results: list[LabelScore]
 
 

runner/gateway.openapi.yaml

@@ -303,6 +303,64 @@ paths:
       security:
       - HTTPBearer: []
       x-speakeasy-name-override: segmentAnything2
+  /text-sentiment-analysis:
+    post:
+      tags:
+      - analysis
+      summary: Text Sentiment Analysis
+      description: Analyze the sentiment of a given text inputs.
+      operationId: analyzeSentiment
+      security:
+      - HTTPBearer: []
+      parameters:
+      - name: model_id
+        in: query
+        required: true
+        schema:
+          type: string
+          description: Hugging Face model ID used for text classsification.
+          title: Model Id
+      - name: text_input
+        in: query
+        required: true
+        schema:
+          type: string
+          description: Text to analyze. For multiple sentences, separate them with
+            commas.
+          title: Text Input
+      responses:
+        '200':
+          description: Successful Response
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/TextSentimentAnalysisResponse'
+                x-speakeasy-name-override: data
+        '400':
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/HTTPError'
+          description: Bad Request
+        '401':
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/HTTPError'
+          description: Unauthorized
+        '500':
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/HTTPError'
+          description: Internal Server Error
+        '422':
+          description: Validation Error
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/HTTPValidationError'
+      x-speakeasy-name-override: textSentimentAnalysis
 components:
   schemas:
     APIError:
@@ -586,6 +644,23 @@ components:
       - images
       title: ImageResponse
       description: Response model for image generation.
+    LabelScore:
+      properties:
+        label:
+          type: string
+          enum:
+          - negative
+          - neutral
+          - positive
+          title: Label
+        score:
+          type: number
+          title: Score
+      type: object
+      required:
+      - label
+      - score
+      title: LabelScore
     MasksResponse:
       properties:
         masks:
@@ -646,6 +721,17 @@ components:
       - chunks
       title: TextResponse
       description: Response model for text generation.
+    TextSentimentAnalysisResponse:
+      properties:
+        results:
+          items:
+            $ref: '#/components/schemas/LabelScore'
+          type: array
+          title: Results
+      type: object
+      required:
+      - results
+      title: TextSentimentAnalysisResponse
     TextToImageParams:
       properties:
         model_id:

runner/gen_openapi.py

@@ -10,6 +10,7 @@
     image_to_image,
     image_to_video,
     segment_anything_2,
+    text_sentiment_analysis,
     text_to_image,
     upscale,
 )
@@ -123,7 +124,8 @@ def write_openapi(fname: str, entrypoint: str = "runner", version: str = "0.0.0"
     app.include_router(upscale.router)
     app.include_router(audio_to_text.router)
     app.include_router(segment_anything_2.router)
-
+    app.include_router(text_sentiment_analysis.router)
+
     logger.info(f"Generating OpenAPI schema for '{entrypoint}' entrypoint...")
     openapi = get_openapi(
         title="Livepeer AI Runner",

runner/openapi.yaml

@@ -314,6 +314,64 @@ paths:
       security:
       - HTTPBearer: []
       x-speakeasy-name-override: segmentAnything2
+  /text-sentiment-analysis:
+    post:
+      tags:
+      - analysis
+      summary: Text Sentiment Analysis
+      description: Analyze the sentiment of a given text inputs.
+      operationId: analyzeSentiment
+      security:
+      - HTTPBearer: []
+      parameters:
+      - name: model_id
+        in: query
+        required: true
+        schema:
+          type: string
+          description: Hugging Face model ID used for text classsification.
+          title: Model Id
+      - name: text_input
+        in: query
+        required: true
+        schema:
+          type: string
+          description: Text to analyze. For multiple sentences, separate them with
+            commas.
+          title: Text Input
+      responses:
+        '200':
+          description: Successful Response
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/TextSentimentAnalysisResponse'
+                x-speakeasy-name-override: data
+        '400':
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/HTTPError'
+          description: Bad Request
+        '401':
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/HTTPError'
+          description: Unauthorized
+        '500':
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/HTTPError'
+          description: Internal Server Error
+        '422':
+          description: Validation Error
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/HTTPValidationError'
+      x-speakeasy-name-override: textSentimentAnalysis
 components:
   schemas:
     APIError:
@@ -600,6 +658,23 @@ components:
       - images
       title: ImageResponse
       description: Response model for image generation.
+    LabelScore:
+      properties:
+        label:
+          type: string
+          enum:
+          - negative
+          - neutral
+          - positive
+          title: Label
+        score:
+          type: number
+          title: Score
+      type: object
+      required:
+      - label
+      - score
+      title: LabelScore
     MasksResponse:
       properties:
         masks:
@@ -660,6 +735,17 @@ components:
       - chunks
       title: TextResponse
       description: Response model for text generation.
+    TextSentimentAnalysisResponse:
+      properties:
+        results:
+          items:
+            $ref: '#/components/schemas/LabelScore'
+          type: array
+          title: Results
+      type: object
+      required:
+      - results
+      title: TextSentimentAnalysisResponse
     TextToImageParams:
       properties:
         model_id: