erdo 0.1.4__py3-none-any.whl → 0.1.5__py3-none-any.whl

erdo/_generated/types.py

@@ -71,14 +71,20 @@ class ExecutionModeType(str, Enum):
 class LlmModel(str, Enum):
     """Enum for LlmModel values"""
 
-    CLAUDE_3_7_SONNET_20250219 = "claude-3-7-sonnet-20250219"
-    CLAUDE_SONNET_4_20250514 = "claude-sonnet-4-20250514"
-    CLAUDE_3_5_HAIKU_20241022 = "claude-3-5-haiku-20241022"
+    CLAUDE_SONNET_4 = "claude-sonnet-4"
+    CLAUDE_SONNET_4_5 = "claude-sonnet-4-5"
+    CLAUDE_HAIKU_4_5 = "claude-haiku-4-5"
     GPT_4O = "gpt-4o"
     GPT_4O_MINI = "gpt-4o-mini"
     GPT_4_1 = "gpt-4.1"
     GPT_4_1_MINI = "gpt-4.1-mini"
     GPT_4_1_NANO = "gpt-4.1-nano"
+    GPT_5 = "gpt-5"
+    GPT_5_MINI = "gpt-5-mini"
+    GPT_5_NANO = "gpt-5-nano"
+    GEMINI_2_5_PRO = "gemini-2.5-pro"
+    GEMINI_2_5_FLASH = "gemini-2.5-flash"
+    GEMINI_2_5_FLASH_LITE = "gemini-2.5-flash-lite"
 
 
 # OutputBehaviorType (from OutputBehaviorType in Go)
@@ -145,6 +151,7 @@ class InvocationEventType(str, Enum):
     STATUS = "status"
     RESULT = "result"
     DATASET_CREATED = "dataset created"
+    LOG = "log"
 
 
 # Status (from Status in Go)
@@ -168,6 +175,7 @@ class Error(str, Enum):
     INFO_NEEDED = "info needed"
     TERMINATED = "terminated"
     BAD_REQUEST = "bad request"
+    TIMEOUT = "timeout"
 
 
 # DatasetType (from DatasetType in Go)
@@ -280,6 +288,7 @@ class Bot:
     organization_id: str
     visibility: str
     source: str
+    key: Optional[str] = None
     persona: Optional[str] = None
     running_message: Optional[str] = None
     finished_message: Optional[str] = None
@@ -294,6 +303,8 @@ class Bot:
         result["description"] = self.description
         result["code"] = self.code
         result["file_path"] = self.file_path
+        if self.key is not None:
+            result["key"] = self.key
         if self.persona is not None:
             result["persona"] = self.persona
         if self.running_message is not None:
@@ -355,7 +366,7 @@ class ParameterValueSource:
     """ParameterValueSource matching Go backend structure."""
 
     type: Any
-    parameters: Any
+    parameters: Dict[str, Any]
     id: Optional[UUID] = None
     parameter_definition_id: Optional[UUID] = None
     created_at: Optional[datetime] = None
@@ -385,7 +396,7 @@ class ParameterValueSourceHandler:
     """ParameterValueSourceHandler matching Go backend structure."""
 
     action_type: str
-    parameters: Any
+    parameters: Dict[str, Any]
     execution_mode: str
     id: Optional[UUID] = None
     parameter_value_source_id: Optional[UUID] = None
@@ -414,7 +425,7 @@ class ParameterInterpreter:
     """ParameterInterpreter matching Go backend structure."""
 
     action_type: str
-    parameters: Any
+    parameters: Dict[str, Any]
     interpreter_order: int
     id: Optional[UUID] = None
     parameter_definition_id: Optional[UUID] = None
@@ -463,7 +474,7 @@ class AgentDiscovery:
 class ExecutionModeConfig:
     """ExecutionMode matching Go backend structure."""
 
-    mode: str
+    mode: Any
     data: Any
     if_condition: Optional["ConditionDefinition"] = None
 
@@ -564,11 +575,11 @@ class ActionDefinition:
 class ResultSchema:
     """ResultSchema matching Go backend structure."""
 
-    properties: Any
+    properties: Dict[str, Any]
     description: str = None
     required_fields: List[str] = None
     optional_fields: List[str] = None
-    examples: List[Any] = None
+    examples: List[Dict[str, Any]] = None
 
     def to_dict(self) -> Dict[str, Any]:
         """Convert to dict format expected by backend."""
@@ -592,7 +603,7 @@ class PropertySchema:
     type: Any
     description: str = None
     items: Optional["PropertySchema"] = None
-    properties: Any = None
+    properties: Dict[str, Any] = None
     example: Any = None
     enum: List[str] = None
 
@@ -622,7 +633,7 @@ class IntegrationSchema:
     description: str
     type: str
     auth_types: List[str]
