versionhq 1.2.2.0__py3-none-any.whl → 1.2.2.1__py3-none-any.whl

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/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:

versionhq/agent_network/model.py

@@ -255,44 +255,6 @@ class AgentNetwork(BaseModel):
                 return member.agent
 
 
-    # task execution
-    # def _process_async_tasks(self, futures: List[Tuple[Task, Future[TaskOutput], int]], was_replayed: bool = False) -> List[TaskOutput]:
-    #     """
-    #     When we have `Future` tasks, updated task outputs and task execution logs accordingly.
-    #     """
-
-    #     task_outputs: List[TaskOutput] = []
-
-    #     for future_task, future, task_index in futures:
-    #         task_output = future.result()
-    #         task_outputs.append(task_output)
-    #         future_task._store_execution_log(task_index, was_replayed)
-
-    #     return task_outputs
-
-
-    # def _calculate_usage_metrics(self) -> UsageMetrics:
-    #     """
-    #     Calculate and return the usage metrics that consumed by the agent network.
-    #     """
-    #     total_usage_metrics = UsageMetrics()
-
-    #     for member in self.members:
-    #         agent = member.agent
-    #         if hasattr(agent, "_token_process"):
-    #             token_sum = agent._token_process.get_summary()
-    #             total_usage_metrics.add_usage_metrics(token_sum)
-
-    #     if self.managers:
-    #         for manager in self.managers:
-    #             if hasattr(manager.agent, "_token_process"):
-    #                 token_sum = manager.agent._token_process.get_summary()
-    #                 total_usage_metrics.add_usage_metrics(token_sum)
-
-    #     self.usage_metrics = total_usage_metrics
-    #     return total_usage_metrics
-
-
     def _execute_tasks(self, tasks: List[Task], start_index: Optional[int] = None) -> Tuple[TaskOutput, TaskGraph]:
         """Executes tasks and returns TaskOutput object as concl or latest response in the network."""
         res, task_graph = None, None
@@ -303,7 +265,6 @@ class AgentNetwork(BaseModel):
             res = task.execute(agent=responsible_agent)
             return res, task_graph
 
