versionhq 1.2.1.0__py3-none-any.whl → 1.2.1.2__py3-none-any.whl

versionhq/__init__.py

@@ -8,6 +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.llm.model import LLM
 from versionhq.llm.llm_vars import LLM_CONTEXT_WINDOW_SIZES, PARAMS, PROVIDERS, MODELS
 from versionhq.clients.customer.model import Customer
@@ -16,10 +17,9 @@ from versionhq.clients.workflow.model import MessagingWorkflow, MessagingCompone
 from versionhq.knowledge.model import Knowledge, KnowledgeStorage
 from versionhq.knowledge.source import PDFKnowledgeSource, CSVKnowledgeSource, JSONKnowledgeSource, TextFileKnowledgeSource, ExcelKnowledgeSource, StringKnowledgeSource
 from versionhq.knowledge.source_docling import DoclingSource
-from versionhq.graph.model import TaskStatus, TaskGraph, Node, Edge, DependencyType
+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.team.model import Team, TeamOutput, Formation, Member, TaskHandlingProcess
 from versionhq.tool.model import Tool, ToolSet
 from versionhq.tool.cache_handler import CacheHandler
 from versionhq.tool.tool_handler import ToolHandler
@@ -30,10 +30,16 @@ from versionhq.memory.model import ShortTermMemory,LongTermMemory, UserMemory, M
 from versionhq.task.formation import form_agent_network
 
 
