iointel 1.3.0__tar.gz → 1.3.2__tar.gz

iointel-1.3.2/.github/workflows/linting.yaml

@@ -0,0 +1,16 @@
+name: Linting
+ 
+on:
+  pull_request:
+  push:
+    branches: [main, dev]
+ 
+jobs:
+  linting:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v3
+ 
+      - name: Lint
+        uses: pre-commit/action@v3.0.0

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iointel
-Version: 1.3.0
+Version: 1.3.2
 Summary: A framework to create agents, tasks, and workflows.
 License-File: LICENSE
 Keywords: AI,Agents,LLM,framework,tools
@@ -10,19 +10,15 @@ Requires-Dist: duckduckgo-search==8.0.0
 Requires-Dist: firecrawl-py>=2.2.0
 Requires-Dist: firecrawl>=2.2.0
 Requires-Dist: httpx
-Requires-Dist: langchain-core>=0.3.28
-Requires-Dist: langchain-openai>=0.2.14
 Requires-Dist: pgvector>=0.4.0
 Requires-Dist: pydantic-ai>=0.1.0
 Requires-Dist: pydantic>=2.10.3
 Requires-Dist: python-dotenv>=1.0.1
 Requires-Dist: python-multipart>=0.0.20
 Requires-Dist: pyyaml>=6.0.2
-Requires-Dist: r2r>=3.4.4
 Requires-Dist: rich>=13.9.4
 Requires-Dist: sqlalchemy-utils>=0.41.2
 Requires-Dist: sqlalchemy>=2.0.36
-Requires-Dist: wolframalpha>=5.1.3
 Provides-Extra: dev
 Requires-Dist: pre-commit>=4.1.0; extra == 'dev'
 Requires-Dist: pytest-asyncio>=0.21.2; extra == 'dev'