-    credential_schema: Any
+    credential_schema: Dict[str, Any]
     available_scopes: List[str]
     documentation_url: str
     codegen_details: Optional[CodegenDetails] = None
@@ -711,8 +722,8 @@ class CredentialSchema:
 class ExportActionsResponse:
     """ExportActionsResponse matching Go backend structure."""
 
-    services: Any
-    integrations: Any
+    services: Dict[str, Any]
+    integrations: Dict[str, Any]
 
     def to_dict(self) -> Dict[str, Any]:
         """Convert to dict format expected by backend."""
@@ -728,7 +739,7 @@ class ConditionDefinition:
 
     type: str
     conditions: List["ConditionDefinition"] = None
-    leaf: Any = None
+    leaf: Dict[str, Any] = None
 
     def to_dict(self) -> Dict[str, Any]:
         """Convert to dict format expected by backend."""
@@ -747,7 +758,7 @@ class TempCondition:
 
     type: str
     conditions: Any = None
-    leaf: Any = None
+    leaf: Dict[str, Any] = None
 
     def to_dict(self) -> Dict[str, Any]:
         """Convert to dict format expected by backend."""
@@ -829,7 +840,7 @@ class JSONSchema:
     """JSONSchema matching Go backend structure."""
 
     type: Any
-    properties: Any = None
+    properties: Dict[str, Any] = None
     required: List[str] = None
     items: Optional["JSONSchemaProperty"] = None
 
@@ -854,11 +865,13 @@ class Tool:
     description: str
     input_schema: "JSONSchema"
     action_type: str
-    parameters: Any
+    parameters: Dict[str, Any]
     bot_output_visibility: str = None
     history_content_type: str = None
     ui_content_type: str = None
     as_root: bool = None
+    running_message: Optional[str] = None
+    finished_message: Optional[str] = None
 
     def to_dict(self) -> Dict[str, Any]:
         """Convert to dict format expected by backend."""
@@ -876,6 +889,10 @@ class Tool:
             result["ui_content_type"] = self.ui_content_type
         if self.as_root is not None:
             result["as_root"] = self.as_root
+        if self.running_message is not None:
+            result["running_message"] = self.running_message
+        if self.finished_message is not None:
+            result["finished_message"] = self.finished_message
         return result
 
 
@@ -929,8 +946,8 @@ class Result:
     """Result matching Go backend structure."""
 
     status: Any
-    parameters: Optional[Any] = None
-    output: Optional[Any] = None
+    parameters: Optional[Dict[str, Any]] = None
+    output: Optional[Dict[str, Any]] = None
     message: Optional[str] = None
     error: Optional[Any] = None
 
@@ -964,13 +981,13 @@ class APIStep:
     execution_mode: "APIExecutionMode"
     output_behaviour: Any
     output_channels: List[str]
-    parameter_hydration_behaviour: Any
     depends_on: Optional[List[str]] = None
     key: Optional[str] = None
     running_message: Optional[str] = None
     finished_message: Optional[str] = None
     history_content_type: Optional[str] = None
     ui_content_type: Optional[str] = None
+    parameter_hydration_behaviour: Optional[Any] = None
     result_handler_id: Optional[str] = None
     created_at: Optional[datetime] = None
     updated_at: Optional[datetime] = None
@@ -1001,7 +1018,8 @@ class APIStep:
             result["history_content_type"] = self.history_content_type
         if self.ui_content_type is not None:
             result["ui_content_type"] = self.ui_content_type
-        result["parameter_hydration_behaviour"] = self.parameter_hydration_behaviour
+        if self.parameter_hydration_behaviour is not None:
+            result["parameter_hydration_behaviour"] = self.parameter_hydration_behaviour
         if self.result_handler_id is not None:
             result["result_handler_id"] = self.result_handler_id
         if self.created_at is not None:
@@ -1103,7 +1121,7 @@ class APIParameterInterpreter:
 class APIExecutionMode:
     """APIExecutionMode matching Go backend structure."""
 
-    mode: str
+    mode: Any
     data: Any
     if_condition: Optional["APIConditionDefinition"] = None
 
@@ -1149,6 +1167,8 @@ class APITool:
     history_content_type: str = None
     ui_content_type: str = None
     as_root: bool = None
+    running_message: Optional[str] = None
+    finished_message: Optional[str] = None
 
     def to_dict(self) -> Dict[str, Any]:
         """Convert to dict format expected by backend."""
@@ -1166,6 +1186,10 @@ class APITool:
             result["ui_content_type"] = self.ui_content_type
         if self.as_root is not None:
             result["as_root"] = self.as_root
+        if self.running_message is not None:
+            result["running_message"] = self.running_message
+        if self.finished_message is not None:
+            result["finished_message"] = self.finished_message
         return result
 
 
