versionhq 1.2.2.1__tar.gz → 1.2.2.2__tar.gz

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

@@ -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 → versionhq-1.2.2.2}/pyproject.toml

@@ -15,7 +15,7 @@ exclude = ["test*", "__pycache__", "*.egg-info"]
 
 [project]
 name = "versionhq"
-version = "1.2.2.1"
+version = "1.2.2.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.2.1 → versionhq-1.2.2.2}/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.1"
+__version__ = "1.2.2.2"
 __all__ = [
     "Agent",
 

{versionhq-1.2.2.1 → versionhq-1.2.2.2}/src/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-1.2.2.1 → versionhq-1.2.2.2}/src/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-1.2.2.1 → versionhq-1.2.2.2}/src/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-1.2.2.1 → versionhq-1.2.2.2}/src/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 → versionhq-1.2.2.2}/src/versionhq.egg-info/PKG-INFO

@@ -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 → versionhq-1.2.2.2}/tests/usecase_test.py

@@ -55,3 +55,18 @@ def test_demo_agent_customization():
     assert agent.llm.temperature == 1 and agent.llm.top_p == 0.1 and agent.llm.n == 1 and agent.llm.stop == "test"
     assert agent.knowledge_sources == ['https://business.linkedin.com', f'{current_path}/demo.csv',]
     assert agent.with_memory == True
+
+
+def test_solo_tg_eval():
+    import versionhq as vhq
+
+    network = vhq.form_agent_network(
+        task="test",
+        expected_outcome="test"
+    )
+
+    res, tg = network.launch()
+    eval = tg.evaluate(
+        eval_criteria=["cost", "", "{criteria_3}"]
+    )
+    assert isinstance(eval, vhq.Evaluation)