agent-lab-sdk 0.1.44__py3-none-any.whl → 0.1.45__py3-none-any.whl

agent_lab_sdk/schema/__init__.py

@@ -10,13 +10,14 @@ from .input_types import (
     SwitchInput,
     FileInput,
     FilesInput,
-    SelectOption
+    SelectOption,
+    Visibility
 )
 
 __all__ = [
     "LogMessage",
     "MainInput",
-    "StringInput", 
+    "StringInput",
     "StringArrayInput",
     "StringArrayInputInline",
     "NumberInput",
@@ -25,5 +26,6 @@ __all__ = [
     "SwitchInput",
     "FileInput",
     "FilesInput",
-    "SelectOption"
+    "SelectOption",
+    "Visibility"
 ]

agent_lab_sdk/schema/input_types.py

@@ -1,35 +1,54 @@
 from typing import List, Optional, Any
+from enum import Enum
 from pydantic import BaseModel
 from pydantic.json_schema import WithJsonSchema
 
 
-def MainInput(placeholder: str | None = None) -> type:
+class Visibility(str, Enum):
+    """
+    Enum representing visibility states for input fields.
+
+    Attributes:
+        ALWAYS: Field is always visible
+        START: Field is visible only at start
+        AFTER_START: Field is visible after start
+    """
+    ALWAYS = "always"
+    START = "start"
+    AFTER_START = "after_start"
+
+
+def MainInput(placeholder: str | None = None, visibility: str | Visibility = Visibility.ALWAYS) -> type:
     """
     Factory function for creating a main input type with optional placeholder.
-    
+
     Args:
         placeholder: Optional placeholder text for the input field
-        
+        visibility: Visibility state of the field (default: "always")
+
     Returns:
         Type annotation for main input field
     """
     return WithJsonSchema({
         "type": "main-input",
         "placeholder": placeholder,
+        "visibility": visibility.value if isinstance(visibility, Visibility) else visibility,
     })
 
 
-def StringInput(default: str | None = None, title: str | None = None, description: str | None = None, hidden: bool | None = False, depends: str | None = None) -> type:
+def StringInput(default: str | None = None, title: str | None = None, description: str | None = None,
+                hidden: bool | None = False, depends: str | None = None, visibility: str | Visibility = Visibility.ALWAYS) -> type:
     """
     Factory function for creating a string input type.
-    
+
     Args:
         default: Default value for the string input
         title: Title for the string input
         description: Description text for the string input
         hidden: Whether the input should be hidden in the UI
         depends: Specifies the parameter that this value depends on or is derived from.
-        
+        visibility: Visibility state of the field (default: "always")
+
     Returns:
         Type annotation for string input field
     """
@@ -40,13 +59,15 @@ def StringInput(default: str | None = None, title: str | None = None, descriptio
         "description": description,
         "hidden": hidden,
         "depends": depends,
+        "visibility": visibility.value if isinstance(visibility, Visibility) else visibility,
     })
 
 
-def StringArrayInput(placeholder: str | None = None, title: str | None = None, description: str | None = None, group: str | None = None, hidden: bool | None = False, depends: str | None = None) -> type:
+def StringArrayInput(placeholder: str | None = None, title: str | None = None, description: str | None = None,
+                     group: str | None = None, hidden: bool | None = False, depends: str | None = None, visibility: str | Visibility = Visibility.ALWAYS) -> type:
     """
     Factory function for creating a string array input type.
-    
+
     Args:
         placeholder: Placeholder text for the input field
         title: Title for the string array input
@@ -54,7 +75,8 @@ def StringArrayInput(placeholder: str | None = None, title: str | None = None, d
         group: Group name for organizing inputs in the UI
         hidden: Whether the input should be hidden in the UI
         depends: Specifies the parameter that this value depends on or is derived from.
-        
+        visibility: Visibility state of the field (default: "always")
+
     Returns:
         Type annotation for string array input field
     """
