versionhq 1.1.7.9__tar.gz → 1.1.8__tar.gz

{versionhq-1.1.7.9 → versionhq-1.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: versionhq
-Version: 1.1.7.9
+Version: 1.1.8
 Summary: LLM orchestration frameworks for model-agnostic AI agents that handle complex outbound workflows
 Author-email: Kuriko Iwai <kuriko@versi0n.io>
 License: MIT License

{versionhq-1.1.7.9 → versionhq-1.1.8}/pyproject.toml

@@ -15,7 +15,7 @@ exclude = ["test*", "__pycache__"]
 
 [project]
 name = "versionhq"
-version = "1.1.7.9"
+version = "1.1.8"
 authors = [{ name = "Kuriko Iwai", email = "kuriko@versi0n.io" }]
 description = "LLM orchestration frameworks for model-agnostic AI agents that handle complex outbound workflows"
 readme = "README.md"

{versionhq-1.1.7.9 → versionhq-1.1.8}/src/versionhq/__init__.py

@@ -17,7 +17,7 @@ from versionhq.team.model import Team, TeamOutput
 from versionhq.tool.model import Tool
 
 
-__version__ = "1.1.7.9"
+__version__ = "1.1.8"
 __all__ = [
     "Agent",
     "Customer",

{versionhq-1.1.7.9 → versionhq-1.1.8}/src/versionhq/storage/task_output_storage.py

@@ -11,11 +11,9 @@ from versionhq._utils.logger import Logger
 
 load_dotenv(override=True)
 
-
 def fetch_db_storage_path():
-    project_directory_name = os.environ.get("STORAGE_DIR", Path.cwd().name)
-    app_author = "versionhq"
-    data_dir = Path(appdirs.user_data_dir(project_directory_name, app_author))
+    directory_name = Path.cwd().name
+    data_dir = Path(appdirs.user_data_dir(appname=directory_name, appauthor="Version IO Sdn Bhd.", version=None, roaming=False))
     data_dir.mkdir(parents=True, exist_ok=True)
     return data_dir
 
@@ -39,25 +37,25 @@ class TaskOutputSQLiteStorage:
         Initializes the SQLite database and creates LTM table.
         """
 
-        try:
-            with sqlite3.connect(self.db_path) as conn:
-                cursor = conn.cursor()
-                cursor.execute(
-                    """
-                    CREATE TABLE IF NOT EXISTS task_outputs (
-                        task_id TEXT PRIMARY KEY,
-                        output JSON,
-                        task_index INTEGER,
-                        inputs JSON,
-                        was_replayed BOOLEAN,
-                        timestamp DATETIME DEFAULT CURRENT_TIMESTAMP
-                    )
+        # try:
+        with sqlite3.connect(self.db_path) as conn:
+            cursor = conn.cursor()
+            cursor.execute(
                 """
+                CREATE TABLE IF NOT EXISTS task_outputs (
+                    task_id TEXT PRIMARY KEY,
+                    output JSON,
+                    task_index INTEGER,
+                    inputs JSON,
+                    was_replayed BOOLEAN,
+                    timestamp DATETIME DEFAULT CURRENT_TIMESTAMP
                 )
-                conn.commit()
+            """
+            )
+            conn.commit()
 
-        except sqlite3.Error as e:
-            self._logger.log(level="error", message=f"DATABASE INITIALIZATION ERROR: {e}", color="red")
+        # except sqlite3.Error as e:
+        #     self._logger.log(level="error", message=f"DATABASE INITIALIZATION ERROR: {e}", color="red")
 
 
     def add(self, task, output: Dict[str, Any], task_index: int, was_replayed: bool = False, inputs: Dict[str, Any] = {}):
@@ -81,13 +79,12 @@ class TaskOutputSQLiteStorage:
         try:
             with sqlite3.connect(self.db_path) as conn:
                 cursor = conn.cursor()
-
                 fields, values = [], []
                 for k, v in kwargs.items():
                     fields.append(f"{k} = ?")
                     values.append(json.dumps(v) if isinstance(v, dict) else v)
 
-                query = f"UPDATE latest_kickoff_task_outputs SET {', '.join(fields)} WHERE task_index = ?"  # nosec
+                query = f"UPDATE latest_kickoff_task_outputs SET {', '.join(fields)} WHERE task_index = ?"
                 values.append(task_index)
                 cursor.execute(query, tuple(values))
                 conn.commit()

{versionhq-1.1.7.9 → versionhq-1.1.8}/src/versionhq/task/model.py

@@ -206,10 +206,10 @@ Your outputs MUST adhere to the following format and should NOT include any irre
     @property
     def summary(self) -> str:
         return f"""
-        Task: {self.id} - {self.description}
-        "task_description": {self.description}
-        "task_expected_output": {self.output_prompt}
-        "task_tools": {", ".join([tool_called.tool.name for tool_called in self.tools_called])}
+        Task ID: {str(self.id)}
+        "Description": {self.description}
+        "Prompt": {self.output_prompt}
+        "Tools": {", ".join([tool_called.tool.name for tool_called in self.tools_called])}
         """
 
 
@@ -382,11 +382,7 @@ Your outputs MUST adhere to the following format and should NOT include any irre
         """
 
         future: Future[TaskOutput] = Future()
-        threading.Thread(
-            daemon=True,
-            target=self._execute_task_async,
-            args=(agent, context, tools, future),
-        ).start()
+        threading.Thread(daemon=True, target=self._execute_task_async, args=(agent, context, tools, future)).start()
         return future
 
 

{versionhq-1.1.7.9 → versionhq-1.1.8}/src/versionhq/team/model.py

@@ -123,7 +123,7 @@ class Team(BaseModel):
     __hash__ = object.__hash__
     _execution_span: Any = PrivateAttr()
     _logger: Logger = PrivateAttr()
-    # _inputs: Optional[Dict[str, Any]] = PrivateAttr(default=None)
+    _inputs: Optional[Dict[str, Any]] = PrivateAttr(default=None)
 
     id: UUID4 = Field(default_factory=uuid.uuid4, frozen=True)
     name: Optional[str] = Field(default=None)
@@ -382,14 +382,13 @@ class Team(BaseModel):
                 futures.append((task, future, task_index))
             else:
                 context = create_raw_outputs(tasks=[task,], task_outputs=([last_sync_output,] if last_sync_output else [] ))
-                task_output = task.execute_sync(agent=responsible_agent, context=context, tools=responsible_agent.tools
-                                                )
+                task_output = task.execute_sync(agent=responsible_agent, context=context, tools=responsible_agent.tools)
                 if responsible_agent is self.manager_agent:
                     lead_task_output = task_output
 
                 task_outputs.append(task_output)
                 # self._process_task_result(task, task_output)
-                task._store_execution_log(task_index, was_replayed)
+                task._store_execution_log(task_index, was_replayed, self._inputs)
 
 
         if futures:

{versionhq-1.1.7.9 → versionhq-1.1.8}/src/versionhq.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: versionhq
-Version: 1.1.7.9
+Version: 1.1.8
 Summary: LLM orchestration frameworks for model-agnostic AI agents that handle complex outbound workflows
 Author-email: Kuriko Iwai <kuriko@versi0n.io>
 License: MIT License

{versionhq-1.1.7.9 → versionhq-1.1.8}/tests/task/task_test.py

@@ -277,4 +277,11 @@ def test_conditional_task():
         assert conditional_res.task_id is conditional_task.id
 
 
+def test_store_task_log():
+    task = Task(
+        description="Analyze the client's business model.",
+        output_field_list=[ResponseField(title="task_1", type=str, required=True),],
+    )
+    assert task._task_output_handler.load() is not None
+
 # tools, token usage

{versionhq-1.1.7.9 → versionhq-1.1.8}/tests/team/team_test.py

@@ -127,28 +127,17 @@ def test_form_team_without_leader():
         assert item.id is task_1.id or item.id is task_2.id
 
 
-def test_kickoff_team_without_leader():
-    agent_a = Agent(
-        role="agent a",
-        goal="My amazing goals",
-        llm=MODEL_NAME
-    )
-
-    agent_b = Agent(
-        role="agent b",
-        goal="My amazing goals",
-        llm=MODEL_NAME
-    )
-
+def test_kickoff_without_leader():
+    agent_a = Agent(role="agent a", goal="My amazing goals", llm=MODEL_NAME)
+    agent_b = Agent(role="agent b", goal="My amazing goals", llm=MODEL_NAME)
     task_1 = Task(
-            description="Analyze the client's business model.",
-            expected_output_json=True,
-            output_field_list=[
-                ResponseField(title="test1", type=str, required=True),
-                ResponseField(title="test2", type=list, required=True),
-            ],
+        description="Analyze the client's business model.",
+        expected_output_json=True,
+        output_field_list=[
+            ResponseField(title="test1", type=str, required=True),
+            ResponseField(title="test2", type=list, required=True),
+        ],
     )
-
     task_2 = Task(
         description="Define the cohort.",
         expected_output_json=True,
@@ -157,7 +146,6 @@ def test_kickoff_team_without_leader():
             ResponseField(title="test2", type=list, required=True),
         ],
     )
-
     team = Team(
         members=[
             TeamMember(agent=agent_a, is_manager=False, task=task_1),
@@ -185,7 +173,6 @@ def test_kickoff_team_without_leader():
     assert res.token_usage.total_tokens == 0 # as we dont set token usage on agent
 
 
-
 def team_kickoff_with_task_callback():
     """
     Each task has callback with callback kwargs.
@@ -236,7 +223,6 @@ def team_kickoff_with_task_callback():
     assert "pytest" in demo_list[1]
 
 
-
 def test_delegate_in_team():
     """
     When the agent belongs to the team, the team manager or peers are prioritized to delegete the task.
@@ -279,7 +265,37 @@ def test_delegate_in_team():
     assert "agent b" in task_1.processed_by_agents
 
 
+def test_kickoff_with_leader():
+    agent_a = Agent(role="agent a", goal="My amazing goals", llm=MODEL_NAME)
+    agent_b = Agent(role="agent b", goal="My amazing goals", llm=MODEL_NAME)
+    task_1 = Task(
+        description="Analyze the client's business model.",
+        output_field_list=[ResponseField(title="task_1", type=str, required=True),],
+    )
+    task_2 = Task(
+        description="Define the cohort timeframe.",
+        output_field_list=[
+            ResponseField(title="task_2_1", type=int, required=True),
+            ResponseField(title="task_2_2", type=list, required=True),
+        ],
+    )
+    team = Team(
+        members=[
+            TeamMember(agent=agent_a, is_manager=False, task=task_1),
+            TeamMember(agent=agent_b, is_manager=True, task=task_2),
+        ],
+    )
+    res = team.kickoff()
+
+    assert isinstance(res, TeamOutput)
+    assert res.team_id is team.id
+    assert res.raw is not None
+    assert res.json_dict is not None
+    assert team.manager_agent.id is agent_b.id
+    assert len(res.task_output_list) == 2
+    assert [item.raw is not None for item in res.task_output_list]
+    assert len(team.tasks) == 2
+    assert team.tasks[0].output.raw == res.raw
+
 
-if __name__ == "__main__":
-    test_kickoff_team_without_leader()
-# kickoff with teamleader, async, task handling process
+# async, task handling process

{versionhq-1.1.7.9 → versionhq-1.1.8}/uv.lock

@@ -1644,7 +1644,7 @@ wheels = [
 
 [[package]]
 name = "versionhq"
-version = "1.1.7.8"
+version = "1.1.8"
 source = { editable = "." }
 dependencies = [
     { name = "appdirs" },