versionhq 1.1.7.5__tar.gz → 1.1.7.7__tar.gz

{versionhq-1.1.7.5 → versionhq-1.1.7.7}/.gitignore

@@ -1,4 +1,5 @@
 knowledge/
+memory/
 
 composio.py
 memo.txt
@@ -8,6 +9,7 @@ build/
 .pypirc
 uploads/
 sample_dataset/
+chroma.sqlite3
 *egg-info/
 
 __pycache__/

{versionhq-1.1.7.5 → versionhq-1.1.7.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: versionhq
-Version: 1.1.7.5
+Version: 1.1.7.7
 Summary: LLM orchestration frameworks for model-agnostic AI agents that handle complex outbound workflows
 Author-email: Kuriko Iwai <kuriko@versi0n.io>
 License: MIT License
@@ -32,6 +32,9 @@ Keywords: orchestration framework,orchestration,ai agent,multi-agent system,RAG,
 Classifier: Programming Language :: Python
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Topic :: Software Development :: Build Tools
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
 License-File: LICENSE
@@ -43,12 +46,12 @@ Requires-Dist: typing
 Requires-Dist: json-repair>=0.31.0
 Requires-Dist: litellm>=1.55.8
 Requires-Dist: openai>=1.57.0
-Requires-Dist: composio-openai>=0.6.0
-Requires-Dist: pre-commit>=4.0.1
-Requires-Dist: gunicorn>=23.0.0
+Requires-Dist: composio-openai>=0.6.9
 Requires-Dist: composio>=0.1.0
 Requires-Dist: setuptools>=75.6.0
 Requires-Dist: wheel>=0.45.1
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: appdirs>=1.4.4
 
 # Overview
 
@@ -56,7 +59,7 @@ Requires-Dist: wheel>=0.45.1
 [![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)
 ![python ver](https://img.shields.io/badge/Python-3.12/3.13-purple)
-![pyenv ver](https://img.shields.io/badge/pyenv-2.4.23-orange)
+![pyenv ver](https://img.shields.io/badge/pyenv-2.5.0-orange)
 
 
 An LLM orchestration frameworks for multi-agent systems with RAG to autopilot outbound workflows.
@@ -276,8 +279,7 @@ src/
    ```
    uv venv
    source .venv/bin/activate
-
-   uv pip install -r requirements.txt -v
+   uv pip sync
    ```
 
 * In case of AssertionError/module mismatch, run Python version control using `.pyenv`

{versionhq-1.1.7.5 → versionhq-1.1.7.7}/README.md

@@ -4,7 +4,7 @@
 [![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)
 ![python ver](https://img.shields.io/badge/Python-3.12/3.13-purple)
-![pyenv ver](https://img.shields.io/badge/pyenv-2.4.23-orange)
+![pyenv ver](https://img.shields.io/badge/pyenv-2.5.0-orange)
 
 
 An LLM orchestration frameworks for multi-agent systems with RAG to autopilot outbound workflows.
@@ -224,8 +224,7 @@ src/
    ```
    uv venv
    source .venv/bin/activate
-
-   uv pip install -r requirements.txt -v
+   uv pip sync
    ```
 
 * In case of AssertionError/module mismatch, run Python version control using `.pyenv`

{versionhq-1.1.7.5 → versionhq-1.1.7.7}/pyproject.toml

@@ -15,7 +15,7 @@ exclude = ["test*", "__pycache__"]
 
 [project]
 name = "versionhq"
-version = "1.1.7.5"
+version = "1.1.7.7"
 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"
@@ -31,17 +31,20 @@ dependencies = [
   "json-repair>=0.31.0",
   "litellm>=1.55.8",
   "openai>=1.57.0",
-  "composio-openai>=0.6.0",
-  "pre-commit>=4.0.1",
-  "gunicorn>=23.0.0",
+  "composio-openai>=0.6.9",
   "composio>=0.1.0",
   "setuptools>=75.6.0",
   "wheel>=0.45.1",
+  "python-dotenv>=1.0.0",
+  "appdirs>=1.4.4",
 ]
 classifiers = [
   "Programming Language :: Python",
   "License :: OSI Approved :: MIT License",
   "Operating System :: OS Independent",
+  "Development Status :: 3 - Alpha",
+  "Intended Audience :: Developers",
+  "Topic :: Software Development :: Build Tools",
 ]
 
 [project.urls]
@@ -52,10 +55,9 @@ Issues = "https://github.com/versionHQ/multi-agent-system/issues"
 [tool.uv]
 dev-dependencies = [
   "mypy>=1.10.0",
-  "pre-commit>=3.6.0",
+  "pre-commit>=4.0.1",
   "pytest>=8.0.0",
   "pytest-vcr>=1.0.2",
-  "python-dotenv>=1.0.0",
   "black",
   "bandit",
   "twine",

versionhq-1.1.7.7/requirements-dev.txt

@@ -0,0 +1,9 @@
+mypy>=1.10.0
+pre-commit>=4.0.1
+pytest>=8.0.0
+pytest-vcr>=1.0.2
+python-dotenv>=1.0.0
+black
+bandit
+twine
+pytest>=8.3.4

versionhq-1.1.7.7/requirements.txt

@@ -0,0 +1,220 @@
+# This file was autogenerated by uv via the following command:
+#    uv pip compile pyproject.toml -o requirements.txt
+aiohappyeyeballs==2.4.4
+    # via aiohttp
+aiohttp==3.11.11
+    # via
+    #   composio-core
+    #   litellm
+aiosignal==1.3.2
+    # via aiohttp
+annotated-types==0.7.0
+    # via pydantic
+anyio==4.7.0
+    # via
+    #   httpx
+    #   openai
+    #   starlette
+appdirs==1.4.4
+    # via versionhq (pyproject.toml)
+attrs==24.3.0
+    # via
+    #   aiohttp
+    #   jsonschema
+    #   referencing
+bcrypt==4.2.1
+    # via paramiko
+certifi==2024.12.14
+    # via
+    #   httpcore
+    #   httpx
+    #   requests
+    #   sentry-sdk
+cffi==1.17.1
+    # via
+    #   cryptography
+    #   pynacl
+charset-normalizer==3.4.1
+    # via requests
+click==8.1.8
+    # via
+    #   composio-core
+    #   litellm
+    #   uvicorn
+composio==0.1.0
+    # via versionhq (pyproject.toml)
+composio-core==0.6.9
+    # via composio-openai
+composio-openai==0.6.9
+    # via versionhq (pyproject.toml)
+cryptography==44.0.0
+    # via paramiko
+distro==1.9.0
+    # via openai
+fastapi==0.115.6
+    # via composio-core
+filelock==3.16.1
+    # via huggingface-hub
+frozenlist==1.5.0
+    # via
+    #   aiohttp
+    #   aiosignal
+fsspec==2024.12.0
+    # via huggingface-hub
+h11==0.14.0
+    # via
+    #   httpcore
+    #   uvicorn
+httpcore==1.0.7
+    # via httpx
+httpx==0.27.2
+    # via
+    #   litellm
+    #   openai
+huggingface-hub==0.27.0
+    # via tokenizers
+idna==3.10
+    # via
+    #   anyio
+    #   httpx
+    #   requests
+    #   yarl
+importlib-metadata==8.5.0
+    # via
+    #   composio-core
+    #   litellm
+inflection==0.5.1
+    # via composio-core
+jinja2==3.1.5
+    # via litellm
+jiter==0.8.2
+    # via openai
+json-repair==0.35.0
+    # via versionhq (pyproject.toml)
+jsonref==1.1.0
+    # via composio-core
+jsonschema==4.23.0
+    # via
+    #   composio-core
+    #   litellm
+jsonschema-specifications==2024.10.1
+    # via jsonschema
+litellm==1.56.5
+    # via versionhq (pyproject.toml)
+markdown-it-py==3.0.0
+    # via rich
+markupsafe==3.0.2
+    # via
+    #   jinja2
+    #   werkzeug
+mdurl==0.1.2
+    # via markdown-it-py
+multidict==6.1.0
+    # via
+    #   aiohttp
+    #   yarl
+openai==1.58.1
+    # via
+    #   versionhq (pyproject.toml)
+    #   composio-openai
+    #   litellm
+packaging==24.2
+    # via huggingface-hub
+paramiko==3.5.0
+    # via composio-core
+propcache==0.2.1
+    # via
+    #   aiohttp
+    #   yarl
+pycparser==2.22
+    # via cffi
+pydantic==2.10.4
+    # via
+    #   versionhq (pyproject.toml)
+    #   composio-core
+    #   fastapi
+    #   litellm
+    #   openai
+pydantic-core==2.27.2
+    # via pydantic
+pygments==2.18.0
+    # via rich
+pynacl==1.5.0
+    # via paramiko
+pyperclip==1.9.0
+    # via composio-core
+pysher==1.0.8
+    # via composio-core
+python-dotenv==1.0.1
+    # via
+    #   versionhq (pyproject.toml)
+    #   litellm
+pyyaml==6.0.2
+    # via huggingface-hub
+referencing==0.35.1
+    # via
+    #   jsonschema
+    #   jsonschema-specifications
+regex==2024.11.6
+    # via
+    #   versionhq (pyproject.toml)
+    #   tiktoken
+requests==2.32.3
+    # via
+    #   versionhq (pyproject.toml)
+    #   composio-core
+    #   huggingface-hub
+    #   pysher
+    #   tiktoken
+rich==13.9.4
+    # via composio-core
+rpds-py==0.22.3
+    # via
+    #   jsonschema
+    #   referencing
+semver==3.0.2
+    # via composio-core
+sentry-sdk==2.19.2
+    # via composio-core
+setuptools==75.6.0
+    # via versionhq (pyproject.toml)
+sniffio==1.3.1
+    # via
+    #   anyio
+    #   httpx
+    #   openai
+starlette==0.41.3
+    # via fastapi
+tiktoken==0.8.0
+    # via litellm
+tokenizers==0.21.0
+    # via litellm
+tqdm==4.67.1
+    # via
+    #   huggingface-hub
+    #   openai
+typing==3.10.0.0
+    # via versionhq (pyproject.toml)
+typing-extensions==4.12.2
+    # via
+    #   fastapi
+    #   huggingface-hub
+    #   openai
+    #   pydantic
+    #   pydantic-core
+urllib3==2.3.0
+    # via
+    #   requests
+    #   sentry-sdk
+uvicorn==0.34.0
+    # via composio-core
+websocket-client==1.8.0
+    # via pysher
+werkzeug==3.1.3
+    # via versionhq (pyproject.toml)
+wheel==0.45.1
+    # via versionhq (pyproject.toml)
+yarl==1.18.3
+    # via aiohttp
+zipp==3.21.0
+    # via importlib-metadata

{versionhq-1.1.7.5 → versionhq-1.1.7.7}/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.5"
+__version__ = "1.1.7.7"
 __all__ = [
     "Agent",
     "Customer",

{versionhq-1.1.7.5 → versionhq-1.1.7.7}/src/versionhq/agent/model.py

@@ -85,7 +85,7 @@ class Agent(ABC, BaseModel):
     """
 
     __hash__ = object.__hash__
-    _logger: Logger = PrivateAttr(default_factory=lambda: Logger(verbose=False))
+    _logger: Logger = PrivateAttr(default_factory=lambda: Logger(verbose=True))
     _rpm_controller: Optional[RPMController] = PrivateAttr(default=None)
     _request_within_rpm_limit: Any = PrivateAttr(default=None)
     _token_process: TokenProcess = PrivateAttr(default_factory=TokenProcess)
@@ -327,7 +327,7 @@ class Agent(ABC, BaseModel):
         messages = []
         messages.append({"role": "user", "content": prompts})  #! REFINEME
         messages.append({"role": "assistant", "content": self.backstory})
-        print("Messages sent to the model:", messages)
+        self._logger.log(level="info", message=f"Messages sent to the model: {messages}", color="blue")
 
         callbacks = kwargs.get("callbacks", None)
 
@@ -338,7 +338,7 @@ class Agent(ABC, BaseModel):
             callbacks=callbacks,
         )
         task_execution_counter += 1
-        print("Agent's #1 res: ", response)
+        self._logger.log(level="info", message=f"Agent's first response: {response}", color="blue")
 
         if (response is None or response == "") and task_execution_counter < self.max_retry_limit:
             while task_execution_counter <= self.max_retry_limit:
@@ -349,10 +349,10 @@ class Agent(ABC, BaseModel):
                     callbacks=callbacks,
                 )
                 task_execution_counter += 1
-                print(f"Agent's #{task_execution_counter} res: ", response)
+                self._logger.log(level="info", message=f"Agent's next response: {response}", color="blue")
 
         elif response is None or response == "":
-            print("Received None or empty response from LLM call.")
+            self._logger.log(level="error", message="Received None or empty response from the model", color="red")
             raise ValueError("Invalid response from LLM call - None or empty.")
 
         return {"output": response.output if hasattr(response, "output") else response}

versionhq-1.1.7.7/src/versionhq/storage/task_output_storage.py

@@ -0,0 +1,141 @@
+import appdirs
+import os
+import json
+import sqlite3
+import datetime
+from typing import Any, Dict, List, Optional
+from dotenv import load_dotenv
+from pathlib import Path
+
+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))
+    data_dir.mkdir(parents=True, exist_ok=True)
+    return data_dir
+
+storage_path = fetch_db_storage_path()
+default_db_name = "task_outputs"
+
+
+class TaskOutputSQLiteStorage:
+    """
+    An SQLite storage class to handle storing task outputs.
+    """
+
+    def __init__(self, db_path: str = f"{storage_path}/{default_db_name}.db") -> None:
+        self.db_path = db_path
+        self._logger = Logger(verbose=True)
+        self._initialize_db()
+
+
+    def _initialize_db(self):
+        """
+        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
+                    )
+                """
+                )
+                conn.commit()
+
+        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] = {}):
+        try:
+            with sqlite3.connect(self.db_path) as conn:
+                cursor = conn.cursor()
+                cursor.execute(
+                        """INSERT OR REPLACE INTO task_outputs
+                            (task_id, output, task_index, inputs, was_replayed, timestamp)
+                            VALUES (?, ?, ?, ?, ?, ?)
+                        """,
+                     (str(task.id), json.dumps(output), task_index, json.dumps(inputs), was_replayed, datetime.datetime.now())
+                )
+                conn.commit()
+
+        except sqlite3.Error as e:
+            self._logger.log(level="error", message=f"SAVING TASK OUTPUTS ERROR: {e}", color="red")
+
+
+    def update(self, task_index: int, **kwargs):
+        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
+                values.append(task_index)
+                cursor.execute(query, tuple(values))
+                conn.commit()
+
+                if cursor.rowcount == 0:
+                    self._logger.log(
+                        level="info", message=f"No row found with task_index {task_index}. No update performed.", color="yellow",
+                    )
+
+        except sqlite3.Error as e:
+            self._logger.log(level="error", message=f"UPDATE TASK OUTPUTS ERROR: {e}", color="red")
+
+
+    def load(self) -> Optional[List[Dict[str, Any]]]:
+        try:
+            with sqlite3.connect(self.db_path) as conn:
+                cursor = conn.cursor()
+                cursor.execute("""
+                SELECT *
+                FROM task_outputs
+                ORDER BY task_index
+                """)
+
+                rows = cursor.fetchall()
+                results = []
+                for row in rows:
+                    result = {
+                        "task_id": row[0],
+                        "output": json.loads(row[1]),
+                        "task_index": row[2],
+                        "inputs": json.loads(row[3]),
+                        "was_replayed": row[4],
+                        "timestamp": row[5],
+                    }
+                    results.append(result)
+                return results
+
+        except sqlite3.Error as e:
+            self._logger.log(level="error", message=f"LOADING TASK OUTPUTS ERROR: {e}", color="red")
+            return None
+
+
+    def delete_all(self):
+        try:
+            with sqlite3.connect(self.db_path) as conn:
+                cursor = conn.cursor()
+                cursor.execute("DELETE FROM task_outputs")
+                conn.commit()
+
+        except sqlite3.Error as e:
+            self._logger.log(level="error", message=f"ERROR: Failed to delete all: {e}", color="red")

versionhq-1.1.7.7/src/versionhq/task/log_handler.py

@@ -0,0 +1,59 @@
+from datetime import datetime
+from typing import Any, Dict, List, Optional
+
+from pydantic import BaseModel, Field
+
+from versionhq.storage.task_output_storage import TaskOutputSQLiteStorage
+
+
+class ExecutionLog(BaseModel):
+    task_id: str
+    output: Dict[str, Any]
+    timestamp: datetime = Field(default_factory=datetime.now)
+    task_index: int
+    inputs: Dict[str, Any] = Field(default_factory=dict)
+    was_replayed: bool = False
+
+    def __getitem__(self, key: str) -> Any:
+        return getattr(self, key)
+
+
+
+class TaskOutputStorageHandler:
+
+    def __init__(self):
+        self.storage = TaskOutputSQLiteStorage()
+
+
+    def update(self, task, task_index: int, was_replayed: bool = False, inputs: Dict[str, Any] = {}) -> None:
+        """
+        task: task instance
+        """
+        saved_outputs = self.load()
+        if saved_outputs is None:
+            raise ValueError("Logs cannot be None")
+
+        self.add(task, task_index, was_replayed, inputs)
+
+
+    def add(self, task, task_index: int, was_replayed: bool = False, inputs: Dict[str, Any] = {}) -> None:
+        from versionhq.task.model import Task
+
+        output_to_store = dict()
+
+        if isinstance(task, Task):
+            output_to_store = dict(
+                description=str(task.description),
+                raw=str(task.output.raw),
+                responsible_agent=str(task.processed_by_agents),
+            )
+
+        self.storage.add(task=task, output=output_to_store, task_index=task_index, was_replayed=was_replayed, inputs=inputs)
+
+
+    def reset(self) -> None:
+        self.storage.delete_all()
+
+
+    def load(self) -> Optional[List[Dict[str, Any]]]:
+        return self.storage.load()