update models with required

Tauffer-Consulting · Oct 24, 2023 · 5aa9c24 · 5aa9c24
1 parent 68b20cd
commit 5aa9c24
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 22 deletions.
diff --git a/.domino/compiled_metadata.json b/.domino/compiled_metadata.json
@@ -38,15 +38,8 @@
                 "file_path": {
                     "title": "File Path",
                     "description": "The path to the text file to process.",
-                    "anyOf": [
-                        {
-                            "type": "string",
-                            "format": "file-path"
-                        },
-                        {
-                            "type": "string"
-                        }
-                    ]
+                    "required": true,
+                    "type": "string"
                 },
                 "model_size": {
                     "description": "The size of the model to use.",
@@ -128,7 +121,7 @@
             }
         },
         "secrets_schema": null,
-        "source_url": "https://github.com/Tauffer-Consulting/ml_domino_pieces/tree/main/pieces/WhisperPiece"
+        "source_url": "Tauffer-Consulting/ml_domino_pieces//tree/main/pieces/WhisperPiece"
     },
     "TextSummarizerPiece": {
         "name": "TextSummarizerPiece",
@@ -170,15 +163,7 @@
                     "title": "Input File Path",
                     "description": "The path to the text file to summarize.",
                     "default": "",
-                    "anyOf": [
-                        {
-                            "type": "string",
-                            "format": "file-path"
-                        },
-                        {
-                            "type": "string"
-                        }
-                    ]
+                    "type": "string"
                 },
                 "input_text": {
                     "title": "Input Text",
@@ -248,6 +233,6 @@
             }
         },
         "secrets_schema": null,
-        "source_url": "https://github.com/Tauffer-Consulting/ml_domino_pieces/tree/main/pieces/TextSummarizerPiece"
+        "source_url": "Tauffer-Consulting/ml_domino_pieces//tree/main/pieces/TextSummarizerPiece"
     }
 }
diff --git a/pieces/TextSummarizerPiece/models.py b/pieces/TextSummarizerPiece/models.py
@@ -12,7 +12,7 @@ class InputModel(BaseModel):
     """
     Input data for TextSummarizerPiece
     """
-    input_file_path: Union[FilePath, str] = Field(
+    input_file_path: str = Field(
         description='The path to the text file to summarize.',
         default=""
     )

diff --git a/pieces/WhisperPiece/models.py b/pieces/WhisperPiece/models.py
@@ -20,8 +20,9 @@ class InputModel(BaseModel):
     """
     Input data for WhisperPiece
     """
-    file_path: Union[FilePath, str] = Field(
+    file_path: str = Field(
         description='The path to the text file to process.',
+        required=True
     )
     model_size: ModelSizeType = Field(
         description='The size of the model to use.',