versionhq 1.2.2.0__tar.gz → 1.2.2.1__tar.gz

{versionhq-1.2.2.0 → versionhq-1.2.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.2.2.0
+Version: 1.2.2.1
 Summary: An agentic orchestration framework for building agent networks that handle task automation.
 Author-email: Kuriko Iwai <kuriko@versi0n.io>
 License: MIT License
@@ -84,7 +84,7 @@ Requires-Dist: pygraphviz>=1.14; extra == "pygraphviz"
 [![DL](https://img.shields.io/badge/Download-20K+-red)](https://clickpy.clickhouse.com/dashboard/versionhq)
 ![MIT license](https://img.shields.io/badge/License-MIT-green)
 [![Publisher](https://github.com/versionHQ/multi-agent-system/actions/workflows/publish.yml/badge.svg)](https://github.com/versionHQ/multi-agent-system/actions/workflows/publish.yml)
-![PyPI](https://img.shields.io/badge/PyPI-v1.2.1+-blue)
+![PyPI](https://img.shields.io/badge/PyPI-v1.2.2+-blue)
 ![python ver](https://img.shields.io/badge/Python-3.11/3.12-purple)
 ![pyenv ver](https://img.shields.io/badge/pyenv-2.5.0-orange)
 
@@ -255,14 +255,14 @@ agent.update(
    ```python
    import versionhq as vhq
 
-   network = vhq.form_agent_network(
+   network = work(
       task="YOUR AMAZING TASK OVERVIEW",
       expected_outcome="YOUR OUTCOME EXPECTATION",
    )
-   res, _ = network.launch()
+   res, tg = 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 agent network with multiple agents on `Formation` and return response in `TaskOutput` object and `TaskGraph` that connects multiple tasks as nodes.
 
 
 ### Executing tasks
@@ -295,7 +295,6 @@ res = task.execute(context="context to consider")
 assert isinstance(res, vhq.TaskOutput)
 ```
 
-
 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
@@ -338,11 +337,13 @@ network =vhq.AgentNetwork(
       vhq.Member(agent=agent_b, is_manager=True, tasks=[task_2]), # Agent B as a manager
    ],
 )
-res, _ = network.launch()
+res, tg = network.launch()
 
 assert isinstance(res, vhq.TaskOutput)
-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]
+assert agent_b.key in task_1.processed_agents
+assert agent_b.key in task_2.processed_agents
+
+assert isinstance(tg, vhq.TaskGraph)
 ```
 
 This will return a list with dictionaries with keys defined in the `ResponseField` of each task.

{versionhq-1.2.2.0 → versionhq-1.2.2.1}/README.md

@@ -3,7 +3,7 @@
 [![DL](https://img.shields.io/badge/Download-20K+-red)](https://clickpy.clickhouse.com/dashboard/versionhq)
 ![MIT license](https://img.shields.io/badge/License-MIT-green)
 [![Publisher](https://github.com/versionHQ/multi-agent-system/actions/workflows/publish.yml/badge.svg)](https://github.com/versionHQ/multi-agent-system/actions/workflows/publish.yml)
-![PyPI](https://img.shields.io/badge/PyPI-v1.2.1+-blue)
+![PyPI](https://img.shields.io/badge/PyPI-v1.2.2+-blue)
 ![python ver](https://img.shields.io/badge/Python-3.11/3.12-purple)
 ![pyenv ver](https://img.shields.io/badge/pyenv-2.5.0-orange)
 
@@ -174,14 +174,14 @@ agent.update(
    ```python
    import versionhq as vhq
 
-   network = vhq.form_agent_network(
+   network = work(
       task="YOUR AMAZING TASK OVERVIEW",
       expected_outcome="YOUR OUTCOME EXPECTATION",
    )
-   res, _ = network.launch()
+   res, tg = 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 agent network with multiple agents on `Formation` and return response in `TaskOutput` object and `TaskGraph` that connects multiple tasks as nodes.
 
 
 ### Executing tasks
@@ -214,7 +214,6 @@ res = task.execute(context="context to consider")
 assert isinstance(res, vhq.TaskOutput)
 ```
 
-
 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
@@ -257,11 +256,13 @@ network =vhq.AgentNetwork(
       vhq.Member(agent=agent_b, is_manager=True, tasks=[task_2]), # Agent B as a manager
    ],
 )
-res, _ = network.launch()
+res, tg = network.launch()
 
 assert isinstance(res, vhq.TaskOutput)
-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]
+assert agent_b.key in task_1.processed_agents
+assert agent_b.key in task_2.processed_agents
+
+assert isinstance(tg, vhq.TaskGraph)
 ```
 
 This will return a list with dictionaries with keys defined in the `ResponseField` of each task.

versionhq-1.2.2.1/docs/core/agent-network/config.md

@@ -0,0 +1,102 @@
+---
+tags:
+  - Agent Network
+---
+
+
+## Adding Members
+
+<class>`class` versionhq.agent_network.model.<bold>Member<bold></class>
+
+You can simply add an agent as a member using `members` field.
+
+```python
+import versionhq as vhq
+
+network = vhq.AgentNetwork(
+    members=[
+        vhq.Member(
+            agent=vhq.Agent(role="new member", goal="work in the network"),
+            is_manager=False  # explicitly mentioned. Setting `True` makes this member a manager of the network.
+        ),
+    ]
+)
+assert isinstance(network.members[0].agent, vhq.Agent)
+```
+
+Ref. <a href="/core/agent-network/ref/#member">`Member`</a> class.
+
+<hr />
+
+## Changing Formation
+
+The formation of network members will be automatically assigned based on the task goals, but you can explicitly define it by using `formation` field.
+
+
+```python
+import versionhq as vhq
+
+network = vhq.AgentNetwork(
+  members=[
+    vhq.Member(agent=vhq.Agent(role="member 1", goal="work in the network")),
+    vhq.Member(agent=vhq.Agent(role="member 2", goal="work in the network")),
+    vhq.Member(agent=vhq.Agent(role="member 3", goal="work in the network")),
+  ],
+  formation=vhq.Formation.SQUAD,
+)
+
+assert network.formation == vhq.Formation.SQUAD
+```
+
+Ref. Enum <a href="/core/agent-network/ref/#enum-formation">`Formation`</a>
+
+
+<hr >
+
+## Task Handling
+
+The class method `.launch()` will automatically decide the best task handling process and execute the tasks accordingly.
+
+
+```python
+import versionhq as vhq
+
+network = vhq.AgentNetwork(
+  members=[
+      vhq.Member(agent=vhq.Agent(role="member 1", goal="work in the network"), tasks=[vhq.Task(description="Run a demo 1")]),
+      vhq.Member(agent=vhq.Agent(role="member 2", goal="work in the network"), tasks=[vhq.Task(description="Run a demo")]),
+      vhq.Member(agent=vhq.Agent(role="member 3", goal="work in the network")),
+  ],
+)
+
+res, tg = network.launch()
+
+assert isinstance(res, vhq.TaskOutput)
+assert isinstance(tg, vhq.TaskGraph)
+
+```
+
+<hr >
+
+You can also specify the process using `process` field.
+
+```python
+import versionhq as vhq
+
+network = vhq.AgentNetwork(
+  members=[
+      vhq.Member(agent=vhq.Agent(role="member 1", goal="work in the network"), tasks=[vhq.Task(description="Run a demo 1")]),
+      vhq.Member(agent=vhq.Agent(role="member 2", goal="work in the network"), tasks=[vhq.Task(description="Run a demo 2")]),
+      vhq.Member(agent=vhq.Agent(role="member 3", goal="work in the network")),
+  ],
+  process=vhq.TaskHandlingProcess.CONSENSUAL,
+  consent_trigger=lambda x: True, # consent trigger event is a MUST for TaskHandlingProcess.CONSENSUAL
+)
+
+res, tg = network.launch()
+
+assert isinstance(res, vhq.TaskOutput)
+assert isinstance(tg, vhq.TaskGraph)
+```
+
+Ref. Enum <a href="/core/agent-network/ref/#enum-taskhandlingprocess">`TaskHandlingProcess`</a>

versionhq-1.2.2.1/docs/core/agent-network/form.md

@@ -0,0 +1,117 @@
+---
+tags:
+  - Agent Network
+---
+
+
+You can generate an `AgentNetwork` by using `form_agent_network` method with a concise `task` description and `expected_outcome` args.
+
+```python
+import versionhq as vhq
+
+network = vhq.form_agent_network(
+    task="Find the best trip destination this summer.",
+    expected_outcome="a list of destinations and why it's suitable",
+    context="planning a suprise trip for my friend", # optional
+)
+
+assert isinstance(network, vhq.AgentNetwork)
+assert network.members # auto-generated agents as network members
+assert network.tasks # auto-defined sub-tasks to achieve the main task goal
+```
+
+
+<hr>
+
+**Strucured Output**
+
+To generate structured output, you can add a JSON dict or Pydantic class as `expected_outcome` args instead of plane text.
+
+```python
+import versionhq as vhq
+from pydantic import BaseModel
+
+class Outcome(BaseModel):
+    destinations: list[str]
+    why_suitable: list[str]
+
+
+network = vhq.form_agent_network(
+    task="Find the best trip destination this summer.",
+    expected_outcome=Outcome,
+    context="planning a suprise trip for my friend", # optional
+)
+
+assert isinstance(network, vhq.AgentNetwork)
+assert network.members
+assert network.tasks
+```
+
+<hr >
+
+**Agents**
+
+You can use `agents` args to add existing agents to the network.
+
+```python
+import versionhq as vhq
+from pydantic import BaseModel
+
+my_agent = vhq.Agent(
+    role="Travel Agent",
+    goal="select best trip destination",
+    knowledge_sources=[".....","url1",]
+)
+
+class Outcome(BaseModel):
+    destinations: list[str]
+    why_suitable: list[str]
+
+network = vhq.form_agent_network(
+    task="Find the best trip destination this summer.",
+    expected_outcome=Outcome,
+    context="planning a suprise trip for my friend",
+    agents=[my_agent,]
+)
+
+assert isinstance(network, vhq.AgentNetwork)
+assert [member for member in network.members if member.agent == my_agent]
+assert network.tasks
+```
+
+<hr>
+
+**Formation**
+
+Similar to `agents`, you can define `formation` args to specify the network formation:
+
+
+```python
+import versionhq as vhq
+from pydantic import BaseModel
+
+my_agent = vhq.Agent(
+    role="Travel Agent",
+    goal="select best trip destination",
+    knowledge_sources=[".....","url1",]
+)
+
+class Outcome(BaseModel):
+    destinations: list[str]
+    why_suitable: list[str]
+
+network = vhq.form_agent_network(
+    task="Find the best trip destination this summer.",
+    expected_outcome=Outcome,
+    context="planning a suprise trip for my friend",
+    agents=[my_agent,],
+    formation=vhq.Formation.SUPERVISING
+)
+
+assert isinstance(network, vhq.AgentNetwork)
+assert [member for member in network.members if member.agent == my_agent]
+assert network.tasks
+assert network.formation == vhq.Formation.SUPERVISING
+```
+
+Ref. Enum <a href="core/agent-network/ref/#enum-formation">Formation</a>

{versionhq-1.2.2.0 → versionhq-1.2.2.1}/docs/core/agent-network/index.md

@@ -23,7 +23,7 @@ You can specify a desired formation or allow the agents to determine it autonomo
 
 By default, lead agents will determine the best network formation autonomously based on the given task and its goal.
 
-Calling `.launch()` method can start executing tasks by the network, then generate a response in text and JSON formats stored in the `TaskOutput` object.
+Calling `.launch()` method can start executing tasks and generate a tuple of response as a `TaskOutput` object and `TaskGraph` object.
 
 ```python
 import versionhq as vhq
@@ -33,11 +33,12 @@ network = vhq.form_agent_network(
   expected_outcome='media mix, key messages, and CTA targets.'
 )
 
-res, _ = network.launch()
+res, tg = network.launch()
 
 assert isinstance(res, vhq.TaskOutput)
+assert isinstance(tg, vhq.TaskGraph)
 ```
 
-Ref. <a href="/core/task-output">TaskOutput</a> class
+Ref. <a href="/core/task-output">TaskOutput</a> / <a href="/core/task-graph">TaskGraph </a> class.
 
 Visit <a href="https://versi0n.io">Playground</a>.

versionhq-1.2.2.1/docs/core/agent-network/ref.md

@@ -0,0 +1,141 @@
+---
+tags:
+  - Agent Network
+---
+
+
+## Class `AgentNetwork`
+
+### Variable
+
+| <div style="width:200px">**Variable**</div> | **Data Type** | **Default** | **Nullable** | **Description** |
+| :---               | :---  | :--- | :--- | :--- |
+| **`id`**   | UUID4 | uuid.uuid4() | False | Stores auto-generated ID as identifier. Not editable. |
+| **`name`**   | str | None | True | Stores a name of the network. |
+| **`members`**   | List[`Member`] | list() | False | Stores a list of `Member` objects. |
+| **`formation`**  | `Formation`| None | True | Stores `Formation` enum. |
+| **`should_reform`**  | bool | False | - | Whether to reform the network during the activation. |
+| **`network_tasks`**  | List[`Task`] | list() | - | Stores a list of `Task` objects not assigned to any network members. |
+| **`prompt_file`** | str | None | True | Stores absolute file path to the prompt file with JSON formatted prompt |
+| **`process`** | `TaskHandlingProcess` | TaskHandlingProcess.SEQUENTIAL | - | Enum of the task handling process. |
+| **`consent_trigger`** | Callable[..., Any] | None | True | Stores a trigger event (func) for consentual processing. |
+| **`pre_launch_callbacks`** | List[Callable[..., Any]] | list() | - | Stores callbacks to run before the network launch. |
+| **`post_launch_callbacks`** | List[Callable[..., Any]] | list() | - | Stores callbacks to run after the network launch. |
+| **`step_callbacks`** | Callable[..., Any] | None | True | Stores callbacks to run at every step of each member agent takes during the activation. |
+| **`cache`** | bool | True | - | Whether to store cache. |
+| **`execution_logs`** |  List[Dict[str, Any]] | list() | - | Stores a list of execution logs of all the tasks in the network. |
+
+
+### Class Methods
+
+| <div style="width:120px">**Method**</div> | **Params** | **Returns** | **Description** |
+| :---               | :---  | :--- | :--- |
+| **`launch`** | kwargs_pre: Optional[Dict[str, str]] = None <br> kwargs_post: Optional[Dict[str, Any]] = None <br> start_index: int = None |  Tuple[TaskOutput, TaskGraph]: | Core method to launch the network and execute tasks |
+
+
+### Properties
+
+| <div style="width:120px">**Property**</div> | **Returns** | **Description** |
+| :---               | :---  | :--- |
+| **`key`** | str | Unique identifier of the netowrk. |
+| **`managers`** | List[`Member`] | A list of manager members. |
+| **`manager_tasks`** | List[`Task`] | A list of tasks handled by managers. |
+| **`tasks`** | List[`Task`] | All the tasks in the network.|
+| **`unassigned_member_tasks`** | List[Task] | Unassigned member-level tasks. |
+
+
+<hr>
+
+## Class `Member`
+
+### Variable
+
+| <div style="width:200px">**Variable**</div> | **Data Type** | **Default** | **Nullable** | **Description** |
+| :---               | :---  | :--- | :--- | :--- |
+| **`agent`**   | `Agent` | None | True | Agent as a member |
+| **`is_manager`**   | bool | False | - | Whether the member is a manager. |
+| **`can_share_knowledge`**   | bool | True | - | Whether the member can share its knowledge among the other members in the network. |
+| **`can_share_memory`**   | bool | True | - | Whether the member can share its memories among the other members in the network. |
+| **`tasks`**  | List[`Task`]| list() | - | Assinged tasks. |
+
+
+### Properties
+
+| <div style="width:120px">**Property**</div> | **Returns** | **Description** |
+| :---               | :---  | :--- |
+| **`is_idling`** | bool | Whether it has unprocessed assgined task/s |
+
+
+
+<hr>
+
+## Class `Agent`
+
+### Variables
+
+| <div style="width:200px">**Variable**</div> | **Data Type** | **Default** | **Nullable** | **Description** |
+| :---               | :---  | :--- | :--- | :--- |
+| **`id`**   | UUID4 | uuid.uuid4() | False | Stores auto-generated ID as identifier. Not editable. |
+| **`role`**   | str | None | False | Stores a role of the agent. |
+| **`goal`**   | str | None | False | Stores a goal of the agent. |
+| **`backstory`**  | str | None | True | Stores backstory of the agent. Utilized as system prompt. |
+| **`tools`**  | List[InstanceOf[`Tool` \| `ToolSet`] \| Type[`Tool`]] | None | True | Stores tools to be used when executing a task. |
+| **`knowledge_sources`**  | List[`BaseKnowledgeSource` \| Any] | None | True | Stores knowledge sources in text, file path, or url. |
+| **`embedder_config`** | Dict[str, Any] | None | True | Stores embedding configuration for storing knowledge sources. |
+| **`with_memory`** | bool | False | - | Whether to store tasks and results in memory. |
+| **`memory_config`** |  Dict[str, Any] | None | True | Stores configuration of the memory. |
+| **`short_term_memory`** | InstanceOf[`ShortTermMemory`] | None | True | Stores `ShortTermMemory` object. |
+| **`long_term_memory`** | InstanceOf[`LongTermMemory`] | None | True | Stores `LongTermMemory` object. |
+| **`user_memory`** | InstanceOf[`UserMemory`] | None | True | Stores `UserMemory` object. |
+| **`use_developer_prompt`** |  bool | True | - | Whether to use the system (developer) prompt when calling the model. |
+| **`developer_promt_template`** | str | None | True | File path to the prompt template. |
+| **`user_promt_template`** | str | None | True | File path to the prompt template. |
+| **`networks`** | List[Any] | list() | True | Stores a list of agent networks that the agent belongs to. |
+| **`allow_delegation`** | bool | False | - | Whether the agent can delegate assinged tasks to another agent. |
+| **`max_retry_limit`** | int | 2 | - | Maximum number of retries when the task execution failed. |
+| **`maxit`** | int | 25 | - | Maximum number of total optimization loops conducted when an error occues during the task execution. |
+| **`callbacks`** | List[Callabale] | None | True | Stores a list of callback functions that must be called after every task execution completed.|
+| **`llm`** | str \| InstanceOf[`LLM`] \| Dict[str, Any] | None | False | Stores the main model that the agent runs on. |
+| **`func_calling_llm`** | str \| InstanceOf[`LLM`] \| Dict[str, Any] | None | False | Stores the function calling model that the agent runs on. |
+| **`respect_context_window`** | bool | True | - | Whether to follow the main model's maximum context window size. |
+| **`max_execution_time`** | int | None | True | Stores maximum execution time in seconds. |
+| **`max_rpm`** | int | None | True | Stores maximum number of requests per minute. |
+| **`llm_config`** | Dict[str, Any] | None | True | Stores configuration of `LLM` object. |
+| **`config`** | Dict[str, Any] | None | True | Stores model config. |
+
+
+### Class Methods
+
+| <div style="width:120px">**Method**</div> | **Params** | **Returns** | **Description** |
+| :---               | :---  | :--- | :--- |
+| **`update`** | **kwargs: Any  | Self | Updates agents with given kwargs. Invalid keys will be ignored. |
+| **`start`** | context: Any = None <br> tool_res_as_final: bool = False  | `TaskOutput` \| None  | Starts to operate the agent. |
+| **`execute_task`** | task: [Task] <br> context: Any = None <br> task_tools: Optional[List[Tool \| ToolSet]] = list() | str | Returns response from the model in plane text format. |
+
+
+### Properties
+
+| <div style="width:120px">**Property**</div> | **Returns** | **Description** |
+| :---               | :---  | :--- |
+| **`key`** | str | Unique identifier of the agent using its ID and sanitized role. |
+
+
+## ENUM `Formation`
+
+```python
+class Formation(str, Enum):
+    SOLO = 1
+    SUPERVISING = 2
+    SQUAD = 3
+    RANDOM = 4
+    HYBRID = 10
+```
+
+## ENUM `TaskHandlingProcess`
+
+```python
+class TaskHandlingProcess(str, Enum):
+    HIERARCHY = 1
+    SEQUENTIAL = 2
+    CONSENSUAL = 3
+```

{versionhq-1.2.2.0 → versionhq-1.2.2.1}/docs/index.md

@@ -3,7 +3,7 @@
 [![DL](https://img.shields.io/badge/Download-20K+-red)](https://clickpy.clickhouse.com/dashboard/versionhq)
 ![MIT license](https://img.shields.io/badge/License-MIT-green)
 [![Publisher](https://github.com/versionHQ/multi-agent-system/actions/workflows/publish.yml/badge.svg)](https://github.com/versionHQ/multi-agent-system/actions/workflows/publish.yml)
-![PyPI](https://img.shields.io/badge/PyPI-v1.2.1+-blue)
+![PyPI](https://img.shields.io/badge/PyPI-v1.2.2+-blue)
 ![python ver](https://img.shields.io/badge/Python-3.11/3.12-purple)
 ![pyenv ver](https://img.shields.io/badge/pyenv-2.5.0-orange)
 

{versionhq-1.2.2.0 → versionhq-1.2.2.1}/mkdocs.yml

@@ -112,16 +112,15 @@ nav:
   - Quick Start: 'quickstart.md'
   - Agent Network:
     - 'core/agent-network/index.md'
-    - Configuration: 'core/agent-network/config.md'
-    - Task Handling: 'core/agent-network/task-handling.md'
-    - Reference: 'core/agent-network/ref.md'
+    - Generating: 'core/agent-network/form.md'
+    - Configuring: 'core/agent-network/config.md'
     - Agent:
       - 'core/agent/index.md'
       - Configuration: 'core/agent/config.md'
       - Task Handling: 'core/agent/task-handling.md'
-      - Reference: 'core/agent/ref.md'
     - LLM:
       - 'core/llm/index.md'
+    - Reference: 'core/agent-network/ref.md'
   - Task Graph:
     - 'core/task-graph/index.md'
     - Task:

{versionhq-1.2.2.0 → versionhq-1.2.2.1}/pyproject.toml

@@ -15,7 +15,7 @@ exclude = ["test*", "__pycache__", "*.egg-info"]
 
 [project]
 name = "versionhq"
-version = "1.2.2.0"
+version = "1.2.2.1"
 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.2.0 → versionhq-1.2.2.1}/src/versionhq/__init__.py

@@ -31,7 +31,7 @@ from versionhq.agent_network.formation import form_agent_network
 from versionhq.task_graph.draft import workflow
 
 
-__version__ = "1.2.2.0"
+__version__ = "1.2.2.1"
 __all__ = [
     "Agent",
 

{versionhq-1.2.2.0 → versionhq-1.2.2.1}/src/versionhq/agent_network/formation.py

@@ -61,7 +61,7 @@ def form_agent_network(
     # try:
     prompt_formation = formation.name if formation and isinstance(formation, Formation) else f"Select the best formation to effectively execute the tasks from the given Enum sets: {str(Formation.__dict__)}."
 
-    prompt_expected_outcome = expected_outcome if isinstance(expected_outcome, str) else expected_outcome.model_dump_json()
+    prompt_expected_outcome = expected_outcome if isinstance(expected_outcome, str) else str(expected_outcome.model_dump()) if type(expected_outcome) == BaseModel else ""
 
     class Outcome(BaseModel):
         formation: Enum
@@ -76,7 +76,7 @@ def form_agent_network(
     )
 
     if agents:
-        vhq_task.description += "Consider adding following agents in the formation: " + ", ".join([agent.role for agent in agents if isinstance(agent, Agent)])
+        vhq_task.description += "You MUST add the following agents' roles in the network formation: " + ", ".join([agent.role for agent in agents if isinstance(agent, Agent)])
 
     res = vhq_task.execute(agent=vhq_formation_planner, context=context)
 
@@ -88,7 +88,17 @@ def form_agent_network(
     members = []
     leader = str(res.pydantic.leader_agent) if res.pydantic else str(res.json_dict["leader_agent"])
 
-    created_agents = [Agent(role=str(item), goal=str(item)) for item in res.pydantic.agent_roles]
+    agent_roles = res.pydantic.agent_roles if res.pydantic else res.json_dict["agent_roles"]
+    created_agents = [Agent(role=str(item), goal=str(item)) for item in agent_roles]
+
+    if agents:
+        for i, agent in enumerate(created_agents):
+            matches = [item for item in agents if item.role == agent.role]
+            if matches:
+                created_agents[i] = matches[0]
+            else:
+                pass
+
     created_tasks = []
 
     if res.pydantic: