versionhq 1.2.1.5__tar.gz → 1.2.1.6__tar.gz

versionhq-1.2.1.6/.env.sample

@@ -0,0 +1,18 @@
+DEFAULT_MODEL_NAME=
+
+LITELLM_API_KEY=
+LITELLM_API_BASE=
+
+OPENAI_API_KEY=
+GEMINI_API_KEY=
+OPENROUTER_API_KEY=
+ANTHROPIC_API_KEY=
+HUGGINGFACE_API_BASE=
+HUGGINGFACE_API_KEY=
+
+UPSTAGE_API_KEY=
+
+COMPOSIO_API_KEY=
+COMPOSIO_CLI_KEY=
+
+MEM0_API_KEY=

{versionhq-1.2.1.5 → versionhq-1.2.1.6}/.gitignore

@@ -4,8 +4,9 @@ destinations.py
 train.py
 
 _memo
+memo.txt
 _idea.py
-
+site/
 
 dist/
 lib/
@@ -13,7 +14,7 @@ build/
 .pypirc
 *egg-info/
 
-uploads/
+.diagrams/
 sample_dataset/
 chroma.sqlite3
 
@@ -24,8 +25,8 @@ temp/*
 .vscode/*
 assets/*
 
-memo.txt
-site/
+
+
 
 _logs/
 .logs/

{versionhq-1.2.1.5 → versionhq-1.2.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.2.1.5
+Version: 1.2.1.6
 Summary: An agentic orchestration framework for building agent networks that handle task automation.
 Author-email: Kuriko Iwai <kuriko@versi0n.io>
 License: MIT License
@@ -89,7 +89,7 @@ Requires-Dist: pygraphviz>=1.14; extra == "pygraphviz"
 ![pyenv ver](https://img.shields.io/badge/pyenv-2.5.0-orange)
 
 
-Agentic orchestration framework for multi-agent networks and task graph for complex task automation.
+Agentic orchestration framework for multi-agent networks and task graphs for complex task automation.
 
 **Visit:**
 
@@ -361,7 +361,7 @@ Tasks can be delegated to a manager, peers within the agent network, or a comple
 
 **Workflow, Task Graph**
 
-* [NetworkX](https://networkx.org/documentation/stable/reference/introduction.html): A Python package to analyze, create, and manipulate complex graph networks.
+* [NetworkX](https://networkx.org/documentation/stable/reference/introduction.html): A Python package to analyze, create, and manipulate complex graph networks. Ref. [Gallary](https://networkx.org/documentation/latest/auto_examples/index.html)
 * [Matplotlib](https://matplotlib.org/stable/index.html): For graph visualization.
 * [Graphviz](https://graphviz.org/about/): For graph visualization.
 
@@ -397,7 +397,7 @@ Tasks can be delegated to a manager, peers within the agent network, or a comple
 .github
 └── workflows/                # Github actions
 │
-docs/                         # Documentation built by MkDocs
+docs/                         # Documentation
 mkdocs.yml                    # MkDocs config
 │
 src/
@@ -413,12 +413,13 @@ src/
 │     └── llm/
 │     └── ...
 │
-└── uploads/  [.gitignore]    # Local directory to store uploaded files such as graphviz diagrams generatd by `Network` class
+└── .diagrams/  [.gitignore]  # Local directory to store graph diagrams
 │
-└── .logs/    [.gitignore]    # Local directory to store error/warning logs for debugging
+└── .logs/      [.gitignore]  # Local directory to store error/warning logs for debugging
 │
 │
 pyproject.toml                # Project config
+.env.sample                   # sample .env file
 
 ```
 
@@ -475,16 +476,7 @@ pyproject.toml                # Project config
 
 ### Adding env secrets to .env file
 
-Create `.env` file in the project root and add following:
-
-   ```
-   OPENAI_API_KEY=your-openai-api-key
-   GEMINI_API_KEY=your-gemini-api-key
-   LITELLM_API_KEY=your-litellm-api-key
-   COMPOSIO_API_KEY=your-composio-api-key
-   COMPOSIO_CLI_KEY=your-composio-cli-key
-   [OTHER_LLM_INTERFACE_PROVIDER_OF_YOUR_CHOICE]_API_KEY=your-api-key
-   ```
+Create `.env` file in the project root and add secret vars following `.env.sample` file.
 
 
 <hr />

{versionhq-1.2.1.5 → versionhq-1.2.1.6}/README.md

@@ -8,7 +8,7 @@
 ![pyenv ver](https://img.shields.io/badge/pyenv-2.5.0-orange)
 
 
-Agentic orchestration framework for multi-agent networks and task graph for complex task automation.
+Agentic orchestration framework for multi-agent networks and task graphs for complex task automation.
 
 **Visit:**
 
@@ -280,7 +280,7 @@ Tasks can be delegated to a manager, peers within the agent network, or a comple
 
 **Workflow, Task Graph**
 
-* [NetworkX](https://networkx.org/documentation/stable/reference/introduction.html): A Python package to analyze, create, and manipulate complex graph networks.
+* [NetworkX](https://networkx.org/documentation/stable/reference/introduction.html): A Python package to analyze, create, and manipulate complex graph networks. Ref. [Gallary](https://networkx.org/documentation/latest/auto_examples/index.html)
 * [Matplotlib](https://matplotlib.org/stable/index.html): For graph visualization.
 * [Graphviz](https://graphviz.org/about/): For graph visualization.
 
@@ -316,7 +316,7 @@ Tasks can be delegated to a manager, peers within the agent network, or a comple
 .github
 └── workflows/                # Github actions
 │
-docs/                         # Documentation built by MkDocs
+docs/                         # Documentation
 mkdocs.yml                    # MkDocs config
 │
 src/
@@ -332,12 +332,13 @@ src/
 │     └── llm/
 │     └── ...
 │
-└── uploads/  [.gitignore]    # Local directory to store uploaded files such as graphviz diagrams generatd by `Network` class
+└── .diagrams/  [.gitignore]  # Local directory to store graph diagrams
 │
-└── .logs/    [.gitignore]    # Local directory to store error/warning logs for debugging
+└── .logs/      [.gitignore]  # Local directory to store error/warning logs for debugging
 │
 │
 pyproject.toml                # Project config
+.env.sample                   # sample .env file
 
 ```
 
@@ -394,16 +395,7 @@ pyproject.toml                # Project config
 
 ### Adding env secrets to .env file
 
-Create `.env` file in the project root and add following:
-
-   ```
-   OPENAI_API_KEY=your-openai-api-key
-   GEMINI_API_KEY=your-gemini-api-key
-   LITELLM_API_KEY=your-litellm-api-key
-   COMPOSIO_API_KEY=your-composio-api-key
-   COMPOSIO_CLI_KEY=your-composio-cli-key
-   [OTHER_LLM_INTERFACE_PROVIDER_OF_YOUR_CHOICE]_API_KEY=your-api-key
-   ```
+Create `.env` file in the project root and add secret vars following `.env.sample` file.
 
 
 <hr />

{versionhq-1.2.1.5 → versionhq-1.2.1.6}/mkdocs.yml

@@ -109,15 +109,15 @@ nav:
   - Home: 'index.md'
   - Quick Start: 'quickstart.md'
   - Core Components:
-      - Agent: 'core/agent.md'
-      - Task Graph: 'core/task-graph.md'
-      - Task:
-        - Task: 'core/task.md'
-        - ResponseField: 'core/task/response-field.md'
-        - TaskOutput: 'core/task/task-output.md'
-        - Evaluation: 'core/task/evaluation.md'
-      - Tool: 'core/tool.md'
-    #   - Compoio Tools: 'core/composio-tool.md'
+    - Agent: 'core/agent.md'
+    - Task Graph: 'core/task-graph.md'
+    - Task:
+      - Task: 'core/task.md'
+      - ResponseField: 'core/task/response-field.md'
+      - TaskOutput: 'core/task/task-output.md'
+      - Evaluation: 'core/task/evaluation.md'
+    - Tool: 'core/tool.md'
+  #   - Compoio Tools: 'core/composio-tool.md'
   - Tags: 'tags.md'
   - Examples:
     - Playground: https://versi0n.io

{versionhq-1.2.1.5 → versionhq-1.2.1.6}/pyproject.toml

@@ -15,7 +15,7 @@ exclude = ["test*", "__pycache__", "*.egg-info"]
 
 [project]
 name = "versionhq"
-version = "1.2.1.5"
+version = "1.2.1.6"
 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.5 → versionhq-1.2.1.6}/src/versionhq/__init__.py

@@ -28,9 +28,10 @@ from versionhq.memory.contextual_memory import ContextualMemory
 from versionhq.memory.model import ShortTermMemory,LongTermMemory, UserMemory, MemoryItem
 
 from versionhq.task.formation import form_agent_network
+from versionhq.task_graph.draft import workflow
 
 
-__version__ = "1.2.1.5"
+__version__ = "1.2.1.6"
 __all__ = [
     "Agent",
 
@@ -88,5 +89,6 @@ __all__ = [
     "UserMemory",
     "MemoryItem",
 
-    "form_agent_network"
+    "form_agent_network",
+    "workflow",
 ]

{versionhq-1.2.1.5 → versionhq-1.2.1.6}/src/versionhq/task/formation.py

@@ -39,7 +39,7 @@ def form_agent_network(
                         pass
 
                 case str():
-                    matched = [item for item in Formation._member_names_ if item == formation.upper()]
+                    matched = [item for item in Formation.s_ if item == formation.upper()]
                     if matched:
                         formation = getattr(Formation, matched[0])
                     else:

{versionhq-1.2.1.5 → versionhq-1.2.1.6}/src/versionhq/task/model.py

@@ -4,6 +4,7 @@ import datetime
 import uuid
 import inspect
 import enum
+from textwrap import dedent
 from concurrent.futures import Future
 from hashlib import md5
 from typing import Any, Dict, List, Set, Optional, Callable, Type
@@ -288,7 +289,7 @@ class Task(BaseModel):
     should_evaluate: bool = Field(default=False, description="True to run the evaluation flow")
     eval_criteria: Optional[List[str]] = Field(default_factory=list, description="criteria to evaluate the outcome. i.e., fit to the brand tone")
 
-    # recording
+    # recording !# REFINEME - eval_callbacks
     processed_agents: Set[str] = Field(default_factory=set, description="store roles of the agents that executed the task")
     tool_errors: int = 0
     delegations: int = 0
@@ -364,7 +365,7 @@ 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."
 
-        return output_prompt
+        return dedent(output_prompt)
 
 
     def _draft_context_prompt(self, context: Any) -> str:
@@ -374,7 +375,7 @@ Ref. Output image: {output_formats_to_follow}
 
         context_to_add = None
         if not context:
-            Logger().log(level="error", color="red", message="Missing a context to add to the prompt. We'll return ''.")
+            # Logger().log(level="error", color="red", message="Missing a context to add to the prompt. We'll return ''.")
             return context_to_add
 
         match context:
@@ -403,7 +404,7 @@ Ref. Output image: {output_formats_to_follow}
             case _:
                 pass
 
-        return context_to_add
+        return dedent(context_to_add)
 
 
     def _prompt(self, model_provider: str = None, context: Optional[Any] = None) -> str:
@@ -593,7 +594,6 @@ Ref. Output image: {output_formats_to_follow}
                 return self._execute_async(agent=agent, context=context)
 
 
-
     def _execute_sync(self, agent, context: Optional[Any] = None) -> TaskOutput:
         """Executes the task synchronously."""
         return self._execute_core(agent, context)

versionhq-1.2.1.6/src/versionhq/task_graph/colors.py

@@ -0,0 +1,28 @@
+white = "#fafafa"
+black = "#191b1b"
+darkgrey = "#363636"
+grey = "#414141"
+primary = "#B9005B"
+orange = "#ffb713"
+lightgreen = "#b8fff5"
+green = "#00d1b2"
+darkgreen = "#169d87"
+darkergreen = "#026b5d"
+
+
+# $black: #191b1b;
+# $true-white: #ffffff;
+# $white: #fafafa;
+# $grey: #414141;
+# $light-grey: #f9f9f9;
+# $border-color: #dedede;
+# $primary: #B9005B;
+# $primary-hovered: #9D034F;
+# $gradient-main: linear-gradient(to right $black $primary);
+# $linkedin-blue: #0077B5;
+# $bulma-primary: #00d1b2;
+# $bluma-primary-dark: #169d87;
+# $bulma-grey-light: hsl(0 0%, 71%);
+# $bulma-grey: hsl(0, 0%, 48%);
+# $bulma-dark: hsl(0, 0%, 21%); //#363636
+# $bulma-white-ter: #F5F5F5;

versionhq-1.2.1.6/src/versionhq/task_graph/draft.py

@@ -0,0 +1,101 @@
+import sys
+from typing import Type, Any
+from pydantic import BaseModel
+from pydantic._internal._model_construction import ModelMetaclass
+from textwrap import dedent
+if 'pydantic.main' not in sys.modules:
+    import pydantic.main
+
+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._utils.logger import Logger
+
+
+def workflow(final_output: Type[BaseModel], context: Any = None, human: bool = True) -> TaskGraph | None:
+    """
+    Generate a TaskGraph object to generate the givne final_output most resource-efficiently.
+    """
+
+    if not final_output or not isinstance(final_output, ModelMetaclass):
+        Logger().log(level="error", message="Missing an expected output in Pydantic model.", color="red")
+        return None
+
+    final_output_prompt = ", ".join([k for k in final_output.model_fields.keys()])
+
+    if not final_output_prompt:
+        Logger().log(level="error", message="Expected output is in invalid format.", color="red")
+        return None
+
+    context_prompt = f'We are designing a resource-efficient workflow using graph algorithm concepts to achieve the following goal: {final_output_prompt}.'
+
+    dep_type_prompt = ", ".join([k for k in DependencyType._member_map_.keys()])
+
+    graph_expert = Agent(
+        role="vhq-G Expert",
+        goal="design the most resource-efficient workflow graph to achieve the given goal",
+        knowledge_sources=[
+            "https://en.wikipedia.org/wiki/Graph_theory",
+            # "https://www.geeksforgeeks.org/graph-data-structure-and-algorithms/?ref=lbp",
+            "https://www.geeksforgeeks.org/graph-and-its-representations/",
+            ", ".join([k for k in DependencyType._member_map_.keys()]),
+        ],
+        llm="gemini-2.0",
+        use_memory=True,
+        maxit=1,
+        max_retry_limit=1,
+    )
+
+    task = Task(
+        description=f"Design a resource-efficient workflow to achieve the following goal: {final_output_prompt}.  The workflow should consist of a list of tasks, 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. Use 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="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])
+
+    if not res:
+        return None
+
+    task_items = res.json_dict["tasks"]
+    tasks = [Task(name=item["name"], description=item["description"]) for item in task_items]
+    nodes = [Node(task=task) for task in tasks]
+    task_graph = TaskGraph(
+        nodes={node.identifier: node for node in nodes},
+        concl=final_output,
+        should_reform=True,
+    )
+
+    for res in task_items:
+        if res["connections"]:
+            dependency_types = [DependencyType[dt] if DependencyType[dt] else DependencyType.FINISH_TO_START for dt in res["dependency_types"]]
+
+            for i, target_task_name in enumerate(res["connections"]):
+                source = [v for k, v in task_graph.nodes.items() if v.task.name == res["name"]][0]
+                target = [v for k, v in task_graph.nodes.items() if v.task.name == target_task_name][0]
+                dependency_type = dependency_types[i]
+                task_graph.add_dependency(source_node_identifier=source.identifier, target_node_identifier=target.identifier, dependency_type=dependency_type)
+
+    ## test purpose
+    # task_graph.visualize()
+
+    # if human:
+    #     print('Proceed? Y/n:')
+    #     x = input()
+
+    #     if x.lower() == "y":
+    #         print("ok. generating agent network")
+
+    #     else:
+    #         request = input("request?")
+    #         print('ok. regenerating the graph based on your input: ', request)
+
+    return task_graph