{iointel-1.3.0 → 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.0"
+__version__ = "1.3.2"

{iointel-1.3.0 → iointel-1.3.2}/iointel/src/agent_methods/agents/agents_factory.py

@@ -18,15 +18,19 @@ def create_agent(params: AgentParams) -> Agent:
     agent_data["tools"] = resolve_tools(params)
     if output_type := agent_data.get("output_type"):
         if isinstance(output_type, str):
-            agent_data["output_type"] = globals().get(output_type) or getattr(__builtins__, output_type, output_type)
+            agent_data["output_type"] = globals().get(output_type) or getattr(
+                __builtins__, output_type, output_type
+            )
     return Agent(**agent_data)
 
 
-def create_swarm(agents: list[AgentParams]|AgentSwarm):
+def create_swarm(agents: list[AgentParams] | AgentSwarm):
     raise NotImplementedError()
 
 
-def agent_or_swarm(agent_obj: Agent|Sequence[Agent], store_creds: bool) -> list[AgentParams]|AgentSwarm:
+def agent_or_swarm(
+    agent_obj: Agent | Sequence[Agent], store_creds: bool
+) -> list[AgentParams] | AgentSwarm:
     """
     Serializes an agent object into a list of AgentParams.
 
@@ -63,7 +67,9 @@ def agent_or_swarm(agent_obj: Agent|Sequence[Agent], store_creds: bool) -> list[
 
     if isinstance(agent_obj, Sequence):
         # group of agents not packed as a swarm
-        assert all(not hasattr(ag, "members") for ag in agent_obj), "Nested swarms not allowed"
+        assert all(not hasattr(ag, "members") for ag in agent_obj), (
+            "Nested swarms not allowed"
+        )
         return [make_params(ag) for ag in agent_obj]
     if hasattr(agent_obj, "api_key"):
         # Individual agent.

{iointel-1.3.0 → iointel-1.3.2}/iointel/src/agent_methods/agents/tool_factory.py

@@ -97,7 +97,9 @@ def resolve_tools(params: AgentParams) -> List[Optional[Callable]]:
             logger.debug(
                 f"Tool '{tool_obj.name}' found in TOOLS_REGISTRY under the custom name '{registered_tool_name}'."
             )
-            resolved_tools.append(registered_tool.model_copy(update={"fn_self": tool_obj.fn_self}))
+            resolved_tools.append(
+                registered_tool.model_copy(update={"fn_self": tool_obj.fn_self})
+            )
             continue
         else:
             logger.warning(

{iointel-1.3.0 → 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.0 → iointel-1.3.2}/iointel/src/agent_methods/tools/duckduckgo.py

@@ -90,6 +90,7 @@ class DuckDuckGoSearchAPIWrapper(BaseModel):
 
         return results
 
+
 @register_tool
 def search_the_web(text: str, max_results: int = 10):
     """
@@ -99,6 +100,7 @@ def search_the_web(text: str, max_results: int = 10):
     """
     return DuckDuckGoSearchAPIWrapper().results(text, max_results=max_results)
 
+
 @register_tool
 async def search_the_web_async(text: str, max_results: int = 10):
     """

{iointel-1.3.0 → iointel-1.3.2}/iointel/src/agent_methods/tools/utils.py

@@ -2,6 +2,7 @@ from datetime import datetime
 
 from iointel.src.utilities.decorators import register_tool
 
+
 @register_tool
 def what_time_is_it():
     """

{iointel-1.3.0 → 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
@@ -222,13 +222,15 @@ class Agent(BaseModel):
     def add_tool(self, tool):
         registered_tool = self._get_registered_tool(tool)
         self.tools += [registered_tool]
-        self._runner._register_tool(PatchedValidatorTool(registered_tool.get_wrapped_fn()))
+        self._runner._register_tool(
+            PatchedValidatorTool(registered_tool.get_wrapped_fn())
+        )
 
     async def run(
         self,
         query: str,
         conversation_id: Optional[str] = None,
-        pretty=True,
+        pretty=None,
         message_history_limit=100,
         **kwargs,
     ):
@@ -279,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.0 → iointel-1.3.2}/iointel/src/chainables.py

@@ -1,7 +1,6 @@
 from typing import List, Optional, Dict
 
 # from .task import CUSTOM_WORKFLOW_REGISTRY
-from .agent_methods.tools.tools import between
 from .utilities.runners import run_agents
 from .utilities.decorators import register_custom_task
 from .utilities.registries import CHAINABLE_METHODS, CUSTOM_WORKFLOW_REGISTRY
@@ -97,6 +96,7 @@ def execute_summarize_text(
         )
         return summary.execute()
 
+
 @register_custom_task("sentiment")
 async def execute_sentiment(
     task_metadata: dict, objective: str, agents: List[Agent], execution_metadata: dict
@@ -152,14 +152,12 @@ def execute_extract_entities(
         return extracted.execute()
 
 
-
 @register_custom_task("translate_text")
 def execute_translate_text(
     task_metadata: dict, objective: str, agents: List[Agent], execution_metadata: dict
 ):
     target_lang = task_metadata["target_language"]
     from ..client.client import translate_text_task
-    from .agent_methods.data_models.datamodels import TranslationResult
 
     client_mode = execution_metadata.get("client_mode", False)
     if client_mode:
@@ -176,7 +174,6 @@ def execute_translate_text(
         return result
 
 
-
 @register_custom_task("classify")
 def execute_classify(
     task_metadata: dict, objective: str, agents: List[Agent], execution_metadata: dict
@@ -193,7 +190,7 @@ def execute_classify(
             objective=f"""Take this text: {objective}
 
             Classify it into the appropriate category.
-            Category must be one of: {', '.join(classify_by)}.
+            Category must be one of: {", ".join(classify_by)}.
             Return only the determined category, omit the thoughts.""",
             agents=agents,
             output_type=str,

{iointel-1.3.0 → iointel-1.3.2}/iointel/src/memory.py

@@ -13,8 +13,8 @@ from pydantic_ai.messages import (
 )
 
 from datetime import datetime, timezone
-from sqlalchemy import create_engine, Column, String, Text, DateTime, select
-from sqlalchemy.orm import sessionmaker, declarative_base
+from sqlalchemy import Column, String, Text, DateTime, select
+from sqlalchemy.orm import declarative_base
 
 # For async support
 from sqlalchemy.ext.asyncio import create_async_engine, async_sessionmaker

{iointel-1.3.0 → iointel-1.3.2}/iointel/src/task.py

@@ -1,5 +1,4 @@
-from typing import List, Dict, Any, Optional
-import asyncio
+from typing import List, Dict, Any
 
 from .agents import Agent
 from .utilities.helpers import LazyCaller
@@ -86,9 +85,7 @@ class Task:
             )
         )
 
-    async def chain_runs(
-        self, run_specs: List[Dict[str, Any]]
-    ) -> List[Any]:
+    async def chain_runs(self, run_specs: List[Dict[str, Any]]) -> List[Any]:
         """
         Execute multiple runs in sequence. Each element in run_specs is a dict containing parameters for `self.run`.
         The output of one run can be fed into the context of the next run if desired.
@@ -134,8 +131,7 @@ class Task:
                 **{
                     k: v
                     for k, v in spec.items()
-                    if k
-                    not in ["objective", "agents", "context", "conversation_id"]
+                    if k not in ["objective", "agents", "context", "conversation_id"]
                 },
             )
             results.append(result)

