versionhq 1.2.1.22__py3-none-any.whl → 1.2.2.0__py3-none-any.whl

versionhq/__init__.py

@@ -8,7 +8,7 @@ from dotenv import load_dotenv
 load_dotenv(override=True)
 
 from versionhq.agent.model import Agent
-from versionhq.agent_network.model import AgentNetwork, NetworkOutput, Formation, Member, TaskHandlingProcess
+from versionhq.agent_network.model import AgentNetwork, Formation, Member, TaskHandlingProcess
 from versionhq.llm.model import LLM
 from versionhq.llm.llm_vars import LLM_CONTEXT_WINDOW_SIZES, PARAMS, PROVIDERS, MODELS
 from versionhq.clients.customer.model import Customer
@@ -19,7 +19,7 @@ from versionhq.knowledge.source import PDFKnowledgeSource, CSVKnowledgeSource, J
 from versionhq.knowledge.source_docling import DoclingSource
 from versionhq.task_graph.model import TaskStatus, TaskGraph, Node, Edge, DependencyType
 from versionhq.task.model import Task, TaskOutput, ResponseField, TaskExecutionType
-from versionhq.task.evaluate import Evaluation, EvaluationItem
+from versionhq.task.evaluation import Evaluation, EvaluationItem
 from versionhq.tool.model import Tool, ToolSet
 from versionhq.tool.cache_handler import CacheHandler
 from versionhq.tool.tool_handler import ToolHandler
@@ -31,12 +31,11 @@ from versionhq.agent_network.formation import form_agent_network
 from versionhq.task_graph.draft import workflow
 
 
