Merge pull request #19 from workflowai/guillaume/image-file

Export File field

Author: guillaq

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "workflowai"
-version = "0.3.3"
+version = "0.3.4"
 description = ""
 authors = ["Guillaume Aquilina <[email protected]>"]
 readme = "README.md"

workflowai/fields.py

@@ -1,5 +1,6 @@
 from workflowai.core.fields.chat_message import ChatMessage as ChatMessage
 from workflowai.core.fields.email_address import EmailAddressStr as EmailAddressStr
+from workflowai.core.fields.file import File as File
 from workflowai.core.fields.html_string import HTMLString as HTMLString
 from workflowai.core.fields.http_url import HttpUrl as HttpUrl
 from workflowai.core.fields.image import Image as Image

workflowai/fields_test.py

@@ -0,0 +1,18 @@
+import os
+
+from workflowai import fields
+
+
+def test_exported_fields_match_core_fields():
+    # Get the number of exported fields in fields.py
+    exported_fields = {attr for attr in dir(fields) if not attr.startswith("_")}
+
+    # Get the number of files in workflowai/core/fields
+    core_fields_dir = os.path.join(os.path.dirname(__file__), "core", "fields")
+    core_field_files = {
+        "".join(word.capitalize() for word in f.removesuffix(".py").split("_"))
+        for f in os.listdir(core_fields_dir)
+        if f.endswith(".py") and f != "__init__.py" and not f.endswith("_test.py")
+    }
+
+    assert len(core_field_files) == len(exported_fields)