-
         nodes = [
             Node(
                 task=task,
@@ -332,47 +293,6 @@ class AgentNetwork(BaseModel):
         else:
             res, _ = task_graph.activate()
 
-        # task_outputs: List[TaskOutput] = []
-        # lead_task_output: TaskOutput = None
-        # futures: List[Tuple[Task, Future[TaskOutput], int]] = []
-        # last_sync_output: Optional[TaskOutput] = None
-
-        # for task_index, task in enumerate(tasks):
-        #     if start_index is not None and task_index < start_index:
-        #         if task.output:
-        #             if task.execution_type == TaskExecutionType.ASYNC:
-        #                 task_outputs.append(task.output)
-        #             else:
-        #                 task_outputs = [task.output]
-        #                 last_sync_output = task.output
-        #         continue
-
-        #     responsible_agent = self._get_responsible_agent(task=task)
-
-        #     ## commented out - this will be handled by node objects
-        #     # if isinstance(task, ConditionalTask):
-        #     #     skipped_task_output = task._handle_conditional_task(task_outputs, futures, task_index, was_replayed)
-        #     #     if skipped_task_output:
-        #     #         continue
-
-        #     # self._log_task_start(task, responsible_agent)
-
-        #     context = create_raw_outputs(tasks=[task, ], task_outputs=([last_sync_output,] if last_sync_output else []))
-        #     res = task.execute(agent=responsible_agent, context=context)
-
-        #     if isinstance(res, Future):
-        #         futures.append((task, res, task_index))
-        #     else:
-        #         # if self.managers and responsible_agent in [manager.agent for manager in self.managers]:
-        #         #     lead_task_output = task_output
-
-        #         task_outputs.append(res)
-        #         task._store_log(task_index, was_replayed, self._inputs)
-
-
-        # if futures:
-        #     task_outputs = self._process_async_tasks(futures, was_replayed)
-
         if not res:
             Logger().log(level="error", message="Missing task outputs.", color="red")
             raise ValueError("Missing task outputs")
@@ -380,7 +300,6 @@ class AgentNetwork(BaseModel):
         return res, task_graph
 
 
-
     def launch(
             self, kwargs_pre: Optional[Dict[str, str]] = None, kwargs_post: Optional[Dict[str, Any]] = None, start_index: int = None
         ) -> Tuple[TaskOutput, TaskGraph]:
@@ -391,7 +310,7 @@ class AgentNetwork(BaseModel):
         self._assign_tasks()
 
         if kwargs_pre:
-            for func in self.pre_launch_callbacks: # signature check
+            for func in self.pre_launch_callbacks: #! REFINEME -  signature check
                 func(**kwargs_pre)
 
         for member in self.members:
@@ -420,12 +339,6 @@ class AgentNetwork(BaseModel):
                 case _:
                     pass
 
-        # metrics += [member.agent._token_process.get_summary() for member in self.members]
-
-        # self.usage_metrics = UsageMetrics()
-        # for metric in metrics:
-        #     self.usage_metrics.add_usage_metrics(metric)
-
         return result, tg
 
 

{versionhq-1.2.2.0.dist-info → versionhq-1.2.2.1.dist-info}/METADATA

@@ -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.dist-info → versionhq-1.2.2.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-versionhq/__init__.py,sha256=uh8uc3vSkOPeiz6MBNYdjJeCzH_KPMN-x5Xp5MWyOPo,2857
+versionhq/__init__.py,sha256=6i-QHTxLobpE8uqqoiPoPnfeTOqY4J8i3ptRaGfo_c0,2857
 versionhq/_utils/__init__.py,sha256=d-vYVcORZKG-kkLe_fzE8VbViDpAk9DDOKe2fVK25ew,178
 versionhq/_utils/i18n.py,sha256=TwA_PnYfDLA6VqlUDPuybdV9lgi3Frh_ASsb_X8jJo8,1483
 versionhq/_utils/logger.py,sha256=iHxGjm3BvUo5dHKLU88_pc0Z45wzSHOjyJGQkb7OADk,3255
@@ -13,8 +13,8 @@ versionhq/agent/rpm_controller.py,sha256=grezIxyBci_lDlwAlgWFRyR5KOocXeOhYkgN02d
 versionhq/agent/TEMPLATES/Backstory.py,sha256=IAhGnnt6VUMe3wO6IzeyZPDNu7XE7Uiu3VEXUreOcKs,532
 versionhq/agent/TEMPLATES/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/agent_network/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-versionhq/agent_network/formation.py,sha256=7iGw20Dj2sFLAho6yfrdmonAwFcxINBDGFr2RU-Qz3s,7505
-versionhq/agent_network/model.py,sha256=93VVOtON6_TEren4FnWfLtUGl_T6DLVA0BhZgY9yhIA,19059
+versionhq/agent_network/formation.py,sha256=CYKNUeKC392wW3leIDIAaGiKADSsumC_vTe0VOnNwRs,7901
+versionhq/agent_network/model.py,sha256=f2i26vyVagzq4g2lIxNivhab45i1dnK8HZV8mVO6Og8,15564
 versionhq/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/clients/customer/__init__.py,sha256=-YXh1FQfvpfLacK8SUC7bD7Wx_eIEi4yrkCC_cUasFg,217
@@ -60,8 +60,8 @@ versionhq/tool/composio_tool_vars.py,sha256=FvBuEXsOQUYnN7RTFxT20kAkiEYkxWKkiVtg
 versionhq/tool/decorator.py,sha256=C4ZM7Xi2gwtEMaSeRo-geo_g_MAkY77WkSLkAuY0AyI,1205
 versionhq/tool/model.py,sha256=PO4zNWBZcJhYVur381YL1dy6zqurio2jWjtbxOxZMGI,12194
 versionhq/tool/tool_handler.py,sha256=2m41K8qo5bGCCbwMFferEjT-XZ-mE9F0mDUOBkgivOI,1416
-versionhq-1.2.2.0.dist-info/LICENSE,sha256=cRoGGdM73IiDs6nDWKqPlgSv7aR4n-qBXYnJlCMHCeE,1082
-versionhq-1.2.2.0.dist-info/METADATA,sha256=dEEdYYA25P_WeaIrGwXDKdXmlWSgWcl-pMGEleVuaU0,22059
-versionhq-1.2.2.0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-versionhq-1.2.2.0.dist-info/top_level.txt,sha256=DClQwxDWqIUGeRJkA8vBlgeNsYZs4_nJWMonzFt5Wj0,10
-versionhq-1.2.2.0.dist-info/RECORD,,
+versionhq-1.2.2.1.dist-info/LICENSE,sha256=cRoGGdM73IiDs6nDWKqPlgSv7aR4n-qBXYnJlCMHCeE,1082
+versionhq-1.2.2.1.dist-info/METADATA,sha256=WIONV5Bz91sfCUeeiFQP4GU9TnIletGgOYW-f3mjbEo,22015
+versionhq-1.2.2.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+versionhq-1.2.2.1.dist-info/top_level.txt,sha256=DClQwxDWqIUGeRJkA8vBlgeNsYZs4_nJWMonzFt5Wj0,10
+versionhq-1.2.2.1.dist-info/RECORD,,