versionhq 1.2.3.6__tar.gz → 1.2.3.8__tar.gz

{versionhq-1.2.3.6 → versionhq-1.2.3.8}/PKG-INFO

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

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

@@ -32,7 +32,7 @@ from versionhq.agent_network.formation import form_agent_network
 from versionhq.task_graph.draft import workflow
 
 
-__version__ = "1.2.3.6"
+__version__ = "1.2.3.8"
 __all__ = [
     "Agent",
 

{versionhq-1.2.3.6 → versionhq-1.2.3.8}/src/versionhq/agent/model.py

@@ -7,13 +7,12 @@ from dotenv import load_dotenv
 from pydantic import UUID4, BaseModel, Field, InstanceOf, PrivateAttr, model_validator, field_validator
 from pydantic_core import PydanticCustomError
 
-from versionhq.llm.model import LLM, DEFAULT_CONTEXT_WINDOW_SIZE, DEFAULT_MODEL_NAME, PROVIDERS
+from versionhq.agent.rpm_controller import RPMController
 from versionhq.tool.model import Tool, ToolSet, BaseTool
 from versionhq.knowledge.model import BaseKnowledgeSource, Knowledge
 from versionhq.memory.contextual_memory import ContextualMemory
 from versionhq.memory.model import ShortTermMemory, LongTermMemory, UserMemory
 from versionhq._utils.logger import Logger
-from versionhq.agent.rpm_controller import RPMController
 from versionhq._utils.process_config import process_config
 
 
@@ -68,8 +67,8 @@ class Agent(BaseModel):
     callbacks: Optional[List[Callable]] = Field(default_factory=list, description="callback functions to execute after any task execution")
 
     # llm settings cascaded to the LLM model
-    llm: str | InstanceOf[LLM] | Dict[str, Any] = Field(default=None)
-    func_calling_llm: str | InstanceOf[LLM] | Dict[str, Any] = Field(default=None)
+    llm: Any = Field(default=None, description="store LLM object")
+    func_calling_llm: Any = Field(default=None, description="store LLM object")
     respect_context_window: bool = Field(default=True, description="keep messages under the context window size")
     max_execution_time: Optional[int] = Field(default=None, description="max. task execution time in seconds")
     max_rpm: Optional[int] = Field(default=None, description="max. number of requests per minute")
@@ -78,7 +77,6 @@ class Agent(BaseModel):
     # # cache, error, ops handling
     # formatting_errors: int = Field(default=0, description="number of formatting errors.")
 
-
     @field_validator("id", mode="before")
     @classmethod
     def _deny_user_set_id(cls, v: Optional[UUID4]) -> None:
@@ -106,6 +104,8 @@ class Agent(BaseModel):
         """
         Set up `llm` and `func_calling_llm` as valid LLM objects using the given kwargs.
         """
+        from versionhq.llm.model import LLM, DEFAULT_MODEL_NAME
+
         self.llm = self._convert_to_llm_object(llm=self.llm)
 
         func_calling_llm = self.func_calling_llm if self.func_calling_llm else self.llm if self.llm else None
@@ -151,7 +151,7 @@ class Agent(BaseModel):
                                 setattr(tool, k, v)
                         tool_list.append(tool)
 
-                case callable():
+                case func if callable(func):
                     tool = Tool(func=item)
                     tool_list.append(tool)
 
@@ -267,11 +267,13 @@ class Agent(BaseModel):
         return self
 
 
-    def _convert_to_llm_object(self, llm: Any = None) -> LLM:
+    def _convert_to_llm_object(self, llm: Any = None): # returns LLM object
         """
         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.
         """
+        from versionhq.llm.model import LLM, DEFAULT_MODEL_NAME
+
         llm = llm if llm else self.llm if self.llm else DEFAULT_MODEL_NAME
 
         if not llm:
@@ -279,11 +281,11 @@ class Agent(BaseModel):
 
         match llm:
             case LLM():
-                return self._set_llm_params(llm=llm, config=self.llm_config)
+                return self._set_llm_params(llm_obj=llm, config=self.llm_config)
 
             case str():
                 llm = LLM(model=llm)
-                return self._set_llm_params(llm=llm, config=self.llm_config)
+                return self._set_llm_params(llm_obj=llm, config=self.llm_config)
 
             case dict():
                 model_name = llm.pop("model_name", llm.pop("deployment_name", str(llm)))
@@ -301,13 +303,20 @@ class Agent(BaseModel):
                     "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, config=config)
+                return self._set_llm_params(llm_obj=llm, config=config)
 
 
-    def _set_llm_params(self, llm: LLM, config: Dict[str, Any] = None) -> LLM:
+    def _set_llm_params(self, llm_obj, config: Dict[str, Any] = None):  # returns LLM object
         """
         Add valid params to the LLM object.
         """
+        from versionhq.llm.model import LLM, DEFAULT_CONTEXT_WINDOW_SIZE
+        from versionhq.llm.llm_vars import PROVIDERS
+
+        llm = llm_obj if isinstance(llm_obj, LLM) else None
+
+        if not llm: return
+
         if llm.provider is None:
             provider_name = llm.model.split("/")[0]
             valid_provider = provider_name if provider_name in PROVIDERS else None
@@ -352,6 +361,8 @@ class Agent(BaseModel):
         """
         Fine-tuned the base model using OpenAI train framework.
         """
+        from versionhq.llm.model import LLM
+
         if not isinstance(self.llm, LLM):
             pass
 

{versionhq-1.2.3.6 → versionhq-1.2.3.8}/src/versionhq/tool/model.py

@@ -144,6 +144,11 @@ class BaseTool(ABC, BaseModel):
 class Tool(BaseTool):
     func: Callable = Field(default=None)
 
+    # def __init__(self, func: Callable = None, *args, **kwargs):
+    #     self.func = func
+    #     super().__init__(*args, **kwargs)
+
+
     @model_validator(mode="after")
     def validate_func(self) -> Self:
         if not self.func and not self._run:

{versionhq-1.2.3.6 → versionhq-1.2.3.8}/src/versionhq.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.2.3.6
+Version: 1.2.3.8
 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.3.6 → versionhq-1.2.3.8}/tests/agent/agent_test.py