@@ -66,11 +88,12 @@ def StringArrayInput(placeholder: str | None = None, title: str | None = None, d
         "group": group,
         "hidden": hidden,
         "depends": depends,
+        "visibility": visibility.value if isinstance(visibility, Visibility) else visibility,
     })
 
 
 def StringArrayInputInline(placeholder: str | None = None, title: str | None = None, description: str | None = None,
-                     group: str | None = None, hidden: bool | None = False, depends: str | None = None) -> type:
+                     group: str | None = None, hidden: bool | None = False, depends: str | None = None, visibility: str | Visibility = Visibility.ALWAYS) -> type:
     """
     Factory function for creating a string array input inline type.
 
@@ -81,6 +104,7 @@ def StringArrayInputInline(placeholder: str | None = None, title: str | None = N
         group: Group name for organizing inputs in the UI
         hidden: Whether the input should be hidden in the UI
         depends: Specifies the parameter that this value depends on or is derived from.
+        visibility: Visibility state of the field (default: "always")
 
     Returns:
         Type annotation for string array input field
@@ -93,20 +117,23 @@ def StringArrayInputInline(placeholder: str | None = None, title: str | None = N
         "group": group,
         "hidden": hidden,
         "depends": depends,
+        "visibility": visibility.value if isinstance(visibility, Visibility) else visibility,
     })
 
 
-def NumberInput(default: float | None = None, title: str | None = None, description: str | None = None, hidden: bool | None = False, depends: str | None = None) -> type:
+def NumberInput(default: float | None = None, title: str | None = None, description: str | None = None,
+                hidden: bool | None = False, depends: str | None = None, visibility: str | Visibility = Visibility.ALWAYS) -> type:
     """
     Factory function for creating a number input type.
-    
+
     Args:
         default: Default value for the number input
         title: Title for the number input
         description: Description text for the number input
         hidden: Whether the input should be hidden in the UI
         depends: Specifies the parameter that this value depends on or is derived from.
-        
+        visibility: Visibility state of the field (default: "always")
+
     Returns:
         Type annotation for number input field
     """
@@ -117,6 +144,7 @@ def NumberInput(default: float | None = None, title: str | None = None, descript
         "description": description,
         "hidden": hidden,
         "depends": depends,
+        "visibility": visibility.value if isinstance(visibility, Visibility) else visibility,
     })
 
 
@@ -134,10 +162,11 @@ class SelectOption(BaseModel):
     description: Optional[str] = None
 
 
-def SelectInput(items: List[Any] = [], title: str | None = None, group: str | None = None, default: str | None = None, hidden: bool | None = False, depends: str | None = None) -> type:
+def SelectInput(items: List[Any] = [], title: str | None = None, group: str | None = None, default: str | None = None,
+                hidden: bool | None = False, depends: str | None = None, visibility: str | Visibility = Visibility.ALWAYS) -> type:
     """
     Factory function for creating a select input type.
-    
+
     Args:
         items: List of SelectOption objects or dictionaries
         title: Title for the select input
