versionhq 1.1.11.4__py3-none-any.whl → 1.1.11.5__py3-none-any.whl

versionhq/__init__.py

@@ -10,15 +10,14 @@ warnings.filterwarnings(
 from versionhq.agent.model import Agent
 from versionhq.clients.customer.model import Customer
 from versionhq.clients.product.model import Product, ProductProvider
-from versionhq.clients.workflow.model import MessagingWorkflow, MessagingComponent, Score, ScoreFormat
-from versionhq.llm.model import LLM
+from versionhq.clients.workflow.model import MessagingWorkflow, MessagingComponent
 from versionhq.task.model import Task, TaskOutput
 from versionhq.team.model import Team, TeamOutput
 from versionhq.tool.model import Tool
 from versionhq.tool.composio_tool import ComposioHandler
 
 
-__version__ = "1.1.11.4"
+__version__ = "1.1.11.5"
 __all__ = [
     "Agent",
     "Customer",
@@ -26,8 +25,6 @@ __all__ = [
     "ProductProvider",
     "MessagingWorkflow",
     "MessagingComponent",
-    "Score",
-    "ScoreFormat",
     "LLM",
     "Task",
     "TaskOutput",

versionhq/agent/model.py

@@ -345,14 +345,13 @@ class Agent(BaseModel):
     @model_validator(mode="after")
     def set_up_knowledge(self) -> Self:
         if self.knowledge_sources:
-            knowledge_agent_name = f"{self.role.replace(' ', '_')}"
-
-            if isinstance(self.knowledge_sources, list) and all(isinstance(k, BaseKnowledgeSource) for k in self.knowledge_sources):
-                self._knowledge = Knowledge(
-                    sources=self.knowledge_sources,
-                    embedder_config=self.embedder_config,
-                    collection_name=knowledge_agent_name,
-                )
+            collection_name = f"{self.role.replace(' ', '_')}"
+
+            self._knowledge = Knowledge(
+                sources=self.knowledge_sources,
+                embedder_config=self.embedder_config,
+                collection_name=collection_name,
+            )
 
         return self
 

versionhq/knowledge/model.py

@@ -11,11 +11,11 @@ class Knowledge(BaseModel):
     """
     Knowlede class for collection of sources and setup for the vector store to query relevant context.
     """
+    collection_name: Optional[str] = None
     sources: List[BaseKnowledgeSource] = Field(default_factory=list)
-    model_config = ConfigDict(arbitrary_types_allowed=True)
     storage: KnowledgeStorage = Field(default_factory=KnowledgeStorage)
     embedder_config: Optional[Dict[str, Any]] = None
-    collection_name: Optional[str] = None
+    model_config = ConfigDict(arbitrary_types_allowed=True)
 
     def __init__(
         self,
@@ -27,16 +27,12 @@ class Knowledge(BaseModel):
     ):
         super().__init__(**data)
 
-
-        if storage:
-            self.storage = storage
-        else:
-            self.storage = KnowledgeStorage(embedder_config=embedder_config, collection_name=collection_name)
-
-        self.storage._set_embedding_function(embedder_config=embedder_config)
+        self.collection_name = collection_name
+        self.sources = sources
+        self.embedder_config = embedder_config
+        self.storage = storage if storage else KnowledgeStorage(embedder_config=embedder_config, collection_name=self.collection_name)
         self.storage.initialize_knowledge_storage()
 
-        self.sources = sources
         for source in sources:
             source.storage = self.storage
             source.add()

versionhq/knowledge/source.py

@@ -237,7 +237,10 @@ class PDFKnowledgeSource(BaseFileKnowledgeSource):
         """
         Load and preprocess PDF file content.
         """
-        pdfplumber = self._import_pdfplumber()
+        self._import_pdfplumber()
+
+        import pdfplumber
+
         content = {}
         for path in self.valid_file_paths:
             text = ""
@@ -257,9 +260,12 @@ class PDFKnowledgeSource(BaseFileKnowledgeSource):
         """
         try:
             import pdfplumber
-            return pdfplumber
         except ImportError:
-            raise ImportError("pdfplumber is not installed. Please install it with: pip install pdfplumber")
+            try:
+                import os
+                os.system("uv add pdfplumber --optional pdfplumber")
+            except:
+                raise ImportError("pdfplumber is not installed. Please install it with: uv add pdfplumber")
 
 
     def add(self) -> None:
@@ -381,10 +387,16 @@ class ExcelKnowledgeSource(BaseFileKnowledgeSource):
             import pandas as pd
             return pd
         except ImportError as e:
-            missing_package = str(e).split()[-1]
-            raise ImportError(
-                f"{missing_package} is not installed. Please install it with: pip install {missing_package}"
-            )
+            try:
+                import os
+                os.system("uv add pandas --optional pandas")
+                import pandas as pd
+                return pd
+            except:
+                missing_package = str(e).split()[-1]
+                raise ImportError(
+                    f"{missing_package} is not installed. Please install it with: pip install {missing_package}"
+                )
 
 
     def add(self) -> None:

versionhq/knowledge/source_docling.py

@@ -3,6 +3,7 @@ from typing import Iterator, List, Optional
 from urllib.parse import urlparse
 
 try:
+    import docling
     from docling.datamodel.base_models import InputFormat
     from docling.document_converter import DocumentConverter
     from docling.exceptions import ConversionError
@@ -10,6 +11,17 @@ try:
     from docling_core.types.doc.document import DoclingDocument
     DOCLING_AVAILABLE = True
 except ImportError:
+    import envoy
+    r = envoy.run("uv add docling --optional docling")
+
+    import docling
+    from docling.datamodel.base_models import InputFormat
+    from docling.document_converter import DocumentConverter
+    from docling.exceptions import ConversionError
+    from docling_core.transforms.chunker.hierarchical_chunker import HierarchicalChunker
+    from docling_core.types.doc.document import DoclingDocument
+    DOCLING_AVAILABLE = True
+except:
     DOCLING_AVAILABLE = False
 
 from pydantic import Field, InstanceOf
@@ -25,30 +37,27 @@ class DoclingSource(BaseKnowledgeSource):
     Support PDF, DOCX, TXT, XLSX, PPTX, MD, Images, and HTML files without any additional dependencies.
     """
 
-    def __init__(self, *args, **kwargs):
-        if not DOCLING_AVAILABLE:
-            raise ImportError("The docling package is required. Please install the package using: $ uv add docling.")
-
-        super().__init__(*args, **kwargs)
-
-
     file_paths: List[Path | str] = Field(default_factory=list)
     valid_file_paths: List[Path | str] = Field(default_factory=list)
     content: List["DoclingDocument"] = Field(default_factory=list)
-    document_converter: "DocumentConverter" = Field(
-        default_factory=lambda: DocumentConverter(
-            allowed_formats=[
-                InputFormat.MD,
-                InputFormat.ASCIIDOC,
-                InputFormat.PDF,
-                InputFormat.DOCX,
-                InputFormat.HTML,
-                InputFormat.IMAGE,
-                InputFormat.XLSX,
-                InputFormat.PPTX,
-            ]
-        )
-    )
+    document_converter: "DocumentConverter" = Field(default_factory=lambda: DocumentConverter(
+        allowed_formats=[
+            InputFormat.MD,
+            InputFormat.ASCIIDOC,
+            InputFormat.PDF,
+            InputFormat.DOCX,
+            InputFormat.HTML,
+            InputFormat.IMAGE,
+            InputFormat.XLSX,
+            InputFormat.PPTX,
+        ]
+    ))
+
+    def __init__(self, *args, **kwargs):
+        if not DOCLING_AVAILABLE:
+            raise ImportError("The docling package is required. Please install the package using: $ uv add docling.")
+        else:
+            super().__init__(*args, **kwargs)
 
 
     def _convert_source_to_docling_documents(self) -> List["DoclingDocument"]:

versionhq/knowledge/storage.py

@@ -65,15 +65,15 @@ class KnowledgeStorage(BaseKnowledgeStorage):
     A class to store ChromaDB Storage vals that handles embeddings, ChromaClient, and Collection.
     """
 
-    collection: Optional[chromadb.Collection] = None
+    app: Optional[ClientAPI] = None # store ChromaDBClient object
     collection_name: Optional[str] = "knowledge"
-    app: Optional[ClientAPI] = None
-    embedding_function: Optional[Any] = None # store ChromaDB's EmbeddingFunction instance
+    collection: Optional[chromadb.Collection] = None
+    embedding_function: Optional[Any] = None # store ChromaDB's EmbeddingFunction object
     embedder_config: Optional[Dict[str, Any]] = None # store config dict for embedding_function
 
 
     def __init__(self, embedder_config: Optional[Dict[str, Any]] = None, collection_name: Optional[str] = None):
-        self.collection_name = collection_name
+        self.collection_name = collection_name if collection_name else "knowledge"
         self.embedder_config = embedder_config
         self.initialize_knowledge_storage()
 
@@ -100,13 +100,11 @@ class KnowledgeStorage(BaseKnowledgeStorage):
         base_path = os.path.join(fetch_db_storage_path(), "knowledge")
         chroma_client = chromadb.PersistentClient(path=base_path, settings=Settings(allow_reset=True))
         self.app = chroma_client
-
-        self._set_embedding_function(self.embedder_config)
+        self._set_embedding_function(embedder_config=self.embedder_config)
 
         try:
-            collection_name = f"knowledge_{self.collection_name}" if self.collection_name else "knowledge"
             if self.app:
-                self.collection = self.app.get_or_create_collection(name=collection_name, embedding_function=self.embedding_function)
+                self.collection = self.app.get_or_create_collection(name=self.collection_name, embedding_function=self.embedding_function)
             else:
                 raise Exception("Vector Database Client not initialized")
         except Exception:

versionhq/memory/model.py

@@ -55,7 +55,13 @@ class ShortTermMemory(Memory):
             try:
                 from versionhq.storage.mem0_storage import Mem0Storage
             except ImportError:
-                raise ImportError("Mem0 is not installed. Please install it with `uv pip install mem0ai`.")
+                try:
+                    import os
+                    os.system("uv add mem0ai --optional mem0ai")
+
+                    from versionhq.storage.mem0_storage import Mem0Storage
+                except:
+                    raise ImportError("Mem0 is not installed. Please install it with `uv pip install mem0ai`.")
 
             storage = Mem0Storage(type="stm", agent=agent)
         else:

{versionhq-1.1.11.4.dist-info → versionhq-1.1.11.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.1.11.4
+Version: 1.1.11.5
 Summary: LLM orchestration frameworks for model-agnostic AI agents that handle complex outbound workflows
 Author-email: Kuriko Iwai <kuriko@versi0n.io>
 License: MIT License
@@ -31,12 +31,16 @@ Project-URL: Repository, https://github.com/versionHQ/multi-agent-system
 Project-URL: Issues, https://github.com/versionHQ/multi-agent-system/issues
 Keywords: orchestration framework,orchestration,ai agent,multi-agent system,RAG,agent
 Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Information Technology
 Classifier: Topic :: Software Development :: Build Tools
-Requires-Python: >=3.12
+Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: regex==2024.11.6
@@ -58,6 +62,7 @@ Requires-Dist: langchain-openai>=0.2.14
 Requires-Dist: composio-langchain>=0.6.12
 Requires-Dist: chromadb>=0.6.3
 Requires-Dist: wheel>=0.45.1
+Requires-Dist: envoy>=0.0.3
 Provides-Extra: docling
 Requires-Dist: docling>=2.17.0; extra == "docling"
 Provides-Extra: mem0ai
@@ -66,12 +71,14 @@ Provides-Extra: pdfplumber
 Requires-Dist: pdfplumber>=0.11.5; extra == "pdfplumber"
 Provides-Extra: pandas
 Requires-Dist: pandas>=2.2.3; extra == "pandas"
+Provides-Extra: numpy
+Requires-Dist: numpy>=1.26.4; extra == "numpy"
 
 # Overview
 
 ![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.11.1-blue)
+![PyPI](https://img.shields.io/badge/PyPI->=v1.1.11.4-blue)
 ![python ver](https://img.shields.io/badge/Python->=3.12-purple)
 ![pyenv ver](https://img.shields.io/badge/pyenv-2.5.0-orange)
 
@@ -142,7 +149,7 @@ You can specify which formation you want them to generate, or let the agent deci
    pip install versionhq
    ```
 
-(Python >= 3.12)
+(Python 3.11 or higher)
 
 
 ### Case 1. Solo Agent:
@@ -285,11 +292,20 @@ src/
 
 ## Setup
 
-1. Install the `uv` package manager:
+1. Install `uv` package manager:
+
+   For MacOS:
+
    ```
    brew install uv
    ```
 
+   For Ubuntu/Debian:
+
+   ```
+   sudo apt-get install uv
+   ```
+
 2. Install dependencies:
    ```
    uv venv

{versionhq-1.1.11.4.dist-info → versionhq-1.1.11.5.dist-info}/RECORD

@@ -1,4 +1,4 @@
-versionhq/__init__.py,sha256=b8BjYKXpj6Dw8_k9zl1oysFiKt44MCiBc0bb8XeynD0,951
+versionhq/__init__.py,sha256=2Zt2x3RvXctSU64Bv3Omw9EFjs5WQ3xKOijkiHNt4tU,863
 versionhq/_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/_utils/i18n.py,sha256=TwA_PnYfDLA6VqlUDPuybdV9lgi3Frh_ASsb_X8jJo8,1483
 versionhq/_utils/logger.py,sha256=U-MpeGueA6YS8Ptfy0VnU_ePsZP-8Pvkvi0tZ4s_UMg,1438
@@ -7,7 +7,7 @@ versionhq/_utils/usage_metrics.py,sha256=hhq1OCW8Z4V93vwW2O2j528EyjOlF8wlTsX5IL-
 versionhq/_utils/vars.py,sha256=bZ5Dx_bFKlt3hi4-NNGXqdk7B23If_WaTIju2fiTyPQ,57
 versionhq/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/agent/default_agents.py,sha256=Sea3xDswxxMccer1vVDhp1E5etXW3ddf2n20JTMHgqs,503
-versionhq/agent/model.py,sha256=F_VkSQ6G6mJvDWrRBILZ6KjtlCpm0r_8bMN73sDKKGc,22921
+versionhq/agent/model.py,sha256=kL949N0MELAZz58XB_vlQfE1YQU_o4iHggNDn_h_7yo,22758
 versionhq/agent/parser.py,sha256=riG0dkdQCxH7uJ0AbdVdg7WvL0BXhUgJht0VtQvxJBc,4082
 versionhq/agent/rpm_controller.py,sha256=7AKIEPbWBq_ESOZCaiKVOGjfSPHd2qwg6-wbBlhqC0g,2367
 versionhq/agent/TEMPLATES/Backstory.py,sha256=IAhGnnt6VUMe3wO6IzeyZPDNu7XE7Uiu3VEXUreOcKs,532
@@ -23,16 +23,16 @@ versionhq/clients/workflow/model.py,sha256=FNftenLLoha0bkivrjId32awLHAkBwIT8iNlj
 versionhq/knowledge/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/knowledge/_utils.py,sha256=YWRF8U533cfZes_gZqUvdj-K24MD2ri1R0gjc_aPYyc,402
 versionhq/knowledge/embedding.py,sha256=KfHc__1THxb5jrg1EMrF-v944RDuIr2hE0l-MtM3Bp0,6826
-versionhq/knowledge/model.py,sha256=_liwQoS_VJlJgVSwAb7Y68SwbPuU0QBY_q0cA8x7dCo,1862
-versionhq/knowledge/source.py,sha256=yUwOds0zc8oPLvtV_hIE4P7k9BjQ9vc4MbbGorv_H6I,13292
-versionhq/knowledge/source_docling.py,sha256=fGfa3NntjH5tzpmWSoLsSgKJxBvTEKwl1egAlo67qnA,4935
-versionhq/knowledge/storage.py,sha256=vg7NEi19b47QaxXQxx2BLag3hjUZAQnwPqUifzhWCvQ,7373
+versionhq/knowledge/model.py,sha256=n7kU4jQ24BUIxwosSVRK8tYhAFYhgc4yf7e4Q-bq4bk,1832
+versionhq/knowledge/source.py,sha256=WOARChmm_cNtBD-xGo4RoYmcuodzdalctXI-gDBCW6k,13610
+versionhq/knowledge/source_docling.py,sha256=Lv7PDE97pwV5_3SPcIgTzEu3H4obRbPfY3NALvfkgX8,5364
+versionhq/knowledge/storage.py,sha256=uesUF6zFhXcgadrgzPVzLf9kBgvb3jMjWJIJzBlnhFk,7350
 versionhq/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/llm/llm_vars.py,sha256=PO__b-h5e-6oQ-uoIgXx3lPSAUPUwXYfdVRW73fvX14,8761
 versionhq/llm/model.py,sha256=1uaBxT10GIlUl-BtE8Mfux-ZRcScp4HUIas_fD_cdWQ,14471
 versionhq/memory/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/memory/contextual_memory.py,sha256=tCsOOAUnfrOL7YiakqGoi3uShzzS870TmGnlGd3z_A4,3556
-versionhq/memory/model.py,sha256=3VDaCHqZ4iLQcMkKRmRVbv8db7E-rXOxYFzJ6vQpsSI,5535
+versionhq/memory/model.py,sha256=6Sy-cnrhHNIx3ZN38uNO7d8YywIl_uo_OvDVzVM-w14,5755
 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=LMitExpDATg__Oc7Hvl6h6uwQ7vUKQ34Eyp1BSt0DfY,4260
@@ -57,8 +57,8 @@ versionhq/tool/composio_tool_vars.py,sha256=FvBuEXsOQUYnN7RTFxT20kAkiEYkxWKkiVtg
 versionhq/tool/decorator.py,sha256=C4ZM7Xi2gwtEMaSeRo-geo_g_MAkY77WkSLkAuY0AyI,1205
 versionhq/tool/model.py,sha256=7ccEnje_8LuxLVeog6pL38nToArXQXk4KY7A9hfprDo,12239
 versionhq/tool/tool_handler.py,sha256=2m41K8qo5bGCCbwMFferEjT-XZ-mE9F0mDUOBkgivOI,1416
-versionhq-1.1.11.4.dist-info/LICENSE,sha256=7CCXuMrAjPVsUvZrsBq9DsxI2rLDUSYXR_qj4yO_ZII,1077
-versionhq-1.1.11.4.dist-info/METADATA,sha256=H3fBLb0rTLGR5EL7yvNyzekPsa6Iu1SNpqOJbUD3uMw,18251
-versionhq-1.1.11.4.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-versionhq-1.1.11.4.dist-info/top_level.txt,sha256=DClQwxDWqIUGeRJkA8vBlgeNsYZs4_nJWMonzFt5Wj0,10
-versionhq-1.1.11.4.dist-info/RECORD,,
+versionhq-1.1.11.5.dist-info/LICENSE,sha256=7CCXuMrAjPVsUvZrsBq9DsxI2rLDUSYXR_qj4yO_ZII,1077
+versionhq-1.1.11.5.dist-info/METADATA,sha256=kqMA6vNrLp3wgffnEfUJjPT2Jr3eylK-guAj8g4QXt8,18638
+versionhq-1.1.11.5.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+versionhq-1.1.11.5.dist-info/top_level.txt,sha256=DClQwxDWqIUGeRJkA8vBlgeNsYZs4_nJWMonzFt5Wj0,10
+versionhq-1.1.11.5.dist-info/RECORD,,