versionhq 1.1.7.1__tar.gz → 1.1.7.2__tar.gz

{versionhq-1.1.7.1 → versionhq-1.1.7.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: versionhq
-Version: 1.1.7.1
+Version: 1.1.7.2
 Summary: LLM orchestration frameworks for model-agnostic AI agents that handle complex outbound workflows
 Author-email: Kuriko Iwai <kuriko@versi0n.io>
 License: MIT License

{versionhq-1.1.7.1 → versionhq-1.1.7.2}/pyproject.toml

@@ -15,7 +15,7 @@ exclude = ["test*", "__pycache__"]
 
 [project]
 name = "versionhq"
-version = "1.1.7.1"
+version = "1.1.7.2"
 authors = [{ name = "Kuriko Iwai", email = "kuriko@versi0n.io" }]
 description = "LLM orchestration frameworks for model-agnostic AI agents that handle complex outbound workflows"
 readme = "README.md"

{versionhq-1.1.7.1 → versionhq-1.1.7.2}/src/versionhq/__init__.py

@@ -17,7 +17,7 @@ from versionhq.team.model import Team, TeamOutput
 from versionhq.tool.model import Tool
 
 
-__version__ = "1.1.7.1"
+__version__ = "1.1.7.2"
 __all__ = [
     "Agent",
     "Customer",

{versionhq-1.1.7.1 → versionhq-1.1.7.2}/src/versionhq/agent/model.py

@@ -367,7 +367,7 @@ class Agent(ABC, BaseModel):
         """
 
         task_prompt = task.prompt()
-        if context:
+        if context is not task.prompt_context:  # as `task.prompt()` includes adding `task.prompt_context` to the prompt.
             task_prompt += context
 
         tool_results = []

{versionhq-1.1.7.1 → versionhq-1.1.7.2}/src/versionhq/task/model.py

@@ -153,7 +153,7 @@ class Task(BaseModel):
     tools_called: Optional[List[ToolCalled]] = Field(default_factory=list, description="tools that the agent can use for this task")
     take_tool_res_as_final: bool = Field(default=False,description="when set True, tools res will be stored in the `TaskOutput`")
 
-    prompt_context: Optional[str] = None
+    prompt_context: Optional[str] = Field(default=None)
     async_execution: bool = Field(default=False,description="whether the task should be executed asynchronously or not")
     config: Optional[Dict[str, Any]] = Field(default=None, description="configuration for the agent")
     callback: Optional[Any] = Field(default=None, description="callback to be executed after the task is completed.")
@@ -278,17 +278,20 @@ Your outputs MUST adhere to the following format and should NOT include any irre
         When we have cusotmer/product info, add them to the prompt.
         """
 
-        task_slices = [self.description, f"{self.output_prompt}"]
+        task_slices = [self.description, f"{self.output_prompt}", f"Take the following context into consideration: "]
 
         if self.context:
             context_outputs = "\n".join([task.output.context_prompting() if hasattr(task, "output") else "" for task in self.context])
-            task_slices.insert(1,  f"Take the following context into consideration: {context_outputs}")
+            task_slices.insert(len(task_slices),  context_outputs)
 
         if customer:
-            task_slices.insert(1,  f"customer overview: {customer}")
+            task_slices.insert(len(task_slices),  f"Customer overview: {customer}")
 
         if product_overview:
-            task_slices.insert(1, f"Product overview: {product_overview}")
+            task_slices.insert(len(task_slices), f"Product overview: {product_overview}")
+
+        if self.prompt_context:
+            task_slices.insert(len(task_slices), self.prompt_context)
 
         return "\n".join(task_slices)
 

{versionhq-1.1.7.1 → versionhq-1.1.7.2}/src/versionhq.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: versionhq
-Version: 1.1.7.1
+Version: 1.1.7.2
 Summary: LLM orchestration frameworks for model-agnostic AI agents that handle complex outbound workflows
 Author-email: Kuriko Iwai <kuriko@versi0n.io>
 License: MIT License

{versionhq-1.1.7.1 → versionhq-1.1.7.2}/tests/task/task_test.py

@@ -51,7 +51,7 @@ def test_sync_execute_task():
             assert type(res.pydantic.test2) == list
 
 
-def test_sync_execute_task_with_context():
+def test_sync_execute_task_with_task_context():
     """
     Use case = One agent handling multiple tasks sequentially using context set in the main task.
     """
@@ -109,4 +109,60 @@ def test_sync_execute_task_with_context():
     assert "result" in main_task.prompt()
 
 
+def test_sync_execute_task_with_prompt_context():
+    agent = Agent(
+        role="demo agent 001",
+        goal="My amazing goals",
+        backstory="My amazing backstory",
+        verbose=True,
+        llm=DEFAULT_MODEL_NAME,
+        max_tokens=3000,
+    )
+
+    sub_task = Task(
+        description="analyze the client's business model",
+        expected_output_json=True,
+        expected_output_pydantic=False,
+        output_field_list=[
+            ResponseField(title="result", type=str, required=True),
+        ]
+    )
+    main_task = Task(
+        description="Define the optimal cohort timeframe in days and target audience.",
+        expected_output_json=True,
+        expected_output_pydantic=True,
+        output_field_list=[
+            ResponseField(title="test1", type=int, required=True),
+            ResponseField(title="test2", type=str, required=True),
+        ],
+        context=[sub_task]
+    )
+    res = main_task.execute_sync(agent=agent, context="consider building black friday marketing campaign.")
+
+    assert isinstance(res, TaskOutput)
+    assert res.task_id is main_task.id
+    assert res.raw is not None
+    assert isinstance(res.raw, str)
+    assert res.json_dict is not None
+    assert isinstance(res.json_dict, dict)
+    assert res.pydantic is not None
+
+    if hasattr(res.pydantic, "output"):
+        assert res.pydantic.output is not None
+    else:
+        assert hasattr(res.pydantic, "test1")
+        if res.pydantic.test1:
+            assert type(res.pydantic.test1) == Union[int, str]
+
+        assert hasattr(res.pydantic, "test2")
+        if res.pydantic.test2:
+            assert type(res.pydantic.test2) == Union[list, str]
+
+    assert sub_task.output is not None
+    assert sub_task.output.json_dict is not None
+    assert "result" in main_task.prompt()
+    assert main_task.prompt_context == "consider building black friday marketing campaign."
+    assert "consider building black friday marketing campaign." in main_task.prompt()
+
+
 # CALLBACKS, tools, FUTURE, ASYNC, CONDITIONAL, token usage