versionhq 1.1.7.7__py3-none-any.whl → 1.1.7.8__py3-none-any.whl

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/_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/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/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/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/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/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/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/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/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.dist-info → versionhq-1.1.7.8.dist-info}/METADATA

@@ -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.dist-info → versionhq-1.1.7.8.dist-info}/RECORD

@@ -1,14 +1,14 @@
-versionhq/__init__.py,sha256=UNk3K1Icj6U6bX4hmcE-7PSzT3mYI9f_i_gXybscmWc,871
+versionhq/__init__.py,sha256=KTM1HF1lTl5AiAuFQuGwltg4LEZjHnac9mz8h12MtGg,871
 versionhq/_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/_utils/cache_handler.py,sha256=zDQKzIn7vp-M2-uepHFxgJstjfftZS5mzXKL_-4uVvI,370
 versionhq/_utils/i18n.py,sha256=TwA_PnYfDLA6VqlUDPuybdV9lgi3Frh_ASsb_X8jJo8,1483
 versionhq/_utils/logger.py,sha256=lqRYH45KHMQ4mwE1woa5xNmngYu4O749AYECsnWWpmA,1851
-versionhq/_utils/process_config.py,sha256=ogrhovLbwe0ocQlcohRgBBRtww7C3pk9hikjvgDzT5U,919
+versionhq/_utils/process_config.py,sha256=UqoWD5IR4VLxEDGxIyVUylw_ppXwk8Wx1ynVuD-pUSg,822
 versionhq/_utils/rpm_controller.py,sha256=T7waIGeblu5K58erY4lqVLcPsWM7W9UFdU3DG9Dsk0w,2214
 versionhq/_utils/usage_metrics.py,sha256=c33a_28y8ECUgflsKN3mkNm0fNkWgZmXwybMwIqoKXA,1098
 versionhq/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-versionhq/agent/model.py,sha256=fK7rb10IFrgDDCD78owNRdMsIQ_aj7nmIQdp1WbWJc8,18621
-versionhq/agent/parser.py,sha256=GhoNQo4WloVM3vGnAmt9lnEOTARX7nWMhJE55rF_5Rs,5500
+versionhq/agent/model.py,sha256=tv14XkjrgsFsryFWzaw0w2h0X1T0ffipVK9l4kYbpIE,18598
+versionhq/agent/parser.py,sha256=db5kfk-lR1Ph9-rsTvSeW1NjR6GJ00iaqTNYxJy3N8o,5487
 versionhq/agent/TEMPLATES/Backstory.py,sha256=cdngBx1GEv7nroR46FEhnysnBJ9mEVL763_9np6Skkc,395
 versionhq/agent/TEMPLATES/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -18,25 +18,25 @@ versionhq/clients/customer/model.py,sha256=rQnCv_wdCdrYAsUjyB6X6ULiuWfqcBBoarXcQ
 versionhq/clients/product/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/clients/product/model.py,sha256=Us3UnzYlub6ipBislMN-JrvxZx0ocl9PtQJINJ8XtBA,2385
 versionhq/clients/workflow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-versionhq/clients/workflow/model.py,sha256=qpRCDwULhSLWDFkSYrvXW5m07bKDrwttTmqsYA9ZVP4,5727
+versionhq/clients/workflow/model.py,sha256=GI-cSw-7zOFzLC2Xa4YfCowu3MI0JyH33qDFTqbJLmg,5725
 versionhq/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/llm/llm_vars.py,sha256=YZoXqFBW7XpclUZ14_AAz7WOjoyCXnGcI959GSpX2q0,5343
-versionhq/llm/model.py,sha256=PdwisrlrsDqd6gXwXCyGbGTRTeGZ8SXpt_gfua8qunk,8266
+versionhq/llm/model.py,sha256=mXzSuf1s6MebGT7_yqgNppde0NIlAF8bjIXAp2MZ9Uw,8247
 versionhq/storage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/storage/task_output_storage.py,sha256=RxvF_lSRUuo2Af3zMAuc1aOymx1e6e_5VJ2y757_Hu0,4910
 versionhq/task/__init__.py,sha256=g4mCATnn1mUXxsfQ5p6IpPawr8O421wVIT8kMKEcxQw,180
 versionhq/task/formatter.py,sha256=N8Kmk9vtrMtBdgJ8J7RmlKNMdZWSmV8O1bDexmCWgU0,643
 versionhq/task/log_handler.py,sha256=KJRrcNZgFSKhlNzvtYFnvtp6xukaF1s7ifX9u4zWrN8,1683
-versionhq/task/model.py,sha256=QCNRHulD4RdOVZAUWft4rAqlTIqcbZTqSA-lDUoSfpQ,19314
+versionhq/task/model.py,sha256=bUmERE6AZfs8qh2Hb9LES5BLUlXrkrOUyWxCU27M1ic,19317
 versionhq/team/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-versionhq/team/model.py,sha256=M99sMAlKyS9jQsufH2xoGrv2dDIvaJWsmIQgW-0pT0M,18305
+versionhq/team/model.py,sha256=T_71FarXEzAxrTn_8yYVWMwLS9p-UxSovexZtHYnYn0,18066
 versionhq/team/team_planner.py,sha256=B1UOn_DYVVterUn2CAd80jfO4sViJCCXPJA3abSSugg,2143
 versionhq/tool/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/tool/decorator.py,sha256=Y-j4jkoujD5LUvpe8uf3p5Zagk2XVaRKC9rkIE-2geo,1189
-versionhq/tool/model.py,sha256=JZOEcZRIEfcrjL8DgrFYDt4YNgMF8rXS26RK6D2x9mc,6906
-versionhq/tool/tool_handler.py,sha256=e-2VfG9zFpfPG_oMoPXye93GDovs7FuUASWQwUTLrJ0,1498
-versionhq-1.1.7.7.dist-info/LICENSE,sha256=7CCXuMrAjPVsUvZrsBq9DsxI2rLDUSYXR_qj4yO_ZII,1077
-versionhq-1.1.7.7.dist-info/METADATA,sha256=tHhEWrbEEt-NyUVKKQwMImjFByi5LJiCBVLzZc86L_E,15919
-versionhq-1.1.7.7.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-versionhq-1.1.7.7.dist-info/top_level.txt,sha256=DClQwxDWqIUGeRJkA8vBlgeNsYZs4_nJWMonzFt5Wj0,10
-versionhq-1.1.7.7.dist-info/RECORD,,
+versionhq/tool/model.py,sha256=s-y8323ikd5m5U2HG59ATgFW6L7yIFiPLLdOpeXQ8RI,6874
+versionhq/tool/tool_handler.py,sha256=esUqGp8HoREesai8fmh2klAf04Sjpsacmb03C7F6sNQ,1541
+versionhq-1.1.7.8.dist-info/LICENSE,sha256=7CCXuMrAjPVsUvZrsBq9DsxI2rLDUSYXR_qj4yO_ZII,1077
+versionhq-1.1.7.8.dist-info/METADATA,sha256=xsI-HWU0_Rb0OLH_tSDvIk7E8SWP0U-SjTQfoomW6zU,15919
+versionhq-1.1.7.8.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+versionhq-1.1.7.8.dist-info/top_level.txt,sha256=DClQwxDWqIUGeRJkA8vBlgeNsYZs4_nJWMonzFt5Wj0,10
+versionhq-1.1.7.8.dist-info/RECORD,,