versionhq 1.2.1.12__py3-none-any.whl → 1.2.1.14__py3-none-any.whl

versionhq/__init__.py

@@ -31,7 +31,7 @@ from versionhq.task.formation import form_agent_network
 from versionhq.task_graph.draft import workflow
 
 
-__version__ = "1.2.1.12"
+__version__ = "1.2.1.14"
 __all__ = [
     "Agent",
 

versionhq/agent/model.py

@@ -3,7 +3,6 @@ import uuid
 from typing import Any, Dict, List, Optional, TypeVar, Callable, Type
 from typing_extensions import Self
 from dotenv import load_dotenv
-import litellm
 
 from pydantic import UUID4, BaseModel, Field, InstanceOf, PrivateAttr, model_validator, field_validator
 from pydantic_core import PydanticCustomError
@@ -57,9 +56,7 @@ class TokenProcess:
 # @track_agent()
 class Agent(BaseModel):
     """
-    A class to store agent information.
-    Agents must have `role`, `goal`, and `llm` = DEFAULT_MODEL_NAME as default.
-    Then run validation on `backstory`, `llm`, `tools`, `rpm` (request per min), `knowledge`, and `memory`.
+    A Pydantic class to store an agent object. Agents must have `role` and `goal` to start.
     """
 
     __hash__ = object.__hash__
@@ -453,7 +450,6 @@ class Agent(BaseModel):
         return self
 
 
-
     def invoke(
         self,
         prompts: str,
@@ -515,8 +511,6 @@ class Agent(BaseModel):
                 raise ValueError("Invalid response from LLM call - None or empty.")
 
 
-
-
     def execute_task(self, task, context: Optional[Any] = None, task_tools: Optional[List[Tool | ToolSet]] = list()) -> str:
         """
         Format a task prompt, adding context from knowledge and memory (if given), and invoke LLM.
@@ -582,5 +576,27 @@ class Agent(BaseModel):
         return raw_response
 
 
+    def start(self, context: Any = None) -> Any | None:
+        """
+        Defines and executes a task when it is not given and returns TaskOutput object.
+        """
+
+        if not self.goal or not self.role:
+            return None
+
+        from versionhq.task.model import Task, ResponseField
+
+        class Output(BaseModel):
+            result: str
+            steps: list[str]
+
+        task = Task(
+            description=f"Generate a simple result in a sentence to achieve the goal: {self.goal}. If needed, list up necessary steps in concise manner.",
+            pydantic_output=Output
+        )
+        res = task.execute(agent=self, context=context)
+        return res
+
+
     def __repr__(self):
         return f"Agent(role={self.role}, goal={self.goal}"

versionhq/task/model.py

@@ -363,7 +363,7 @@ Response format: {response_format}
 Ref. Output image: {output_formats_to_follow}
 """
         else:
-            output_prompt = "Return your response as a valid JSON serializable string, enclosed in double quotes. Do not use single quotes, trailing commas, or other non-standard JSON syntax."
+            output_prompt = "You MUST Return your response as a valid JSON serializable string, enclosed in double quotes. Do not use single quotes, trailing commas, or other non-standard JSON syntax."
 
         return dedent(output_prompt)
 

versionhq/task_graph/draft.py

@@ -1,6 +1,6 @@
 import sys
 from typing import Type, Any
-from pydantic import BaseModel, create_model
+from pydantic import BaseModel
 from pydantic._internal._model_construction import ModelMetaclass
 from textwrap import dedent
 if 'pydantic.main' not in sys.modules:
@@ -10,11 +10,11 @@ sys.modules['pydantic.main'].ModelMetaclass = ModelMetaclass
 
 from versionhq.agent.model import Agent
 from versionhq.task.model import ResponseField
-from versionhq.task_graph.model import TaskGraph, Task, DependencyType, Node, TaskStatus
+from versionhq.task_graph.model import TaskGraph, Task, DependencyType, Node
 from versionhq._utils.logger import Logger
 
 
-def workflow(final_output: Type[BaseModel], context: Any = None, human: bool = True) -> TaskGraph | None:
+def workflow(final_output: Type[BaseModel], context: Any = None, human: bool = False) -> TaskGraph | None:
     """
     Generate a TaskGraph object to generate the givne final_output most resource-efficiently.
     """
@@ -52,13 +52,12 @@ def workflow(final_output: Type[BaseModel], context: Any = None, human: bool = T
         description=dedent(f"Design a resource-efficient workflow to achieve the following goal: {final_output_prompt}. The workflow should consist of a list of detailed tasks that represent decision making points, each with the following information:\nname: A concise name of the task\ndescription: A concise description of the task.\nconnections: A list of target tasks that this task connects to.\ndependency_types: The type of dependency between this task and each of its connected task. \noutput: key output from the task in a word.\n\nUse the following dependency types: {dep_type_prompt}.\n\nPrioritize minimizing resource consumption (computation, memory, and data transfer) when defining tasks, connections, and dependencies.  Consider how data is passed between tasks and aim to reduce unnecessary data duplication or transfer. Explain any design choices made to optimize resource usage."),
         response_fields=[
             ResponseField(title="tasks", data_type=list, items=dict, properties=[
-                    ResponseField(title="name", data_type=str),
-                    ResponseField(title="description", data_type=str),
-                    ResponseField(title="output", data_type=str),
-                    ResponseField(title="connections", data_type=list, items=str),
-                    ResponseField(title="dependency_types", data_type=list, items=str),
-                ]
-            )
+                ResponseField(title="name", data_type=str),
+                ResponseField(title="description", data_type=str),
+                ResponseField(title="output", data_type=str),
+                ResponseField(title="connections", data_type=list, items=str),
+                ResponseField(title="dependency_types", data_type=list, items=str),
+            ])
         ]
     )
     res = task.execute(agent=graph_expert, context=[context_prompt, context])

