versionhq 1.1.7.7__tar.gz → 1.1.7.8__tar.gz

{versionhq-1.1.7.7 → versionhq-1.1.7.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: versionhq
-Version: 1.1.7.7
+Version: 1.1.7.8
 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.7.7 → versionhq-1.1.7.8}/pyproject.toml

@@ -15,7 +15,7 @@ exclude = ["test*", "__pycache__"]
 
 [project]
 name = "versionhq"
-version = "1.1.7.7"
+version = "1.1.7.8"
 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.7.7 → versionhq-1.1.7.8}/src/versionhq/__init__.py

@@ -17,7 +17,7 @@ from versionhq.team.model import Team, TeamOutput
 from versionhq.tool.model import Tool
 
 
-__version__ = "1.1.7.7"
+__version__ = "1.1.7.8"
 __all__ = [
     "Agent",
     "Customer",

{versionhq-1.1.7.7 → versionhq-1.1.7.8}/src/versionhq/_utils/process_config.py

@@ -2,9 +2,7 @@ from typing import Any, Dict, Type
 from pydantic import BaseModel
 
 
-def process_config(
-    values_to_update: Dict[str, Any], model_class: Type[BaseModel]
-) -> Dict[str, Any]:
+def process_config(values_to_update: Dict[str, Any], model_class: Type[BaseModel]) -> Dict[str, Any]:
     """
     Process the config dictionary and update the values accordingly.
     Refer to the Pydantic model class for field validation.
@@ -15,7 +13,7 @@ def process_config(
     else:
         return values_to_update
 
-    # copy values from config to the model's attributes if the attribute isn't already set.
+
     for key, value in config.items():
         if key not in model_class.model_fields or values_to_update.get(key) is not None:
             continue

{versionhq-1.1.7.7 → versionhq-1.1.7.8}/src/versionhq/agent/model.py

@@ -1,7 +1,7 @@
 import os
 import uuid
 from abc import ABC
-from typing import Any, Dict, List, Optional, TypeVar, Union
+from typing import Any, Dict, List, Optional, TypeVar
 from dotenv import load_dotenv
 from pydantic import UUID4, BaseModel, Field, InstanceOf, PrivateAttr, model_validator, field_validator
 from pydantic_core import PydanticCustomError
@@ -22,7 +22,7 @@ load_dotenv(override=True)
 T = TypeVar("T", bound="Agent")
 
 
-# def _format_answer(agent, answer: str) -> Union[AgentAction, AgentFinish]:
+# def _format_answer(agent, answer: str) -> AgentAction | AgentFinish:
 #     return AgentParser(agent=agent).parse(answer)
 
 # def mock_agent_ops_provider():
@@ -111,8 +111,8 @@ class Agent(ABC, BaseModel):
     step_callback: Optional[Any] = Field(default=None,description="Callback to be executed after each step of the agent execution")
 
     # llm settings cascaded to the LLM model
-    llm: Union[str, InstanceOf[LLM], Any] = Field(default=None)
-    function_calling_llm: Union[str, InstanceOf[LLM], Any] = Field(default=None)
+    llm: str | InstanceOf[LLM] | Any = Field(default=None)
+    function_calling_llm: str | InstanceOf[LLM] | Any = Field(default=None)
     respect_context_window: bool = Field(default=True,description="Keep messages under the context window size by summarizing content")
     max_tokens: Optional[int] = Field(default=None, description="max. number of tokens for the agent's execution")
     max_execution_time: Optional[int] = Field(default=None, description="max. execution time for an agent to execute a task")

{versionhq-1.1.7.7 → versionhq-1.1.7.8}/src/versionhq/agent/parser.py

@@ -1,5 +1,5 @@
 import re
-from typing import Any, Union
+from typing import Any
 from json_repair import repair_json
 
 from versionhq._utils.i18n import I18N
@@ -70,7 +70,7 @@ class AgentParser:
     def __init__(self, agent: Any):
         self.agent = agent
 
-    def parse(self, text: str) -> Union[AgentAction, AgentFinish]:
+    def parse(self, text: str) -> AgentAction | AgentFinish:
         thought = self._extract_thought(text)
         includes_answer = FINAL_ANSWER_ACTION in text
         regex = (

{versionhq-1.1.7.7 → versionhq-1.1.7.8}/src/versionhq/clients/workflow/model.py

@@ -1,8 +1,8 @@
 import uuid
 from abc import ABC
 from datetime import date, datetime, time, timedelta
-from typing import Any, Dict, List, Union, Callable, Type, Optional, get_args, get_origin
-from pydantic import UUID4, InstanceOf, BaseModel, ConfigDict, Field, create_model, field_validator, model_validator
+from typing import Any, Dict, List, Callable, Type, Optional, get_args, get_origin
+from pydantic import UUID4, InstanceOf, BaseModel, ConfigDict, Field, field_validator, model_validator
 from pydantic_core import PydanticCustomError
 
 from versionhq.clients.product.model import Product
@@ -12,7 +12,7 @@ from versionhq.team.model import Team
 
 
 class ScoreFormat:
-    def __init__(self, rate: Union[float, int] = 0, weight: int = 1):
+    def __init__(self, rate: float | int = 0, weight: int = 1):
         self.rate = rate
         self.weight = weight
         self.aggregate = rate * weight
@@ -39,7 +39,7 @@ class Score:
 
 
     def result(self) -> int:
-        aggregate_score = self.brand_tone.aggregate + self.audience.aggregate + self.track_record.aggregate
+        aggregate_score = int(self.brand_tone.aggregate) + int(self.audience.aggregate) + int(self.track_record.aggregate)
         denominator = self.brand_tone.weight + self.audience.weight + self.track_record.weight
 
         for k, v in self.kwargs.items():
@@ -57,11 +57,12 @@ class MessagingComponent(ABC, BaseModel):
     layer_id: int = Field(default=0, description="add id of the layer: 0, 1, 2")
     message: str = Field(default=None, max_length=1024, description="text message content to be sent")
     interval: Optional[str] = Field(
-        default=None,description="interval to move on to the next layer. if this is the last layer, set as `None`")
-    score: Union[float, InstanceOf[Score]] = Field(default=None)
+        default=None, description="interval to move on to the next layer. if this is the last layer, set as `None`"
+    )
+    score: float | InstanceOf[Score] = Field(default=None)
 
 
-    def store_scoring_result(self, scoring_subject: str, score_raw: Union[int, Score, ScoreFormat] = None):
+    def store_scoring_result(self, scoring_subject: str, score_raw: int | Score | ScoreFormat = None):
         """
         Set up the `score` field
         """
@@ -109,8 +110,11 @@ class MessagingWorkflow(ABC, BaseModel):
     product: InstanceOf[Product] = Field(default=None)
     customer: InstanceOf[Customer] = Field(default=None)
 
-    metrics: Union[List[Dict[str, Any]], List[str]] = Field(
-        default=None, max_length=256, description="store metrics that used to predict and track the performance of this workflow.")
+    metrics: List[Dict[str, Any]] | List[str] = Field(
+        default=None,
+        max_length=256,
+        description="store metrics that used to predict and track the performance of this workflow."
+    )
 
 
     @property

{versionhq-1.1.7.7 → versionhq-1.1.7.8}/src/versionhq/llm/model.py

@@ -7,7 +7,7 @@ import litellm
 from dotenv import load_dotenv
 from litellm import get_supported_openai_params
 from contextlib import contextmanager
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Dict, List, Optional
 
 from versionhq.llm.llm_vars import LLM_CONTEXT_WINDOW_SIZES
 from versionhq.task import TaskOutputFormat
@@ -103,7 +103,7 @@ class LLM:
     def __init__(
         self,
         model: str,
-        timeout: Optional[Union[float, int]] = None,
+        timeout: Optional[float | int] = None,
         max_tokens: Optional[int] = None,
         max_completion_tokens: Optional[int] = None,
         context_window_size: Optional[int] = DEFAULT_CONTEXT_WINDOW,
@@ -111,7 +111,7 @@ class LLM:
         temperature: Optional[float] = None,
         top_p: Optional[float] = None,
         n: Optional[int] = None,
-        stop: Optional[Union[str, List[str]]] = None,
+        stop: Optional[str | List[str]] = None,
         presence_penalty: Optional[float] = None,
         frequency_penalty: Optional[float] = None,
         logit_bias: Optional[Dict[int, float]] = None,

{versionhq-1.1.7.7 → versionhq-1.1.7.8}/src/versionhq/task/model.py

@@ -3,10 +3,10 @@ import threading
 import uuid
 from concurrent.futures import Future
 from hashlib import md5
-from typing import Any, Dict, List, Set, Optional, Tuple, Callable, Union, Type
+from typing import Any, Dict, List, Set, Optional, Tuple, Callable, Type
 from typing_extensions import Annotated
 
-from pydantic import UUID4, BaseModel, Field, PrivateAttr, field_validator, model_validator, create_model
+from pydantic import UUID4, BaseModel, Field, PrivateAttr, field_validator, model_validator, create_model, InstanceOf
 from pydantic_core import PydanticCustomError
 
 from versionhq._utils.process_config import process_config
@@ -49,7 +49,7 @@ class ResponseField(BaseModel):
             return value
 
 
-    def create_pydantic_model(self, result: Dict, base_model: Union[BaseModel | Any]) -> Any:
+    def create_pydantic_model(self, result: Dict, base_model: InstanceOf[BaseModel] | Any) -> Any:
         for k, v in result.items():
             if k is not self.title:
                 pass
@@ -70,7 +70,7 @@ class TaskOutput(BaseModel):
 
     task_id: UUID4 = Field(default_factory=uuid.uuid4, frozen=True, description="store Task ID")
     raw: str = Field(default="", description="Raw output of the task")
-    json_dict: Union[Dict[str, Any]] = Field(default=None, description="`raw` converted to dictionary")
+    json_dict: Dict[str, Any] = Field(default=None, description="`raw` converted to dictionary")
     pydantic: Optional[Any] = Field(default=None, description="`raw` converted to the abs. pydantic model")
 
     def __str__(self) -> str:

{versionhq-1.1.7.7 → versionhq-1.1.7.8}/src/versionhq/team/model.py

@@ -6,7 +6,7 @@ from enum import Enum
 from dotenv import load_dotenv
 from concurrent.futures import Future
 from hashlib import md5
-from typing import Any, Dict, List, TYPE_CHECKING, Callable, Optional, Tuple, Union
+from typing import Any, Dict, List, TYPE_CHECKING, Callable, Optional, Tuple
 from pydantic import UUID4, InstanceOf, Json, BaseModel, Field, PrivateAttr, field_validator, model_validator
 from pydantic._internal._generate_schema import GenerateSchema
 from pydantic_core import PydanticCustomError, core_schema
@@ -61,7 +61,7 @@ class TeamOutput(BaseModel):
     team_id: UUID4 = Field(default_factory=uuid.uuid4, frozen=True, description="store the team ID that generate the TeamOutput")
     raw: str = Field(default="", description="raw output of the team lead task handled by the team leader")
     pydantic: Optional[Any] = Field(default=None, description="`raw` converted to the abs. pydantic model")
-    json_dict: Union[Dict[str, Any]] = Field(default=None, description="`raw` converted to dictionary")
+    json_dict: Dict[str, Any] = Field(default=None, description="`raw` converted to dictionary")
     task_output_list: list[TaskOutput] = Field(default=list, description="store output of all the tasks that the team has executed")
     token_usage: UsageMetrics = Field(default=dict, description="processed token summary")
 
@@ -200,10 +200,6 @@ class Team(BaseModel):
         if v:
             raise PydanticCustomError("may_not_set_field", "The 'id' field cannot be set by the user.", {})
 
-    # @field_validator("config", mode="before")
-    # @classmethod
-    # def check_config_type(cls, v: Union[Json, Dict[str, Any]]) -> Union[Json, Dict[str, Any]]:
-    #     return json.loads(v) if isinstance(v, Json) else v
 
     @model_validator(mode="after")
     def check_manager_llm(self):

{versionhq-1.1.7.7 → versionhq-1.1.7.8}/src/versionhq/tool/model.py

@@ -1,15 +1,7 @@
 from abc import ABC
 from inspect import signature
 from typing import Any, Dict, Callable, Type, Optional, get_args, get_origin
-from pydantic import (
-    InstanceOf,
-    BaseModel,
-    ConfigDict,
-    Field,
-    create_model,
-    field_validator,
-    model_validator,
-)
+from pydantic import InstanceOf, BaseModel, ConfigDict, Field, create_model, field_validator, model_validator
 
 from versionhq._utils.cache_handler import CacheHandler
 
@@ -135,6 +127,7 @@ class Tool(ABC, BaseModel):
 
         return cls(name=tool_name, func=func, args_schema=args_schema)
 
+
     def run(self, *args, **kwargs) -> Any:
         """
         Use the tool.

{versionhq-1.1.7.7 → versionhq-1.1.7.8}/src/versionhq/tool/tool_handler.py

@@ -1,4 +1,5 @@
-from typing import Any, Optional, Union
+from typing import Any, Optional
+from pydantic import InstanceOf
 from versionhq.tool.model import ToolCalled, InstructorToolCalled, CacheTool
 from versionhq._utils.cache_handler import CacheHandler
 
@@ -23,7 +24,7 @@ class ToolHandler:
 
     def record_last_tool_used(
         self,
-        last_used_tool: Union[ToolCalled, InstructorToolCalled],
+        last_used_tool: InstanceOf[ToolCalled] | InstanceOf[InstructorToolCalled],
         output: str,
         should_cache: bool = True,
     ) -> Any:

{versionhq-1.1.7.7 → versionhq-1.1.7.8}/src/versionhq.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: versionhq
-Version: 1.1.7.7
+Version: 1.1.7.8
 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.7.7 → versionhq-1.1.7.8}/tests/clients/workflow_test.py

