versionhq 1.2.1.17__tar.gz → 1.2.1.19__tar.gz

{versionhq-1.2.1.17 → versionhq-1.2.1.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.2.1.17
+Version: 1.2.1.19
 Summary: An agentic orchestration framework for building agent networks that handle task automation.
 Author-email: Kuriko Iwai <kuriko@versi0n.io>
 License: MIT License

{versionhq-1.2.1.17 → versionhq-1.2.1.19}/pyproject.toml

@@ -15,7 +15,7 @@ exclude = ["test*", "__pycache__", "*.egg-info"]
 
 [project]
 name = "versionhq"
-version = "1.2.1.17"
+version = "1.2.1.19"
 authors = [{ name = "Kuriko Iwai", email = "kuriko@versi0n.io" }]
 description = "An agentic orchestration framework for building agent networks that handle task automation."
 readme = "README.md"

{versionhq-1.2.1.17 → versionhq-1.2.1.19}/src/versionhq/__init__.py

@@ -31,7 +31,7 @@ from versionhq.agent_network.formation import form_agent_network
 from versionhq.task_graph.draft import workflow
 
 
-__version__ = "1.2.1.17"
+__version__ = "1.2.1.19"
 __all__ = [
     "Agent",
 

{versionhq-1.2.1.17 → versionhq-1.2.1.19}/src/versionhq/_utils/process_config.py

@@ -8,7 +8,7 @@ def process_config(values_to_update: Dict[str, Any], model_class: Type[BaseModel
     Refer to the Pydantic model class for field validation.
     """
 
-    config = values_to_update.pop("config", {})
+    config = values_to_update.pop("config") if "config" in values_to_update else {}
 
     if config:
         for k, v in config.items():

{versionhq-1.2.1.17 → versionhq-1.2.1.19}/src/versionhq/agent/model.py

@@ -149,94 +149,6 @@ class Agent(BaseModel):
         return self
 
 
-    def _convert_to_llm_object(self, llm: Any = None) -> LLM:
-        """
-        Convert the given value to LLM object.
-        When `llm` is dict or self.llm_config is not None, add these values to the LLM object after validating them.
-        """
-        llm = llm if llm else self.llm if self.llm else DEFAULT_MODEL_NAME
-
-        if not llm:
-            pass
-
-        match llm:
-            case LLM():
-                return self._set_llm_params(llm=llm, config=self.llm_config)
-
-            case str():
-                llm_obj = LLM(model=llm)
-                return self._set_llm_params(llm=llm_obj, config=self.llm_config)
-
-            case dict():
-                model_name = llm.pop("model_name", llm.pop("deployment_name", str(llm)))
-                llm_obj = LLM(model=model_name if model_name else DEFAULT_MODEL_NAME)
-                config = llm.update(self.llm_config) if self.llm_config else llm
-                return self._set_llm_params(llm_obj, config=config)
-
-            case _:
-                model_name = (getattr(self.llm, "model_name") or getattr(self.llm, "deployment_name") or str(self.llm))
-                llm_obj = LLM(model=model_name if model_name else DEFAULT_MODEL_NAME)
-                llm_params = {
-                    "max_tokens": (getattr(llm, "max_tokens") or 3000),
-                    "timeout": getattr(llm, "timeout", self.max_execution_time),
-                    "callbacks": getattr(llm, "callbacks", None),
-                    "temperature": getattr(llm, "temperature", None),
-                    "logprobs": getattr(llm, "logprobs", None),
-                    "api_key": getattr(llm, "api_key", os.environ.get("LITELLM_API_KEY", None)),
-                    "base_url": getattr(llm, "base_url", None),
-                }
-                config = llm_params.update(self.llm_config) if self.llm_config else llm_params
-                return self._set_llm_params(llm=llm_obj, config=config)
-
-
-    def _set_llm_params(self, llm: LLM, config: Dict[str, Any] = None) -> LLM:
-        """
-        Add valid params to the LLM object.
-        """
-
-        import litellm
-        from versionhq.llm.llm_vars import PARAMS
-
-        valid_config = {k: v for k, v in config.items() if v} if config else {}
-
-        if valid_config:
-            valid_keys = list()
-            try:
-                valid_keys = litellm.get_supported_openai_params(model=llm.model, custom_llm_provider=self.endpoint_provider, request_type="chat_completion")
-                if not valid_keys:
-                    valid_keys = PARAMS.get("common")
-            except:
-                valid_keys = PARAMS.get("common")
-
-            valid_keys += PARAMS.get("litellm")
-
-            for key in valid_keys:
-                if key in valid_config and valid_config[key]:
-                    val = valid_config[key]
-                    if [key == k for k, v in LLM.model_fields.items()]:
-                        setattr(llm, key, val)
-                    else:
-                        llm.other_valid_config.update({ key: val})
-
-
-        llm.timeout = self.max_execution_time if llm.timeout is None else llm.timeout
-        # llm.max_tokens = self.max_tokens if self.max_tokens else llm.max_tokens
-
-        if llm.provider is None:
-            provider_name = llm.model.split("/")[0]
-            valid_provider = provider_name if provider_name in PROVIDERS else None
-            llm.provider = valid_provider
-
-        if self.callbacks:
-            llm.callbacks = self.callbacks
-            llm._set_callbacks(llm.callbacks)
-
-        if self.respect_context_window == False:
-            llm.context_window_size = DEFAULT_CONTEXT_WINDOW_SIZE
-
-        return llm
-
-
     @model_validator(mode="after")
     def set_up_tools(self) -> Self:
         """
@@ -310,7 +222,7 @@ class Agent(BaseModel):
 
         if self.knowledge_sources:
             try:
-                collection_name = f"{self.role.replace(' ', '_')}"
+                collection_name = f"{self.role.replace(' ', '_')}-{str(self.id)}"
                 knowledge_sources = []
                 docling_fp, txt_fp, json_fp, excel_fp, csv_fp, pdf_fp = [], [], [], [], [], []
                 str_cont = ""
@@ -369,6 +281,94 @@ class Agent(BaseModel):
         return self
 
 
+    def _convert_to_llm_object(self, llm: Any = None) -> LLM:
+        """
+        Convert the given value to LLM object.
+        When `llm` is dict or self.llm_config is not None, add these values to the LLM object after validating them.
+        """
+        llm = llm if llm else self.llm if self.llm else DEFAULT_MODEL_NAME
+
+        if not llm:
+            pass
+
+        match llm:
+            case LLM():
+                return self._set_llm_params(llm=llm, config=self.llm_config)
+
+            case str():
+                llm_obj = LLM(model=llm)
+                return self._set_llm_params(llm=llm_obj, config=self.llm_config)
+
+            case dict():
+                model_name = llm.pop("model_name", llm.pop("deployment_name", str(llm)))
+                llm_obj = LLM(model=model_name if model_name else DEFAULT_MODEL_NAME)
+                config = llm.update(self.llm_config) if self.llm_config else llm
+                return self._set_llm_params(llm_obj, config=config)
+
+            case _:
+                model_name = (getattr(self.llm, "model_name") or getattr(self.llm, "deployment_name") or str(self.llm))
+                llm_obj = LLM(model=model_name if model_name else DEFAULT_MODEL_NAME)
+                llm_params = {
+                    "max_tokens": (getattr(llm, "max_tokens") or 3000),
+                    "timeout": getattr(llm, "timeout", self.max_execution_time),
+                    "callbacks": getattr(llm, "callbacks", None),
+                    "temperature": getattr(llm, "temperature", None),
+                    "logprobs": getattr(llm, "logprobs", None),
+                    "api_key": getattr(llm, "api_key", os.environ.get("LITELLM_API_KEY", None)),
+                    "base_url": getattr(llm, "base_url", None),
+                }
+                config = llm_params.update(self.llm_config) if self.llm_config else llm_params
+                return self._set_llm_params(llm=llm_obj, config=config)
+
+
+    def _set_llm_params(self, llm: LLM, config: Dict[str, Any] = None) -> LLM:
+        """
+        Add valid params to the LLM object.
+        """
+
+        import litellm
+        from versionhq.llm.llm_vars import PARAMS
+
+        valid_config = {k: v for k, v in config.items() if v} if config else {}
+
+        if valid_config:
+            valid_keys = list()
+            try:
+                valid_keys = litellm.get_supported_openai_params(model=llm.model, custom_llm_provider=self.endpoint_provider, request_type="chat_completion")
+                if not valid_keys:
+                    valid_keys = PARAMS.get("common")
+            except:
+                valid_keys = PARAMS.get("common")
+
+            valid_keys += PARAMS.get("litellm")
+
+            for key in valid_keys:
+                if key in valid_config and valid_config[key]:
+                    val = valid_config[key]
+                    if [key == k for k, v in LLM.model_fields.items()]:
+                        setattr(llm, key, val)
+                    else:
+                        llm.other_valid_config.update({ key: val})
+
+
+        llm.timeout = self.max_execution_time if llm.timeout is None else llm.timeout
+        # llm.max_tokens = self.max_tokens if self.max_tokens else llm.max_tokens
+
+        if llm.provider is None:
+            provider_name = llm.model.split("/")[0]
+            valid_provider = provider_name if provider_name in PROVIDERS else None
+            llm.provider = valid_provider
+
+        if self.callbacks:
+            llm.callbacks = self.callbacks
+            llm._set_callbacks(llm.callbacks)
+
+        if self.respect_context_window == False:
+            llm.context_window_size = DEFAULT_CONTEXT_WINDOW_SIZE
+
+        return llm
+
+
     def _update_llm(self, llm: Any = None, llm_config: Optional[Dict[str, Any]] = None) -> Self:
         """
         Update llm and llm_config of the exsiting agent. (Other conditions will remain the same.)
@@ -599,3 +599,6 @@ class Agent(BaseModel):
 
     def __repr__(self):
         return f"Agent(role={self.role}, goal={self.goal}"
+
+    def __str__(self):
+        return super().__str__()

{versionhq-1.2.1.17 → versionhq-1.2.1.19}/src/versionhq/agent_network/formation.py

@@ -9,6 +9,9 @@ from versionhq.agent_network.model import AgentNetwork, Member, Formation
 from versionhq.agent.inhouse_agents  import vhq_formation_planner
 from versionhq._utils import Logger
 
+import chromadb
+chromadb.api.client.SharedSystemClient.clear_system_cache()
+
 
 def form_agent_network(
         task: str,
@@ -85,7 +88,7 @@ def form_agent_network(
     members = []
     leader = str(res.pydantic.leader_agent) if res.pydantic else str(res.json_dict["leader_agent"])
 
-    created_agents = [Agent(role=item, goal=item) for item in res.pydantic.agent_roles]
+    created_agents = [Agent(role=str(item), goal=str(item)) for item in res.pydantic.agent_roles]
     created_tasks = []
 
     if res.pydantic:
@@ -94,7 +97,7 @@ def form_agent_network(
                 fields = {}
                 for ob in item:
                     try:
-                        field_name = str(ob).lower().replace(" ", "_").replace(":", "_")[0: 10]
+                        field_name = str(ob).lower().split(":")[0].replace(" ", "_")[0: 16]
                         fields[field_name] = (str, Field(default=None))
                     except:
                         pass
@@ -108,7 +111,7 @@ def form_agent_network(
                 fields = {}
                 for ob in item:
                     try:
-                        field_name = str(ob).lower().replace(" ", "_").replace(":", "_")[0: 10]
+                        field_name = str(ob).lower().split(":")[0].replace(" ", "_")[0: 16]
                         fields[field_name] = (str, Field(default=None))
                     except:
                         pass
@@ -137,7 +140,7 @@ def form_agent_network(
             member = Member(agent=created_agents[i], is_manager=is_manager, tasks=[created_tasks[i]])
             members.append(member)
 
-        network_tasks.append(created_tasks[len(created_agents):len(created_tasks)])
+        network_tasks.extend(created_tasks[len(created_agents):len(created_tasks)])
 
 
     if _formation == Formation.SUPERVISING and not [member for member in members if member.is_manager]:

{versionhq-1.2.1.17 → versionhq-1.2.1.19}/src/versionhq/knowledge/model.py

@@ -9,6 +9,7 @@ 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)
     storage: KnowledgeStorage = Field(default_factory=KnowledgeStorage)

{versionhq-1.2.1.17 → versionhq-1.2.1.19}/src/versionhq/storage/rag_storage.py

@@ -129,16 +129,16 @@ class RAGStorage(BaseRAGStorage):
         from chromadb.config import Settings
 
         self._set_embedder_config()
-        chroma_client = chromadb.PersistentClient(
-            path=self.path if self.path else self.storage_file_name,
-            settings=Settings(allow_reset=self.allow_reset),
-        )
-        self.app = chroma_client
-
         try:
+            chroma_client = chromadb.PersistentClient(
+                path=self.path if self.path else self.storage_file_name,
+                settings=Settings(allow_reset=self.allow_reset),
+            )
+            self.app = chroma_client
             self.collection = self.app.get_collection(name=self.type, embedding_function=self.embedder_config)
         except Exception:
-            self.collection = self.app.create_collection(name=self.type, embedding_function=self.embedder_config)
+            if self.app:
+                self.collection = self.app.create_collection(name=self.type, embedding_function=self.embedder_config)
 
 
     def _sanitize_role(self, role: str) -> str:

{versionhq-1.2.1.17 → versionhq-1.2.1.19}/src/versionhq.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.2.1.17
+Version: 1.2.1.19
 Summary: An agentic orchestration framework for building agent networks that handle task automation.
 Author-email: Kuriko Iwai <kuriko@versi0n.io>
 License: MIT License

{versionhq-1.2.1.17 → versionhq-1.2.1.19}/tests/agent/agent_test.py

@@ -244,10 +244,12 @@ def test_agent_with_knowledge_sources():
 
     agent = Agent(role="Information Agent", goal="Provide information based on knowledge sources", knowledge_sources=knowledge_sources)
 
-    assert agent._knowledge.collection_name == f"{agent.role.replace(' ', '_')}"
+    assert  f"{agent.role.replace(' ', '_')}" in agent._knowledge.collection_name
     assert [isinstance(item, StringKnowledgeSource | DoclingSource) for item in agent.knowledge_sources]
     assert agent._knowledge.embedder_config == agent.embedder_config
-    assert agent._knowledge.storage and agent._knowledge.storage.embedding_function and  agent._knowledge.storage.app is not None and agent._knowledge.storage.collection_name is not None
+    assert agent._knowledge.storage and agent._knowledge.storage.embedding_function
+    assert agent._knowledge.storage.app is not None
+    assert agent._knowledge.storage.collection_name is not None
 
     task = Task(description="Answer the following question: What is Kuriko's favorite color?")
 

{versionhq-1.2.1.17 → versionhq-1.2.1.19}/tests/task/task_test.py

@@ -1,19 +1,11 @@
-import sys
-import threading
 from unittest.mock import patch
 from typing import Callable
 
-from pydantic import BaseModel, Field
-
 from versionhq.agent.model import Agent, LLM
 from versionhq.task.model import Task, ResponseField, TaskOutput, TaskExecutionType
 from versionhq.tool.model import Tool, ToolSet
 from versionhq.tool.decorator import tool
 
-sys.setrecursionlimit(2097152)
-threading.stack_size(134217728)
-
-
 
 # def test_async_execute_task():
 #     task = Task(description="Return string: 'test'", type=TaskExecutionType.ASYNC)

{versionhq-1.2.1.17 → versionhq-1.2.1.19}/uv.lock

@@ -3982,7 +3982,7 @@ wheels = [
 
 [[package]]
 name = "versionhq"
-version = "1.2.1.17"
+version = "1.2.1.18"
 source = { editable = "." }
 dependencies = [
     { name = "appdirs" },