Merge pull request #13 from workflowai/spatika/file-input

Syncing file input based updates

Author: Spatika-Ganesh

workflowai/core/domain/errors.py

@@ -31,6 +31,9 @@
     "provider_unavailable",
     # The request timed out
     "read_timeout",
+    # The requested model does not support the requested generation mode
+    # (e-g a model that does not support images generation was sent an image)
+    "model_does_not_support_mode",
 ]
 
 ErrorCode = Union[
@@ -43,9 +46,6 @@
         "no_provider_supporting_model",
         # The requested provider does not support the model
         "provider_does_not_support_model",
-        # The requested model does not support the requested generation mode
-        # (e-g a model that does not support images generation was sent an image)
-        "model_does_not_support_mode",
         # Run properties are invalid, for example the model does not exist
         "invalid_run_properties",
         # An internal error occurred

workflowai/core/fields/file.py

@@ -0,0 +1,10 @@
+from pydantic import BaseModel, Field
+
+
+class File(BaseModel):
+    name: str = Field(..., description="An optional name for the image")
+    content_type: str = Field(..., description="The content type of the image", examples=["image/png", "image/jpeg"])
+    data: str = Field(..., description="The base64 encoded data of the image")
+
+    def to_url(self) -> str:
+        return f"data:{self.content_type};base64,{self.data}"