-__version__ = "1.2.1.22"
+__version__ = "1.2.2.0"
 __all__ = [
     "Agent",
 
     "AgentNetwork",
-    "NetworkOutput",
     "Formation",
     "Member",
     "TaskHandlingProcess",

versionhq/_utils/__init__.py

@@ -1,3 +1,3 @@
 from versionhq._utils.logger import Logger
 from versionhq._utils.process_config import process_config
-from versionhq._utils.usage_metrics import UsageMetrics
+from versionhq._utils.vars import KNOWLEDGE_DIRECTORY, MAX_FILE_NAME_LENGTH

versionhq/_utils/usage_metrics.py

@@ -21,3 +21,35 @@ class UsageMetrics(BaseModel):
         self.cached_prompt_tokens += usage_metrics.cached_prompt_tokens
         self.completion_tokens += usage_metrics.completion_tokens
         self.successful_requests += usage_metrics.successful_requests
+
+
+
+# class TokenProcess:
+#     total_tokens: int = 0
+#     prompt_tokens: int = 0
+#     cached_prompt_tokens: int = 0
+#     completion_tokens: int = 0
+#     successful_requests: int = 0
+
+#     def sum_prompt_tokens(self, tokens: int) -> None:
+#         self.prompt_tokens = self.prompt_tokens + tokens
+#         self.total_tokens = self.total_tokens + tokens
+
+#     def sum_completion_tokens(self, tokens: int) -> None:
+#         self.completion_tokens = self.completion_tokens + tokens
+#         self.total_tokens = self.total_tokens + tokens
+
+#     def sum_cached_prompt_tokens(self, tokens: int) -> None:
+#         self.cached_prompt_tokens = self.cached_prompt_tokens + tokens
+
+#     def sum_successful_requests(self, requests: int) -> None:
+#         self.successful_requests = self.successful_requests + requests
+
+#     def get_summary(self) -> UsageMetrics:
+#         return UsageMetrics(
+#             total_tokens=self.total_tokens,
+#             prompt_tokens=self.prompt_tokens,
+#             cached_prompt_tokens=self.cached_prompt_tokens,
+#             completion_tokens=self.completion_tokens,
+#             successful_requests=self.successful_requests,
+#         )

versionhq/agent/inhouse_agents.py

@@ -10,13 +10,15 @@ vhq_client_manager = Agent(
     role="vhq-Client Manager",
     goal="Efficiently communicate with the client on the task progress",
     llm=DEFAULT_MODEL_NAME,
+    maxit=1,
+    max_retry_limit=1,
     with_memory=True,
 )
 
 
 vhq_task_evaluator = Agent(
     role="vhq-Task Evaluator",
-    goal="score the output according to the given evaluation criteria.",
+    goal="score the output according to the given evaluation criteria, taking a step by step approach.",
     llm=DEFAULT_MODEL_NAME,
     llm_config=dict(top_p=0.8, top_k=30, max_tokens=5000, temperature=0.9),
     maxit=1,
@@ -45,4 +47,6 @@ vhq_agent_creator = Agent(
     role="vhq-Agent Creator",
     goal="build an agent that can handle the given task",
     llm="gemini/gemini-2.0-flash-exp",
+    maxit=1,
+    max_retry_limit=1,
 )

versionhq/agent/model.py

@@ -14,7 +14,6 @@ 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.usage_metrics import UsageMetrics
 from versionhq._utils.process_config import process_config
 
 
@@ -22,37 +21,6 @@ load_dotenv(override=True)
 T = TypeVar("T", bound="Agent")
 
 
-class TokenProcess:
-    total_tokens: int = 0
-    prompt_tokens: int = 0
-    cached_prompt_tokens: int = 0
-    completion_tokens: int = 0
-    successful_requests: int = 0
-
-    def sum_prompt_tokens(self, tokens: int) -> None:
-        self.prompt_tokens = self.prompt_tokens + tokens
-        self.total_tokens = self.total_tokens + tokens
-
-    def sum_completion_tokens(self, tokens: int) -> None:
-        self.completion_tokens = self.completion_tokens + tokens
-        self.total_tokens = self.total_tokens + tokens
-
-    def sum_cached_prompt_tokens(self, tokens: int) -> None:
-        self.cached_prompt_tokens = self.cached_prompt_tokens + tokens
-
-    def sum_successful_requests(self, requests: int) -> None:
-        self.successful_requests = self.successful_requests + requests
-
-    def get_summary(self) -> UsageMetrics:
-        return UsageMetrics(
-            total_tokens=self.total_tokens,
-            prompt_tokens=self.prompt_tokens,
-            cached_prompt_tokens=self.cached_prompt_tokens,
-            completion_tokens=self.completion_tokens,
-            successful_requests=self.successful_requests,
-        )
-
-
 # @track_agent()
 class Agent(BaseModel):
     """
@@ -62,7 +30,6 @@ class Agent(BaseModel):
     __hash__ = object.__hash__
     _rpm_controller: Optional[RPMController] = PrivateAttr(default=None)
     _request_within_rpm_limit: Any = PrivateAttr(default=None)
-    _token_process: TokenProcess = PrivateAttr(default_factory=TokenProcess)
     _times_executed: int = PrivateAttr(default=0)
     _logger_config: Dict[str, Any] = PrivateAttr(default=dict(verbose=True, info_file_save=True))
     config: Optional[Dict[str, Any]] = Field(default=None, exclude=True, description="values to add to the Agent class")
@@ -92,7 +59,7 @@ class Agent(BaseModel):
     user_prompt_template: Optional[str] = Field(default=None, description="abs. file path to user prompt template")
 
     # task execution rules
-    networks: Optional[List[Any]] = Field(default_factory=list, description="store a list of agent networks that the agent belong as a member")
+    networks: Optional[List[Any]] = Field(default_factory=list, description="store a list of agent networks that the agent belongs to as a member")
     allow_delegation: bool = Field(default=False, description="whether to delegate the task to another agent")
     max_retry_limit: int = Field(default=2, description="max. number of task retries when an error occurs")
     maxit: Optional[int] = Field(default=25, description="max. number of total optimization loops conducted when an error occurs")
@@ -428,10 +395,10 @@ class Agent(BaseModel):
 
             if tool_res_as_final:
                 raw_response = self.func_calling_llm.call(messages=messages, tools=tools, tool_res_as_final=True)
-                task.tokens = self.func_calling_llm._tokens
+                task._tokens = self.func_calling_llm._tokens
             else:
                 raw_response = self.llm.call(messages=messages, response_format=response_format, tools=tools)
-                task.tokens = self.llm._tokens
+                task._tokens = self.llm._tokens
 
             task_execution_counter += 1
             Logger(**self._logger_config, filename=self.key).log(level="info", message=f"Agent response: {raw_response}", color="green")
@@ -446,7 +413,7 @@ class Agent(BaseModel):
                         self._rpm_controller.check_or_wait()
 
                     raw_response = self.llm.call(messages=messages, response_format=response_format, tools=tools)
-                    task.tokens = self.llm._tokens
+                    task._tokens = self.llm._tokens
                     iterations += 1
 
                 task_execution_counter += 1