versionhq 1.2.2.0__py3-none-any.whl → 1.2.2.2__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.2"
 __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
@@ -301,9 +263,10 @@ class AgentNetwork(BaseModel):
             task = self.tasks[0]
             responsible_agent = self._get_responsible_agent(task=task)
             res = task.execute(agent=responsible_agent)
+            node = Node(task=task)
+            task_graph = TaskGraph(nodes={ node.identifier: node, }, concl=res if res else None)
             return res, task_graph
 
-
         nodes = [
             Node(
                 task=task,
@@ -332,47 +295,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 +302,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 +312,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 +341,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/knowledge/source_docling.py

@@ -9,10 +9,10 @@ try:
     from docling_core.transforms.chunker.hierarchical_chunker import HierarchicalChunker
     from docling_core.types.doc.document import DoclingDocument
     DOCLING_AVAILABLE = True
-except ImportError:
-    import envoy
-    envoy.run("uv add docling --optional docling")
-    DOCLING_AVAILABLE = True
+# except ImportError:
+    # import envoy
+    # envoy.run("uv add docling --optional docling")
+    # DOCLING_AVAILABLE = True
 except:
     DOCLING_AVAILABLE = False
 

versionhq/memory/contextual_memory.py

@@ -23,25 +23,25 @@ class ContextualMemory:
         self.um = um
 
 
-    def build_context_for_task(self, query: str = None) -> str:
-        """
-        Automatically builds a minimal, highly relevant set of contextual information for a given task.
-        """
+    def _sanitize_query(self, query: str = None) -> str:
         if not query:
             return ""
 
-        context = []
-        context.append(self._fetch_stm_context(query))
-        if self.memory_provider == "mem0":
-            context.append(self._fetch_user_context(query))
-        return "\n".join(filter(None, context))
+        return query.replace("{", "").replace("}", "")
 
 
-    def _fetch_stm_context(self, query) -> str:
+    def _fetch_stm_context(self, query: str = None) -> str:
         """
-        Fetches recent relevant insights from STM related to the task's description and expected_output, formatted as bullet points.
+        Fetches recent relevant insights from STM
         """
+        if not query:
+            return ""
+
+
         stm_results = self.stm.search(query)
+        if not stm_results:
+            return ""
+
         formatted_results = "\n".join(
             [
                 f"- {result['memory'] if self.memory_provider == 'mem0' else result['context']}"
@@ -51,13 +51,16 @@ class ContextualMemory:
         return f"Recent Insights:\n{formatted_results}" if stm_results else ""
 
 
-    def _fetch_ltm_context(self, task) -> Optional[str]:
+    def _fetch_ltm_context(self, query: str = None) -> Optional[str]:
         """
         Fetches historical data or insights from LTM that are relevant to the task's description and expected_output, formatted as bullet points.
         """
-        ltm_results = self.ltm.search(task, latest_n=2)
+        if not query:
+            return ""
+
+        ltm_results = self.ltm.search(query, latest_n=2)
         if not ltm_results:
-            return None
+            return ""
 
         formatted_results = [suggestion for result in ltm_results for suggestion in result["metadata"]["suggestions"]]
         formatted_results = list(dict.fromkeys(formatted_results))
@@ -65,10 +68,12 @@ class ContextualMemory:
         return f"Historical Data:\n{formatted_results}" if ltm_results else ""
 
 
-    def _fetch_user_context(self, query: str) -> str:
+    def _fetch_user_context(self, query: str = None) -> str:
         """
         Fetches and formats relevant user information from User Memory.
         """
+        if not query:
+            return ""
 
         user_memories = self.um.search(query)
         if not user_memories:
@@ -78,6 +83,23 @@ class ContextualMemory:
         return f"User memories/preferences:\n{formatted_memories}"
 
 
+    def build_context_for_task(self, query: str = None) -> str:
+        """
+        Automatically builds a minimal, highly relevant set of contextual information for a given task.
+        """
+        if not query:
+            return ""
+
+        query = self._sanitize_query(query=query)
+
+        context = []
+        context.append(self._fetch_stm_context(query))
+        context.append(self._fetch_ltm_context(query))
+        if self.memory_provider == "mem0":
+            context.append(self._fetch_user_context(query))
+        return "\n".join(filter(None, context))
+
+
     # def _fetch_entity_context(self, query) -> str:
     #     """
     #     Fetches relevant entity information from Entity Memory related to the task's description and expected_output,

versionhq/storage/ltm_sqlite_storage.py

@@ -97,7 +97,7 @@ class LTMSQLiteStorage:
                     ]
 
         except sqlite3.Error as e:
-            self._logger.log(level="error", message=f"MEMORY ERROR: An error occurred while querying LTM: {e}",color="red")
+            self._logger.log(level="error", message=f"MEMORY ERROR: An error occurred while querying LTM: {str(e)}",color="red")
             return None
 
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.2.2.0
+Version: 1.2.2.2
 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.2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-versionhq/__init__.py,sha256=uh8uc3vSkOPeiz6MBNYdjJeCzH_KPMN-x5Xp5MWyOPo,2857
+versionhq/__init__.py,sha256=vGvxfqX6PuindBNVOX_OoewD27MqktESYpdzvLvkX_U,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=lr63cmH7ecQrfVAtnN44mtijrnOsfKe-8xOVNBXv-3Q,15696
 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
@@ -28,17 +28,17 @@ versionhq/knowledge/_utils.py,sha256=YWRF8U533cfZes_gZqUvdj-K24MD2ri1R0gjc_aPYyc
 versionhq/knowledge/embedding.py,sha256=KfHc__1THxb5jrg1EMrF-v944RDuIr2hE0l-MtM3Bp0,6826
 versionhq/knowledge/model.py,sha256=ixH8n5kLtJEp1nPAFYA0piYm-n0nnFDtWFp0r9YEVAs,1787
 versionhq/knowledge/source.py,sha256=-hEUPtJUHHMx4rUKtiHl19J8xAMw-WVBw34zwa2jZ08,13630
-versionhq/knowledge/source_docling.py,sha256=mg7bgvKePHn2LlA_XzSFCbS0zOo9xfu_aNOf5cEo6c4,5421
+versionhq/knowledge/source_docling.py,sha256=OUdWUZ6CCaddvmOKNYpYVzXCzqy9kwuTqsl0supT6GI,5429
 versionhq/knowledge/storage.py,sha256=7oxCg3W9mFjYH1YmuH9kFtTbNxquzYFjuUjd_TlsB9E,8170
 versionhq/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/llm/llm_vars.py,sha256=wjQK20cKvph6Vq1v71o4d16zBGcHlwq0bzOT_zWno7w,7041
 versionhq/llm/model.py,sha256=HIBmf8FYV6-cDbZK1ZBu6z3dmF0ZUbKbCelfwxMlgyY,17177
 versionhq/memory/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-versionhq/memory/contextual_memory.py,sha256=WeDujcEp4oud30OusXSPPNrMEQP-vGrt1mcfYopQruU,3483
+versionhq/memory/contextual_memory.py,sha256=QEMVvHuEXxY7M6-12S8HhyFKf108KfX8Zzt7paPW048,3882
 versionhq/memory/model.py,sha256=VQR1229t7GQPMItlGAHLtJrb6LrZfSoRA1DRW4z0SOU,8234
 versionhq/storage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/storage/base.py,sha256=p-Jas0fXQan_qotnRD6seQxrT2lj-uw9-SmHQhdppcs,355
-versionhq/storage/ltm_sqlite_storage.py,sha256=wdUiuwHfJocdk0UGqyrdU4S5Nae1rgsoRNu3LWmGFcI,3951
+versionhq/storage/ltm_sqlite_storage.py,sha256=K0Assani1QfFuq5FMxkcMmRlmLUOvw-RNkKyMV70U3g,3956
 versionhq/storage/mem0_storage.py,sha256=ZY8MELBWaINRv9YuRW5MxH7dj2cII-L0i3xSD6o1-2M,3781
 versionhq/storage/rag_storage.py,sha256=bS2eE874obarYl-4hT6ZWYWTRsqtfuGpKgKzERmM6Uo,7433
 versionhq/storage/task_output_storage.py,sha256=M8vInLJ5idGAq17w1juHKXtyPyF-B-rK_P8UcqD-Px8,5357
@@ -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.2.dist-info/LICENSE,sha256=cRoGGdM73IiDs6nDWKqPlgSv7aR4n-qBXYnJlCMHCeE,1082
+versionhq-1.2.2.2.dist-info/METADATA,sha256=XZ5QDEnQP1G4U-VFHBsGKBPvVc4vqZMafEp6rZvj89s,22015
+versionhq-1.2.2.2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+versionhq-1.2.2.2.dist-info/top_level.txt,sha256=DClQwxDWqIUGeRJkA8vBlgeNsYZs4_nJWMonzFt5Wj0,10
+versionhq-1.2.2.2.dist-info/RECORD,,