@@ -145,7 +174,8 @@ def SelectInput(items: List[Any] = [], title: str | None = None, group: str | No
         default: Default selected value
         hidden: Whether the input should be hidden in the UI
         depends: Specifies the parameter that this value depends on or is derived from.
-        
+        visibility: Visibility state of the field (default: "always")
+
     Returns:
         Type annotation for select input field
     """
@@ -157,13 +187,15 @@ def SelectInput(items: List[Any] = [], title: str | None = None, group: str | No
         "default": default,
         "hidden": hidden,
         "depends": depends,
+        "visibility": visibility.value if isinstance(visibility, Visibility) else visibility,
     })
 
 
-def CheckboxInput(title: str | None = None, group: str | None = None, description: str | None = None, default: bool | None = False, hidden: bool | None = False, depends: str | None = None) -> type:
+def CheckboxInput(title: str | None = None, group: str | None = None, description: str | None = None,
+                  default: bool | None = False, hidden: bool | None = False, depends: str | None = None, visibility: str | Visibility = Visibility.ALWAYS) -> type:
     """
     Factory function for creating a checkbox input type.
-    
+
     Args:
         title: Title for the checkbox
         group: Group name for organizing inputs in the UI
@@ -171,7 +203,8 @@ def CheckboxInput(title: str | None = None, group: str | None = None, descriptio
         default: Default checked state
         hidden: Whether the input should be hidden in the UI
         depends: Specifies the parameter that this value depends on or is derived from.
-        
+        visibility: Visibility state of the field (default: "always")
+
     Returns:
         Type annotation for checkbox input field
     """
@@ -183,11 +216,12 @@ def CheckboxInput(title: str | None = None, group: str | None = None, descriptio
         "default": default,
         "hidden": hidden,
         "depends": depends,
+        "visibility": visibility.value if isinstance(visibility, Visibility) else visibility,
     })
 
 
 def SwitchInput(title: str | None = None, group: str | None = None, description: str | None = None,
-                  default: bool | None = False, hidden: bool | None = False, depends: str | None = None) -> type:
+                  default: bool | None = False, hidden: bool | None = False, depends: str | None = None, visibility: str | Visibility = Visibility.ALWAYS) -> type:
     """
     Factory function for creating a switch input type.
 
@@ -198,6 +232,7 @@ def SwitchInput(title: str | None = None, group: str | None = None, description:
         default: Default checked state
         hidden: Whether the input should be hidden in the UI
         depends: Specifies the parameter that this value depends on or is derived from.
+        visibility: Visibility state of the field (default: "always")
 
     Returns:
         Type annotation for switch input field
@@ -210,20 +245,23 @@ def SwitchInput(title: str | None = None, group: str | None = None, description:
         "default": default,
         "hidden": hidden,
         "depends": depends,
+        "visibility": visibility.value if isinstance(visibility, Visibility) else visibility,
     })
 
 
-def FileInput(title: str | None = None, file_extensions: str | None = None, group: str | None = None, hidden: bool | None = False, depends: str | None = None) -> type:
+def FileInput(title: str | None = None, file_extensions: str | None = None, group: str | None = None,
+              hidden: bool | None = False, depends: str | None = None, visibility: str | Visibility = Visibility.ALWAYS) -> type:
     """
     Factory function for creating a single file input type.
-    
+
     Args:
         title: Title for the file input
         file_extensions: Comma-separated list of allowed file extensions (e.g., ".pdf,.txt")
         group: Group name for organizing inputs in the UI
         hidden: Whether the input should be hidden in the UI
         depends: Specifies the parameter that this value depends on or is derived from.
-        
+        visibility: Visibility state of the field (default: "always")
+
     Returns:
         Type annotation for file input field
     """
@@ -234,14 +272,15 @@ def FileInput(title: str | None = None, file_extensions: str | None = None, grou
         "group": group,
         "hidden": hidden,
         "depends": depends,
+        "visibility": visibility.value if isinstance(visibility, Visibility) else visibility,
     })
 
 
 def FilesInput(title: str | None = None, file_extensions: str | None = None, group: str | None = None,
-               hidden: bool | None = False, depends: str | None = None, limit: int | None = 10) -> type:
+               hidden: bool | None = False, depends: str | None = None, limit: int | None = 10, visibility: str | Visibility = Visibility.ALWAYS) -> type:
     """
     Factory function for creating a multiple files input type.
-    
+
     Args:
         title: Title for the files input
         file_extensions: Comma-separated list of allowed file extensions (e.g., ".pdf,.txt")
@@ -249,6 +288,7 @@ def FilesInput(title: str | None = None, file_extensions: str | None = None, gro
         hidden: Whether the input should be hidden in the UI
         depends: Specifies the parameter that this value depends on or is derived from.
         limit: Limit count files.
+        visibility: Visibility state of the field (default: "always")
 
     Returns:
         Type annotation for files input field
@@ -261,4 +301,5 @@ def FilesInput(title: str | None = None, file_extensions: str | None = None, gro
         "hidden": hidden,
         "depends": depends,
         "limit": limit,
+        "visibility": visibility.value if isinstance(visibility, Visibility) else visibility,
     })

{agent_lab_sdk-0.1.44.dist-info → agent_lab_sdk-0.1.45.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-lab-sdk
-Version: 0.1.44
+Version: 0.1.45
 Summary: SDK для работы с Agent Lab
 Author-email: Andrew Ohurtsov <andermirik@yandex.com>
 License: Proprietary and Confidential — All Rights Reserved
@@ -262,7 +262,7 @@ from typing import List, Annotated
 from pydantic import BaseModel, Field
 from agent_lab_sdk.schema import (
     MainInput, StringInput, StringArrayInput, NumberInput,
-    SelectInput, CheckboxInput, FileInput, FilesInput, SelectOption
+    SelectInput, CheckboxInput, FileInput, FilesInput, SelectOption, Visibility
 )
 
 class AgentState(BaseModel):
@@ -273,7 +273,8 @@ class AgentState(BaseModel):
     title: Annotated[str, StringInput(
         default="Без названия",
         title="Заголовок",
-        description="Название для вашего запроса"
+        description="Название для вашего запроса",
+        visibility=Visibility.ALWAYS  # или visibility="always"
     )]
     
     # Массив строк
@@ -327,18 +328,18 @@ class AgentState(BaseModel):
 
 #### Доступные фабричные функции
 
-| Тип                      | Описание                          | Основные параметры                                                  |
-|--------------------------|-----------------------------------|---------------------------------------------------------------------|
-| `MainInput`              | Основное поле ввода               | `placeholder`                                                       |
-| `StringInput`            | Текстовое поле                    | `default`, `title`, `description`, `hidden`, `depends`              |
-| `StringArrayInput`       | Массив строк                      | `placeholder`, `title`, `description`, `group`, `hidden`, `depends` |
-| `StringArrayInputInline` | Массив строк в одной строке ввода | `placeholder`, `title`, `description`, `group`, `hidden`, `depends` |
-| `NumberInput`            | Числовое поле                     | `default`, `title`, `description`, `hidden`, `depends`              |
-| `SelectInput`            | Выпадающий список                 | `items`, `title`, `group`, `default`, `hidden`, `depends`           |
-| `CheckboxInput`          | Чекбокс                           | `title`, `group`, `description`, `default`, `hidden`, `depends`     |
-| `SwitchInput`            | Switch                            | `title`, `group`, `description`, `default`, `hidden`, `depends`     |
-| `FileInput`              | Загрузка одного файла             | `title`, `file_extensions`, `group`, `hidden`, `depends`            |
-| `FilesInput`             | Загрузка нескольких файлов        | `title`, `file_extensions`, `group`, `hidden`, `depends`, `limit`   |
+| Тип                      | Описание                          | Основные параметры                                                                |
+|--------------------------|-----------------------------------|-----------------------------------------------------------------------------------|
+| `MainInput`              | Основное поле ввода               | `placeholder`, `visibility`                                                       |
+| `StringInput`            | Текстовое поле                    | `default`, `title`, `description`, `hidden`, `depends`, `visibility`              |
+| `StringArrayInput`       | Массив строк                      | `placeholder`, `title`, `description`, `group`, `hidden`, `depends`, `visibility` |
+| `StringArrayInputInline` | Массив строк в одной строке ввода | `placeholder`, `title`, `description`, `group`, `hidden`, `depends`, `visibility` |
+| `NumberInput`            | Числовое поле                     | `default`, `title`, `description`, `hidden`, `depends`, `visibility`              |
+| `SelectInput`            | Выпадающий список                 | `items`, `title`, `group`, `default`, `hidden`, `depends`, `visibility`           |
+| `CheckboxInput`          | Чекбокс                           | `title`, `group`, `description`, `default`, `hidden`, `depends`, `visibility`     |
+| `SwitchInput`            | Switch                            | `title`, `group`, `description`, `default`, `hidden`, `depends`, `visibility`     |
+| `FileInput`              | Загрузка одного файла             | `title`, `file_extensions`, `group`, `hidden`, `depends`, `visibility`            |
+| `FilesInput`             | Загрузка нескольких файлов        | `title`, `file_extensions`, `group`, `hidden`, `depends`, `limit`, `visibility`   |
 
 #### Группировка полей
 
@@ -371,6 +372,54 @@ class TaskConfig(BaseModel):
     )]
 ```
 
+#### Управление видимостью полей
+
+Параметр `visibility` контролирует, когда поле отображается в интерфейсе. Доступные значения:
+
+```python
+from agent_lab_sdk.schema import Visibility
+
+# Enum с тремя значениями:
+Visibility.ALWAYS       # "always" - поле всегда доступно для ввода (по умолчанию)
+Visibility.START        # "start" - поле доступно для ввода только при старте
+Visibility.AFTER_START  # "after_start" - поле доступно для ввода после старта
+```
+
+**Пример использования:**
+
+```python
+class AgentConfig(BaseModel):
+    # Всегда доступно для ввода поле
+    query: Annotated[str, MainInput(
+        placeholder="Введите запрос",
+        visibility=Visibility.ALWAYS
+    )]
+
+    # Поле доступно для ввода только при первом запуске
+    api_key: Annotated[str, StringInput(
+        title="API ключ",
+        description="Ключ для доступа к внешнему API",
+        visibility=Visibility.START
+    )]
+
+    # Поле появляется после первого сообщения
+    session_id: Annotated[str, StringInput(
+        title="ID сессии",
+        description="Идентификатор текущей сессии",
+        visibility=Visibility.AFTER_START,
+        hidden=True
+    )]
+```
+
+Можно также передавать строковые значения напрямую:
+
+```python
+title: Annotated[str, StringInput(
+    title="Заголовок",
+    visibility="always"  # эквивалентно Visibility.ALWAYS
+)]
+```
+
 ### 5.2. LogMessage
 
 `LogMessage` — вспомогательное сообщение для потоковой передачи логов из узлов LangGraph / LangChain. Экземпляры создаются как обычные сообщения чата, но получают тип `log`, поэтому фронтенд может отображать их отдельно от ответов модели.

{agent_lab_sdk-0.1.44.dist-info → agent_lab_sdk-0.1.45.dist-info}/RECORD

@@ -9,14 +9,14 @@ agent_lab_sdk/llm/llm.py,sha256=xL6FYyzjx22w_HnGK0ygGJTGr_OEz9EB6JWyYztprg4,2018
 agent_lab_sdk/llm/throttled.py,sha256=faccDXiKkmFFfZkVrxFQDHxHutAAivsKDEpYlygfuqU,8019
 agent_lab_sdk/metrics/__init__.py,sha256=G4VSlzKwupPMM4c6vZaF1rnd0KusKarezDMjli9pVFw,57
 agent_lab_sdk/metrics/metrics.py,sha256=_XTT9vMG7T0u_D2pL371wm8GoBU5fodJ45D2RACnBJw,3439
-agent_lab_sdk/schema/__init__.py,sha256=bHSyXQYkcB9fWBlziWodXR_IzC5nKrdKzrCpyVWNY9o,521
-agent_lab_sdk/schema/input_types.py,sha256=e75nRW7Dz_RHk5Yia8DkFfbqMafsLQsQrJPfzQhpIBw,9123
+agent_lab_sdk/schema/__init__.py,sha256=cDVmQG5eYd2qO7DtDTt_YCISQyUXjLprUZ6KvKESKtU,554
+agent_lab_sdk/schema/input_types.py,sha256=ziazdlwmOXQv7d8syaT5D1wulgHEpxaQAThGxwHwU2k,11630
 agent_lab_sdk/schema/log_message.py,sha256=nadi6lZGRuDSPmfbYs9QPpRJUT9Pfy8Y7pGCvyFF5Mw,638
 agent_lab_sdk/storage/__init__.py,sha256=HAtUoqg3k0irqPMewayadVA9aXJOmYSxRr6a5J1scT0,174
 agent_lab_sdk/storage/storage.py,sha256=ELpt7GRwFD-aWa6ctinfA_QwcvzWLvKS0Wz8FlxVqAs,2075
 agent_lab_sdk/storage/storage_v2.py,sha256=ONseynX59xzWK17dfzxZvnii2rpz3Oo2Zo9Ck-lcGnw,1997
-agent_lab_sdk-0.1.44.dist-info/licenses/LICENSE,sha256=_TRXHkF3S9ilWBPdZcHLI_S-PRjK0L_SeOb2pcPAdV4,417
-agent_lab_sdk-0.1.44.dist-info/METADATA,sha256=gbWuS14rM1KtS4W8f6Jze8orWGmjzcWgqJpdnfE7NzY,20766
-agent_lab_sdk-0.1.44.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agent_lab_sdk-0.1.44.dist-info/top_level.txt,sha256=E1efqkJ89KNmPBWdLzdMHeVtH0dYyCo4fhnSb81_15I,14
-agent_lab_sdk-0.1.44.dist-info/RECORD,,
+agent_lab_sdk-0.1.45.dist-info/licenses/LICENSE,sha256=_TRXHkF3S9ilWBPdZcHLI_S-PRjK0L_SeOb2pcPAdV4,417
+agent_lab_sdk-0.1.45.dist-info/METADATA,sha256=yG9kDYbpWPtiLfSHPhoPZYHL2Bx1_l8M2RNW3f_cS_0,22820
+agent_lab_sdk-0.1.45.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agent_lab_sdk-0.1.45.dist-info/top_level.txt,sha256=E1efqkJ89KNmPBWdLzdMHeVtH0dYyCo4fhnSb81_15I,14
+agent_lab_sdk-0.1.45.dist-info/RECORD,,