@@ -1264,12 +1288,12 @@ class Dataset:
     integration_id: Optional[UUID] = None
     integration_config_id: Optional[UUID] = None
     integration_config: Optional[IntegrationConfig] = None
-    encrypted_integration_credentials: Optional[Any] = None
-    credential_schema: Optional[Any] = None
+    encrypted_integration_credentials: Optional[Dict[str, str]] = None
+    credential_schema: Optional[Dict[str, Any]] = None
     codegen_details: Optional[CodegenDetails] = None
     analysis_details: Optional[AnalysisDetails] = None
     available_scopes: Optional[List[str]] = None
-    parameters: Optional[Any] = None
+    parameters: Optional[Dict[str, Any]] = None
 
     def to_dict(self) -> Dict[str, Any]:
         """Convert to dict format expected by backend."""
@@ -1325,7 +1349,7 @@ class BotResource:
 
     id: int
     created_by: str
-    extra: Any
+    extra: Dict[str, Any]
     dataset: Optional[Dataset] = None
 
     def to_dict(self) -> Dict[str, Any]:
@@ -1557,7 +1581,7 @@ class SegmentOption:
     id: str
     name: str
     type: str
-    metadata: Any
+    metadata: Dict[str, str]
     children: List["SegmentOption"]
     parent_id: Optional[str] = None
 
@@ -1598,14 +1622,14 @@ class IntegrationDefinition:
     scope_separator: Optional[str] = None
     api_version: Optional[str] = None
     api_version_description: Optional[str] = None
-    credential_schema: Any = None
+    credential_schema: Dict[str, Any] = None
     ui_config: Optional["UIConfig"] = None
     codegen_details: Optional[CodegenDetails] = None
     analysis_details: Optional[AnalysisDetails] = None
     segment_config: Optional["SegmentConfig"] = None
     dataset_resource_discovery_config: Optional["DatasetResourceDiscoveryConfig"] = None
     expiry_config: Optional["ExpiryConfig"] = None
-    provider_credentials: Any = None
+    provider_credentials: Dict[str, Any] = None
     default_resource_attach_type: Optional[str] = None
 
     def to_dict(self) -> Dict[str, Any]:
@@ -1692,7 +1716,7 @@ class PythonIntegrationInstance:
     """PythonIntegrationInstance matching Go backend structure."""
 
     config: IntegrationConfig
-    credentials: Any
+    credentials: Dict[str, Any]
     methods: List[str] = None
 
     def to_dict(self) -> Dict[str, Any]:
@@ -1772,7 +1796,7 @@ class ListIntegrationsResponse:
 class ExportIntegrationsResponse:
     """ExportIntegrationsResponse matching Go backend structure."""
 
-    integrations: Any
+    integrations: Dict[str, Any]
 
     def to_dict(self) -> Dict[str, Any]:
         """Convert to dict format expected by backend."""
@@ -1788,20 +1812,20 @@ class Memory:
     id: str
     content: str
     description: str
+    tags: List[str]
     state: str
     created_by_entity_type: str
     created_by_id: str
+    extra: Dict[str, Any]
     created_at: datetime
     updated_at: datetime
     type: str
     current_version: int
     approval_status: str
-    tags: List[str] = None
     estimated_stale_at: Optional[datetime] = None
     stale_when_text: Optional[str] = None
     created_from: Optional[str] = None
     organization_id: Optional[str] = None
-    extra: Any = None
     dataset_id: Optional[str] = None
     integration_config_id: Optional[str] = None
     user_id: Optional[str] = None
@@ -1813,8 +1837,7 @@ class Memory:
         result["id"] = self.id
         result["content"] = self.content
         result["description"] = self.description
-        if self.tags is not None:
-            result["tags"] = self.tags
+        result["tags"] = self.tags
         result["state"] = self.state
         if self.estimated_stale_at is not None:
             result["estimated_stale_at"] = self.estimated_stale_at
@@ -1826,8 +1849,7 @@ class Memory:
             result["created_from"] = self.created_from
         if self.organization_id is not None:
             result["organization_id"] = self.organization_id
-        if self.extra is not None:
-            result["extra"] = self.extra
+        result["extra"] = self.extra
         result["created_at"] = self.created_at
         result["updated_at"] = self.updated_at
         result["type"] = self.type
@@ -1844,6 +1866,25 @@ class Memory:
         return result
 
 
