iointel 1.3.1__tar.gz → 1.3.2__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iointel
-Version: 1.3.1
+Version: 1.3.2
 Summary: A framework to create agents, tasks, and workflows.
 License-File: LICENSE
 Keywords: AI,Agents,LLM,framework,tools

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

@@ -1,6 +1,3 @@
-from .src.utilities.magic import (
-    UNUSED as UNUSED,
-)  # this performs some magic to stop prefect mucking with logging setup
 from .src.agents import Agent
 from .src.memory import AsyncMemory
 from .src.workflow import Workflow
@@ -8,6 +5,7 @@ from .src.utilities.runners import run_agents, run_agents_stream
 from .src.agent_methods.tools.wolfram import query_wolfram
 from .src.agent_methods.data_models.datamodels import PersonaConfig
 from .src.utilities.decorators import register_custom_task, register_tool
+from .src.utilities.rich import pretty_output
 
 from .src.code_parsers.pycode_parser import (
     PythonModule,
@@ -42,7 +40,9 @@ __all__ = [
     "PythonCodeGenerator",
     ###tools###
     "query_wolfram",
+    ###configuration###
+    "pretty_output",
 ]
 
 
-__version__ = "1.3.1"
+__version__ = "1.3.2"

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

@@ -406,7 +406,7 @@ class AgentParams(BaseModel):
         description="Context to be passed to the agent.",
     )
     # memories: Optional[list[Memory]] = Field(default_factory=list)
-    memory: Optional[AsyncMemory] = Field(default_factory=list)
+    memory: Optional[AsyncMemory] = None
 
     model_settings: Optional[Dict[str, Any]] = Field(default_factory=dict)
     output_type: Optional[Any] = str

{iointel-1.3.1 → iointel-1.3.2}/iointel/src/agents.py

@@ -1,6 +1,6 @@
 from .memory import AsyncMemory
 from .agent_methods.data_models.datamodels import PersonaConfig, Tool
-from .utilities.rich import console
+from .utilities.rich import console, pretty_output
 from .utilities.constants import get_api_url, get_base_model, get_api_key
 from .utilities.registries import TOOLS_REGISTRY
 from .utilities.helpers import supports_tool_choice_required
@@ -230,7 +230,7 @@ class Agent(BaseModel):
         self,
         query: str,
         conversation_id: Optional[str] = None,
-        pretty=True,
+        pretty=None,
         message_history_limit=100,
         **kwargs,
     ):
@@ -281,7 +281,7 @@ class Agent(BaseModel):
                 await self.memory.store_run_history(conversation_id, result)
             except Exception as e:
                 print("Error storing run history:", e)
-        if pretty:
+        if pretty or (pretty is None and pretty_output.is_enabled):
             task_header = Text(
                 f" Objective: {query} ", style="bold white on dark_green"
             )

iointel-1.3.2/iointel/src/utilities/rich.py

@@ -0,0 +1,39 @@
+from contextlib import contextmanager
+from rich.console import Console
+from rich.theme import Theme
+
+custom_theme = Theme({"electric_blue": "#7DF9FF"})
+
+console = Console(theme=custom_theme)
+
+
+class PrettyController:
+    def __init__(self):
+        self._enabled = True
+
+    @property
+    def is_enabled(self):
+        return self._enabled
+
+    def _onoff(self, enabled: bool):
+        @contextmanager
+        def inner(this=self):
+            old_value, this._enabled = this._enabled, enabled
+            try:
+                yield this
+            finally:
+                this._enabled = old_value
+
+        return inner()
+
+    def enabled(self):
+        return self._onoff(True)
+
+    def disabled(self):
+        return self._onoff(False)
+
+    def global_onoff(self, enabled: bool):
+        self._enabled = enabled
+
+
+pretty_output = PrettyController()

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

@@ -28,6 +28,7 @@ from .utilities.stages import (
     WhileStage,
     FallbackStage,
 )
+from .utilities.rich import pretty_output
 
 from .utilities.graph_nodes import WorkflowState, TaskNode, make_task_node
 from pydantic_graph import Graph, End, GraphRunContext
@@ -225,7 +226,11 @@ class Workflow:
                 result = await result.execute()
             return result
 
-    async def execute_graph_streaming(self, graph, initial_state):
+    async def execute_graph_streaming(
+        self, graph, initial_state, pretty: Optional[bool] = None
+    ):
+        if pretty is None:
+            pretty = pretty_output.is_enabled
         nodes = list(graph.node_defs.values())
         total_tasks = len(nodes)
 
@@ -240,6 +245,7 @@ class Workflow:
             TaskProgressColumn(),
             TimeElapsedColumn(),
             transient=True,
+            disable=not pretty,
         ) as progress:
             task_progress = progress.add_task(
                 "[cyan]Executing Tasks...", total=total_tasks
@@ -295,7 +301,11 @@ class Workflow:
 
         return state
 
-    async def execute_graph(self, graph: Graph, initial_state: WorkflowState):
+    async def execute_graph(
+        self, graph: Graph, initial_state: WorkflowState, pretty: Optional[bool] = None
+    ):
+        if pretty is None:
+            pretty = pretty_output.is_enabled
         nodes = list(graph.node_defs.values())
         total_tasks = len(nodes)
 
@@ -306,6 +316,7 @@ class Workflow:
             TaskProgressColumn(),
             TimeElapsedColumn(),
             transient=True,
+            disable=not pretty,
         ) as progress:
             task_progress = progress.add_task(
                 "[cyan]Executing Tasks...", total=total_tasks

iointel-1.3.2/tests/test_utilities.py

@@ -0,0 +1,18 @@
+from iointel import pretty_output
+
+
+def test_pretty_context():
+    assert pretty_output.is_enabled, "Pretty output not enabled"
+    with pretty_output.disabled() as ctx:
+        assert not pretty_output.is_enabled, "Pretty output not disabled"
+        assert ctx is pretty_output, "Wrong context manager instance"
+    assert pretty_output.is_enabled, "Pretty output not enabled"
+
+
+def test_pretty_global():
+    assert pretty_output.is_enabled, "Pretty output not enabled"
+    try:
+        pretty_output.global_onoff(False)
+        assert not pretty_output.is_enabled, "Pretty output not disabled"
+    finally:
+        pretty_output.global_onoff(True)

iointel-1.3.1/iointel/src/utilities/magic.py

@@ -1,9 +0,0 @@
-import os
-
-
-# turn off most prefect log messages, as they aren't useful
-# to end user, but might hurt UX for inexperienced ones
-for env_name in ("PREFECT_LOGGING_SERVER_LEVEL", "PREFECT_LOGGING_LEVEL"):
-    os.environ[env_name] = os.environ.get(env_name, "CRITICAL")
-
-UNUSED = None

iointel-1.3.1/iointel/src/utilities/rich.py

@@ -1,6 +0,0 @@
-from rich.console import Console
-from rich.theme import Theme
-
-custom_theme = Theme({"electric_blue": "#7DF9FF"})
-
-console = Console(theme=custom_theme)