{iointel-1.3.0 → iointel-1.3.2}/iointel/src/utilities/graph_nodes.py

@@ -9,7 +9,7 @@ from dataclasses import dataclass, field, make_dataclass
 class WorkflowState:
     initial_text: str = ""
     conversation_id: str = ""
-    results: Dict[str, Any] = field(default_factory=dict) 
+    results: Dict[str, Any] = field(default_factory=dict)
 
     @classmethod
     def get_id(cls) -> str:
@@ -23,17 +23,15 @@ class TaskNode(BaseNode[WorkflowState]):
     default_agents: list
     conversation_id: str
     unique_id: str = field(init=False)
-    next_task: Optional['TaskNode'] = None
+    next_task: Optional["TaskNode"] = None
 
     def __post_init__(self):
         self.unique_id = (
-            self.task.get("task_id")
-            or self.task.get("name")
-            or str(uuid.uuid4())
+            self.task.get("task_id") or self.task.get("name") or str(uuid.uuid4())
         )
 
         self.conversation_id = self.conversation_id
-        
+
     @classmethod
     def get_node_def(cls, local_ns: Optional[Dict[str, Any]] = None):
         next_edges: dict[str, Any] = {}
@@ -49,11 +47,16 @@ class TaskNode(BaseNode[WorkflowState]):
             returns_base_node=True,
         )
 
-    async def run(self, context: GraphRunContext[WorkflowState]) -> "TaskNode" | End[WorkflowState]:
-        from ..workflow import Workflow, _get_task_key  # import must happen here, or circular issue occurs
+    async def run(
+        self, context: GraphRunContext[WorkflowState]
+    ) -> "TaskNode" | End[WorkflowState]:
+        from ..workflow import (
+            Workflow,
+            _get_task_key,
+        )  # import must happen here, or circular issue occurs
 
         wf = Workflow()
-        state = context.state 
+        state = context.state
 
         if not state.conversation_id:
             state.conversation_id = self.conversation_id or str(uuid.uuid4())
@@ -64,27 +67,25 @@ class TaskNode(BaseNode[WorkflowState]):
         task_key = _get_task_key(self.task)
 
         result = await wf.run_task(
-            self.task,
-            self.default_text,
-            self.default_agents,
-            self.conversation_id
+            self.task, self.default_text, self.default_agents, self.conversation_id
         )
 
-        state.results[task_key] = result.get("data", result) if isinstance(result, dict) else result
+        state.results[task_key] = (
+            result.get("data", result) if isinstance(result, dict) else result
+        )
         return self.next_task if self.next_task else End(state)
 
 
-def make_task_node(task: dict,
-                   default_text: str,
-                   default_agents: list,
-                   conv_id: str) -> type[BaseNode]:
+def make_task_node(
+    task: dict, default_text: str, default_agents: list, conv_id: str
+) -> type[BaseNode]:
     """
     Return a brand‑new subclass of TaskNode whose `get_id()` is unique.
     The task parameters are stored on the *class* so the graph can later
     instantiate it.
     """
-    uid       = task.get("task_id") or task.get("name") or str(uuid.uuid4())
-    cls_name  = f'{task.get("name","task")}_{uid}'.replace('-', '_')
+    uid = task.get("task_id") or task.get("name") or str(uuid.uuid4())
+    cls_name = f"{task.get('name', 'task')}_{uid}".replace("-", "_")
 
     # Every field you’d pass to TaskNode’s __init__ becomes a *class*
     # attribute; pydantic‑graph will pass them to the generated __init__.