+@dataclass
+class ResourceAnalysis:
+    """ResourceAnalysis matching Go backend structure."""
+
+    entity_key: str
+    summary: Optional[str] = None
+    last_analyzed: Optional[datetime] = None
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Convert to dict format expected by backend."""
+        result = {}
+        if self.summary is not None:
+            result["summary"] = self.summary
+        if self.last_analyzed is not None:
+            result["last_analyzed"] = self.last_analyzed
+        result["entity_key"] = self.entity_key
+        return result
+
+
 @dataclass
 class Resource:
     """Resource matching Go backend structure."""
@@ -1852,15 +1893,17 @@ class Resource:
     key: str
     name: str
     type: Any
+    use_cases: List[str]
+    tags: List[str]
     created_at: datetime
     updated_at: datetime
     integration_config_id: str
     state: Any
     attach_type: Any
     dataset_id: str
+    analyses: List["ResourceAnalysis"]
+    related_resources: List[Resource]
     description: Optional[str] = None
-    use_cases: List[str] = None
-    tags: List[str] = None
     instructions: Optional[str] = None
 
     def to_dict(self) -> Dict[str, Any]:
@@ -1872,10 +1915,8 @@ class Resource:
         result["type"] = self.type
         if self.description is not None:
             result["description"] = self.description
-        if self.use_cases is not None:
-            result["use_cases"] = self.use_cases
-        if self.tags is not None:
-            result["tags"] = self.tags
+        result["use_cases"] = self.use_cases
+        result["tags"] = self.tags
         result["created_at"] = self.created_at
         result["updated_at"] = self.updated_at
         result["integration_config_id"] = self.integration_config_id
@@ -1884,6 +1925,8 @@ class Resource:
         result["dataset_id"] = self.dataset_id
         if self.instructions is not None:
             result["instructions"] = self.instructions
+        result["analyses"] = self.analyses
+        result["related_resources"] = self.related_resources
         return result
 
 
@@ -2106,7 +2149,7 @@ class SegmentOption:
     id: str
     name: str
     type: str
-    metadata: Any
+    metadata: Dict[str, str]
     children: List["SegmentOption"]
     parent_id: Optional[str] = None
 
@@ -2147,14 +2190,14 @@ class IntegrationDefinition:
     scope_separator: Optional[str] = None
     api_version: Optional[str] = None
     api_version_description: Optional[str] = None
-    credential_schema: Any = None
+    credential_schema: Dict[str, Any] = None
     ui_config: Optional["UIConfig"] = None
     codegen_details: Optional[CodegenDetails] = None
     analysis_details: Optional[AnalysisDetails] = None
     segment_config: Optional["SegmentConfig"] = None
     dataset_resource_discovery_config: Optional["DatasetResourceDiscoveryConfig"] = None
     expiry_config: Optional["ExpiryConfig"] = None
-    provider_credentials: Any = None
+    provider_credentials: Dict[str, Any] = None
     default_resource_attach_type: Optional[str] = None
 
     def to_dict(self) -> Dict[str, Any]:
@@ -2241,7 +2284,7 @@ class PythonIntegrationInstance:
     """PythonIntegrationInstance matching Go backend structure."""
 
     config: IntegrationConfig
-    credentials: Any
+    credentials: Dict[str, Any]
     methods: List[str] = None
 
     def to_dict(self) -> Dict[str, Any]:
@@ -2321,7 +2364,7 @@ class ListIntegrationsResponse:
 class ExportIntegrationsResponse:
     """ExportIntegrationsResponse matching Go backend structure."""
 
-    integrations: Any
+    integrations: Dict[str, Any]
 
     def to_dict(self) -> Dict[str, Any]:
         """Convert to dict format expected by backend."""
@@ -2415,6 +2458,7 @@ __all__ = [
     "ListIntegrationsResponse",
     "ExportIntegrationsResponse",
     "Memory",
+    "ResourceAnalysis",
     "Resource",
 ]
 

erdo/actions/__init__.py

@@ -14,27 +14,26 @@ from .._generated.actions import codeexec  # noqa: F401
 from .._generated.actions import llm  # noqa: F401
 from .._generated.actions import memory  # noqa: F401
 from .._generated.actions import resource_definitions  # noqa: F401
+from .._generated.actions import sqlexec  # noqa: F401
 from .._generated.actions import utils  # noqa: F401
 from .._generated.actions import webparser  # noqa: F401
 from .._generated.actions import websearch  # noqa: F401
 
-# Import internal actions explicitly
-from .._generated.internal_actions import (  # noqa: F401
-    checkpoint_attempt,
-    refresh_resource,
-)
+# Import internal actions first
+from .._generated.internal_actions import *  # noqa: F403,F401
 
 # Make all services available for import
 __all__ = [
     "checkpoint_attempt",
     "refresh_resource",
+    "resource_definitions",
+    "webparser",
+    "websearch",
     "analysis",
-    "bot",
+    "codeexec",
     "llm",
-    "resource_definitions",
+    "sqlexec",
     "utils",
-    "codeexec",
+    "bot",
     "memory",
-    "webparser",
-    "websearch",
 ]