versionhq 1.2.2.1__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.1"
+__version__ = "1.2.2.2"
 __all__ = [
     "Agent",
 

versionhq/agent_network/model.py

@@ -263,6 +263,8 @@ 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 = [

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.1.dist-info → versionhq-1.2.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.2.2.1
+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

{versionhq-1.2.2.1.dist-info → versionhq-1.2.2.2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-versionhq/__init__.py,sha256=6i-QHTxLobpE8uqqoiPoPnfeTOqY4J8i3ptRaGfo_c0,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
@@ -14,7 +14,7 @@ versionhq/agent/TEMPLATES/Backstory.py,sha256=IAhGnnt6VUMe3wO6IzeyZPDNu7XE7Uiu3V
 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=CYKNUeKC392wW3leIDIAaGiKADSsumC_vTe0VOnNwRs,7901
-versionhq/agent_network/model.py,sha256=f2i26vyVagzq4g2lIxNivhab45i1dnK8HZV8mVO6Og8,15564
+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.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,,
+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,,