versionhq 1.1.11.5__tar.gz → 1.1.11.6__tar.gz

{versionhq-1.1.11.5 → versionhq-1.1.11.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.1.11.5
+Version: 1.1.11.6
 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.11.5 → versionhq-1.1.11.6}/pyproject.toml

@@ -15,7 +15,7 @@ exclude = ["test*", "__pycache__", "*.egg-info"]
 
 [project]
 name = "versionhq"
-version = "1.1.11.5"
+version = "1.1.11.6"
 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.11.5 → versionhq-1.1.11.6}/src/versionhq/__init__.py

@@ -17,7 +17,7 @@ from versionhq.tool.model import Tool
 from versionhq.tool.composio_tool import ComposioHandler
 
 
-__version__ = "1.1.11.5"
+__version__ = "1.1.11.6"
 __all__ = [
     "Agent",
     "Customer",

{versionhq-1.1.11.5 → versionhq-1.1.11.6}/src/versionhq/knowledge/storage.py

@@ -73,11 +73,21 @@ class KnowledgeStorage(BaseKnowledgeStorage):
 
 
     def __init__(self, embedder_config: Optional[Dict[str, Any]] = None, collection_name: Optional[str] = None):
-        self.collection_name = collection_name if collection_name else "knowledge"
+        self.collection_name = self._validate_collection_name(collection_name) if collection_name else "knowledge"
         self.embedder_config = embedder_config
         self.initialize_knowledge_storage()
 
 
+    def _validate_collection_name(self, collection_name: str = None) -> str:
+        """
+        Return a valid collection name from the given collection name.
+        Expected collection name (1) contains 3-63 characters, (2) starts and ends with an alphanumeric character, (3) otherwise contains only alphanumeric characters, underscores or hyphens (-), (4) contains no two consecutive periods (..) and (5) is not a valid IPv4 address.
+        """
+        collection_name = collection_name if collection_name else self.collection_name
+        valid_collection_name = collection_name.replace(' ', "-").replace("(", "-").replace(")", "").replace("..", "")
+        return valid_collection_name
+
+
     def _create_default_embedding_function(self) -> Any:
         from chromadb.utils.embedding_functions.openai_embedding_function import OpenAIEmbeddingFunction
 
@@ -101,6 +111,7 @@ class KnowledgeStorage(BaseKnowledgeStorage):
         chroma_client = chromadb.PersistentClient(path=base_path, settings=Settings(allow_reset=True))
         self.app = chroma_client
         self._set_embedding_function(embedder_config=self.embedder_config)
+        self.collection_name = self.collection_name if self.collection_name else "knowledge"
 
         try:
             if self.app:

{versionhq-1.1.11.5 → versionhq-1.1.11.6}/src/versionhq.egg-info/PKG-INFO

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