same as https://github.com/Agenta-AI/agenta/pull/949/files

agenta-cli/agenta/__init__.py

@@ -9,6 +9,7 @@
     MultipleChoiceParam,
     MessagesInput,
     TextParam,
+    FileInputURL,
 )
 from .sdk.utils.preinit import PreInitObject
 from .sdk.agenta_init import Config, init

agenta-cli/agenta/sdk/__init__.py

@@ -11,6 +11,7 @@
     MultipleChoiceParam,
     TextParam,
     MessagesInput,
+    FileInputURL,
 )
 from .agenta_init import Config, init
 

agenta-cli/agenta/sdk/agenta_decorator.py

@@ -25,6 +25,7 @@
     MultipleChoiceParam,
     TextParam,
     MessagesInput,
+    FileInputURL,
 )
 
 app = FastAPI()
@@ -314,6 +315,7 @@ def override_schema(openapi_schema: dict, func_name: str, endpoint: str, params:
     - The min and max values for each IntParam instance
     - The default value for DictInput instance
     - The default value for MessagesParam instance
+    - The default value for FileInputURL instance
     - ... [PLEASE ADD AT EACH CHANGE]
 
     Args:
@@ -380,3 +382,9 @@ def find_in_schema(schema: dict, param_name: str, xparam: str):
         ):
             subschema = find_in_schema(schema_to_override, param_name, "messages")
             subschema["default"] = param_val.default
+        if (
+            isinstance(param_val, inspect.Parameter)
+            and param_val.annotation is FileInputURL
+        ):
+            subschema = find_in_schema(schema_to_override, param_name, "file_url")
+            subschema["default"] = "https://example.com"

agenta-cli/agenta/sdk/types.py

@@ -1,7 +1,7 @@
 import json
 from typing import Any, Dict, List
 
-from pydantic import BaseModel, Extra
+from pydantic import BaseModel, Extra, HttpUrl
 
 
 class InFile:
@@ -125,6 +125,12 @@ def __modify_schema__(cls, field_schema: dict[str, Any]):
         field_schema.update({"x-parameter": "messages", "type": "array"})
 
 
+class FileInputURL(HttpUrl):
+    @classmethod
+    def __modify_schema__(cls, field_schema: Dict[str, Any]) -> None:
+        field_schema.update({"x-parameter": "file_url", "type": "string"})
+
+
 class Context(BaseModel):
     class Config:
         extra = Extra.allow

agenta-cli/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "agenta"
-version = "0.6.1"
+version = "0.6.2"
 description = "The SDK for agenta is an open-source LLMOps platform."
 readme = "README.md"
 authors = ["Mahmoud Mabrouk <[email protected]>"]