versionhq 1.2.1.17__tar.gz → 1.2.1.18__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.2.1.17
+Version: 1.2.1.18
 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.18}/pyproject.toml

@@ -15,7 +15,7 @@ exclude = ["test*", "__pycache__", "*.egg-info"]
 
 [project]
 name = "versionhq"
-version = "1.2.1.17"
+version = "1.2.1.18"
 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.18}/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.18"
 __all__ = [
     "Agent",
 

{versionhq-1.2.1.17 → versionhq-1.2.1.18}/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.18}/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:
         """
@@ -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.18}/src/versionhq/agent_network/formation.py

@@ -94,7 +94,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 +108,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 +137,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.18}/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.18
 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.18}/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)