-__version__ = "1.2.1.0"
+__version__ = "1.2.1.2"
 __all__ = [
     "Agent",
 
+    "AgentNetwork",
+    "NetworkOutput",
+    "Formation",
+    "Member",
+    "TaskHandlingProcess",
+
     "LLM",
     "LLM_CONTEXT_WINDOW_SIZES",
     "PARAMS",
@@ -70,12 +76,6 @@ __all__ = [
     "Evaluation",
     "EvaluationItem",
 
-    "Team",
-    "TeamOutput",
-    "Formation",
-    "Member",
-    "TaskHandlingProcess",
-
     "Tool",
     "ToolSet",
     "CacheHandler",

versionhq/_utils/logger.py

@@ -1,3 +1,4 @@
+import logging
 from datetime import datetime
 from typing import Optional
 
@@ -42,9 +43,55 @@ class Logger(BaseModel):
     """
 
     verbose: bool = Field(default=True)
+    info_file_save: bool = Field(default=False, description="whether to save INFO logs")
     _printer: Printer = PrivateAttr(default_factory=Printer)
 
-    def log(self, level, message, color="yellow"):
+
+    def log(self, level: str, message: str, color="yellow"):
         if self.verbose:
             timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
             self._printer.print(f"\n{timestamp} - versionHQ [{level.upper()}]: {message}", color=color)
+
+        self._save(level=level, message=message)
+
+
+    def _save(self, level: str, message: str, filename: str = None):
+        import os
+        from pathlib import Path
+
+        if level.lower() == "info" and self.info_file_save == False:
+            return
+
+        logging_level = logging.INFO
+        match level:
+            case "warning":
+                logging_level = logging.WARNING
+            case "error":
+                logging_level = logging.ERROR
+            case _:
+                pass
+
+        cwd = Path.cwd()
+        log_file_dir = f"{cwd}/_logs"
+        os.makedirs(log_file_dir, exist_ok=True)
+        filename = filename if filename else datetime.now().strftime('%H_%M_%S_%d_%m_%Y')
+        abs_dir = f"{log_file_dir}/{filename}.log"
+
+        logging.basicConfig(filename=abs_dir, filemode='w', level=logging_level)
+        logger = logging.getLogger(__name__)
+        file_handler = logging.FileHandler(filename=abs_dir)
+        formatter = logging.Formatter('%(asctime)s : %(levelname)s : %(name)s : %(message)s')
+        file_handler.setFormatter(formatter)
+
+        for handler in logging.root.handlers[:]:
+            logging.root.removeHandler(handler)
+
+        logger.addHandler(file_handler)
+
+        match logging_level:
+            case logging.WARNING:
+                logger.warning(message)
+            case logging.ERROR:
+                logger.error(message)
+            case _:
+               logger.info(message)

versionhq/agent/inhouse_agents.py

@@ -39,3 +39,10 @@ vhq_formation_planner = Agent(
         "Random is a formation where a single agent handles tasks, asking help from other agents without sharing its memory or knowledge. Typical usecase is that an email agent drafts promo message for the given audience, asking insights on tones from other email agents which oversee other customer clusters, or an agent calls the external, third party agent to deploy the campaign. ",
     ]
 )
+
+
+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",
+)

versionhq/agent/model.py

@@ -95,7 +95,7 @@ class Agent(BaseModel):
     user_prompt_template: Optional[str] = Field(default=None, description="user prompt template")
 
     # task execution rules
-    team: Optional[List[Any]] = Field(default=None, description="Team to which the agent belongs")
+    network: Optional[List[Any]] = Field(default=None, description="store a list of agent networks that the agent belong as a member")
     allow_delegation: bool = Field(default=False,description="if the agent can delegate the task to another agent or ask some help")
     max_retry_limit: int = Field(default=2 ,description="max. number of retry for the task execution when an error occurs")
     maxit: Optional[int] = Field(default=25,description="max. number of total optimization loops conducted when an error occurs")
@@ -553,7 +553,7 @@ class Agent(BaseModel):
 
 
         ## comment out for now
-        # if self.team and self.team._train:
+        # if self.network and self.network._train:
         #     task_prompt = self._training_handler(task_prompt=task_prompt)
         # else:
         #     task_prompt = self._use_trained_data(task_prompt=task_prompt)

versionhq/{team → agent_network}/model.py

@@ -1,7 +1,6 @@
 import uuid
 import warnings
 from enum import Enum
-from dotenv import load_dotenv
 from concurrent.futures import Future
 from hashlib import md5
 from typing import Any, Dict, List, Callable, Optional, Tuple
@@ -10,9 +9,8 @@ from pydantic._internal._generate_schema import GenerateSchema
 from pydantic_core import PydanticCustomError, core_schema
 
 from versionhq.agent.model import Agent
-from versionhq.task.model import Task, TaskOutput, TaskExecutionType
+from versionhq.task.model import Task, TaskOutput, TaskExecutionType, ResponseField
 from versionhq.task.formatter import create_raw_outputs
-from versionhq.team.team_planner import TeamPlanner
 from versionhq._utils.logger import Logger
 from versionhq._utils.usage_metrics import UsageMetrics
 
@@ -41,21 +39,21 @@ class Formation(str, Enum):
 class TaskHandlingProcess(str, Enum):
     """
     A class representing task handling processes to tackle multiple tasks.
-    When the team has multiple tasks that connect with edges, follow the edge conditions.
+    When the agent network has multiple tasks that connect with edges, follow the edge conditions.
     """
     SEQUENT = 1
     HIERARCHY = 2
     CONSENSUAL = 3 # either from managers or peers or (human) - most likely controlled by edge
 
 
-class TeamOutput(TaskOutput):
+class NetworkOutput(TaskOutput):
     """
-    A class to store output from the team, inherited from TaskOutput class.
+    A class to store output from the network, inherited from TaskOutput class.
     """
 
-    team_id: UUID4 = Field(default_factory=uuid.uuid4, frozen=True, description="store the team ID that generate the TeamOutput")
+    network_id: UUID4 = Field(default_factory=uuid.uuid4, frozen=True, description="store the network ID as an identifier that generate the output")
     task_description: str = Field(default=None, description="store initial request (task description) from the client")
-    task_outputs: list[TaskOutput] = Field(default=list, description="store TaskOutput objects of all tasks that the team has executed")
+    task_outputs: list[TaskOutput] = Field(default=list, description="store TaskOutput objects of all tasks that the network has executed")
     # token_usage: UsageMetrics = Field(default=dict, description="processed token summary")
 
     def return_all_task_outputs(self) -> List[Dict[str, Any]]:
@@ -71,18 +69,18 @@ class TeamOutput(TaskOutput):
         elif self.json_dict and key in self.json_dict:
             return self.json_dict[key]
         else:
-            raise KeyError(f"Key '{key}' not found in the team output.")
+            raise KeyError(f"Key '{key}' not found in the output.")
 
 
 
 class Member(BaseModel):
     """
-    A class to store a member in the team and connect the agent as a member with tasks and memory/knowledge share settings.
+    A class to store a member (agent) in the network, with its tasks and memory/knowledge share settings.
     """
     agent: Agent | None = Field(default=None)
     is_manager: bool = Field(default=False)
-    can_share_knowledge: bool = Field(default=True, description="whether to share the agent's knowledge in the team")
-    can_share_memory: bool = Field(default=True, description="whether to share the agent's memory in the team")
+    can_share_knowledge: bool = Field(default=True, description="whether to share the agent's knowledge in the network")
+    can_share_memory: bool = Field(default=True, description="whether to share the agent's memory in the network")
     tasks: Optional[List[Task]] = Field(default_factory=list, description="tasks explicitly assigned to the agent")
 
     @property
@@ -90,10 +88,10 @@ class Member(BaseModel):
         return bool(self.tasks)
 
 
-class Team(BaseModel):
+class AgentNetwork(BaseModel):
     """
-    A class to store a team with members and tasks.
-    Tasks can be 1. multiple individual tasks, 2. multiple dependant tasks connected via Graph, and 3. hybrid.
+    A class to store a agent network with agent members and their tasks.
+    Tasks can be 1. multiple individual tasks, 2. multiple dependant tasks connected via TaskGraph, and 3. hybrid.
     """
 
     __hash__ = object.__hash__
@@ -108,8 +106,7 @@ class Team(BaseModel):
     should_reform: bool = Field(default=False, description="True if task exe. failed or eval scores below threshold")
 
     # formation planning
-    planner_llm: Optional[Any] = Field(default=None, description="llm to generate and evaluate formation")
-    team_tasks: Optional[List[Task]] = Field(default_factory=list, description="tasks without dedicated agents to handle")
+    network_tasks: Optional[List[Task]] = Field(default_factory=list, description="tasks without dedicated agents")
 
     # task execution rules
     prompt_file: str = Field(default="", description="absolute path to the prompt json file")
@@ -118,11 +115,11 @@ class Team(BaseModel):
     # callbacks
     pre_launch_callbacks: List[Callable[[Optional[Dict[str, Any]]], Optional[Dict[str, Any]]]] = Field(
         default_factory=list,
-        description="list of callback functions to be executed before the team launch. i.e., adjust inputs"
+        description="list of callback functions to be executed before the network launch. i.e., adjust inputs"
     )
-    post_launch_callbacks: List[Callable[[TeamOutput], TeamOutput]] = Field(
+    post_launch_callbacks: List[Callable[[NetworkOutput], NetworkOutput]] = Field(
         default_factory=list,
-        description="list of callback functions to be executed after the team launch. i.e., store the result in repo"
+        description="list of callback functions to be executed after the network launch. i.e., store the result in repo"
     )
     step_callback: Optional[Any] = Field(default=None, description="callback to be executed after each step for all agents execution")
 
@@ -146,19 +143,19 @@ class Team(BaseModel):
     @model_validator(mode="after")
     def assess_tasks(self):
         """
-        Validates if the model recognize all tasks that the team needs to handle.
+        Validates if the model recognize all tasks that the network needs to handle.
         """
 
         if self.tasks:
-            if all(task in self.tasks for task in self.team_tasks) == False:
-                raise PydanticCustomError("task_validation_error", "`team_tasks` needs to be recognized in the task.", {})
+            if all(task in self.tasks for task in self.network_tasks) == False:
+                raise PydanticCustomError("task_validation_error", "`network_tasks` needs to be recognized in the task.", {})
 
 
             num_member_tasks = 0
             for member in self.members:
                 num_member_tasks += len(member.tasks)
 
-            # if len(self.tasks) != len(self.team_tasks) + num_member_tasks:
+            # if len(self.tasks) != len(self.network_tasks) + num_member_tasks:
             #     raise PydanticCustomError("task_validation_error", "Some tasks are missing.", {})
         return self
 
@@ -166,7 +163,7 @@ class Team(BaseModel):
     @model_validator(mode="after")
     def check_manager(self):
         """
-        Check if the team has a manager
+        Check if the agent network has a manager
         """
         if self.process == TaskHandlingProcess.HIERARCHY or self.formation == Formation.SUPERVISING:
             if not self.managers:
@@ -174,9 +171,9 @@ class Team(BaseModel):
                 raise PydanticCustomError("missing_manager", "`manager` is required when using hierarchical process.", {})
 
         ## comment out for the formation flexibilities
-        # if self.managers and (self.manager_tasks is None or self.team_tasks is None):
+        # if self.managers and (self.manager_tasks is None or self.network_tasks is None):
         #     self._logger.log(level="error", message="The manager is idling. At least 1 task needs to be assigned to the manager.", color="red")
-        #     raise PydanticCustomError("missing_manager_task", "manager needs to have at least one manager task or team task.", {})
+        #     raise PydanticCustomError("missing_manager_task", "manager needs to have at least one manager task or network task.", {})
 
         return self
 
@@ -184,9 +181,9 @@ class Team(BaseModel):
     @model_validator(mode="after")
     def validate_task_member_paring(self):
         """
-        Sequential task processing without any team tasks require a task-agent pairing.
+        Sequential task processing without any network_tasks require a task-agent pairing.
         """
-        if self.process == TaskHandlingProcess.SEQUENT and self.team_tasks is None:
+        if self.process == TaskHandlingProcess.SEQUENT and self.network_tasks is None:
             for task in self.tasks:
                 if not [member.task == task for member in self.members]:
                     self._logger.log(level="error", message=f"The following task needs a dedicated agent to be assinged: {task.description}", color="red")
@@ -196,7 +193,7 @@ class Team(BaseModel):
     @model_validator(mode="after")
     def validate_end_with_at_most_one_async_task(self):
         """
-        Validates that the team completes max. one asynchronous task by counting tasks traversed backward
+        Validates that the agent network completes max. one asynchronous task by counting tasks traversed backward
         """
 
         async_task_count = 0
@@ -209,10 +206,43 @@ class Team(BaseModel):
                 break
 
         if async_task_count > 1:
-            raise PydanticCustomError("async_task_count", "The team must end with max. one asynchronous task.", {})
+            raise PydanticCustomError("async_task_count", "The agent network must end with at maximum one asynchronous task.", {})
         return self
 
 
+    @staticmethod
+    def handle_assigning_agents(unassigned_tasks: List[Task]) -> List[Member]:
+        """
+        Build an agent and assign it with a task. Return a list of Member connecting the agent created and the task given.
+        """
+
+        from versionhq.agent.inhouse_agents import vhq_agent_creator
+
+        new_member_list: List[Member] = []
+
+        for unassgined_task in unassigned_tasks:
+            task = Task(
+                description=f"""
+                    Based on the following task summary, draft a AI agent's role and goal in concise manner.
+                    Task summary: {unassgined_task.summary}
+                """,
+                response_fields=[
+                    ResponseField(title="goal", data_type=str, required=True),
+                    ResponseField(title="role", data_type=str, required=True),
+                ],
+            )
+            res = task.execute(agent=vhq_agent_creator)
+            agent = Agent(
+                role=res.json_dict["role"] if "role" in res.json_dict else res.raw,
+                goal=res.json_dict["goal"] if "goal" in res.json_dict else task.description
+            )
+            if agent.id:
+                member = Member(agent=agent, tasks=[unassgined_task], is_manager=False)
+                new_member_list.append(member)
+
+        return new_member_list
+
+
     def _get_responsible_agent(self, task: Task) -> Agent | None:
         if task is None:
             return None
@@ -226,17 +256,16 @@ class Team(BaseModel):
 
     def _assign_tasks(self) -> None:
         """
-        Form a team considering agents and tasks given, and update `self.members` field:
-            1. Idling managers to take the team tasks.
-            2. Idling members to take the remaining tasks starting from the team tasks to member tasks.
+        Form a agent network considering given agents and tasks, and update `self.members` field:
+            1. Idling managers to take the network tasks.
+            2. Idling members to take the remaining tasks starting from the network tasks to member tasks.
             3. Create agents to handle the rest tasks.
         """
 
-        team_planner = TeamPlanner(tasks=self.tasks, planner_llm=self.planner_llm)
         idling_managers: List[Member] = [member for member in self.members if member.is_idling and member.is_manager == True]
         idling_members: List[Member] =  [member for member in self.members if member.is_idling and member.is_manager == False]
-        unassigned_tasks: List[Task] = self.team_tasks + self.member_tasks_without_agent if self.team_tasks else self.member_tasks_without_agent
-        new_team_members: List[Member] = []
+        unassigned_tasks: List[Task] = self.network_tasks + self.member_tasks_without_agent if self.network_tasks else self.member_tasks_without_agent
+        new_members: List[Member] = []
 
         if idling_managers:
             idling_managers[0].tasks.extend(unassigned_tasks)
@@ -245,9 +274,9 @@ class Team(BaseModel):
             idling_members[0].tasks.extend(unassigned_tasks)
 
         else:
-            new_team_members = team_planner._handle_assign_agents(unassigned_tasks=unassigned_tasks)
-            if new_team_members:
-                self.members += new_team_members
+            new_members = self.handle_assigning_agents(unassigned_tasks=unassigned_tasks)
+            if new_members:
+                self.members += new_members
 
 
     # task execution
@@ -268,7 +297,7 @@ class Team(BaseModel):
 
     def _calculate_usage_metrics(self) -> UsageMetrics:
         """
-        Calculate and return the usage metrics that consumed by the team.
+        Calculate and return the usage metrics that consumed by the agent network.
         """
         total_usage_metrics = UsageMetrics()
 
@@ -288,12 +317,12 @@ class Team(BaseModel):
         return total_usage_metrics
 
 
-    def _execute_tasks(self, tasks: List[Task], start_index: Optional[int] = 0, was_replayed: bool = False) -> TeamOutput:
+    def _execute_tasks(self, tasks: List[Task], start_index: Optional[int] = 0, was_replayed: bool = False) -> NetworkOutput:
         """
-        Executes tasks sequentially and returns the final output in TeamOutput class.
+        Executes tasks sequentially and returns the final output in NetworkOutput class.
         When we have a manager agent, we will start from executing manager agent's tasks.
         Priority:
-        1. Team tasks > 2. Manager task > 3. Member tasks (in order of index)
+        1. Network tasks > 2. Manager task > 3. Member tasks (in order of index)
         """
 
         task_outputs: List[TaskOutput] = []
@@ -330,6 +359,7 @@ class Team(BaseModel):
             else:
                 context = create_raw_outputs(tasks=[task,], task_outputs=([last_sync_output,] if last_sync_output else [] ))
                 task_output = task.execute(agent=responsible_agent, context=context)
+
                 if self.managers and responsible_agent in [manager.agent for manager in self.managers]:
                     lead_task_output = task_output
 
@@ -341,15 +371,15 @@ class Team(BaseModel):
             task_outputs = self._process_async_tasks(futures, was_replayed)
 
         if not task_outputs:
-            self._logger.log(level="error", message="Missing task outcomes. Failed to launch the task.", color="red")
-            raise ValueError("Failed to launch tasks")
+            self._logger.log(level="error", message="Missing task outputs.", color="red")
+            raise ValueError("Missing task outputs")
 
         final_task_output = lead_task_output if lead_task_output is not None else task_outputs[0] #! REFINEME
 
         # token_usage = self._calculate_usage_metrics() #! combining with Eval
 
-        return TeamOutput(
-            team_id=self.id,
+        return NetworkOutput(
+            network_id=self.id,
             raw=final_task_output.raw,
             json_dict=final_task_output.json_dict,
             pydantic=final_task_output.pydantic,
@@ -358,33 +388,23 @@ class Team(BaseModel):
         )
 
 
-
-    def launch(self, kwargs_pre: Optional[Dict[str, str]] = None, kwargs_post: Optional[Dict[str, Any]] = None) -> TeamOutput:
+    def launch(self, kwargs_pre: Optional[Dict[str, str]] = None, kwargs_post: Optional[Dict[str, Any]] = None) -> NetworkOutput:
         """
-        Confirm and launch the formation - execute tasks and record outputs.
-        0. Assign an agent to a task - using conditions (manager prioritizes team_tasks) and planner_llm.
-        1. Address `pre_launch_callbacks` if any.
-        2. Handle team members' tasks in accordance with the process.
-        3. Address `post_launch_callbacks` if any.
+        Launch the agent network - executing tasks and recording their outputs.
         """
 
         metrics: List[UsageMetrics] = []
 
-        if self.team_tasks or self.member_tasks_without_agent:
+        if self.network_tasks or self.member_tasks_without_agent:
             self._assign_tasks()
 
         if kwargs_pre is not None:
             for func in self.pre_launch_callbacks:
                 func(**kwargs_pre)
 
-        # self._execution_span = self._telemetry.team_execution_span(self, inputs)
-        # self._task_output_handler.reset()
-        # self._logging_color = "bold_purple"
-        # i18n = I18N(prompt_file=self.prompt_file)
-
         for member in self.members:
             agent = member.agent
-            agent.team = self
+            agent.network = self
 
             if self.step_callback:
                 agent.callbacks.append(self.step_callback)
@@ -421,7 +441,7 @@ class Team(BaseModel):
     @property
     def manager_tasks(self) -> List[Task]:
         """
-        Tasks (incl. team tasks) handled by managers in the team.
+        Tasks (incl. network tasks) handled by managers in the agent network.
         """
         res = list()
 
@@ -436,22 +456,22 @@ class Team(BaseModel):
     @property
     def tasks(self) -> List[Task]:
         """
-        Return all the tasks that the team needs to handle in order of priority:
-        1. team tasks, -> assigned to the member
+        Return all the tasks that the agent network needs to handle in order of priority:
+        1. network_tasks, -> assigned to the member
         2. manager_task,
         3. members' tasks
         """
 
-        team_tasks = self.team_tasks
+        network_tasks = self.network_tasks
         manager_tasks = self.manager_tasks
         member_tasks = []
 
         for member in self.members:
             if member.is_manager == False and member.tasks:
-                a = [item for item in member.tasks if item not in team_tasks and item not in manager_tasks]
+                a = [item for item in member.tasks if item not in network_tasks and item not in manager_tasks]
                 member_tasks += a
 
-        return team_tasks + manager_tasks + member_tasks
+        return network_tasks + manager_tasks + member_tasks
 
 
     @property

versionhq/clients/workflow/model.py

@@ -1,7 +1,7 @@
 import uuid
 from abc import ABC
-from datetime import date, datetime, time, timedelta
-from typing import Any, Dict, List, Callable, Type, Optional, get_args, get_origin
+from datetime import datetime
+from typing import Any, Dict, List, Optional
 from typing_extensions import Self
 from pydantic import UUID4, InstanceOf, BaseModel, ConfigDict, Field, field_validator, model_validator
 from pydantic_core import PydanticCustomError
@@ -9,7 +9,7 @@ from pydantic_core import PydanticCustomError
 from versionhq.clients.product.model import Product
 from versionhq.clients.customer.model import Customer
 from versionhq.agent.model import Agent
-from versionhq.team.model import Team
+from versionhq.agent_network.model import AgentNetwork
 from versionhq.tool.composio_tool_vars import ComposioAppName
 
 
@@ -102,8 +102,8 @@ class MessagingWorkflow(ABC, BaseModel):
     messaging_components: List[MessagingComponent] = Field(default_factory=list, description="store messaging components in the workflow")
 
     # responsible tean or agents
-    team: Optional[Team] = Field(default=None, description="store a responsibile team to autopilot the workflow")
-    agents: Optional[List[Agent]] = Field(default=None, description="store responsible agents. None when the team exists")
+    agent_network: Optional[AgentNetwork] = Field(default=None, description="store a responsibile agent network to autopilot the workflow")
+    agents: Optional[List[Agent]] = Field(default=None, description="store responsible agents. None when the `agent_network` fields has a value")
 
     # metrics
     destination: Optional[ComposioAppName | str] = Field(default=None, description="destination service to launch the workflow")
@@ -140,16 +140,16 @@ class MessagingWorkflow(ABC, BaseModel):
         return self
 
 
-    def reassign_agent_or_team(self, agents: List[Agent] = None, team: Team = None) -> None:
+    def reassign_agent(self, agents: List[Agent] = None, agent_network: AgentNetwork = None) -> None:
         """
-        Fire unresponsible agents/team and assign new one.
+        Switch agents
         """
 
-        if not agents and not team:
-            raise ValueError("Need to add at least 1 agent or team.")
+        if not agents and not agent_network:
+            raise ValueError("Missing agent or agent network to assign.")
 
         self.agents = agents
-        self.team = team
+        self.agent_network = agent_network
         self.updated_at = datetime.datetime.now()
 
 

versionhq/knowledge/source.py

@@ -265,7 +265,7 @@ class PDFKnowledgeSource(BaseFileKnowledgeSource):
                 import os
                 os.system("uv add pdfplumber --optional pdfplumber")
             except:
-                raise ImportError("pdfplumber is not installed. Please install it with: uv add pdfplumber --optional pdfplumber")
+                raise ImportError("pdfplumber is not installed. Please install it with: $ uv add versionhq[pdfplumber]")
 
 
     def add(self) -> None:
@@ -394,7 +394,7 @@ class ExcelKnowledgeSource(BaseFileKnowledgeSource):
             except:
                 missing_package = str(e).split()[-1]
                 raise ImportError(
-                    f"{missing_package} is not installed. Please install it with: pip install {missing_package}"
+                    f"{missing_package} is not installed. Please install it with: $ uv add versionhq[{missing_package}]"
                 )
 
 

versionhq/knowledge/source_docling.py

@@ -56,7 +56,7 @@ class DoclingSource(BaseKnowledgeSource):
             super().__init__(*args, **kwargs)
 
         else:
-            raise ImportError("The docling package is required. Please install the package using: $ uv add docling.")
+            raise ImportError("The docling package is required. Please install the package using: $ uv add versionhq[docling]")
         # else:
         #     super().__init__(*args, **kwargs)
 

versionhq/memory/model.py

@@ -125,7 +125,7 @@ class ShortTermMemory(Memory):
 
                     from versionhq.storage.mem0_storage import Mem0Storage
                 except:
-                    raise ImportError("Mem0 is not installed. Please install it with `uv pip install mem0ai`.")
+                    raise ImportError("Mem0 is not installed. Please install it with `$ uv add versionhq[mem0ai]`.")
 
             storage = Mem0Storage(type="stm", agent=agent)
         else:

versionhq/task/formation.py

@@ -5,7 +5,7 @@ from pydantic import BaseModel
 
 from versionhq.task.model import Task
 from versionhq.agent.model import Agent
-from versionhq.team.model import Team, Member, Formation
+from versionhq.agent_network.model import AgentNetwork, Member, Formation
 from versionhq.agent.inhouse_agents  import vhq_formation_planner
 from versionhq._utils import Logger
 
@@ -16,7 +16,7 @@ def form_agent_network(
         agents: List[Agent] = None,
         context: str = None,
         formation: Type[Formation] = None
-    ) -> Team | None:
+    ) -> AgentNetwork | None:
     """
     Make a formation of agents from the given task description, expected outcome, agents (optional), and context (optional).
     """
@@ -91,7 +91,7 @@ def form_agent_network(
             created_agents = [Agent(role=item, goal=item) for item in res.pydantic.agent_roles]
             created_tasks = [Task(description=item) for item in res.pydantic.task_descriptions]
 
-            team_tasks = []
+            network_tasks = []
             members = []
             leader = str(res.pydantic.leader_agent)
 
@@ -106,11 +106,11 @@ def form_agent_network(
 
 
             if len(created_agents) < len(created_tasks):
-                team_tasks.extend(created_tasks[len(created_agents):len(created_tasks)])
+                network_tasks.extend(created_tasks[len(created_agents):len(created_tasks)])
 
             members.sort(key=lambda x: x.is_manager == False)
-            team = Team(members=members, formation=_formation, team_tasks=team_tasks, planner_llm=vhq_formation_planner.llm)
-            return team
+            network = AgentNetwork(members=members, formation=_formation, network_tasks=network_tasks)
+            return network
 
         else:
             res = res.json_dict
@@ -122,7 +122,7 @@ def form_agent_network(
             created_agents = [Agent(role=item, goal=item) for item in res["agent_roles"]]
             created_tasks = [Task(description=item) for item in res["task_descriptions"]]
 
-            team_tasks = []
+            network_tasks = []
             members = []
             leader = str(res["leader_agent"])
 
@@ -136,11 +136,11 @@ def form_agent_network(
                 members.append(member)
 
             if len(created_agents) < len(created_tasks):
-                team_tasks.extend(created_tasks[len(created_agents):len(created_tasks)])
+                network_tasks.extend(created_tasks[len(created_agents):len(created_tasks)])
 
             members.sort(key=lambda x: x.is_manager == False)
-            team = Team( members=members, formation=_formation,  team_tasks=team_tasks, planner_llm=vhq_formation_planner.llm)
-            return team
+            network = AgentNetwork(members=members, formation=_formation,  network_tasks=network_tasks)
+            return network
 
 
     except Exception as e:

versionhq/task/model.py

@@ -613,10 +613,11 @@ Ref. Output image: {output_formats_to_follow}
         """
 
         from versionhq.agent.model import Agent
-        from versionhq.team.model import Team
+        from versionhq.agent_network.model import AgentNetwork
 
         self.prompt_context = context
         task_output: InstanceOf[TaskOutput] = None
+        raw_output: str = None
         tool_output: str | list = None
         task_tools: List[List[InstanceOf[Tool]| InstanceOf[ToolSet] | Type[Tool]]] = []
         started_at, ended_at = datetime.datetime.now(), datetime.datetime.now()
@@ -626,15 +627,15 @@ Ref. Output image: {output_formats_to_follow}
                 if isinstance(item, ToolSet) or isinstance(item, Tool) or type(item) == Tool:
                     task_tools.append(item)
 
-        if self.allow_delegation:
+        if self.allow_delegation == True:
             agent_to_delegate = None
 
-            if hasattr(agent, "team") and isinstance(agent.team, Team):
-                if agent.team.managers:
-                    idling_manager_agents = [manager.agent for manager in agent.team.managers if manager.is_idling]
-                    agent_to_delegate = idling_manager_agents[0] if idling_manager_agents else agent.team.managers[0]
+            if hasattr(agent, "network") and isinstance(agent.network, AgentNetwork):
+                if agent.network.managers:
+                    idling_manager_agents = [manager.agent for manager in agent.network.managers if manager.is_idling]
+                    agent_to_delegate = idling_manager_agents[0] if idling_manager_agents else agent.network.managers[0]
                 else:
-                    peers = [member.agent for member in agent.team.members if member.is_manager == False and member.agent.id is not agent.id]
+                    peers = [member.agent for member in agent.network.members if member.is_manager == False and member.agent.id is not agent.id]
                     if len(peers) > 0:
                         agent_to_delegate = peers[0]
             else:
@@ -673,7 +674,7 @@ Ref. Output image: {output_formats_to_follow}
         self.output = task_output
         self.processed_agents.add(agent.role)
 
-        if self.should_evaluate:
+        if self.should_evaluate and raw_output: # eval only when raw output exsits
             task_output.evaluate(task=self)
 
         self._create_short_and_long_term_memories(agent=agent, task_output=task_output)

versionhq/{graph → task_graph}/model.py

@@ -5,7 +5,7 @@ import matplotlib.pyplot as plt
 from abc import ABC
 from typing import List, Any, Optional, Callable, Dict, Type, Tuple
 
-from pydantic import BaseModel, InstanceOf, Field, UUID4, PrivateAttr, field_validator
+from pydantic import BaseModel, InstanceOf, Field, UUID4, field_validator
 from pydantic_core import PydanticCustomError
 
 from versionhq.task.model import Task, TaskOutput
@@ -13,26 +13,6 @@ from versionhq.agent.model import Agent
 from versionhq._utils.logger import Logger
 
 
-def gen_network():
-    goal = "make a promo plan to increase gross sales of Temu, an ecommerce site with affordable items."
-    agent = Agent(
-        role="Network Generator", goal="draft a best graph with nodes (tasks) and edges", llm="gemini-2.0", maxit=1,
-        knowledge_sources=["https://en.wikipedia.org/wiki/Graph_theory", "https://www.temu.com", ]
-    )
-    class Outcome(BaseModel):
-        task_descriptions: list[str]
-
-
-    task = Task(
-        description="    Create a team of specialized agents designed to automate the following task and deliver the expected outcome. Consider the necessary roles for each agent with a clear task description. If you think we neeed a leader to handle the automation, return a leader_agent role as well, but if not, leave the a leader_agent role blank. ",
-        pydantic_output=Outcome
-    )
-    task = Task(
-        description="Form best network with nodes and edges for the task described as following: Draft a promo plan for the client.",
-        pydantic_output=Edge
-    )
-    res = task.execute(agent=agent)
-
 class TaskStatus(enum.Enum):
     """
     Enum to track the task execution status

{versionhq-1.2.1.0.dist-info → versionhq-1.2.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.2.1.0
+Version: 1.2.1.2
 Summary: An agentic orchestration framework for building agent networks that handle task automation.
 Author-email: Kuriko Iwai <kuriko@versi0n.io>
 License: MIT License
@@ -117,10 +117,10 @@ A Python framework for agentic orchestration that handles complex task automatio
   - [Supervising](#supervising)
 - [Technologies Used](#technologies-used)
 - [Project Structure](#project-structure)
-- [Setting Up a Project](#setting-up-a-project)
-  - [1. Installing package manager](#1-installing-package-manager)
-  - [2. Installing dependencies](#2-installing-dependencies)
-  - [3. Adding env secrets to .env file](#3-adding-env-secrets-to-env-file)
+- [Setting Up Your Project](#setting-up-your-project)
+  - [Installing package manager](#installing-package-manager)
+  - [Installing dependencies](#installing-dependencies)
+  - [Adding env secrets to .env file](#adding-env-secrets-to-env-file)
 - [Contributing](#contributing)
   - [Steps](#steps)
   - [Package Management with uv](#package-management-with-uv)
@@ -256,7 +256,7 @@ agent.update(
    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.
+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.
 
 
 ### Executing tasks
@@ -266,81 +266,87 @@ You can simply build an agent using `Agent` model and execute the task using `Ta
 By default, agents prioritize JSON over plane text outputs.
 
 
-   ```python
-   import versionhq as vhq
-   from pydantic import BaseModel
+```python
+import versionhq as vhq
+from pydantic import BaseModel
 
-   class CustomOutput(BaseModel):
-      test1: str
-      test2: list[str]
+class CustomOutput(BaseModel):
+   test1: str
+   test2: list[str]
 
-   def dummy_func(message: str, test1: str, test2: list[str]) -> str:
-      return f"""{message}: {test1}, {", ".join(test2)}"""
+def dummy_func(message: str, test1: str, test2: list[str]) -> str:
+   return f"""{message}: {test1}, {", ".join(test2)}"""
 
-   task = vhq.Task(
-      description="Amazing task",
-      pydantic_output=CustomOutput,
-      callback=dummy_func,
-      callback_kwargs=dict(message="Hi! Here is the result: ")
-   )
+task = vhq.Task(
+   description="Amazing task",
+   pydantic_output=CustomOutput,
+   callback=dummy_func,
+   callback_kwargs=dict(message="Hi! Here is the result: ")
+)
 
-   res = task.execute(context="amazing context to consider.")
-   print(res)
-   ```
+res = task.execute(context="amazing context to consider.")
+print(res)
+```
 
 
 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(
-      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']},
-      pydantic=<class '__main__.CustomOutput'>,
-      tool_output=None,
-      callback_output='Hi! Here is the result: random str, str item 1, str item 2, str item 3', # returned a plain text summary
-      evaluation=None
-   )
-   ```
+```python
+res == 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']},
+   pydantic=<class '__main__.CustomOutput'>,
+   tool_output=None,
+   callback_output='Hi! Here is the result: random str, str item 1, str item 2, str item 3', # returned a plain text summary
+   evaluation=None
+)
+```
 
 ### Supervising
 
-   ```python
-   import versionhq as vhq
+To create an agent network with one or more manager agents, designate members using the `is_manager` tag.
 
-   agent_a = vhq.Agent(role="agent a", goal="My amazing goals", llm="llm-of-your-choice")
-   agent_b = vhq.Agent(role="agent b", goal="My amazing goals", llm="llm-of-your-choice")
+```python
+import versionhq as vhq
 
-   task_1 = vhq.Task(
-      description="Analyze the client's business model.",
-      response_fields=[vhq.ResponseField(title="test1", data_type=str, required=True),],
-      allow_delegation=True
-   )
+agent_a = vhq.Agent(role="agent a", goal="My amazing goals", llm="llm-of-your-choice")
+agent_b = vhq.Agent(role="agent b", goal="My amazing goals", llm="llm-of-your-choice")
 
-    task_2 = vhq.Task(
-      description="Define the cohort.",
-      response_fields=[ResponseField(title="test1", data_type=int, required=True),],
-      allow_delegation=False
-   )
+task_1 = vhq.Task(
+   description="Analyze the client's business model.",
+   response_fields=[vhq.ResponseField(title="test1", data_type=str, required=True),],
+   allow_delegation=True
+)
 
-   team = vhq.Team(
-      members=[
-         vhq.Member(agent=agent_a, is_manager=False, task=task_1),
-         vhq.Member(agent=agent_b, is_manager=True, task=task_2),
-      ],
-   )
-   res = team.launch()
-   ```
+task_2 = vhq.Task(
+   description="Define a cohort.",
+   response_fields=[vhq.ResponseField(title="test1", data_type=int, required=True),],
+   allow_delegation=False
+)
+
+network =vhq.AgentNetwork(
+   members=[
+      vhq.Member(agent=agent_a, is_manager=False, tasks=[task_1]),
+      vhq.Member(agent=agent_b, is_manager=True, tasks=[task_2]), # Agent B as a manager
+   ],
+)
+res = network.launch()
+
+assert isinstance(res, vhq.NetworkOutput)
+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]
+```
 
 This will return a list with dictionaries with keys defined in the `ResponseField` of each task.
 
-Tasks can be delegated to a team manager, peers in the team, or completely new agent.
+Tasks can be delegated to a manager, peers within the agent network, or a completely new agent.
 
 <hr />
 
 ## Technologies Used
 
-**Graph Theory (Analysis and Visualization)**
+**Task Graph**
 
 * [NetworkX](https://networkx.org/documentation/stable/reference/introduction.html): A Python package to analyze, create, and manipulate complex graph networks.
 * [Matplotlib](https://matplotlib.org/stable/index.html): Visualization library
@@ -401,16 +407,19 @@ src/
 │     └── ...
 │
 └── uploads/  [.gitignore]    # Local directory to store uploaded files such as graphviz diagrams generatd by `Network` class
-|
+│
+└── _logs/    [.gitignore]    # Local directory to store error/warning logs for debugging
+│
+│
 pyproject.toml                # Project config
 
 ```
 
 <hr />
 
-## Setting Up a Project
+## Setting Up Your Project
 
-### 1. Installing package manager
+### Installing package manager
 
    For MacOS:
 
@@ -424,7 +433,7 @@ pyproject.toml                # Project config
    ```
 
 
-### 2. Installing dependencies
+### Installing dependencies
 
    ```
    uv venv
@@ -457,7 +466,7 @@ pyproject.toml                # Project config
 
    - `torch`/`Docling` related errors: Set up default Python version either `3.11.x` or `3.12.x` (same as AssertionError)
 
-### 3. Adding env secrets to .env file
+### Adding env secrets to .env file
 
 Create `.env` file in the project root and add following:
 
@@ -470,6 +479,7 @@ Create `.env` file in the project root and add following:
    [OTHER_LLM_INTERFACE_PROVIDER_OF_YOUR_CHOICE]_API_KEY=your-api-key
    ```
 
+
 <hr />
 
 ## Contributing
@@ -574,7 +584,7 @@ Common issues and solutions:
 
 * Issues related to dependencies: `rm -rf uv.lock`, `uv cache clean`, `uv venv`, and run `uv pip install -r requirements.txt -v`.
 
-* Issues related to the AI agents or RAG system: Check the `output.log` file for detailed error messages and stack traces.
+* Issues related to agents and other systems: Check `_logs` directory for detailed error messages and stack traces.
 
 * Issues related to `Python quit unexpectedly`: Check [this stackoverflow article](https://stackoverflow.com/questions/59888499/macos-catalina-python-quit-unexpectedly-error).
 

{versionhq-1.2.1.0.dist-info → versionhq-1.2.1.2.dist-info}/RECORD

@@ -1,17 +1,19 @@
-versionhq/__init__.py,sha256=OvDVdnABXYcdjKCYL58tOw2duy59rQ2lxc7e1AoxUjc,2781
+versionhq/__init__.py,sha256=nv3StURywg_U4o7_lKbqPuby1xjSphOkY9B6Ra5ORHA,2817
 versionhq/_utils/__init__.py,sha256=dzoZr4cBlh-2QZuPzTdehPUCe9lP1dmRtauD7qTjUaA,158
 versionhq/_utils/i18n.py,sha256=TwA_PnYfDLA6VqlUDPuybdV9lgi3Frh_ASsb_X8jJo8,1483
-versionhq/_utils/logger.py,sha256=2qkODR6y8ApOoMjQZVecTboXvCLrMy2v_mTSOnNMKFY,1581
+versionhq/_utils/logger.py,sha256=IxSlr2Vi7AXaxj5Fuy8LRzEovaIFVwcbWTgJnASsHN8,3155
 versionhq/_utils/process_config.py,sha256=jbPGXK2Kb4iyCugJ3FwRJuU0wL5Trq2x4xFQz2uOyFY,746
 versionhq/_utils/usage_metrics.py,sha256=NXF18dn5NNvGK7EsQ4AAghpR8ppYOjMx6ABenLLHnmM,1066
 versionhq/_utils/vars.py,sha256=bZ5Dx_bFKlt3hi4-NNGXqdk7B23If_WaTIju2fiTyPQ,57
 versionhq/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-versionhq/agent/inhouse_agents.py,sha256=vSobrH1gXDWlaNsiges3sqETeUrEssRzQvCZCY2hQZA,2374
-versionhq/agent/model.py,sha256=kSYCAiRxF62ztj-S-_KfN0mhPKikjbFZ0fy2Yj5J7Jo,25400
+versionhq/agent/inhouse_agents.py,sha256=snDtgDmvZB2bZKH_RTcz5uFOMl3MTjLJwTQBebFt8hk,2532
+versionhq/agent/model.py,sha256=HiEQulVjXx6fWGaP-DBWmiZv5uwXFPd4S8jyBU3ydm0,25442
 versionhq/agent/parser.py,sha256=riG0dkdQCxH7uJ0AbdVdg7WvL0BXhUgJht0VtQvxJBc,4082
 versionhq/agent/rpm_controller.py,sha256=grezIxyBci_lDlwAlgWFRyR5KOocXeOhYkgN02dNFNE,2360
 versionhq/agent/TEMPLATES/Backstory.py,sha256=IAhGnnt6VUMe3wO6IzeyZPDNu7XE7Uiu3VEXUreOcKs,532
 versionhq/agent/TEMPLATES/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+versionhq/agent_network/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+versionhq/agent_network/model.py,sha256=P3Ntr_OIK60EXjEfaEm31HTBFypZfVr__0aC4VBt9G0,19353
 versionhq/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/clients/customer/__init__.py,sha256=-YXh1FQfvpfLacK8SUC7bD7Wx_eIEi4yrkCC_cUasFg,217
@@ -19,22 +21,20 @@ versionhq/clients/customer/model.py,sha256=_AtaVVMm9MgCwrQ-HTRQ2oXUMKrSCEfZwE2Jd
 versionhq/clients/product/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/clients/product/model.py,sha256=3w__pug9XRe4LIm9wX8C8WKqi40r081Eb1q2vWk9UaU,3694
 versionhq/clients/workflow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-versionhq/clients/workflow/model.py,sha256=FNftenLLoha0bkivrjId32awLHAkBwIT8iNljdic_bw,6003
-versionhq/graph/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-versionhq/graph/model.py,sha256=k4NbNn0D3r3ujfkQHOsfiiEJRxZx7iK5DczBnOyn1ik,23912
+versionhq/clients/workflow/model.py,sha256=fqmTx8Y6P6i-sK045ENO88GDbf4WYY4hUA26RfEH8Ek,6011
 versionhq/knowledge/__init__.py,sha256=qW7IgssTA4_bFFV9ziOcYRfGjlq1c8bkb-HnfWknpuQ,567
 versionhq/knowledge/_utils.py,sha256=YWRF8U533cfZes_gZqUvdj-K24MD2ri1R0gjc_aPYyc,402
 versionhq/knowledge/embedding.py,sha256=KfHc__1THxb5jrg1EMrF-v944RDuIr2hE0l-MtM3Bp0,6826
 versionhq/knowledge/model.py,sha256=w29mrJv1kiznCh4P4yJMUQxIuyRw1Sk0XYtBXzCxaG4,1786
-versionhq/knowledge/source.py,sha256=HxlwK7E5qWMknJeswnFCCdTIOB01hV4JrEiMGkp7j1E,13631
-versionhq/knowledge/source_docling.py,sha256=hhHn3rS4KVsFKEPWcfllM8VxSL86PckZdAHDZNQNOq8,5411
+versionhq/knowledge/source.py,sha256=-hEUPtJUHHMx4rUKtiHl19J8xAMw-WVBw34zwa2jZ08,13630
+versionhq/knowledge/source_docling.py,sha256=mg7bgvKePHn2LlA_XzSFCbS0zOo9xfu_aNOf5cEo6c4,5421
 versionhq/knowledge/storage.py,sha256=7oxCg3W9mFjYH1YmuH9kFtTbNxquzYFjuUjd_TlsB9E,8170
 versionhq/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/llm/llm_vars.py,sha256=wjQK20cKvph6Vq1v71o4d16zBGcHlwq0bzOT_zWno7w,7041
 versionhq/llm/model.py,sha256=wlzDUMEyIOm808d1vzqu9gmbB4ch-s_EUvwFR60gR80,17177
 versionhq/memory/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/memory/contextual_memory.py,sha256=tCsOOAUnfrOL7YiakqGoi3uShzzS870TmGnlGd3z_A4,3556
-versionhq/memory/model.py,sha256=4wow2O3UuMZ0AbC2NyxddGZac3-_GjNZbK9wsA015NA,8145
+versionhq/memory/model.py,sha256=MPO8dDP5eAuk9td6bMOq5j2huLzCADFJDrTujHhcWQY,8150
 versionhq/storage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/storage/base.py,sha256=p-Jas0fXQan_qotnRD6seQxrT2lj-uw9-SmHQhdppcs,355
 versionhq/storage/ltm_sqlite_storage.py,sha256=wdUiuwHfJocdk0UGqyrdU4S5Nae1rgsoRNu3LWmGFcI,3951
@@ -44,15 +44,14 @@ versionhq/storage/task_output_storage.py,sha256=E1t_Fkt78dPYIOl3MP7LfQ8oGtjlzxBu
 versionhq/storage/utils.py,sha256=ByYXPoEIGJYLUqz-DWjbCAnneNrH1otiYbp12SCILpM,747
 versionhq/task/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/task/evaluate.py,sha256=WdUgjbZL62XrxyWe5MTz29scfzwmuAHGxJ7GvAB8Fmk,3954
-versionhq/task/formation.py,sha256=8YQxGZ7cs7Q-a7UnwheHxXei9LSt1AWpCQmQiN1ZkgI,6423
+versionhq/task/formation.py,sha256=gBrFmMHvoyqdK9wEa9aBkZImEQV3GMDDX6-RDpw0pEo,6421
 versionhq/task/formatter.py,sha256=N8Kmk9vtrMtBdgJ8J7RmlKNMdZWSmV8O1bDexmCWgU0,643
 versionhq/task/log_handler.py,sha256=LT7YnO7gcPR9IZS7eRvMjnHh8crMBFtqduxd8dxIbkk,1680
-versionhq/task/model.py,sha256=aXsFRhpmIyfoYpF8BU2cBU4nh4OsaSxc9D98DJ9auqE,30624
+versionhq/task/model.py,sha256=H7ZYZpp_1mrw_MlKjLr6gMJztqyhPkdeoolIzZSF68Y,30756
 versionhq/task/structured_response.py,sha256=4q-hQPu7oMMHHXEzh9YW4SJ7N5eCZ7OfZ65juyl_jCI,5000
 versionhq/task/TEMPLATES/Description.py,sha256=V-4kh8xpQTKOcDMi2xnuP-fcNk6kuoz1_5tYBlDLQWQ,420
-versionhq/team/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-versionhq/team/model.py,sha256=88l2lxlAZGIauRJr607IMbV7jP3FIvveVHrT7O4AVpk,18614
-versionhq/team/team_planner.py,sha256=Zc3zvhPR7T0O8RQoq9CKOrvrll4klemY1rONLFeJ96M,3663
+versionhq/task_graph/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+versionhq/task_graph/model.py,sha256=TsszoTJFeJ4TfSdLAjcGa9QSQyf9Uy3WmechTw0qL3k,22832
 versionhq/tool/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/tool/cache_handler.py,sha256=iL8FH7X0G-cdT0uhJwzuhLDaadTXOdfybZcDy151-es,1085
 versionhq/tool/composio_tool.py,sha256=38mEiVvTkuw1BLD233Bl1Gwxbpss1yfQiZLTWwX6BdA,8648
@@ -60,8 +59,8 @@ versionhq/tool/composio_tool_vars.py,sha256=FvBuEXsOQUYnN7RTFxT20kAkiEYkxWKkiVtg
 versionhq/tool/decorator.py,sha256=C4ZM7Xi2gwtEMaSeRo-geo_g_MAkY77WkSLkAuY0AyI,1205
 versionhq/tool/model.py,sha256=PO4zNWBZcJhYVur381YL1dy6zqurio2jWjtbxOxZMGI,12194
 versionhq/tool/tool_handler.py,sha256=2m41K8qo5bGCCbwMFferEjT-XZ-mE9F0mDUOBkgivOI,1416
-versionhq-1.2.1.0.dist-info/LICENSE,sha256=cRoGGdM73IiDs6nDWKqPlgSv7aR4n-qBXYnJlCMHCeE,1082
-versionhq-1.2.1.0.dist-info/METADATA,sha256=Ilxj9ZEYzoFzEnzdDE6umFKruSRN3pmcm_D2LvpVi8A,21854
-versionhq-1.2.1.0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-versionhq-1.2.1.0.dist-info/top_level.txt,sha256=DClQwxDWqIUGeRJkA8vBlgeNsYZs4_nJWMonzFt5Wj0,10
-versionhq-1.2.1.0.dist-info/RECORD,,
+versionhq-1.2.1.2.dist-info/LICENSE,sha256=cRoGGdM73IiDs6nDWKqPlgSv7aR4n-qBXYnJlCMHCeE,1082
+versionhq-1.2.1.2.dist-info/METADATA,sha256=SjmFRj8FztRldIYEtsObQlY5XTk6mWL-zmr6sCaSWAY,22116
+versionhq-1.2.1.2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+versionhq-1.2.1.2.dist-info/top_level.txt,sha256=DClQwxDWqIUGeRJkA8vBlgeNsYZs4_nJWMonzFt5Wj0,10
+versionhq-1.2.1.2.dist-info/RECORD,,

versionhq/team/team_planner.py

@@ -1,92 +0,0 @@
-import os
-from typing import Any, List, Optional, Dict
-from pydantic import BaseModel, Field
-
-
-class TeamPlanner:
-    """
-    A class to handle agent formations based on the given task description.
-    1) complexity
-    2) agent built (or use given agents)
-    3) knowledge, memory sharing
-    4) form a team
-    """
-
-    from versionhq.task.model import Task, ResponseField, TaskOutput
-    from versionhq.agent.model import Agent
-
-
-    def __init__(self, tasks: List[Task], planner_llm: Optional[Any] = None):
-        self.tasks = tasks
-        self.planner_llm = planner_llm if planner_llm else os.environ.get("DEFAULT_MODEL_NAME")
-
-
-    def _handle_assign_agents(self, unassigned_tasks: List[Task]) -> List[Any]:
-        """
-        Build an agent and assign it a task, then return a list of Member connecting the agent created and the task given.
-        """
-
-        from versionhq.agent.model import Agent
-        from versionhq.task.model import Task, ResponseField
-        from versionhq.team.model import Member
-
-        new_member_list: List[Member] = []
-        agent_creator = Agent(
-            role="agent_creator",
-            goal="build an ai agent that can competitively handle the task given",
-            llm=self.planner_llm,
-        )
-
-        for unassgined_task in unassigned_tasks:
-            task = Task(
-                description=f"""
-                    Based on the following task summary, draft a AI agent's role and goal in concise manner.
-                    Task summary: {unassgined_task.summary}
-                """,
-                response_fields=[
-                    ResponseField(title="goal", data_type=str, required=True),
-                    ResponseField(title="role", data_type=str, required=True),
-                ],
-            )
-            res = task.execute(agent=agent_creator)
-            agent = Agent(
-                role=res.json_dict["role"] if "role" in res.json_dict else res.raw,
-                goal=res.json_dict["goal"] if "goal" in res.json_dict else task.description
-            )
-            if agent.id:
-                team_member = Member(agent=agent, tasks=[unassgined_task], is_manager=False)
-                new_member_list.append(team_member)
-
-        return new_member_list
-
-
-    def _handle_task_planning(self, context: Optional[str] = None, tools: Optional[str] = None) -> TaskOutput:
-        """
-        Handles the team planning by creating detailed step-by-step plans for each task.
-        """
-
-        from versionhq.agent.model import Agent
-        from versionhq.task.model import Task
-
-        team_planner = Agent(
-            role="team planner",
-            goal="Plan extremely detailed, step-by-step plan based on the tasks and tools available to each agent so that they can perform the tasks in an exemplary manner and assign a task to each agent.",
-            llm=self.planner_llm,
-        )
-
-        task_summary_list = [task.summary for task in self.tasks]
-
-        class TeamPlanIdea(BaseModel):
-            plan: str | Dict[str, Any] = Field(default=None, description="a decriptive plan to be executed by the team")
-
-
-        task = Task(
-            description=f"""
-                Based on the following task summaries, create the most descriptive plan that the team can execute most efficiently. Take all the task summaries - task's description and tools available - into consideration. Your answer only contains a dictionary.
-
-                Task summaries: {" ".join(task_summary_list)}
-            """,
-            pydantic_output=TeamPlanIdea
-        )
-        output = task.execute(agent=team_planner, context=context, tools=tools)
-        return output