@@ -172,23 +172,22 @@ def test_build_agent_with_llm_and_func_llm_object():
     assert isinstance(agent.func_calling_llm, LLM)
 
 
-test_build_agent_with_llm()
-test_build_agent_with_llm_config()
-test_build_agent_with_llm_object()
-test_build_agent_with_llm_and_func_llm_config()
-test_build_agent_with_llm_and_func_llm_object()
-
-
-def test_agent_with_random_dict_tools():
+def test_agent_with_tools():
     def empty_func():
         return "empty function"
 
-    agent = Agent(role="demo", goal="test a tool", tools=[dict(name="tool 1", func=empty_func), ])
+    def custom_tool(query: str) -> str:
+        return query
+
+    agent = Agent(role="demo", goal="test a tool", tools=[dict(name="tool 1", func=empty_func), lambda x: x, custom_tool,])
 
-    assert [tool._run() == "empty function" for tool in agent.tools]
+    assert agent.tools[0]._run() ==  "empty function"
     assert agent.tools[0].name == "tool 1"
+    assert agent.tools[1]._run(x="hey") == "hey"
+    assert agent.tools[1].name == "random_func"
+    assert agent.tools[2]._run(query="hey") == "hey"
+    assert agent.tools[2].name == "custom_tool"
 
-test_agent_with_random_dict_tools()
 
 def test_agent_with_custom_tools():
     def send_message(message: str) -> str:

{versionhq-1.2.3.6 → versionhq-1.2.3.8}/tests/agent/doc_test.py

@@ -23,7 +23,6 @@ def test_docs_core_agent_b1():
     assert "gemini-2.0" in agent.llm.model and isinstance(agent.llm, LLM)
 
 
-
 def test_docs_core_agent_b3():
     import versionhq as vhq
 
@@ -135,7 +134,6 @@ def test_docs_core_agent_e2():
     assert litellm.callbacks == agent.callbacks and len(agent.callbacks) == 2
 
 
-
 def test_docs_core_agent_f():
     import versionhq as vhq
 
@@ -159,7 +157,6 @@ def test_docs_core_agent_f():
     assert agent.llm.llm_config["stop"] == "answer"
 
 
