versionhq 1.2.3.6__tar.gz → 1.2.3.7__tar.gz

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

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

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

{versionhq-1.2.3.6 → versionhq-1.2.3.7}/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
@@ -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.7}/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.7
 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.7}/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.7}/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

{versionhq-1.2.3.6 → versionhq-1.2.3.7}/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.7"
 source = { editable = "." }
 dependencies = [
     { name = "appdirs" },