versionhq 1.1.7.8__tar.gz → 1.1.8__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: versionhq
-Version: 1.1.7.8
+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
@@ -57,7 +57,7 @@ Requires-Dist: appdirs>=1.4.4
 
 ![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.1.7.5-blue)
+![PyPI](https://img.shields.io/badge/PyPI-v1.1.7.9-blue)
 ![python ver](https://img.shields.io/badge/Python-3.12/3.13-purple)
 ![pyenv ver](https://img.shields.io/badge/pyenv-2.5.0-orange)
 

{versionhq-1.1.7.8 → versionhq-1.1.8}/README.md

@@ -2,7 +2,7 @@
 
 ![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.1.7.5-blue)
+![PyPI](https://img.shields.io/badge/PyPI-v1.1.7.9-blue)
 ![python ver](https://img.shields.io/badge/Python-3.12/3.13-purple)
 ![pyenv ver](https://img.shields.io/badge/pyenv-2.5.0-orange)
 

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

@@ -15,7 +15,7 @@ exclude = ["test*", "__pycache__"]
 
 [project]
 name = "versionhq"
-version = "1.1.7.8"
+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.8 → 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.8"
+__version__ = "1.1.8"
 __all__ = [
     "Agent",
     "Customer",

{versionhq-1.1.7.8 → versionhq-1.1.8}/src/versionhq/clients/product/model.py

@@ -34,7 +34,10 @@ class Product(BaseModel):
     id: UUID4 = Field(default_factory=uuid.uuid4, frozen=True)
     name: Optional[str] = Field(default=None, description="product name")
     description: Optional[str] = Field(
-        default=None,max_length=256,description="product description scraped from landing url or client input. cascade to the agent")
+        default=None,
+        max_length=256,
+        description="product description scraped from landing url or client input. cascade to the agent"
+    )
     provider: Optional[ProductProvider] = Field(default=None)
     audience: Optional[str] = Field(default=None, description="target audience")
     usp: Optional[str] = Field(default=None)

{versionhq-1.1.7.8 → versionhq-1.1.8}/src/versionhq/clients/workflow/model.py

@@ -60,6 +60,7 @@ class MessagingComponent(ABC, BaseModel):
         default=None, description="interval to move on to the next layer. if this is the last layer, set as `None`"
     )
     score: float | InstanceOf[Score] = Field(default=None)
+    condition: str = Field(default=None, max_length=128, description="condition to execute the next messaging component")
 
 
     def store_scoring_result(self, scoring_subject: str, score_raw: int | Score | ScoreFormat = None):
@@ -106,7 +107,7 @@ class MessagingWorkflow(ABC, BaseModel):
         default=None, description="store `Agent` instances responsible for autopiloting this workflow. if the team exsits, this field remains as `None`")
 
     # metrics
-    destination: Optional[str] = Field( default=None, description="destination service to launch this workflow")
+    destination: Optional[str | None] = Field(default=None, description="destination service to launch this workflow")
     product: InstanceOf[Product] = Field(default=None)
     customer: InstanceOf[Customer] = Field(default=None)
 
@@ -142,10 +143,9 @@ class MessagingWorkflow(ABC, BaseModel):
             if self.customer is not None:
                 self.destination = self.customer.on
 
-            else:
-                destination_list = self.product.provider.destinations
-                if destination_list:
-                    self.destination = destination_list[0]
+            elif self.product.provider is not None and self.product.provider.destinations:
+                self.destination = self.product.provider.destinations[0]
+
         return self
 
 

{versionhq-1.1.7.8 → 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.8 → 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.8 → 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.8 → versionhq-1.1.8}/src/versionhq.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: versionhq
-Version: 1.1.7.8
+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
@@ -57,7 +57,7 @@ Requires-Dist: appdirs>=1.4.4
 
 ![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.1.7.5-blue)
+![PyPI](https://img.shields.io/badge/PyPI-v1.1.7.9-blue)
 ![python ver](https://img.shields.io/badge/Python-3.12/3.13-purple)
 ![pyenv ver](https://img.shields.io/badge/pyenv-2.5.0-orange)
 

versionhq-1.1.8/tests/clients/workflow_test.py

@@ -0,0 +1,84 @@
+import os
+import pytest
+import datetime
+
+from versionhq.agent.model import Agent
+from versionhq.llm.model import LLM
+from versionhq.clients.workflow.model import Score, ScoreFormat, MessagingWorkflow, MessagingComponent
+from versionhq.clients.product.model import Product, ProductProvider
+
+
+MODEL_NAME = os.environ.get("LITELLM_MODEL_NAME", "gpt-3.5-turbo")
+LITELLM_API_KEY = os.environ.get("LITELLM_API_KEY")
+
+
+def test_store_scores():
+    """
+    Test if the final result will be calcurated using a random subject
+    """
+
+    messaging_component = MessagingComponent(message="demo")
+    score_raw = 15
+    messaging_component.store_scoring_result("demo", score_raw=score_raw)
+
+    assert messaging_component.score is not None
+    assert messaging_component.score.result() is not None
+
+
+def test_score_result():
+    messaging_component = MessagingComponent(message="demo")
+    score_raw = 15
+    messaging_component.store_scoring_result("demo", score_raw=score_raw)
+
+    result = messaging_component.score.result()
+
+    assert result is not None
+    assert result != 0
+
+
+def test_setup_messaging_workflow_with_anonymous_provider():
+    product = Product(
+        description="demo p",
+        audience="demo audience",
+        usp="demo usp",
+        landing_url="www.com",
+        cohort_timeframe=30,
+    )
+    comp = MessagingComponent(message="demo")
+    messaging_workflow = MessagingWorkflow(
+        _created_at=datetime.datetime.now(),
+        components=[comp, ],
+        agents=[],
+        product=product,
+    )
+
+    assert messaging_workflow.id is not None
+    assert messaging_workflow.destination is None
+
+
+def test_setup_messaging_workflow_with_provider():
+    provider = ProductProvider(
+        name="demo provider",
+        region="US",
+        data_pipeline=["data"],
+        destinations=["email", "linkedin",]
+    )
+    product = Product(
+        description="demo p",
+        audience="demo audience",
+        usp="demo usp",
+        landing_url="www.com",
+        cohort_timeframe=30,
+        provider=provider,
+    )
+    comp = MessagingComponent(message="demo")
+    messaging_workflow = MessagingWorkflow(
+        _created_at=datetime.datetime.now(),
+        components=[comp, ],
+        agents=[],
+        product=product,
+    )
+
+    assert messaging_workflow.id is not None
+    assert messaging_workflow.destination is not None
+    assert messaging_workflow.destination in provider.destinations

{versionhq-1.1.7.8 → 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.8 → 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.8 → 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" },

versionhq-1.1.7.8/tests/clients/workflow_test.py

@@ -1,32 +0,0 @@
-import os
-import pytest
-from versionhq.agent.model import Agent
-from versionhq.llm.model import LLM
-from versionhq.clients.workflow.model import Score, ScoreFormat, MessagingWorkflow, MessagingComponent
-
-MODEL_NAME = os.environ.get("LITELLM_MODEL_NAME", "gpt-3.5-turbo")
-LITELLM_API_KEY = os.environ.get("LITELLM_API_KEY")
-
-
-def test_store_scores():
-    """
-    Test if the final result will be calcurated using a random subject
-    """
-
-    messaging_component = MessagingComponent(message="demo")
-    score_raw = 15
-    messaging_component.store_scoring_result("demo", score_raw=score_raw)
-
-    assert messaging_component.score is not None
-    assert messaging_component.score.result() is not None
-
-
-def test_score_result():
-    messaging_component = MessagingComponent(message="demo")
-    score_raw = 15
-    messaging_component.store_scoring_result("demo", score_raw=score_raw)
-
-    result = messaging_component.score.result()
-
-    assert result is not None
-    assert result != 0