-
 def test_docs_core_agent_g():
     from versionhq import Agent
     from versionhq.task.model import Task

{versionhq-1.2.3.6 → versionhq-1.2.3.8}/tests/agent_network/agent_network_test.py

@@ -326,6 +326,3 @@ def test_network_eval():
 
     res, _ = network.launch()
     assert res._tokens and res.latency
-
-
-test_network_eval()

{versionhq-1.2.3.6 → versionhq-1.2.3.8}/tests/agent_network/doc_test.py

@@ -12,7 +12,6 @@ def test_doc_an_formation_a():
     assert network.tasks # auto-defined sub-tasks to achieve the main task goal
 
 
-
 def test_doc_an_formation_b():
     import versionhq as vhq
     from pydantic import BaseModel
@@ -32,6 +31,7 @@ def test_doc_an_formation_b():
     assert network.members
     assert network.tasks
 
+
 def test_doc_an_formation_c():
     import versionhq as vhq
     from pydantic import BaseModel
@@ -86,7 +86,6 @@ def test_doc_an_formation_d():
     assert network.formation == vhq.Formation.SUPERVISING
 
 
-
 def test_doc_an_config_a():
     import versionhq as vhq
 
@@ -100,6 +99,7 @@ def test_doc_an_config_a():
     )
     assert isinstance(network.members[0].agent, vhq.Agent)
 
+
 def test_doc_an_config_b():
     import versionhq as vhq
 
@@ -114,6 +114,7 @@ def test_doc_an_config_b():
 
     assert network.formation == vhq.Formation.SQUAD
 
+
 def test_doc_an_config_c():
     import versionhq as vhq
 

{versionhq-1.2.3.6 → versionhq-1.2.3.8}/tests/formation_test.py

@@ -14,7 +14,6 @@ def test_minimum_inputs():
     assert network.managers if network.formation in (vhq.Formation.SUPERVISING, vhq.Formation.HYBRID) else not network.managers
 
 
-
 def _test_specific_formation():
     import versionhq as vhq
 

{versionhq-1.2.3.6 → versionhq-1.2.3.8}/tests/knowledge/knowledge_test.py

@@ -282,7 +282,6 @@ def test_pdf_knowledge_source(mock_vector_db):
     mock_vector_db.query.assert_called_once()
 
 
