iointel 1.3.2__tar.gz → 1.3.3__tar.gz

{iointel-1.3.2 → iointel-1.3.3}/.gitignore

@@ -11,4 +11,8 @@ dist
 .idea
 
 *.ipynb
-*.ipynb_checkpoints
+*.ipynb_checkpoints
+
+# virtual environments
+venv
+.venv

{iointel-1.3.2 → iointel-1.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iointel
-Version: 1.3.2
+Version: 1.3.3
 Summary: A framework to create agents, tasks, and workflows.
 License-File: LICENSE
 Keywords: AI,Agents,LLM,framework,tools
@@ -117,7 +117,7 @@ Users can define tasks (like `sentiment`, `translate_text`, etc.) in a **local**
 
 - A **task** is a single step in a workflow, e.g.,  `schedule_reminder`, `sentiment`, `translate_text`, etc.
 - Tasks are managed by the `Workflow` class in `workflow.py`.
-- Tasks can be chained for multi-step logic into a workflow (e.g., `await Workflow(text="...").translate_text().sentiment().run_tasks()`).
+- Tasks can be chained for multi-step logic into a workflow (e.g., `await Workflow(objective="...").translate_text().sentiment().run_tasks()`).
 
 ### Client Mode vs Local Mode<a id="client-mode-vs-local-mode"></a>
 
@@ -199,7 +199,7 @@ In Python code, you can create tasks by instantiating the Tasks class and chaini
 ```python
 from iointel import Workflow
 
-tasks = Workflow(text="This is the text to analyze", client_mode=False)
+tasks = Workflow(objective="This is the text to analyze", client_mode=False)
 (
   tasks
     .sentiment(agents=[my_agent])
@@ -214,14 +214,14 @@ Because client_mode=False, everything runs locally.
 ### Running a Local Workflow<a id="running-a-local-workflow"></a>
 
 ```python
-tasks = Workflow(text="Breaking news: local sports team wins!", client_mode=False)
+tasks = Workflow(objective="Breaking news: local sports team wins!", client_mode=False)
 await tasks.summarize_text(max_words=50).run_tasks()
 ```
 
 ### Running a Remote Workflow (Client Mode)<a id="running-a-remote-workflow-client-mode"></a>
 
 ```python
-tasks = Workflow(text="Breaking news: local sports team wins!", client_mode=True)
+tasks = Workflow(objective="Breaking news: local sports team wins!", client_mode=True)
 await tasks.summarize_text(max_words=50).run_tasks()
 ```
 Now, summarize_text calls the client function (e.g., summarize_task(...)) instead of local logic.

{iointel-1.3.2 → iointel-1.3.3}/README.md

@@ -84,7 +84,7 @@ Users can define tasks (like `sentiment`, `translate_text`, etc.) in a **local**
 
 - A **task** is a single step in a workflow, e.g.,  `schedule_reminder`, `sentiment`, `translate_text`, etc.
 - Tasks are managed by the `Workflow` class in `workflow.py`.
-- Tasks can be chained for multi-step logic into a workflow (e.g., `await Workflow(text="...").translate_text().sentiment().run_tasks()`).
+- Tasks can be chained for multi-step logic into a workflow (e.g., `await Workflow(objective="...").translate_text().sentiment().run_tasks()`).
 
 ### Client Mode vs Local Mode<a id="client-mode-vs-local-mode"></a>
 
@@ -166,7 +166,7 @@ In Python code, you can create tasks by instantiating the Tasks class and chaini
 ```python
 from iointel import Workflow
 
-tasks = Workflow(text="This is the text to analyze", client_mode=False)
+tasks = Workflow(objective="This is the text to analyze", client_mode=False)
 (
   tasks
     .sentiment(agents=[my_agent])
@@ -181,14 +181,14 @@ Because client_mode=False, everything runs locally.
 ### Running a Local Workflow<a id="running-a-local-workflow"></a>
 
 ```python
-tasks = Workflow(text="Breaking news: local sports team wins!", client_mode=False)
+tasks = Workflow(objective="Breaking news: local sports team wins!", client_mode=False)
 await tasks.summarize_text(max_words=50).run_tasks()
 ```
 
 ### Running a Remote Workflow (Client Mode)<a id="running-a-remote-workflow-client-mode"></a>
 
 ```python
-tasks = Workflow(text="Breaking news: local sports team wins!", client_mode=True)
+tasks = Workflow(objective="Breaking news: local sports team wins!", client_mode=True)
 await tasks.summarize_text(max_words=50).run_tasks()
 ```
 Now, summarize_text calls the client function (e.g., summarize_task(...)) instead of local logic.

{iointel-1.3.2 → iointel-1.3.3}/iointel/__init__.py

@@ -45,4 +45,4 @@ __all__ = [
 ]
 
 
-__version__ = "1.3.2"
+__version__ = "1.3.3"

{iointel-1.3.2 → iointel-1.3.3}/iointel/src/agent_methods/data_models/datamodels.py

@@ -446,12 +446,6 @@ class TranslationResult(BaseModel):
 Activation = Annotated[float, Field(ge=0, le=1)]
 
 
-class ModerationException(Exception):
-    """Exception raised when a message is not allowed."""
-
-    ...
-
-
 class ViolationActivation(TypedDict):
     """Violation activation."""
 
@@ -463,6 +457,14 @@ class ViolationActivation(TypedDict):
     dangerous_content: Activation
 
 
+class ModerationException(Exception):
+    """Exception raised when a message is not allowed."""
+
+    def __init__(self, *args, violations: ViolationActivation):
+        super().__init__(*args)
+        self.violations = violations
+
+
 ##### task and workflow models ########
 class BaseStage(BaseModel):
     stage_id: Optional[int] = None

{iointel-1.3.2 → iointel-1.3.3}/iointel/src/chainables.py

@@ -225,17 +225,19 @@ async def execute_moderation(
         ).execute()
 
         if result["extreme_profanity"] > threshold:
-            raise ModerationException("Extreme profanity detected")
+            raise ModerationException("Extreme profanity detected", violations=result)
         elif result["sexually_explicit"] > threshold:
-            raise ModerationException("Sexually explicit content detected")
+            raise ModerationException(
+                "Sexually explicit content detected", violations=result
+            )
         elif result["hate_speech"] > threshold:
-            raise ModerationException("Hate speech detected")
+            raise ModerationException("Hate speech detected", violations=result)
         elif result["harassment"] > threshold:
-            raise ModerationException("Harassment detected")
+            raise ModerationException("Harassment detected", violations=result)
         elif result["self_harm"] > threshold:
-            raise ModerationException("Self harm detected")
+            raise ModerationException("Self harm detected", violations=result)
         elif result["dangerous_content"] > threshold:
-            raise ModerationException("Dangeme profanity detected")
+            raise ModerationException("Dangeme profanity detected", violations=result)
 
         return result
 

{iointel-1.3.2 → iointel-1.3.3}/iointel/src/workflow.py

@@ -1,6 +1,7 @@
 from typing import List, Optional, Any
 import uuid
 import inspect
+import warnings
 import yaml
 from pathlib import Path
 from collections import defaultdict
@@ -59,7 +60,7 @@ class Workflow:
     Manages a chain of tasks and runs them sequentially.
 
     Example usage:
-        workflow = Workflow(text="Some input text", client_mode=False, agents=[swarm])
+        workflow = Workflow(objective="Some input text", client_mode=False, agents=[swarm])
         workflow.summarize_text(max_words=50).custom(name="do-fancy-thing", objective="Fancy step", agents=[my_agent])
         results = await workflow.run_tasks()
     """
@@ -67,9 +68,17 @@ class Workflow:
     def __init__(
         self,
         objective: str = "",
+        text: str | None = None,
         client_mode: bool = True,
         agents: Optional[List[Any]] = None,
     ):
+        if text is not None:
+            if objective:
+                raise ValueError("Both `text` and `objective` parameters set")
+            objective = text
+            warnings.warn(
+                "`text` parameter is deprecated, please use `objective` instead"
+            )
         self.tasks: List[dict] = []
         self.objective = objective
         self.client_mode = client_mode