erdo 0.1.9__tar.gz → 0.1.10__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: erdo
-Version: 0.1.9
+Version: 0.1.10
 Summary: Python SDK for building workflow automation agents with Erdo
 Project-URL: Homepage, https://erdo.ai
 Project-URL: Documentation, https://docs.erdo.ai

{erdo-0.1.9 → erdo-0.1.10}/erdo/invoke/invoke.py

@@ -404,7 +404,7 @@ def invoke(
 def invoke_agent(
     agent: Any,
     parameters: Optional[Dict[str, Any]] = None,
-    dataset_ids: Optional[list] = None,
+    dataset_slugs: Optional[list] = None,
     stream: bool = False,
     print_events: bool = False,
     **kwargs,
@@ -414,7 +414,7 @@ def invoke_agent(
     Args:
         agent: Agent instance with a 'key' attribute
         parameters: Parameters to pass to the agent
-        dataset_ids: Dataset IDs to include
+        dataset_slugs: Dataset slugs to include (e.g. ["my-dataset"] or ["org.my-dataset"])
         stream: Whether to stream events
         print_events: Whether to print events
         **kwargs: Additional arguments (endpoint, auth_token)
@@ -427,14 +427,14 @@ def invoke_agent(
         auth_token=kwargs.get("auth_token"),
         print_events=print_events,
     )
-    return invoke.invoke_agent(agent, parameters, dataset_ids, stream)
+    return invoke.invoke_agent(agent, parameters, dataset_slugs, stream)
 
 
 def invoke_by_key(
     bot_key: str,
     messages: Optional[List[Dict[str, str]]] = None,
     parameters: Optional[Dict[str, Any]] = None,
-    dataset_ids: Optional[list] = None,
+    dataset_slugs: Optional[list] = None,
     mode: Optional[str] = None,
     stream: bool = False,
     output_format: str = "events",
@@ -448,7 +448,7 @@ def invoke_by_key(
         bot_key: Bot key (e.g., "erdo.data-analyzer")
         messages: Messages in format [{"role": "user", "content": "..."}]
         parameters: Parameters to pass to the bot
-        dataset_ids: Dataset IDs to include
+        dataset_slugs: Dataset slugs to include (e.g. ["my-dataset"] or ["org.my-dataset"])
         mode: Invocation mode: "live" (default), "replay" (cached), or "mock" (synthetic)
         stream: Whether to stream events
         output_format: Output format: "events" (raw), "text" (formatted), "json" (summary)
@@ -465,5 +465,5 @@ def invoke_by_key(
         print_events=print_events,
     )
     return invoke.invoke_by_key(
-        bot_key, messages, parameters, dataset_ids, mode, stream, output_format, verbose
+        bot_key, messages, parameters, dataset_slugs, mode, stream, output_format, verbose
     )