erdo 0.1.10__py3-none-any.whl → 0.1.11__py3-none-any.whl

erdo/__init__.py

@@ -10,8 +10,8 @@ from ._generated.condition import __all__ as condition_all
 from ._generated.types import *  # noqa: F403,F401
 from ._generated.types import __all__ as generated_all
 
-# Import invoke function for production and testing
-from .invoke import invoke  # noqa: F401
+# Import invoke functions
+from .invoke.invoke import invoke, invoke_agent, invoke_by_key  # noqa: F401
 
 # Import state object for template support
 from .state import state  # noqa: F401
@@ -28,6 +28,6 @@ __all__.extend(handwritten_all)
 __all__.extend(generated_all)
 __all__.extend(condition_all)
 
-# Add state and invoke to __all__ for explicit import
+# Add state and invoke functions to __all__ for explicit import
 __all__.append("state")
-__all__.append("invoke")
+__all__.extend(["invoke", "invoke_agent", "invoke_by_key"])

erdo/_generated/actions/bot.py

@@ -9,6 +9,8 @@ NOTE: This module is hardcoded because bot.invoke requires bot_name parameter
 but the Go struct uses bot_id. The backend converts bot_name -> bot_id internally.
 """
 
+from __future__ import annotations
+
 from typing import Any, Dict, Optional, Union
 
 from pydantic import BaseModel
@@ -175,6 +177,10 @@ def ask(
     return AskParams(**params_dict)
 
 
+# Rebuild models to resolve forward references (needed for Python 3.10+)
+InvokeParams.model_rebuild()
+AskParams.model_rebuild()
+
 # Associate parameter classes with their result types
 InvokeParams._result = InvokeResult
 AskParams._result = AskResult

erdo/_generated/actions/codeexec.py

@@ -8,7 +8,7 @@ Actual execution happens in the Go backend after syncing.
 
 from typing import Any, Optional, Union
 
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, Field, model_serializer
 
 from erdo.template import TemplateString
 
@@ -41,6 +41,37 @@ class ExecuteParams(BaseActionParams):
     )
     storage_config: Optional[Any] = None  # storage_config parameter
 
+    @model_serializer
+    def _serialize_model(self) -> dict:
+        """Custom serializer to preserve PythonFile._type marker in code_files."""
+        # Serialize code_files first
+        serialized_code_files = None
+        if self.code_files is not None and isinstance(self.code_files, list):
+            serialized_code_files = []
+            for item in self.code_files:
+                if hasattr(item, "model_dump") and callable(
+                    getattr(item, "model_dump")
+                ):
+                    # Use Pydantic's serialization which respects @model_serializer
+                    serialized_code_files.append(item.model_dump())
+                elif hasattr(item, "to_dict") and callable(getattr(item, "to_dict")):
+                    serialized_code_files.append(item.to_dict())
+                else:
+                    serialized_code_files.append(item)
+
+        # Build full serialization
+        return {
+            "name": self.name,
+            "step_metadata": self.step_metadata,
+            "entrypoint": self.entrypoint,
+            "code_files": serialized_code_files,
+            "resources": self.resources,
+            "parameters": self.parameters,
+            "encryption_key": self.encryption_key,
+            "timeout_seconds": self.timeout_seconds,
+            "storage_config": self.storage_config,
+        }
+
 
 class ParseFileAsBotResourceParams(BaseActionParams):
     """Parse a file from code execution results into a bot resource with dataset and analysis parameters"""

erdo/_generated/actions/llm.py

@@ -9,6 +9,8 @@ NOTE: This module is hardcoded to provide bot-compatible parameter names
 that match the exported bot code format.
 """
 
+from __future__ import annotations
+
 from typing import Any, Dict, List, Optional, Union
 
 from pydantic import BaseModel
@@ -139,5 +141,8 @@ def message(
     return MessageParams(**params_dict)
 
 
+# Rebuild models to resolve forward references (needed for Python 3.10+)
+MessageParams.model_rebuild()
+
 # Associate parameter classes with their result types
 MessageParams._result = MessageResult