versionhq 1.2.1.22__tar.gz → 1.2.2.0__tar.gz

{versionhq-1.2.1.22 → versionhq-1.2.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.2.1.22
+Version: 1.2.2.0
 Summary: An agentic orchestration framework for building agent networks that handle task automation.
 Author-email: Kuriko Iwai <kuriko@versi0n.io>
 License: MIT License
@@ -259,7 +259,7 @@ agent.update(
       task="YOUR AMAZING TASK OVERVIEW",
       expected_outcome="YOUR OUTCOME EXPECTATION",
    )
-   res = network.launch()
+   res, _ = network.launch()
    ```
 
 This will form a network with multiple agents on `Formation` and return `TaskOutput` object with output in JSON, plane text, Pydantic model format with evaluation.
@@ -290,15 +290,16 @@ task = vhq.Task(
    callback_kwargs=dict(message="Hi! Here is the result: ")
 )
 
-res = task.execute(context="amazing context to consider.")
-print(res)
+res = task.execute(context="context to consider")
+
+assert isinstance(res, vhq.TaskOutput)
 ```
 
 
 This will return a `TaskOutput` object that stores response in plane text, JSON, and Pydantic model: `CustomOutput` formats with a callback result, tool output (if given), and evaluation results (if given).
 
 ```python
-res == TaskOutput(
+res == vhq.TaskOutput(
    task_id=UUID('<TASK UUID>'),
    raw='{\"test1\":\"random str\", \"test2\":[\"str item 1\", \"str item 2\", \"str item 3\"]}',
    json_dict={'test1': 'random str', 'test2': ['str item 1', 'str item 2', 'str item 3']},
@@ -337,9 +338,9 @@ network =vhq.AgentNetwork(
       vhq.Member(agent=agent_b, is_manager=True, tasks=[task_2]), # Agent B as a manager
    ],
 )
-res = network.launch()
+res, _ = network.launch()
 
-assert isinstance(res, vhq.NetworkOutput)
+assert isinstance(res, vhq.TaskOutput)
 assert not [item for item in task_1.processed_agents if "vhq-Delegated-Agent" == item]
 assert [item for item in task_1.processed_agents if "agent b" == item]
 ```

{versionhq-1.2.1.22 → versionhq-1.2.2.0}/README.md

@@ -178,7 +178,7 @@ agent.update(
       task="YOUR AMAZING TASK OVERVIEW",
       expected_outcome="YOUR OUTCOME EXPECTATION",
    )
-   res = network.launch()
+   res, _ = network.launch()
    ```
 
 This will form a network with multiple agents on `Formation` and return `TaskOutput` object with output in JSON, plane text, Pydantic model format with evaluation.
@@ -209,15 +209,16 @@ task = vhq.Task(
    callback_kwargs=dict(message="Hi! Here is the result: ")
 )
 
-res = task.execute(context="amazing context to consider.")
-print(res)
+res = task.execute(context="context to consider")
+
+assert isinstance(res, vhq.TaskOutput)
 ```
 
 
 This will return a `TaskOutput` object that stores response in plane text, JSON, and Pydantic model: `CustomOutput` formats with a callback result, tool output (if given), and evaluation results (if given).
 
 ```python
-res == TaskOutput(
+res == vhq.TaskOutput(
    task_id=UUID('<TASK UUID>'),
    raw='{\"test1\":\"random str\", \"test2\":[\"str item 1\", \"str item 2\", \"str item 3\"]}',
    json_dict={'test1': 'random str', 'test2': ['str item 1', 'str item 2', 'str item 3']},
@@ -256,9 +257,9 @@ network =vhq.AgentNetwork(
       vhq.Member(agent=agent_b, is_manager=True, tasks=[task_2]), # Agent B as a manager
    ],
 )
-res = network.launch()
+res, _ = network.launch()
 
-assert isinstance(res, vhq.NetworkOutput)
+assert isinstance(res, vhq.TaskOutput)
 assert not [item for item in task_1.processed_agents if "vhq-Delegated-Agent" == item]
 assert [item for item in task_1.processed_agents if "agent b" == item]
 ```

{versionhq-1.2.1.22 → versionhq-1.2.2.0}/docs/core/agent/config.md

@@ -1,6 +1,3 @@
-
-<hr />
-
 ## Optimizing Model
 
 **Model Optimization**

{versionhq-1.2.1.22 → versionhq-1.2.2.0}/docs/core/agent/task-handling.md

@@ -1,7 +1,7 @@
 
-## Prompting
+## Prompt Engineering
 
-**Developer Prompt (System Prompt)**
+**Developer Prompt**
 
 `[var]`<bold>`backstory: Optional[str] = TEMPLATE_BACKSTORY`<bold>
 
@@ -25,9 +25,7 @@ agent = vhq.Agent(
 	goal="Coping with price competition in saturated markets"
 )
 
-print(agent.backstory)
-
-# You are an expert marketing analyst with relevant skillsets and abilities to query relevant information from the given knowledge sources. Leveraging these, you will identify competitive solutions to achieve the following goal: coping with price competition in saturated markets.
+assert agent.backstory == "You are an expert marketing analyst with relevant skillsets and abilities to query relevant information from the given knowledge sources. Leveraging these, you will identify competitive solutions to achieve the following goal: coping with price competition in saturated markets."
 ```
 
 You can also specify your own backstory by simply adding the value to the backstory field of the Agent model:
@@ -41,9 +39,7 @@ agent = vhq.Agent(
     backstory="You are a marketing analyst for a company in a saturated market. The market is becoming increasingly price-competitive, and your company's profit margins are shrinking. Your primary goal is to develop and implement strategies to help your company maintain its market share and profitability in this challenging environment."
 )
 
-print(agent.backstory)
-
-# You are a marketing analyst for a company in a saturated market. The market is becoming increasingly price-competitive, and your company's profit margins are shrinking. Your primary goal is to develop and implement strategies to help your company maintain its market share and profitability in this challenging environment.
+assert agent.backstory == "You are a marketing analyst for a company in a saturated market. The market is becoming increasingly price-competitive, and your company's profit margins are shrinking. Your primary goal is to develop and implement strategies to help your company maintain its market share and profitability in this challenging environment."
 ```
 <hr />
 

versionhq-1.2.2.0/docs/core/agent-network/config.md

@@ -0,0 +1,6 @@
+---
+tags:
+  - Agent Network
+---
+
+# Generating

{versionhq-1.2.1.22 → versionhq-1.2.2.0}/docs/core/agent-network/index.md

@@ -33,7 +33,7 @@ network = vhq.form_agent_network(
   expected_outcome='media mix, key messages, and CTA targets.'
 )
 
-res = network.launch()
+res, _ = network.launch()
 
 assert isinstance(res, vhq.TaskOutput)
 ```

versionhq-1.2.2.0/docs/core/agent-network/ref.md

@@ -0,0 +1,54 @@
+---
+tags:
+  - Agent Network
+---
+
+## Variable `AgentNetwork`
+
+
+| <div style="width:160px">**Variable**</div> | **Data Type** | **Default** | **Nullable** | **Description** |
+| :---               | :---  | :--- | :--- | :--- |
+| **`id`**   | UUID4 | uuid.uuid4() | False | Stores auto-generated ID as identifier. Not editable. |
+| **`role`**   | str | None | False | Stores a role of the agent. |
+| **`goal`**   | str | None | False | Stores a goal of the agent. |
+| **`backstory`**  | str | None | True | Stores backstory of the agent. Utilized as system prompt. |
+| **`tools`**  | List[InstanceOf[`Tool` \| `ToolSet`] \| Type[`Tool`]] | None | True | Stores tools to be used when executing a task. |
+| **`knowledge_sources`**  | List[`BaseKnowledgeSource` \| Any] | None | True | Stores knowledge sources in text, file path, or url. |
+| **`embedder_config`** | Dict[str, Any] | None | True | Stores embedding configuration for storing knowledge sources. |
+| **`with_memory`** | bool | False | - | Whether to store tasks and results in memory. |
+| **`memory_config`** |  Dict[str, Any] | None | True | Stores configuration of the memory. |
+| **`short_term_memory`** | InstanceOf[`ShortTermMemory`] | None | True | Stores `ShortTermMemory` object. |
+| **`long_term_memory`** | InstanceOf[`LongTermMemory`] | None | True | Stores `LongTermMemory` object. |
+| **`user_memory`** | InstanceOf[`UserMemory`] | None | True | Stores `UserMemory` object. |
+| **`use_developer_prompt`** |  bool | True | - | Whether to use the system (developer) prompt when calling the model. |
+| **`developer_promt_template`** | str | None | True | File path to the prompt template. |
+| **`user_promt_template`** | str | None | True | File path to the prompt template. |
+| **`networks`** | List[Any] | list() | True | Stores a list of agent networks that the agent belongs to. |
+| **`allow_delegation`** | bool | False | - | Whether the agent can delegate assinged tasks to another agent. |
+| **`max_retry_limit`** | int | 2 | - | Maximum number of retries when the task execution failed. |
+| **`maxit`** | int | 25 | - | Maximum number of total optimization loops conducted when an error occues during the task execution. |
+| **`callbacks`** | List[Callabale] | None | True | Stores a list of callback functions that must be called after every task execution completed.|
+| **`llm`** | str \| InstanceOf[`LLM`] \| Dict[str, Any] | None | False | Stores the main model that the agent runs on. |
+| **`func_calling_llm`** | str \| InstanceOf[`LLM`] \| Dict[str, Any] | None | False | Stores the function calling model that the agent runs on. |
+| **`respect_context_window`** | bool | True | - | Whether to follow the main model's maximum context window size. |
+| **`max_execution_time`** | int | None | True | Stores maximum execution time in seconds. |
+| **`max_rpm`** | int | None | True | Stores maximum number of requests per minute. |
+| **`llm_config`** | Dict[str, Any] | None | True | Stores configuration of `LLM` object. |
+| **`config`** | Dict[str, Any] | None | True | Stores model config. |
+
+
+## Class Methods `AgentNetwork`
+
+| <div style="width:120px">**Method**</div> | **Params** | **Returns** | **Description** |
+| :---               | :---  | :--- | :--- |
+| **`update`** | **kwargs: Any  | Self | Updates agents with given kwargs. Invalid keys will be ignored. |
+| **`start`** | context: Any = None <br> tool_res_as_final: bool = False  | `TaskOutput` \| None  | Starts to operate the agent. |
+| **`execute_task`** | task: [Task] <br> context: Any = None <br> task_tools: Optional[List[Tool \| ToolSet]] = list() | str | Returns response from the model in plane text format. |
+
+
+
+## Properties `AgentNetwork`
+
+| <div style="width:120px">**Property**</div> | **Returns** | **Description** |
+| :---               | :---  | :--- |
+| **`key`** | str | Unique identifier of the agent using its ID and sanitized role. |

versionhq-1.2.2.0/docs/core/agent-network/task-handling.md

@@ -0,0 +1,6 @@
+---
+tags:
+  - Agent Network
+---
+
+# Task Handling

{versionhq-1.2.1.22 → versionhq-1.2.2.0}/docs/core/task/evaluation.md

@@ -4,5 +4,3 @@ tags:
 ---
 
 # Evaluation
-
-Under drafting

{versionhq-1.2.1.22 → versionhq-1.2.2.0}/docs/core/task/index.md

@@ -15,7 +15,7 @@ Ref. Node / Edge / <a href="/core/task-graph">TaskGraph</a> class
 
 ## Quick Start
 
-Create a task by defining its description in one simple sentence. The `description` will be used for task prompting later.
+Create a task by defining its description in one simple sentence. The `description` will be used in the prompt later.
 
 Each task will be assigned a unique ID as an identifier.
 

{versionhq-1.2.1.22 → versionhq-1.2.2.0}/docs/core/task/task-execution.md

@@ -1,9 +1,7 @@
 # Executing Task
 
 
-## Prompting
-
-`[class method]`<bold>`prompt(self, model_provider: str = None, context: Optional[Any] = None) -> str`</bold>
+## Prompt Engineering
 
 Prompts are generated automatically based on the task `description`, response format, context, agent `role`, and `goal`.
 
@@ -40,9 +38,7 @@ In this scenario, `sub_task_2` executes before the main task. Its string output
 
 <hr>
 
-## Executing
-
-**Agent delegation**
+## Delegation
 
 `[var]`<bold>`allow_delegation: bool = False`</bold>
 
@@ -58,13 +54,13 @@ task = vhq.Task(
 task.execute()
 
 assert task.output is not None
-assert "vhq-Delegated-Agent" in task.processed_agents # delegated agent
+assert task.processed_agents is not None # auto assigned
 assert task.delegations ==1
 ```
 
 <hr>
 
-**SYNC - ASYNC**
+## Sync - Async Execution
 
 `[var]`<bold>`type: bool = False`</bold>
 
@@ -87,7 +83,7 @@ with patch.object(vhq.Agent, "execute_task", return_value="test") as execute:
 
 <hr>
 
-**Using tools**
+## Using Tools
 
 `[var]`<bold>`tools: Optional[List[ToolSet | Tool | Any]] = None`</bold>
 
@@ -146,7 +142,7 @@ assert res.tool_output == "simple func"
 
 <hr>
 
-## Callback
+## Callbacks
 
 `[var]`<bold>`callback: Optional[Callable] = None`</bold>
 

{versionhq-1.2.1.22 → versionhq-1.2.2.0}/docs/core/task/task-ref.md

@@ -16,6 +16,7 @@
 | **`callback_kwargs`** | Optional[Dict[str, Any]] | dict() | True | Args for the callback function (if any)|
 | **`should_evaluate`** | bool | False | - | Whether to evaluate the task output using eval criteria |
 | **`eval_criteria`** | Optional[List[str]] | list() | True | Evaluation criteria given by the human client |
+| **`fsls`** | Optional[List[str]] | None | True | Examples of excellent and weak responses |
 | **`processed_agents`** | Set[str] | set() | True | [Ops] Stores roles of the agents executed the task |
 | **`tool_errors`** | int | 0 | True | [Ops] Stores number of tool errors |
 | **`delegation`** | int | 0 | True | [Ops] Stores number of agent delegations |

{versionhq-1.2.1.22 → versionhq-1.2.2.0}/docs/core/tool.md

@@ -22,8 +22,7 @@ def demo_func(message: str) -> str:
 my_tool = vhq.Tool(func=demo_func)
 res = my_tool.run(params=dict(message="Hi!"))
 
-print(res)
-# Hi!_demo
+assert res == "Hi!_demo"
 ```
 
 e.g. Build an agent with a simple tool
@@ -87,8 +86,7 @@ def demo_func() -> str:
 
 my_tool = Tool(func=demo_func)
 
-print(my_tool.name)
-# demo_func
+assert my_tool.name == "demo_func"
 ```
 
 ```python
@@ -96,8 +94,7 @@ from versionhq import Tool
 
 my_tool = Tool(name="my empty tool", func=lambda x: x)
 
-print(my_tool.name)
-# my empty tool
+assert my_tool.name == "my empty tool"
 ```
 
 Tool names are used to call cached tools later.
@@ -119,8 +116,7 @@ from versionhq import Tool
 my_tool = Tool(func=lambda x: f"demo-{x}")
 res = my_tool._run(x="TESTING")
 
-print(res)
-# demo-TESTING
+assert res == "demo-TESTING"
 ```
 
 Another way to define the logic is to add a function to the `func` field.
@@ -137,8 +133,7 @@ def demo_func() -> str:
 my_tool = Tool(func=demo_func)
 res = my_tool._run()
 
-print(res)
-# demo
+assert res == "demo"
 ```
 
 You can also pass parameters using the class method.
@@ -152,8 +147,7 @@ def demo_func(message: str) -> str:
 my_tool = Tool(func=demo_func)
 res = my_tool._run(message="Hi!")
 
-print(res)
-# Hi!_demo
+assert res == "Hi!_demo"
 ```
 
 ### Custom tool execution
@@ -173,8 +167,7 @@ class MyCustomTool(Tool):
 my_custom_tool = MyCustomTool(func=lambda x: len(x))
 res = my_custom_tool._run(["demo1", "demo2"])
 
-print(res)
-# 2
+assert res == 2
 ```
 
 ### Cached tool execution
@@ -197,11 +190,8 @@ def demo_func(demo_list: List[Any]) -> int:
 my_tool = CustomTool(func=demo_func)
 res = my_tool.run(params=dict(demo_list=["demo1", "demo2"]))
 
-print(res)
-# 2
-
-print(my_tool.tool_handler)
-# ToolHandler object
+assert res == 2
+assert isinstance(my_tool.tool_handler, vhq.ToolHandler)
 ```
 
 *Reference: <bold>`ToolHandler` class</bold>
@@ -252,8 +242,7 @@ task = Task(
 )
 
 res = task.execute(agent=agent)
-print(res)
-# demo func
+assert res == "demo func"
 ```
 
 When the function calling LLM is not provided, we use the main model or default model `gpt-4o` .
@@ -278,8 +267,7 @@ task = Task(
 )
 
 res = task.execute(agent=agent)
-print(res.tool_output)
-# demo func
+assert res.tool_output == "demo func"
 ```
 
 ```python
@@ -300,11 +288,8 @@ task = Task(
 )
 
 res = task.execute(agent=agent)
-print(res.tool_output)
-# "demo func"
-
-print(task.processed_agents)
-# {'Demo Tool Handler'}
+assert res.tool_output == "demo func"
+assert agent.key in task.processed_agents
 ```
 
 <hr />
@@ -320,8 +305,9 @@ If you want to see if the model of your choice supports function calling <bold>e
 ```python
 from versionhq.llm.model import LLM
 llm = LLM(model="<MODEL_NAME_OF_YOUR_CHOICE>")
-llm._supports_function_calling()
-# return bool
+res = llm._supports_function_calling()
+
+assert type(res) == bool
 ```
 
 ### 2. Add tools to the task
@@ -355,8 +341,7 @@ task = Task(
 )
 
 res = task.execute(agent=agent)
-print(res)
-# empty func_demo
+assert res == "empty func_demo"
 ```
 
 ## Decorator
@@ -373,13 +358,7 @@ def my_tool(test_words: str) -> str:
 """Test a tool decorator."""
 return test_words
 
-print(my_tool.name)
-# demo
-
-print(my_tool.description)
-# Tool: demo
-# Args: {'test_words': {'description': '', 'type': 'str'}}
-
-print(my_tool.func("tool's final part!"))
-# tool's final part!
+assert my_tool.name == "demo"
+assert "Tool: demo" in my_tool.description and "'test_words': {'description': '', 'type': 'str'" in my_tool.description
+assert my_tool.func("testing") == "testing"
 ```

{versionhq-1.2.1.22 → versionhq-1.2.2.0}/docs/quickstart.md

@@ -23,7 +23,7 @@ network = vhq.form_agent_network(
    task="YOUR AMAZING TASK OVERVIEW",
    expected_outcome="YOUR OUTCOME EXPECTATION",
 )
-res = network.launch()
+res, _ = network.launch()
 ```
 
 This will form a network with multiple agents on `Formation` and return results as a `TaskOutput` object, storing outputs in JSON, plane text, Pydantic model formats along with evaluation.
@@ -104,7 +104,7 @@ network =vhq.AgentNetwork(
       vhq.Member(agent=agent_b, is_manager=True, tasks=[task_2]), # Agent B as a manager
    ],
 )
