erdo 0.1.9__tar.gz → 0.1.11__tar.gz

{erdo-0.1.9 → erdo-0.1.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: erdo
-Version: 0.1.9
+Version: 0.1.11
 Summary: Python SDK for building workflow automation agents with Erdo
 Project-URL: Homepage, https://erdo.ai
 Project-URL: Documentation, https://docs.erdo.ai
@@ -316,18 +316,46 @@ else:
 
 ### Invocation Modes
 
-- **Live Mode** (default): Runs against real backend with LLM API
-- **Replay Mode**: Uses cached responses - free after first run (perfect for testing!)
-- **Mock Mode**: Returns synthetic responses - always free
+Control how bot actions are executed for testing and development:
+
+| Mode | Description | Cost |
+|------|-------------|------|
+| **live** | Execute with real API calls | $$$ per run |
+| **replay** | Cache responses, replay on subsequent runs | $$$ first run, FREE after |
+| **manual** | Use developer-provided mock responses | FREE always |
 
 ```python
-# Replay mode - free after first run
+# Live mode (default) - real API calls
+response = invoke("my-agent", messages=[...], mode="live")
+
+# Replay mode - cache after first run (recommended for testing!)
 response = invoke("my-agent", messages=[...], mode="replay")
 
-# Mock mode - always free
-response = invoke("my-agent", messages=[...], mode="mock")
+# Replay with refresh - bypass cache, get fresh response
+response = invoke("my-agent", messages=[...], mode={"mode": "replay", "refresh": True})
+
+# Manual mode - use mock responses
+response = invoke(
+    "my-agent",
+    messages=[...],
+    mode="manual",
+    manual_mocks={
+        "llm.message": {
+            "status": "success",
+            "output": {"content": "Mocked response"}
+        }
+    }
+)
 ```
 
+**Replay Mode Refresh:**
+The `refresh` parameter forces a fresh API call while staying in replay mode:
+- First run: Executes and caches the response
+- Subsequent runs: Uses cached response (free!)
+- With refresh: Bypasses cache, gets fresh response, updates cache
+
+Perfect for updating cached responses after bot changes without switching modes.
+
 ## Testing Agents
 
 Write fast, parallel agent tests using `agent_test_*` functions:
@@ -358,8 +386,17 @@ erdo agent-test tests/test_my_agent.py
 
 # Verbose output
 erdo agent-test tests/test_my_agent.py --verbose
+
+# Refresh cached responses (bypass cache for all replay mode tests)
+erdo agent-test tests/test_my_agent.py --refresh
 ```
 
+**Refresh Flag:**
+The `--refresh` flag forces all tests using `mode="replay"` to bypass cache and get fresh responses. Perfect for:
+- Updating cached responses after bot changes
+- Verifying tests with current LLM behavior
+- No code changes needed - just add the flag!
+
 ### Test Helpers
 
 The `erdo.test` module provides assertion helpers:

{erdo-0.1.9 → erdo-0.1.11}/README.md

@@ -277,18 +277,46 @@ else:
 
 ### Invocation Modes
 
-- **Live Mode** (default): Runs against real backend with LLM API
-- **Replay Mode**: Uses cached responses - free after first run (perfect for testing!)
-- **Mock Mode**: Returns synthetic responses - always free
+Control how bot actions are executed for testing and development:
+
+| Mode | Description | Cost |
+|------|-------------|------|
+| **live** | Execute with real API calls | $$$ per run |
+| **replay** | Cache responses, replay on subsequent runs | $$$ first run, FREE after |
+| **manual** | Use developer-provided mock responses | FREE always |
 
 ```python
-# Replay mode - free after first run
+# Live mode (default) - real API calls
+response = invoke("my-agent", messages=[...], mode="live")
+
+# Replay mode - cache after first run (recommended for testing!)
 response = invoke("my-agent", messages=[...], mode="replay")
 
-# Mock mode - always free
-response = invoke("my-agent", messages=[...], mode="mock")
+# Replay with refresh - bypass cache, get fresh response
+response = invoke("my-agent", messages=[...], mode={"mode": "replay", "refresh": True})
+
+# Manual mode - use mock responses
+response = invoke(
+    "my-agent",
+    messages=[...],
+    mode="manual",
+    manual_mocks={
+        "llm.message": {
+            "status": "success",
+            "output": {"content": "Mocked response"}
+        }
+    }
+)
 ```
 
+**Replay Mode Refresh:**
+The `refresh` parameter forces a fresh API call while staying in replay mode:
+- First run: Executes and caches the response
+- Subsequent runs: Uses cached response (free!)
+- With refresh: Bypasses cache, gets fresh response, updates cache
+
+Perfect for updating cached responses after bot changes without switching modes.
+
 ## Testing Agents
 
 Write fast, parallel agent tests using `agent_test_*` functions:
@@ -319,8 +347,17 @@ erdo agent-test tests/test_my_agent.py
 
 # Verbose output
 erdo agent-test tests/test_my_agent.py --verbose
+
+# Refresh cached responses (bypass cache for all replay mode tests)
+erdo agent-test tests/test_my_agent.py --refresh
 ```
 
+**Refresh Flag:**
+The `--refresh` flag forces all tests using `mode="replay"` to bypass cache and get fresh responses. Perfect for:
+- Updating cached responses after bot changes
+- Verifying tests with current LLM behavior
+- No code changes needed - just add the flag!
+
 ### Test Helpers
 
 The `erdo.test` module provides assertion helpers:

{erdo-0.1.9 → erdo-0.1.11}/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-0.1.9 → erdo-0.1.11}/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-0.1.9 → erdo-0.1.11}/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-0.1.9 → erdo-0.1.11}/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