@@ -92,19 +93,18 @@ def make_task_node(task: dict,
     # data‑carrying fields are already inherited from the base ``TaskNode``
     # we can simply pass an **empty list**.  (Leaving it out raises the
     # ``TypeError: make_dataclass() missing 1 required positional argument: 'fields'``)
-    
+
     NewNode = make_dataclass(
         cls_name,
-        [],                       # <‑‑ empty ``fields`` list satisfies the API
+        [],  # <‑‑ empty ``fields`` list satisfies the API
         bases=(TaskNode,),
         namespace={
             # class‑level constants → become default values for the subclass
-            "task"          : task,
-            "default_text"  : default_text,
+            "task": task,
+            "default_text": default_text,
             "default_agents": default_agents,
             "conversation_id": conv_id,
-            "_node_id"      : cls_name,  # used by the custom ``get_id`` below
-    
+            "_node_id": cls_name,  # used by the custom ``get_id`` below
             # Provide a per‑class ``get_id`` so every generated subclass
             # advertises a unique node‑ID to pydantic‑graph
             "get_id": classmethod(lambda cls: cls._node_id),

{iointel-1.3.0 → iointel-1.3.2}/iointel/src/utilities/helpers.py

@@ -17,6 +17,7 @@ def make_logger(name: str, level: str = "INFO"):
 
 logger = make_logger(__name__)
 
+
 class LazyCaller(BaseModel):
     func: Callable
     args: tuple[Any, ...] = ()
@@ -77,13 +78,14 @@ class LazyCaller(BaseModel):
         """Only serialize the name, not the problematic object"""
         return {"name": self.name}
 
+
 def supports_tool_choice_required(model_name: str) -> bool:
     """Temp hack fix to check if the model supports tool choice required."""
     # TODO: Remove this once we have a better way to check if the model supports tool choice required.
     model_name = model_name.lower()
     return (
-        model_name.startswith("gpt-") or
-        model_name.startswith("openai/") or
-        "gpt" in model_name or
-        model_name == "meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8"
+        model_name.startswith("gpt-")
+        or model_name.startswith("openai/")
+        or "gpt" in model_name
+        or model_name == "meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8"
     )

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.0 → iointel-1.3.2}/iointel/src/utilities/runners.py

@@ -1,13 +1,19 @@
 from typing import Sequence
 
-from ..agents import Agent
 from .helpers import LazyCaller
 from ..task import Task
 from ..agent_methods.data_models.datamodels import TaskDefinition
 from ..agent_methods.agents.agents_factory import agent_or_swarm
 
+
 def _to_task_definition(
-    objective: str, agents=None, conversation_id=None, name=None, task_id="some_default", context=None, **kwargs
+    objective: str,
+    agents=None,
+    conversation_id=None,
+    name=None,
+    task_id="some_default",
+    context=None,
+    **kwargs,
 ) -> TaskDefinition:
     """
     Helper that merges the user’s provided fields into a TaskDefinition.
@@ -42,9 +48,8 @@ async def _run_stream(objective: str, output_type=None, **all_kwargs):
 async def _run(objective: str, output_type=None, **all_kwargs):
     definition = _to_task_definition(objective, **all_kwargs)
     agents = definition.agents or []
-    return await Task(agents=agents).run(
-        definition=definition, output_type=output_type
-    )
+    return await Task(agents=agents).run(definition=definition, output_type=output_type)
+
 
 async def _unpack(func, *args, **kwargs):
     result = await (await func(*args, **kwargs)).execute()

{iointel-1.3.0 → iointel-1.3.2}/iointel/src/utilities/stages.py

@@ -1,5 +1,4 @@
 import asyncio
-import concurrent.futures
 from .runners import run_agents
 from typing import Any, get_type_hints
 
@@ -52,13 +51,20 @@ async def _run_sequential(
 ) -> list:
     results = []
     for substage in stage.stages:
-        results.append(await execute_stage(substage, agents, task_metadata, default_text))
+        results.append(
+            await execute_stage(substage, agents, task_metadata, default_text)
+        )
     return results
 
 
 @register_stage_runner
-async def _run_parallel(stage: ParallelStage, agents, task_metadata, default_text) -> list:
-    futures = [execute_stage(substage, agents, task_metadata, default_text) for substage in stage.stages]
+async def _run_parallel(
+    stage: ParallelStage, agents, task_metadata, default_text
+) -> list:
+    futures = [
+        execute_stage(substage, agents, task_metadata, default_text)
+        for substage in stage.stages
+    ]
     return await asyncio.gather(*futures)
 
 
@@ -74,7 +80,9 @@ async def _run_while(stage: WhileStage, agents, task_metadata, default_text) ->
 
 
 @register_stage_runner
-async def _run_fallback(stage: FallbackStage, agents, task_metadata, default_text) -> any:
+async def _run_fallback(
+    stage: FallbackStage, agents, task_metadata, default_text
+) -> any:
     try:
         return await execute_stage(stage.primary, agents, task_metadata, default_text)
     except Exception as e:

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

@@ -5,7 +5,11 @@ import yaml
 from pathlib import Path
 from collections import defaultdict
 
-from .agent_methods.agents.agents_factory import agent_or_swarm, create_agent, create_swarm
+from .agent_methods.agents.agents_factory import (
+    agent_or_swarm,
+    create_agent,
+    create_swarm,
+)
 from .agent_methods.data_models.datamodels import (
     WorkflowDefinition,
     TaskDefinition,
@@ -24,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
@@ -39,6 +44,7 @@ from rich.progress import (
 
 logger = make_logger(__name__)
 
+
 def _get_task_key(task: dict) -> str:
     return (
         task.get("task_id")
@@ -47,6 +53,7 @@ def _get_task_key(task: dict) -> str:
         or "task"
     )
 
+
 class Workflow:
     """
     Manages a chain of tasks and runs them sequentially.
@@ -120,7 +127,7 @@ class Workflow:
             execution_metadata["conversation_id"] = task["conversation_id"]
         if conversation_id:
             execution_metadata["conversation_id"] = conversation_id
-        client_mode = execution_metadata.get("client_mode", self.client_mode)
+        # client_mode = execution_metadata.get("client_mode", self.client_mode)
 
         if stage_defs := execution_metadata.get("stages"):
             stage_objects = []
@@ -219,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)
 
@@ -234,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
@@ -289,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)
 