-res = network.launch()
+res, _ = network.launch()
 
 assert isinstance(res, vhq.NetworkOutput)
 assert "agent b" in task_1.processed_agents # agent_b delegated by agent_a

{versionhq-1.2.1.22 → versionhq-1.2.2.0}/docs/stylesheets/main.css

@@ -6,9 +6,9 @@
 
 
 class {
-    background-color: #363636;
+    background-color: hsl(0, 0%, 21%);
     padding: 8px 10px;
-    border-radius: 5px;
+    border-radius: 3px;
     color: #ededeb;
 }
 

{versionhq-1.2.1.22 → versionhq-1.2.2.0}/mkdocs.yml

@@ -112,6 +112,9 @@ nav:
   - Quick Start: 'quickstart.md'
   - Agent Network:
     - 'core/agent-network/index.md'
+    - Configuration: 'core/agent-network/config.md'
+    - Task Handling: 'core/agent-network/task-handling.md'
+    - Reference: 'core/agent-network/ref.md'
     - Agent:
       - 'core/agent/index.md'
       - Configuration: 'core/agent/config.md'

{versionhq-1.2.1.22 → versionhq-1.2.2.0}/pyproject.toml

@@ -15,7 +15,7 @@ exclude = ["test*", "__pycache__", "*.egg-info"]
 
 [project]
 name = "versionhq"
-version = "1.2.1.22"
+version = "1.2.2.0"
 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.1.22 → versionhq-1.2.2.0}/src/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-1.2.1.22 → versionhq-1.2.2.0}/src/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-1.2.2.0/src/versionhq/_utils/usage_metrics.py

@@ -0,0 +1,55 @@
+from pydantic import BaseModel, Field
+
+
+class UsageMetrics(BaseModel):
+    """
+    Model to track usage
+    """
+
+    total_tokens: int = Field(default=0, description="total number of tokens used")
+    prompt_tokens: int = Field(default=0, description="number of tokens used in prompts")
+    cached_prompt_tokens: int = Field(default=0, description="number of cached prompt tokens used")
+    completion_tokens: int = Field(default=0, description="number of tokens used in completions")
+    successful_requests: int = Field(default=0, description="number of successful requests made")
+
+    def add_usage_metrics(self, usage_metrics: "UsageMetrics") -> None:
+        """
+        Add the usage metrics from another UsageMetrics object.
+        """
+        self.total_tokens += usage_metrics.total_tokens
+        self.prompt_tokens += usage_metrics.prompt_tokens
+        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-1.2.1.22 → versionhq-1.2.2.0}/src/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,
 )