versionhq 1.2.1.0__tar.gz → 1.2.1.2__tar.gz

{versionhq-1.2.1.0 → versionhq-1.2.1.2}/.gitignore

@@ -24,6 +24,7 @@ temp/*
 assets/*
 memo.txt
 site/
+_logs/
 
 __pycache__
 .ruff_cache/

{versionhq-1.2.1.0 → versionhq-1.2.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.2.1.0
+Version: 1.2.1.2
 Summary: An agentic orchestration framework for building agent networks that handle task automation.
 Author-email: Kuriko Iwai <kuriko@versi0n.io>
 License: MIT License
@@ -117,10 +117,10 @@ A Python framework for agentic orchestration that handles complex task automatio
   - [Supervising](#supervising)
 - [Technologies Used](#technologies-used)
 - [Project Structure](#project-structure)
-- [Setting Up a Project](#setting-up-a-project)
-  - [1. Installing package manager](#1-installing-package-manager)
-  - [2. Installing dependencies](#2-installing-dependencies)
-  - [3. Adding env secrets to .env file](#3-adding-env-secrets-to-env-file)
+- [Setting Up Your Project](#setting-up-your-project)
+  - [Installing package manager](#installing-package-manager)
+  - [Installing dependencies](#installing-dependencies)
+  - [Adding env secrets to .env file](#adding-env-secrets-to-env-file)
 - [Contributing](#contributing)
   - [Steps](#steps)
   - [Package Management with uv](#package-management-with-uv)
@@ -256,7 +256,7 @@ agent.update(
    res = network.launch()
    ```
 
-   This will form a network with multiple agents on `Formation` and return `TaskOutput` object with output in JSON, plane text, Pydantic model format with evaluation.
+This will form a network with multiple agents on `Formation` and return `TaskOutput` object with output in JSON, plane text, Pydantic model format with evaluation.
 
 
 ### Executing tasks
@@ -266,81 +266,87 @@ You can simply build an agent using `Agent` model and execute the task using `Ta
 By default, agents prioritize JSON over plane text outputs.
 
 
-   ```python
-   import versionhq as vhq
-   from pydantic import BaseModel
+```python
+import versionhq as vhq
+from pydantic import BaseModel
 
-   class CustomOutput(BaseModel):
-      test1: str
-      test2: list[str]
+class CustomOutput(BaseModel):
+   test1: str
+   test2: list[str]
 
-   def dummy_func(message: str, test1: str, test2: list[str]) -> str:
-      return f"""{message}: {test1}, {", ".join(test2)}"""
+def dummy_func(message: str, test1: str, test2: list[str]) -> str:
+   return f"""{message}: {test1}, {", ".join(test2)}"""
 
-   task = vhq.Task(
-      description="Amazing task",
-      pydantic_output=CustomOutput,
-      callback=dummy_func,
-      callback_kwargs=dict(message="Hi! Here is the result: ")
-   )
+task = vhq.Task(
+   description="Amazing task",
+   pydantic_output=CustomOutput,
+   callback=dummy_func,
+   callback_kwargs=dict(message="Hi! Here is the result: ")
+)
 
-   res = task.execute(context="amazing context to consider.")
-   print(res)
-   ```
+res = task.execute(context="amazing context to consider.")
+print(res)
+```
 
 
 This will return a `TaskOutput` object that stores response in plane text, JSON, and Pydantic model: `CustomOutput` formats with a callback result, tool output (if given), and evaluation results (if given).
 
-   ```python
-   res == TaskOutput(
-      task_id=UUID('<TASK UUID>'),
-      raw='{\"test1\":\"random str\", \"test2\":[\"str item 1\", \"str item 2\", \"str item 3\"]}',
-      json_dict={'test1': 'random str', 'test2': ['str item 1', 'str item 2', 'str item 3']},
-      pydantic=<class '__main__.CustomOutput'>,
-      tool_output=None,
-      callback_output='Hi! Here is the result: random str, str item 1, str item 2, str item 3', # returned a plain text summary
-      evaluation=None
-   )
-   ```
+```python
+res == TaskOutput(
+   task_id=UUID('<TASK UUID>'),
+   raw='{\"test1\":\"random str\", \"test2\":[\"str item 1\", \"str item 2\", \"str item 3\"]}',
+   json_dict={'test1': 'random str', 'test2': ['str item 1', 'str item 2', 'str item 3']},
+   pydantic=<class '__main__.CustomOutput'>,
+   tool_output=None,
+   callback_output='Hi! Here is the result: random str, str item 1, str item 2, str item 3', # returned a plain text summary
+   evaluation=None
+)
+```
 
 ### Supervising
 
-   ```python
-   import versionhq as vhq
+To create an agent network with one or more manager agents, designate members using the `is_manager` tag.
 
-   agent_a = vhq.Agent(role="agent a", goal="My amazing goals", llm="llm-of-your-choice")
-   agent_b = vhq.Agent(role="agent b", goal="My amazing goals", llm="llm-of-your-choice")
+```python
+import versionhq as vhq
 
-   task_1 = vhq.Task(
-      description="Analyze the client's business model.",
-      response_fields=[vhq.ResponseField(title="test1", data_type=str, required=True),],
-      allow_delegation=True
-   )
+agent_a = vhq.Agent(role="agent a", goal="My amazing goals", llm="llm-of-your-choice")
+agent_b = vhq.Agent(role="agent b", goal="My amazing goals", llm="llm-of-your-choice")
 
-    task_2 = vhq.Task(
-      description="Define the cohort.",
-      response_fields=[ResponseField(title="test1", data_type=int, required=True),],
-      allow_delegation=False
-   )
+task_1 = vhq.Task(
+   description="Analyze the client's business model.",
+   response_fields=[vhq.ResponseField(title="test1", data_type=str, required=True),],
+   allow_delegation=True
+)
 
-   team = vhq.Team(
-      members=[
-         vhq.Member(agent=agent_a, is_manager=False, task=task_1),
-         vhq.Member(agent=agent_b, is_manager=True, task=task_2),
-      ],
-   )
-   res = team.launch()
-   ```
+task_2 = vhq.Task(
+   description="Define a cohort.",
+   response_fields=[vhq.ResponseField(title="test1", data_type=int, required=True),],
+   allow_delegation=False
+)
+
+network =vhq.AgentNetwork(
+   members=[
+      vhq.Member(agent=agent_a, is_manager=False, tasks=[task_1]),
+      vhq.Member(agent=agent_b, is_manager=True, tasks=[task_2]), # Agent B as a manager
+   ],
+)
+res = network.launch()
+
+assert isinstance(res, vhq.NetworkOutput)
+assert not [item for item in task_1.processed_agents if "vhq-Delegated-Agent" == item]
+assert [item for item in task_1.processed_agents if "agent b" == item]
+```
 
 This will return a list with dictionaries with keys defined in the `ResponseField` of each task.
 
-Tasks can be delegated to a team manager, peers in the team, or completely new agent.
+Tasks can be delegated to a manager, peers within the agent network, or a completely new agent.
 
 <hr />
 
 ## Technologies Used
 
-**Graph Theory (Analysis and Visualization)**
+**Task Graph**
 
 * [NetworkX](https://networkx.org/documentation/stable/reference/introduction.html): A Python package to analyze, create, and manipulate complex graph networks.
 * [Matplotlib](https://matplotlib.org/stable/index.html): Visualization library
@@ -401,16 +407,19 @@ src/
 │     └── ...
 │
 └── uploads/  [.gitignore]    # Local directory to store uploaded files such as graphviz diagrams generatd by `Network` class
-|
+│
+└── _logs/    [.gitignore]    # Local directory to store error/warning logs for debugging
+│
+│
 pyproject.toml                # Project config
 
 ```
 
 <hr />
 
-## Setting Up a Project
+## Setting Up Your Project
 
-### 1. Installing package manager
+### Installing package manager
 
    For MacOS:
 
@@ -424,7 +433,7 @@ pyproject.toml                # Project config
    ```
 
 
-### 2. Installing dependencies
+### Installing dependencies
 
    ```
    uv venv
@@ -457,7 +466,7 @@ pyproject.toml                # Project config
 
    - `torch`/`Docling` related errors: Set up default Python version either `3.11.x` or `3.12.x` (same as AssertionError)
 
-### 3. Adding env secrets to .env file
+### Adding env secrets to .env file
 
 Create `.env` file in the project root and add following:
 
@@ -470,6 +479,7 @@ Create `.env` file in the project root and add following:
    [OTHER_LLM_INTERFACE_PROVIDER_OF_YOUR_CHOICE]_API_KEY=your-api-key
    ```
 
+
 <hr />
 
 ## Contributing
@@ -574,7 +584,7 @@ Common issues and solutions:
 
 * Issues related to dependencies: `rm -rf uv.lock`, `uv cache clean`, `uv venv`, and run `uv pip install -r requirements.txt -v`.
 
-* Issues related to the AI agents or RAG system: Check the `output.log` file for detailed error messages and stack traces.
+* Issues related to agents and other systems: Check `_logs` directory for detailed error messages and stack traces.
 
 * Issues related to `Python quit unexpectedly`: Check [this stackoverflow article](https://stackoverflow.com/questions/59888499/macos-catalina-python-quit-unexpectedly-error).
 

{versionhq-1.2.1.0 → versionhq-1.2.1.2}/README.md

@@ -36,10 +36,10 @@ A Python framework for agentic orchestration that handles complex task automatio
   - [Supervising](#supervising)
 - [Technologies Used](#technologies-used)
 - [Project Structure](#project-structure)
-- [Setting Up a Project](#setting-up-a-project)
-  - [1. Installing package manager](#1-installing-package-manager)
-  - [2. Installing dependencies](#2-installing-dependencies)
-  - [3. Adding env secrets to .env file](#3-adding-env-secrets-to-env-file)
+- [Setting Up Your Project](#setting-up-your-project)
+  - [Installing package manager](#installing-package-manager)
+  - [Installing dependencies](#installing-dependencies)
+  - [Adding env secrets to .env file](#adding-env-secrets-to-env-file)
 - [Contributing](#contributing)
   - [Steps](#steps)
   - [Package Management with uv](#package-management-with-uv)
@@ -175,7 +175,7 @@ agent.update(
    res = network.launch()
    ```
 
-   This will form a network with multiple agents on `Formation` and return `TaskOutput` object with output in JSON, plane text, Pydantic model format with evaluation.
+This will form a network with multiple agents on `Formation` and return `TaskOutput` object with output in JSON, plane text, Pydantic model format with evaluation.
 
 
 ### Executing tasks
@@ -185,81 +185,87 @@ You can simply build an agent using `Agent` model and execute the task using `Ta
 By default, agents prioritize JSON over plane text outputs.
 
 
-   ```python
-   import versionhq as vhq
-   from pydantic import BaseModel
+```python
+import versionhq as vhq
+from pydantic import BaseModel
 
-   class CustomOutput(BaseModel):
-      test1: str
-      test2: list[str]
+class CustomOutput(BaseModel):
+   test1: str
+   test2: list[str]
 
-   def dummy_func(message: str, test1: str, test2: list[str]) -> str:
-      return f"""{message}: {test1}, {", ".join(test2)}"""
+def dummy_func(message: str, test1: str, test2: list[str]) -> str:
+   return f"""{message}: {test1}, {", ".join(test2)}"""
 
-   task = vhq.Task(
-      description="Amazing task",
-      pydantic_output=CustomOutput,
-      callback=dummy_func,
-      callback_kwargs=dict(message="Hi! Here is the result: ")
-   )
+task = vhq.Task(
+   description="Amazing task",
+   pydantic_output=CustomOutput,
+   callback=dummy_func,
+   callback_kwargs=dict(message="Hi! Here is the result: ")
+)
 
-   res = task.execute(context="amazing context to consider.")
-   print(res)
-   ```
+res = task.execute(context="amazing context to consider.")
+print(res)
+```
 
 
 This will return a `TaskOutput` object that stores response in plane text, JSON, and Pydantic model: `CustomOutput` formats with a callback result, tool output (if given), and evaluation results (if given).
 
-   ```python
-   res == TaskOutput(
-      task_id=UUID('<TASK UUID>'),
-      raw='{\"test1\":\"random str\", \"test2\":[\"str item 1\", \"str item 2\", \"str item 3\"]}',
-      json_dict={'test1': 'random str', 'test2': ['str item 1', 'str item 2', 'str item 3']},
-      pydantic=<class '__main__.CustomOutput'>,
-      tool_output=None,
-      callback_output='Hi! Here is the result: random str, str item 1, str item 2, str item 3', # returned a plain text summary
-      evaluation=None
-   )
-   ```
+```python
+res == TaskOutput(
+   task_id=UUID('<TASK UUID>'),
+   raw='{\"test1\":\"random str\", \"test2\":[\"str item 1\", \"str item 2\", \"str item 3\"]}',
+   json_dict={'test1': 'random str', 'test2': ['str item 1', 'str item 2', 'str item 3']},
+   pydantic=<class '__main__.CustomOutput'>,
+   tool_output=None,
+   callback_output='Hi! Here is the result: random str, str item 1, str item 2, str item 3', # returned a plain text summary
+   evaluation=None
+)
+```
 
 ### Supervising
 
-   ```python
-   import versionhq as vhq
+To create an agent network with one or more manager agents, designate members using the `is_manager` tag.
 
-   agent_a = vhq.Agent(role="agent a", goal="My amazing goals", llm="llm-of-your-choice")
-   agent_b = vhq.Agent(role="agent b", goal="My amazing goals", llm="llm-of-your-choice")
+```python
+import versionhq as vhq
 
-   task_1 = vhq.Task(
-      description="Analyze the client's business model.",
-      response_fields=[vhq.ResponseField(title="test1", data_type=str, required=True),],
-      allow_delegation=True
-   )
+agent_a = vhq.Agent(role="agent a", goal="My amazing goals", llm="llm-of-your-choice")
+agent_b = vhq.Agent(role="agent b", goal="My amazing goals", llm="llm-of-your-choice")
 
-    task_2 = vhq.Task(
-      description="Define the cohort.",
-      response_fields=[ResponseField(title="test1", data_type=int, required=True),],
-      allow_delegation=False
-   )
+task_1 = vhq.Task(
+   description="Analyze the client's business model.",
+   response_fields=[vhq.ResponseField(title="test1", data_type=str, required=True),],
+   allow_delegation=True
+)
 
-   team = vhq.Team(
-      members=[
-         vhq.Member(agent=agent_a, is_manager=False, task=task_1),
-         vhq.Member(agent=agent_b, is_manager=True, task=task_2),
-      ],
-   )
-   res = team.launch()
-   ```
+task_2 = vhq.Task(
+   description="Define a cohort.",
+   response_fields=[vhq.ResponseField(title="test1", data_type=int, required=True),],
+   allow_delegation=False
+)
+
+network =vhq.AgentNetwork(
+   members=[
+      vhq.Member(agent=agent_a, is_manager=False, tasks=[task_1]),
+      vhq.Member(agent=agent_b, is_manager=True, tasks=[task_2]), # Agent B as a manager
+   ],
+)
+res = network.launch()
+
+assert isinstance(res, vhq.NetworkOutput)
+assert not [item for item in task_1.processed_agents if "vhq-Delegated-Agent" == item]
+assert [item for item in task_1.processed_agents if "agent b" == item]
+```
 
 This will return a list with dictionaries with keys defined in the `ResponseField` of each task.
 
-Tasks can be delegated to a team manager, peers in the team, or completely new agent.
+Tasks can be delegated to a manager, peers within the agent network, or a completely new agent.
 
 <hr />
 
 ## Technologies Used
 
-**Graph Theory (Analysis and Visualization)**
+**Task Graph**
 
 * [NetworkX](https://networkx.org/documentation/stable/reference/introduction.html): A Python package to analyze, create, and manipulate complex graph networks.
 * [Matplotlib](https://matplotlib.org/stable/index.html): Visualization library
@@ -320,16 +326,19 @@ src/
 │     └── ...
 │
 └── uploads/  [.gitignore]    # Local directory to store uploaded files such as graphviz diagrams generatd by `Network` class
-|
+│
+└── _logs/    [.gitignore]    # Local directory to store error/warning logs for debugging
+│
+│
 pyproject.toml                # Project config
 
 ```
 
 <hr />
 
-## Setting Up a Project
+## Setting Up Your Project
 
-### 1. Installing package manager
+### Installing package manager
 
    For MacOS:
 
@@ -343,7 +352,7 @@ pyproject.toml                # Project config
    ```
 
 
-### 2. Installing dependencies
+### Installing dependencies
 
    ```
    uv venv
@@ -376,7 +385,7 @@ pyproject.toml                # Project config
 
    - `torch`/`Docling` related errors: Set up default Python version either `3.11.x` or `3.12.x` (same as AssertionError)
 
-### 3. Adding env secrets to .env file
+### Adding env secrets to .env file
 
 Create `.env` file in the project root and add following:
 
@@ -389,6 +398,7 @@ Create `.env` file in the project root and add following:
    [OTHER_LLM_INTERFACE_PROVIDER_OF_YOUR_CHOICE]_API_KEY=your-api-key
    ```
 
+
 <hr />
 
 ## Contributing
@@ -493,7 +503,7 @@ Common issues and solutions:
 
 * Issues related to dependencies: `rm -rf uv.lock`, `uv cache clean`, `uv venv`, and run `uv pip install -r requirements.txt -v`.
 
-* Issues related to the AI agents or RAG system: Check the `output.log` file for detailed error messages and stack traces.
+* Issues related to agents and other systems: Check `_logs` directory for detailed error messages and stack traces.
 
 * Issues related to `Python quit unexpectedly`: Check [this stackoverflow article](https://stackoverflow.com/questions/59888499/macos-catalina-python-quit-unexpectedly-error).
 

{versionhq-1.2.1.0 → versionhq-1.2.1.2}/docs/core/task.md

@@ -9,7 +9,7 @@ tags:
 
 <class>`class` versionhq.task.model.<bold>Task<bold></class>
 
-A class to store and manage information for individual tasks, including their assignment to agents or teams, and dependencies via a node-based system that tracks conditions and status.
+A class to store and manage information for individual tasks, including their assignment to agents or agent networks, and dependencies via a node-based system that tracks conditions and status.
 
 Ref. Node / Edge / TaskGraph class
 

{versionhq-1.2.1.0 → versionhq-1.2.1.2}/docs/index.md

@@ -35,7 +35,7 @@ A Python framework for agentic orchestration that handles complex task automatio
 Agents are model-agnostic, and will improve task output, while oprimizing token cost and job latency, by sharing their memory, knowledge base, and RAG tools with other agents in the network.
 
 
-###  Agent Network
+### Agent Network
 
 Agents adapt their formation based on task complexity.
 
@@ -139,7 +139,7 @@ agent.update(
 
 ## Project Set Up
 
-### 1. Package manager
+**Installing package manager**
 
    For MacOS:
 
@@ -152,7 +152,7 @@ agent.update(
    sudo apt-get install uv
    ```
 
-### 2. Dependencies
+**Installing dependencies**
 
    ```
    uv venv
@@ -186,7 +186,7 @@ agent.update(
 
    - `torch`/`Docling` related errors: Set up default Python version either `3.11.x` or `3.12.x` (same as AssertionError)
 
-### 3. Local .env
+**Setting up a local env file**
 
 Create `.env` file in the project root and add following:
 
@@ -216,7 +216,7 @@ Common issues and solutions:
 
 * Issues related to dependencies: `rm -rf uv.lock`, `uv cache clean`, `uv venv`, and run `uv pip install -r requirements.txt -v`.
 
-* Issues related to the AI agents or RAG system: Check the `output.log` file for detailed error messages and stack traces.
+* Issues related to agents and other systems: Check `_logs` directory for detailed error messages and stack traces.
 
 * Issues related to `Python quit unexpectedly`: Check [this stackoverflow article](https://stackoverflow.com/questions/59888499/macos-catalina-python-quit-unexpectedly-error).
 

{versionhq-1.2.1.0 → versionhq-1.2.1.2}/docs/quickstart.md

@@ -84,33 +84,39 @@ This will return a `TaskOutput` object that stores response in plane text, JSON,
 
 ## Supervising
 
-   ```python
-   import versionhq as vhq
-
-   agent_a = vhq.Agent(role="agent a", goal="My amazing goals", llm="llm-of-your-choice")
-   agent_b = vhq.Agent(role="agent b", goal="My amazing goals", llm="llm-of-your-choice")
-
-   task_1 = vhq.Task(
-      description="Analyze the client's business model.",
-      response_fields=[vhq.ResponseField(title="test1", data_type=str, required=True),],
-      allow_delegation=True
-   )
-
-    task_2 = vhq.Task(
-      description="Define the cohort.",
-      response_fields=[vhq.ResponseField(title="test1", data_type=int, required=True),],
-      allow_delegation=False
-   )
-
-   team =vhq.Team(
-      members=[
-         vhq.Member(agent=agent_a, is_manager=False, task=task_1),
-         vhq.Member(agent=agent_b, is_manager=True, task=task_2),
-      ],
-   )
-   res = team.launch()
-   ```
+To create an agent network with one or more manager agents, designate members using the `is_manager` tag.
+
+```python
+import versionhq as vhq
+
+agent_a = vhq.Agent(role="agent a", goal="My amazing goals", llm="llm-of-your-choice")
+agent_b = vhq.Agent(role="agent b", goal="My amazing goals", llm="llm-of-your-choice")
+
+task_1 = vhq.Task(
+   description="Analyze the client's business model.",
+   response_fields=[vhq.ResponseField(title="test1", data_type=str, required=True),],
+   allow_delegation=True
+)
+
+task_2 = vhq.Task(
+   description="Define a cohort.",
+   response_fields=[vhq.ResponseField(title="test1", data_type=int, required=True),],
+   allow_delegation=False
+)
+
+network =vhq.AgentNetwork(
+   members=[
+      vhq.Member(agent=agent_a, is_manager=False, tasks=[task_1]),
+      vhq.Member(agent=agent_b, is_manager=True, tasks=[task_2]), # Agent B as a manager
+   ],
+)
+res = network.launch()
+
+assert isinstance(res, vhq.NetworkOutput)
+assert not [item for item in task_1.processed_agents if "vhq-Delegated-Agent" == item]
+assert [item for item in task_1.processed_agents if "agent b" == item]
+```
 
 This will return a list with dictionaries with keys defined in the `ResponseField` of each task.
 
-Tasks can be delegated to a team manager, peers in the team, or completely new agent.
+Tasks can be delegated to a manager, peers within the agent network, or a completely new agent.

{versionhq-1.2.1.0 → versionhq-1.2.1.2}/pyproject.toml

@@ -15,7 +15,7 @@ exclude = ["test*", "__pycache__", "*.egg-info"]
 
 [project]
 name = "versionhq"
-version = "1.2.1.0"
+version = "1.2.1.2"
 authors = [{ name = "Kuriko Iwai", email = "kuriko@versi0n.io" }]
 description = "An agentic orchestration framework for building agent networks that handle task automation."
 readme = "README.md"

{versionhq-1.2.1.0 → versionhq-1.2.1.2}/requirements.txt

@@ -19,3 +19,5 @@ chromadb>=0.6.3
 wheel>=0.45.1
 envoy>=0.0.3
 composio-core==0.7.0
+networkx>=3.4.2
+matplotlib>=3.10.0

{versionhq-1.2.1.0 → versionhq-1.2.1.2}/src/versionhq/__init__.py

@@ -8,6 +8,7 @@ from dotenv import load_dotenv
 load_dotenv(override=True)
 
 from versionhq.agent.model import Agent
+from versionhq.agent_network.model import AgentNetwork, NetworkOutput, Formation, Member, TaskHandlingProcess
 from versionhq.llm.model import LLM
 from versionhq.llm.llm_vars import LLM_CONTEXT_WINDOW_SIZES, PARAMS, PROVIDERS, MODELS
 from versionhq.clients.customer.model import Customer
@@ -16,10 +17,9 @@ from versionhq.clients.workflow.model import MessagingWorkflow, MessagingCompone
 from versionhq.knowledge.model import Knowledge, KnowledgeStorage
 from versionhq.knowledge.source import PDFKnowledgeSource, CSVKnowledgeSource, JSONKnowledgeSource, TextFileKnowledgeSource, ExcelKnowledgeSource, StringKnowledgeSource
 from versionhq.knowledge.source_docling import DoclingSource
-from versionhq.graph.model import TaskStatus, TaskGraph, Node, Edge, DependencyType
+from versionhq.task_graph.model import TaskStatus, TaskGraph, Node, Edge, DependencyType
 from versionhq.task.model import Task, TaskOutput, ResponseField, TaskExecutionType
 from versionhq.task.evaluate import Evaluation, EvaluationItem
-from versionhq.team.model import Team, TeamOutput, Formation, Member, TaskHandlingProcess
 from versionhq.tool.model import Tool, ToolSet
 from versionhq.tool.cache_handler import CacheHandler
 from versionhq.tool.tool_handler import ToolHandler
@@ -30,10 +30,16 @@ from versionhq.memory.model import ShortTermMemory,LongTermMemory, UserMemory, M
 from versionhq.task.formation import form_agent_network
 
 
-__version__ = "1.2.1.0"
+__version__ = "1.2.1.2"
 __all__ = [
     "Agent",
 
+    "AgentNetwork",
+    "NetworkOutput",
+    "Formation",
+    "Member",
+    "TaskHandlingProcess",
+
     "LLM",
     "LLM_CONTEXT_WINDOW_SIZES",
     "PARAMS",
@@ -70,12 +76,6 @@ __all__ = [
     "Evaluation",
     "EvaluationItem",
 
-    "Team",
-    "TeamOutput",
-    "Formation",
-    "Member",
-    "TaskHandlingProcess",
-
     "Tool",
     "ToolSet",
     "CacheHandler",