@@ -300,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
@@ -344,14 +361,12 @@ class Workflow:
 
         # pass the class objects to Graph
         return Graph(
-            nodes      = tuple(node_classes),
-            state_type = WorkflowState,
-            run_end_type = WorkflowState,
+            nodes=tuple(node_classes),
+            state_type=WorkflowState,
+            run_end_type=WorkflowState,
         )
 
-    async def run_tasks(
-        self, conversation_id: Optional[str] = None, **kwargs
-    ) -> dict:
+    async def run_tasks(self, conversation_id: Optional[str] = None, **kwargs) -> dict:
         if not conversation_id:
             conversation_id = str(uuid.uuid4())
 
@@ -426,8 +441,8 @@ class Workflow:
                 "memory": True,
                 "memories": True,
                 "agents": {"__all__": {"memory": True}},
-                "tasks": {"__all__": {"agents": {"__all__": {"memory": True}}}}
-            }
+                "tasks": {"__all__": {"agents": {"__all__": {"memory": True}}}},
+            },
         )
         yaml_str = yaml.safe_dump(wf_dict, sort_keys=False)
         if file_path:

{iointel-1.3.0 → iointel-1.3.2}/pyproject.toml

@@ -11,10 +11,6 @@ dependencies = [
     "pydantic-ai>=0.1.0",
     "pyyaml>=6.0.2",
     "duckduckgo-search==8.0.0",
-    "langchain-core>=0.3.28",
-    "langchain-openai>=0.2.14",
-    "r2r>=3.4.4",
-    "wolframalpha>=5.1.3",
     "pgvector>=0.4.0",
     "sqlalchemy-utils>=0.41.2",
     "sqlalchemy>=2.0.36",

{iointel-1.3.0 → iointel-1.3.2}/tests/test_tools_isolated/test_firecrawl_tool.py

@@ -2,5 +2,6 @@ from iointel.src.agent_methods.tools.firecrawl import Crawler
 
 crawler = Crawler()
 
+
 async def test_firecrawl():
     assert crawler.scrape_url(url="https://firecrawl.dev/")

{iointel-1.3.0 → iointel-1.3.2}/tests/test_tools_isolated/test_r2r.py

@@ -43,7 +43,7 @@ async def test_rag_search(r2r_client: R2RClient, sample_document: uuid.UUID):
     result = await r2r_client.rag_search("What do you know about samples?")
     assert result, "Expected non-empty RAG result"
     assert result["answer"], "Expected non-empty RAG answer"
-    assert (
-        "database" in result["answer"]
-    ), "Sample document has to be referenced in RAG answer"
+    assert "database" in result["answer"], (
+        "Sample document has to be referenced in RAG answer"
+    )
     assert result["citations"], "Expected to see at least one citation"

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.0/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.0/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)