{versionhq-1.2.1.12.dist-info → versionhq-1.2.1.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.2.1.12
+Version: 1.2.1.14
 Summary: An agentic orchestration framework for building agent networks that handle task automation.
 Author-email: Kuriko Iwai <kuriko@versi0n.io>
 License: MIT License

{versionhq-1.2.1.12.dist-info → versionhq-1.2.1.14.dist-info}/RECORD

@@ -1,4 +1,4 @@
-versionhq/__init__.py,sha256=uNfcqA6KXR3l8o5Df1_XZCdFKevMxrQ5ri-JJhTYJ1U,2883
+versionhq/__init__.py,sha256=iRSTMITQMFFixAJsBgg0Og100UdL-svAq0H0EoXjM6U,2883
 versionhq/_utils/__init__.py,sha256=dzoZr4cBlh-2QZuPzTdehPUCe9lP1dmRtauD7qTjUaA,158
 versionhq/_utils/i18n.py,sha256=TwA_PnYfDLA6VqlUDPuybdV9lgi3Frh_ASsb_X8jJo8,1483
 versionhq/_utils/logger.py,sha256=zgogTwAY-ujDLrdryAKhdtoaNe1nOFajmEN0V8aMR34,3155
@@ -7,7 +7,7 @@ versionhq/_utils/usage_metrics.py,sha256=NXF18dn5NNvGK7EsQ4AAghpR8ppYOjMx6ABenLL
 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=snDtgDmvZB2bZKH_RTcz5uFOMl3MTjLJwTQBebFt8hk,2532
-versionhq/agent/model.py,sha256=L21COegL50x1sl-1gZqYnICx8Gm276J0GnQx6IthXf8,25343
+versionhq/agent/model.py,sha256=eFmkNuCJhXRhdYJR59J5PFxw5aghLvBrPRj4NXd7J0w,25873
 versionhq/agent/parser.py,sha256=riG0dkdQCxH7uJ0AbdVdg7WvL0BXhUgJht0VtQvxJBc,4082
 versionhq/agent/rpm_controller.py,sha256=grezIxyBci_lDlwAlgWFRyR5KOocXeOhYkgN02dNFNE,2360
 versionhq/agent/TEMPLATES/Backstory.py,sha256=IAhGnnt6VUMe3wO6IzeyZPDNu7XE7Uiu3VEXUreOcKs,532
@@ -47,12 +47,12 @@ versionhq/task/evaluate.py,sha256=WdUgjbZL62XrxyWe5MTz29scfzwmuAHGxJ7GvAB8Fmk,39
 versionhq/task/formation.py,sha256=WH604q9bRmWH7KQCrk2qKJwisCopYX5CjJvsj4TgFjI,6894
 versionhq/task/formatter.py,sha256=N8Kmk9vtrMtBdgJ8J7RmlKNMdZWSmV8O1bDexmCWgU0,643
 versionhq/task/log_handler.py,sha256=LT7YnO7gcPR9IZS7eRvMjnHh8crMBFtqduxd8dxIbkk,1680
-versionhq/task/model.py,sha256=ASc7l5Lhm9tfINYGKtBikqaLAXbobPQNd0xKFv_IcS0,28529
+versionhq/task/model.py,sha256=EcUW7nktGBd9CBenKLh-5HBRa02wrwYF6WeF8ju04tc,28538
 versionhq/task/structured_response.py,sha256=4q-hQPu7oMMHHXEzh9YW4SJ7N5eCZ7OfZ65juyl_jCI,5000
 versionhq/task/TEMPLATES/Description.py,sha256=V-4kh8xpQTKOcDMi2xnuP-fcNk6kuoz1_5tYBlDLQWQ,420
 versionhq/task_graph/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/task_graph/colors.py,sha256=naJCx4Vho4iuJtbW8USUXb-M5uYvd5ds2p8qbjUfRus,669
-versionhq/task_graph/draft.py,sha256=tEymdNbDXPXe88rvTdocpx22djhKV_TmSHa6lqkJ8fY,5158
+versionhq/task_graph/draft.py,sha256=rt41kJ85g5l501K7PC-PVtn1oWmKABKsznorfivcDpc,5096
 versionhq/task_graph/model.py,sha256=njyHQyHrVTZP46iVkC6YvuMnGcS40vOy1wszRtf7DHY,23971
 versionhq/tool/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/tool/cache_handler.py,sha256=iL8FH7X0G-cdT0uhJwzuhLDaadTXOdfybZcDy151-es,1085
@@ -61,8 +61,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.12.dist-info/LICENSE,sha256=cRoGGdM73IiDs6nDWKqPlgSv7aR4n-qBXYnJlCMHCeE,1082
-versionhq-1.2.1.12.dist-info/METADATA,sha256=AMSEfWE5CXam7k5CpDG-3Ne8Ci-_uzpaDalsjSAHUSM,22033
-versionhq-1.2.1.12.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-versionhq-1.2.1.12.dist-info/top_level.txt,sha256=DClQwxDWqIUGeRJkA8vBlgeNsYZs4_nJWMonzFt5Wj0,10
-versionhq-1.2.1.12.dist-info/RECORD,,
+versionhq-1.2.1.14.dist-info/LICENSE,sha256=cRoGGdM73IiDs6nDWKqPlgSv7aR4n-qBXYnJlCMHCeE,1082
+versionhq-1.2.1.14.dist-info/METADATA,sha256=s7vOZmrrn59L5L6ItVtEVzg92yHmrv58PXAr805sbZQ,22033
+versionhq-1.2.1.14.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+versionhq-1.2.1.14.dist-info/top_level.txt,sha256=DClQwxDWqIUGeRJkA8vBlgeNsYZs4_nJWMonzFt5Wj0,10
+versionhq-1.2.1.14.dist-info/RECORD,,