@@ -19,3 +19,14 @@ def test_store_scores():
 
     assert messaging_component.score is not None
     assert messaging_component.score.result() is not None
+
+
+def test_score_result():
+    messaging_component = MessagingComponent(message="demo")
+    score_raw = 15
+    messaging_component.store_scoring_result("demo", score_raw=score_raw)
+
+    result = messaging_component.score.result()
+
+    assert result is not None
+    assert result != 0

{versionhq-1.1.7.7 → versionhq-1.1.7.8}/tests/task/task_test.py

@@ -1,7 +1,7 @@
 import os
 import pytest
 from unittest.mock import patch
-from typing import Union, Dict, Any
+from typing import Dict, Any
 from versionhq.agent.model import Agent
 from versionhq.task.model import Task, ResponseField, TaskOutput, ConditionalTask
 
@@ -119,11 +119,11 @@ def test_sync_execute_task_with_task_context():
     else:
         assert hasattr(res.pydantic, "test1")
         if res.pydantic.test1:
-            assert type(res.pydantic.test1) == Union[int, str]
+            assert type(res.pydantic.test1) == int | str
 
         assert hasattr(res.pydantic, "test2")
         if res.pydantic.test2:
-            assert type(res.pydantic.test2) == Union[list, str]
+            assert type(res.pydantic.test2) == list | str
 
     assert sub_task.output is not None
     assert sub_task.output.json_dict is not None
@@ -179,11 +179,11 @@ def test_sync_execute_task_with_prompt_context():
     else:
         assert hasattr(res.pydantic, "test1")
         if res.pydantic.test1:
-            assert type(res.pydantic.test1) == Union[int, str]
+            assert type(res.pydantic.test1) == int | str
 
         assert hasattr(res.pydantic, "test2")
         if res.pydantic.test2:
-            assert type(res.pydantic.test2) == Union[list, str]
+            assert type(res.pydantic.test2) == list | str
 
     assert sub_task.output is not None
     assert sub_task.output.json_dict is not None

{versionhq-1.1.7.7 → versionhq-1.1.7.8}/uv.lock

@@ -1644,7 +1644,7 @@ wheels = [
 
 [[package]]
 name = "versionhq"
-version = "1.1.7.7"
+version = "1.1.7.8"
 source = { editable = "." }
 dependencies = [
     { name = "appdirs" },