-
 # @pytest.mark.vcr(filter_headers=["authorization"])
 def test_csv_knowledge_source(mock_vector_db, tmpdir):
     csv_content = [

{versionhq-1.2.3.6 → versionhq-1.2.3.8}/tests/llm/llm_connection_test.py

@@ -1,7 +1,8 @@
 import pytest
 from typing import Any, Callable
 
-from versionhq.agent.model import Agent, LLM
+from versionhq.agent.model import Agent
+from versionhq.llm.model import LLM
 from versionhq.task.model import Task
 from versionhq.tool.model import Tool
 

{versionhq-1.2.3.6 → versionhq-1.2.3.8}/tests/task/task_test.py

@@ -1,7 +1,8 @@
 from unittest.mock import patch
 from typing import Callable
 
-from versionhq.agent.model import Agent, LLM
+from versionhq.agent.model import Agent
+from versionhq.llm.model import LLM
 from versionhq.task.model import Task, ResponseField, TaskOutput
 from versionhq.tool.model import Tool, ToolSet
 from versionhq.tool.decorator import tool
@@ -176,7 +177,6 @@ def test_task_with_tools():
     assert res.tool_output is not None
 
 
-
 def test_task_without_response_format():
     task = Task(description="return a simple output with any random values.")
     res = task.execute()

{versionhq-1.2.3.6 → versionhq-1.2.3.8}/tests/tool/tool_test.py

@@ -119,7 +119,6 @@ def test_custom_tool_with_cache():
     assert tool.tool_handler.last_used_tool == ToolSet(tool=tool, kwargs=dict(demo_list=["demo1", "demo2"]))
 
 
-
 def test_cache_tool():
     my_tool = Tool(
         name="my tool",

{versionhq-1.2.3.6 → versionhq-1.2.3.8}/tests/usecase_test.py

@@ -59,7 +59,6 @@ def test_demo_agent_customization():
     assert res.raw is not None
 
 
-
 def test_solo_tg_eval():
     import versionhq as vhq
 

{versionhq-1.2.3.6 → versionhq-1.2.3.8}/uv.lock

@@ -2515,7 +2515,7 @@ name = "nvidia-cudnn-cu12"
 version = "9.1.0.70"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
-    { name = "nvidia-cublas-cu12", marker = "(platform_machine != 'aarch64' and sys_platform == 'linux') or sys_platform == 'win32'" },
+    { name = "nvidia-cublas-cu12", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
 ]
 wheels = [
     { url = "https://files.pythonhosted.org/packages/9f/fd/713452cd72343f682b1c7b9321e23829f00b842ceaedcda96e742ea0b0b3/nvidia_cudnn_cu12-9.1.0.70-py3-none-manylinux2014_x86_64.whl", hash = "sha256:165764f44ef8c61fcdfdfdbe769d687e06374059fbb388b6c89ecb0e28793a6f", size = 664752741 },
@@ -2526,7 +2526,7 @@ name = "nvidia-cufft-cu12"
 version = "11.2.1.3"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
-    { name = "nvidia-nvjitlink-cu12", marker = "(platform_machine != 'aarch64' and sys_platform == 'linux') or sys_platform == 'win32'" },
+    { name = "nvidia-nvjitlink-cu12", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
 ]
 wheels = [
     { url = "https://files.pythonhosted.org/packages/27/94/3266821f65b92b3138631e9c8e7fe1fb513804ac934485a8d05776e1dd43/nvidia_cufft_cu12-11.2.1.3-py3-none-manylinux2014_x86_64.whl", hash = "sha256:f083fc24912aa410be21fa16d157fed2055dab1cc4b6934a0e03cba69eb242b9", size = 211459117 },
@@ -2545,9 +2545,9 @@ name = "nvidia-cusolver-cu12"
 version = "11.6.1.9"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
-    { name = "nvidia-cublas-cu12", marker = "(platform_machine != 'aarch64' and sys_platform == 'linux') or sys_platform == 'win32'" },
-    { name = "nvidia-cusparse-cu12", marker = "(platform_machine != 'aarch64' and sys_platform == 'linux') or sys_platform == 'win32'" },
-    { name = "nvidia-nvjitlink-cu12", marker = "(platform_machine != 'aarch64' and sys_platform == 'linux') or sys_platform == 'win32'" },
+    { name = "nvidia-cublas-cu12", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
+    { name = "nvidia-cusparse-cu12", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
+    { name = "nvidia-nvjitlink-cu12", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
 ]
 wheels = [
     { url = "https://files.pythonhosted.org/packages/3a/e1/5b9089a4b2a4790dfdea8b3a006052cfecff58139d5a4e34cb1a51df8d6f/nvidia_cusolver_cu12-11.6.1.9-py3-none-manylinux2014_x86_64.whl", hash = "sha256:19e33fa442bcfd085b3086c4ebf7e8debc07cfe01e11513cc6d332fd918ac260", size = 127936057 },
@@ -2558,7 +2558,7 @@ name = "nvidia-cusparse-cu12"
 version = "12.3.1.170"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
-    { name = "nvidia-nvjitlink-cu12", marker = "(platform_machine != 'aarch64' and sys_platform == 'linux') or sys_platform == 'win32'" },
+    { name = "nvidia-nvjitlink-cu12", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
 ]
 wheels = [
     { url = "https://files.pythonhosted.org/packages/db/f7/97a9ea26ed4bbbfc2d470994b8b4f338ef663be97b8f677519ac195e113d/nvidia_cusparse_cu12-12.3.1.170-py3-none-manylinux2014_x86_64.whl", hash = "sha256:ea4f11a2904e2a8dc4b1833cc1b5181cde564edd0d5cd33e3c168eff2d1863f1", size = 207454763 },
@@ -4791,7 +4791,7 @@ wheels = [
 
 [[package]]
 name = "versionhq"
-version = "1.2.3.6"
+version = "1.2.3.8"
 source = { editable = "." }
 dependencies = [
     { name = "appdirs" },