versionhq 1.1.12.1__tar.gz → 1.1.12.3__tar.gz

{versionhq-1.1.12.1 → versionhq-1.1.12.3}/.gitignore

@@ -5,6 +5,9 @@ entity_memory.py
 
 train.py
 
+mkdocs.yml
+docs/
+
 dist/
 lib/
 build/

{versionhq-1.1.12.1 → versionhq-1.1.12.3}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.1.12.1
-Summary: LLM orchestration frameworks for model-agnostic AI agents that handle complex outbound workflows
+Version: 1.1.12.3
+Summary: An agentic orchestration framework for multi-agent system that shares memory, knowledge base, and RAG tools.
 Author-email: Kuriko Iwai <kuriko@versi0n.io>
 License: MIT License
         
@@ -29,7 +29,7 @@ Project-URL: Homepage, https://versi0n.io
 Project-URL: Documentation, https://chief-oxygen-8a2.notion.site/Documentation-17e923685cf98001a5fad5c4b2acd79b?pvs=73
 Project-URL: Repository, https://github.com/versionHQ/multi-agent-system
 Project-URL: Issues, https://github.com/versionHQ/multi-agent-system/issues
-Keywords: orchestration framework,orchestration,ai agent,multi-agent system,RAG,agent
+Keywords: orchestration framework,orchestration,ai agent,multi-agent system,RAG,agent,agentic orchestration,llm
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
@@ -63,6 +63,7 @@ Requires-Dist: composio-langchain>=0.6.12
 Requires-Dist: chromadb>=0.6.3
 Requires-Dist: wheel>=0.45.1
 Requires-Dist: envoy>=0.0.3
+Requires-Dist: composio-core==0.7.0
 Provides-Extra: docling
 Requires-Dist: docling>=2.17.0; extra == "docling"
 Provides-Extra: mem0ai
@@ -83,7 +84,7 @@ Requires-Dist: numpy>=1.26.4; extra == "numpy"
 ![pyenv ver](https://img.shields.io/badge/pyenv-2.5.0-orange)
 
 
-LLM orchestration frameworks to deploy multi-agent systems and automate complex tasks with network formations.
+Agentic orchestration framework to deploy agent network and handle complex task automation.
 
 **Visit:**
 
@@ -102,9 +103,10 @@ LLM orchestration frameworks to deploy multi-agent systems and automate complex
 - [Key Features](#key-features)
   - [Agent formation](#agent-formation)
 - [Quick Start](#quick-start)
-  - [Case 1. Solo Agent:](#case-1-solo-agent)
+  - [Generate agent networks and launch task execution:](#generate-agent-networks-and-launch-task-execution)
+  - [Solo Agent:](#solo-agent)
     - [Return a structured output with a summary in string.](#return-a-structured-output-with-a-summary-in-string)
-  - [Case 2. Supervising:](#case-2-supervising)
+  - [Supervising:](#supervising)
 - [Technologies Used](#technologies-used)
 - [Project Structure](#project-structure)
 - [Setup](#setup)
@@ -122,9 +124,9 @@ LLM orchestration frameworks to deploy multi-agent systems and automate complex
 
 ## Key Features
 
-Generate multi-agent systems based on the task complexity, execute tasks, and evaluate output based on the given criteria.
+`versionhq` is a Python framework that can generate agent networks for complex task automation without human interaction.
 
-Agents are model-agnostic, and can handle and share RAG tools, knowledge, memory, and callbacks among other agents. (self-learn)
+Agents are model-agnostic, and will improve task output, while oprimizing token cost and job latency, by sharing their memory, knowledge base, and RAG tools with other agents in the network.
 
 
 ###  Agent formation
@@ -152,15 +154,28 @@ You can specify a desired formation or allow the agents to determine it autonomo
 
 (Python 3.11 or higher)
 
+### Generate agent networks and launch task execution:
 
-### Case 1. Solo Agent:
+   ```
+   from versionhq import form_agent_network
+
+   network = form_agent_network(
+      task="YOUR AMAZING TASK OVERVIEW",
+      expected_outcome="YOUR OUTCOME EXPECTATION",
+   )
+   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.
+
+
+### Solo Agent:
 
 #### Return a structured output with a summary in string.
 
    ```
    from pydantic import BaseModel
-   from versionhq.agent.model import Agent
-   from versionhq.task.model import Task
+   from versionhq import Agent, Task
 
    class CustomOutput(BaseModel):
       test1: str
@@ -197,12 +212,10 @@ This will return `TaskOutput` instance that stores a response in plane text, JSO
    )
    ```
 
-### Case 2. Supervising:
+### Supervising:
 
    ```
-   from versionhq.agent.model import Agent
-   from versionhq.task.model import Task, ResponseField
-   from versionhq.team.model import Team, TeamMember
+   from versionhq import Agent, Task, ResponseField, Team, TeamMember
 
    agent_a = Agent(role="agent a", goal="My amazing goals", llm="llm-of-your-choice")
    agent_b = Agent(role="agent b", goal="My amazing goals", llm="llm-of-your-choice")

{versionhq-1.1.12.1 → versionhq-1.1.12.3}/README.md

@@ -7,7 +7,7 @@
 ![pyenv ver](https://img.shields.io/badge/pyenv-2.5.0-orange)
 
 
-LLM orchestration frameworks to deploy multi-agent systems and automate complex tasks with network formations.
+Agentic orchestration framework to deploy agent network and handle complex task automation.
 
 **Visit:**
 
@@ -26,9 +26,10 @@ LLM orchestration frameworks to deploy multi-agent systems and automate complex
 - [Key Features](#key-features)
   - [Agent formation](#agent-formation)
 - [Quick Start](#quick-start)
-  - [Case 1. Solo Agent:](#case-1-solo-agent)
+  - [Generate agent networks and launch task execution:](#generate-agent-networks-and-launch-task-execution)
+  - [Solo Agent:](#solo-agent)
     - [Return a structured output with a summary in string.](#return-a-structured-output-with-a-summary-in-string)
-  - [Case 2. Supervising:](#case-2-supervising)
+  - [Supervising:](#supervising)
 - [Technologies Used](#technologies-used)
 - [Project Structure](#project-structure)
 - [Setup](#setup)
@@ -46,9 +47,9 @@ LLM orchestration frameworks to deploy multi-agent systems and automate complex
 
 ## Key Features
 
-Generate multi-agent systems based on the task complexity, execute tasks, and evaluate output based on the given criteria.
+`versionhq` is a Python framework that can generate agent networks for complex task automation without human interaction.
 
-Agents are model-agnostic, and can handle and share RAG tools, knowledge, memory, and callbacks among other agents. (self-learn)
+Agents are model-agnostic, and will improve task output, while oprimizing token cost and job latency, by sharing their memory, knowledge base, and RAG tools with other agents in the network.
 
 
 ###  Agent formation
@@ -76,15 +77,28 @@ You can specify a desired formation or allow the agents to determine it autonomo
 
 (Python 3.11 or higher)
 
+### Generate agent networks and launch task execution:
 
-### Case 1. Solo Agent:
+   ```
+   from versionhq import form_agent_network
+
+   network = form_agent_network(
+      task="YOUR AMAZING TASK OVERVIEW",
+      expected_outcome="YOUR OUTCOME EXPECTATION",
+   )
+   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.
+
+
+### Solo Agent:
 
 #### Return a structured output with a summary in string.
 
    ```
    from pydantic import BaseModel
-   from versionhq.agent.model import Agent
-   from versionhq.task.model import Task
+   from versionhq import Agent, Task
 
    class CustomOutput(BaseModel):
       test1: str
@@ -121,12 +135,10 @@ This will return `TaskOutput` instance that stores a response in plane text, JSO
    )
    ```
 
-### Case 2. Supervising:
+### Supervising:
 
    ```
-   from versionhq.agent.model import Agent
-   from versionhq.task.model import Task, ResponseField
-   from versionhq.team.model import Team, TeamMember
+   from versionhq import Agent, Task, ResponseField, Team, TeamMember
 
    agent_a = Agent(role="agent a", goal="My amazing goals", llm="llm-of-your-choice")
    agent_b = Agent(role="agent b", goal="My amazing goals", llm="llm-of-your-choice")

{versionhq-1.1.12.1 → versionhq-1.1.12.3}/pyproject.toml

@@ -15,13 +15,13 @@ exclude = ["test*", "__pycache__", "*.egg-info"]
 
 [project]
 name = "versionhq"
-version = "1.1.12.1"
+version = "1.1.12.3"
 authors = [{ name = "Kuriko Iwai", email = "kuriko@versi0n.io" }]
-description = "LLM orchestration frameworks for model-agnostic AI agents that handle complex outbound workflows"
+description = "An agentic orchestration framework for multi-agent system that shares memory, knowledge base, and RAG tools."
 readme = "README.md"
 requires-python = ">=3.11"
 license = { file = "LICENSE" }
-keywords = ["orchestration framework", "orchestration", "ai agent", "multi-agent system", "RAG", "agent"]
+keywords = ["orchestration framework", "orchestration", "ai agent", "multi-agent system", "RAG", "agent", "agentic orchestration", "llm"]
 dependencies = [
   "regex==2024.11.6",
   "requests>=2.32.3",
@@ -43,6 +43,7 @@ dependencies = [
   "chromadb>=0.6.3",
   "wheel>=0.45.1",
   "envoy>=0.0.3",
+  "composio-core==0.7.0",
 ]
 classifiers = [
   "Programming Language :: Python",
@@ -90,6 +91,7 @@ dev-dependencies = [
   "bandit",
   "twine",
   "pytest>=8.3.4",
+  "mkdocs>=1.6.1",
 ]
 
 [tool.uv.workspace]

{versionhq-1.1.12.1 → versionhq-1.1.12.3}/runtime.txt

@@ -1,2 +1 @@
-python-3.13.1
 python-3.12.8

versionhq-1.1.12.3/src/versionhq/__init__.py

@@ -0,0 +1,74 @@
+# silence some warnings
+import warnings
+warnings.filterwarnings(action="ignore", message="Pydantic serializer warnings:", category=UserWarning, module="pydantic.main")
+warnings.filterwarnings(action="ignore", category=UserWarning, module="pydantic._internal")
+warnings.filterwarnings(action="ignore", module="LiteLLM:utils")
+
+from versionhq.agent.model import Agent
+from versionhq.clients.customer.model import Customer
+from versionhq.clients.product.model import Product, ProductProvider
+from versionhq.clients.workflow.model import MessagingWorkflow, MessagingComponent
+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.task.model import Task, TaskOutput, ConditionalTask, ResponseField
+from versionhq.task.evaluate import Evaluation, EvaluationItem
+from versionhq.team.model import Team, TeamOutput, Formation, TeamMember, TaskHandlingProcess
+from versionhq.tool.model import Tool, ToolSet
+from versionhq.tool.cache_handler import CacheHandler
+from versionhq.tool.tool_handler import ToolHandler
+from versionhq.tool.composio_tool import ComposioHandler
+from versionhq.memory.contextual_memory import ContextualMemory
+from versionhq.memory.model import ShortTermMemory,LongTermMemory, UserMemory, MemoryItem
+
+from versionhq.task.formation import form_agent_network
+
+
+__version__ = "1.1.12.3"
+__all__ = [
+    "Agent",
+
+    "Customer",
+    "Product",
+    "ProductProvider",
+    "MessagingWorkflow",
+    "MessagingComponent",
+
+    "Knowledge",
+    "KnowledgeStorage",
+    "PDFKnowledgeSource",
+    "CSVKnowledgeSource",
+    "JSONKnowledgeSource",
+    "TextFileKnowledgeSource",
+    "ExcelKnowledgeSource",
+    "StringKnowledgeSource",
+    "DoclingSource",
+
+    "Task",
+    "TaskOutput",
+    "ConditionalTask",
+    "ResponseField",
+
+    "Evaluation",
+    "EvaluationItem",
+
+    "Team",
+    "TeamOutput",
+    "Formation",
+    "TeamMember",
+    "TaskHandlingProcess",
+
+    "Tool",
+    "ToolSet",
+    "CacheHandler",
+    "ToolHandler",
+    "ComposioHandler",
+
+    "ContextualMemory",
+    "ShortTermMemory",
+    "LongTermMemory",
+    "UserMemory",
+    "MemoryItem",
+
+    "form_agent_network"
+]

versionhq-1.1.12.3/src/versionhq/_utils/__init__.py

@@ -0,0 +1,3 @@
+from versionhq._utils.logger import Logger
+from versionhq._utils.process_config import process_config
+from versionhq._utils.usage_metrics import UsageMetrics

versionhq-1.1.12.3/src/versionhq/agent/inhouse_agents.py

@@ -0,0 +1,41 @@
+from versionhq.agent.model import Agent
+from versionhq.llm.model import DEFAULT_MODEL_NAME
+
+"""
+In-house agents to be called across the project.
+[Rules] In house agents have names and roles that start with `vhq_`. No customization allowed by client.
+"""
+
+vhq_client_manager = Agent(
+    role="vhq-Client Manager",
+    goal="Efficiently communicate with the client on the task progress",
+    llm=DEFAULT_MODEL_NAME,
+    use_memory=True,
+)
+
+
+vhq_task_evaluator = Agent(
+    role="vhq-Task Evaluator",
+    goal="score the output according to the given evaluation criteria.",
+    llm=DEFAULT_MODEL_NAME,
+    llm_config=dict(top_p=0.8, top_k=30, max_tokens=5000, temperature=0.9),
+    maxit=1,
+    max_retry_limit=1,
+    use_memory=True # refer past eval records of similar tasks
+)
+
+
+vhq_formation_planner = Agent(
+    role="vhq-Formation Planner",
+    goal="Plan a formation of agents based on the given task descirption.",
+    llm="gemini/gemini-2.0-flash-exp",
+    llm_config=dict(top_p=0.8, top_k=30, temperature=0.9),
+    maxit=1,
+    max_retry_limit=1,
+    knowledge_sources=[
+        "Solo is a formation where a single agent with tools, knowledge, and memory handles tasks indivudually. When self-learning mode is on - it will turn into Random formation. Typical usecase is an email agent drafts promo message for the given audience using their own knowledge.",
+        "Supervising is a formation where the leader agent gives directions, while sharing its knowledge and memory with subbordinates.Subordinates can be solo agents or networks. Typical usecase is that the leader agent strategizes an outbound campaign plan and assigns components such as media mix or message creation to subordinate agents.",
+        "Network is a formation where multple agents can share tasks, knowledge, and memory among network members without hierarchy. Typical usecase is that an email agent and social media agent share the product knowledge and deploy multi-channel outbound campaign. ",
+        "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. ",
+    ]
+)

{versionhq-1.1.12.1 → versionhq-1.1.12.3}/src/versionhq/agent/model.py

@@ -1,6 +1,5 @@
 import os
 import uuid
-import datetime
 from typing import Any, Dict, List, Optional, TypeVar, Callable, Type
 from typing_extensions import Self
 from dotenv import load_dotenv
@@ -353,7 +352,7 @@ class Agent(BaseModel):
     @model_validator(mode="after")
     def set_up_memory(self) -> Self:
         """
-        Set up memories: stm, um
+        Set up memories: stm, ltm, and um
         """
 
         if self.use_memory == True:

{versionhq-1.1.12.1 → versionhq-1.1.12.3}/src/versionhq/agent/rpm_controller.py

@@ -5,7 +5,7 @@ from typing_extensions import Self
 
 from pydantic import BaseModel, Field, PrivateAttr, model_validator
 
-from versionhq._utils.logger import Logger
+from versionhq._utils import Logger
 
 
 class RPMController(BaseModel):

{versionhq-1.1.12.1 → versionhq-1.1.12.3}/src/versionhq/clients/product/model.py

@@ -56,7 +56,6 @@ class ProductProvider(ABC, BaseModel):
         return self
 
 
-
 class Product(BaseModel):
     """
     A class to store product information used to create outbound

versionhq-1.1.12.3/src/versionhq/knowledge/__init__.py

@@ -0,0 +1,22 @@
+from versionhq.knowledge.model import Knowledge, KnowledgeStorage
+from versionhq.knowledge.source import (
+    CSVKnowledgeSource,
+    ExcelKnowledgeSource,
+    PDFKnowledgeSource,
+    TextFileKnowledgeSource,
+    JSONKnowledgeSource,
+    StringKnowledgeSource
+)
+from versionhq.knowledge.source_docling import DoclingSource
+
+__all__ = [
+    "Knowledge",
+    "KnowledgeStorage",
+    "DoclingSource",
+    "CSVKnowledgeSource",
+    "ExcelKnowledgeSource",
+    "PDFKnowledgeSource",
+    "TextFileKnowledgeSource",
+    "JSONKnowledgeSource",
+    "StringKnowledgeSource"
+]

{versionhq-1.1.12.1 → versionhq-1.1.12.3}/src/versionhq/knowledge/model.py

@@ -1,5 +1,3 @@
-import os
-from abc import ABC, abstractmethod
 from typing import Any, Dict, List, Optional
 from pydantic import BaseModel, ConfigDict, Field
 

{versionhq-1.1.12.1 → versionhq-1.1.12.3}/src/versionhq/knowledge/source.py

@@ -280,7 +280,6 @@ class PDFKnowledgeSource(BaseFileKnowledgeSource):
 
 
 
-
 class CSVKnowledgeSource(BaseFileKnowledgeSource):
     """
     A knowledge source class that stores and queries CSV file content using embeddings.

{versionhq-1.1.12.1 → versionhq-1.1.12.3}/src/versionhq/memory/model.py

@@ -1,7 +1,53 @@
+import datetime
 from typing import Any, Dict, List, Optional
 
 from versionhq.storage.rag_storage import RAGStorage
 from versionhq.storage.ltm_sqlite_storage import LTMSQLiteStorage
+from versionhq._utils.logger import Logger
+
+
+class MemoryData:
+    """
+    A class to store structured data to store in the memory.
+    """
+    def __init__(
+        self,
+        agent: Optional[str] = None, # task execution agent (core)
+        task_description: Optional[str] = None,
+        task_output: Optional[str] = None,
+        config: Optional[Dict[str, Any]] = None
+    ):
+        self.agent = agent
+        self.task_description = task_description
+        self.task_output = task_output
+
+        if config:
+            for k, v in config.items():
+                setattr(self, k, str(v))
+
+
+
+class MemoryMetadata:
+    """
+     A class to store structured metadata to store in the memory.
+    """
+
+    def __init__(
+        self,
+        eval_criteria: Optional[str] = None,
+        score: Optional[int | float] = None,
+        suggestion: Optional[str] = None,
+        eval_by: Optional[str] = None, # task evaluator agent
+        config: Optional[Dict[str, Any]] = None
+    ):
+        self.eval_criteria = eval_criteria
+        self.score = score
+        self.suggestion = suggestion
+        self.eval_by = eval_by
+
+        if config:
+            for k, v in config.items():
+                setattr(self, k, str(v))
 
 
 class Memory:
@@ -13,28 +59,46 @@ class Memory:
         self.storage = storage
 
 
-    def save(self, value: Any, metadata: Optional[Dict[str, Any]] = None, agent: Optional[str] = None) -> None:
-        metadata = metadata or {}
+    def save(
+            self,
+            data: MemoryData | Dict[str, Any],
+            metadata: Optional[MemoryMetadata | Dict[str, Any]] = None,
+            agent: Optional[str] = None
+        ) -> None:
+
+        """
+        Create a dict for data and metadata without empty values before storing them in the given storage.
+        """
+
+        if not data:
+            Logger(verbose=True).log(level="error", message="Missing data to store. Add either dict or MemoryData object", color="red")
+            return None
 
-        if agent:
-            metadata["agent"] = agent
-        self.storage.save(value, metadata)
+        metadata_dict = metadata if isinstance(metadata, dict) else metadata.__dict__ if isinstance(metadata, MemoryMetadata) else dict()
+        metadata_dict = {k: v for k, v in metadata_dict.items() if v} # remove empty values
+        data_dict = data if isinstance(data, dict) else data.__dict__ if isinstance(data, MemoryData) else dict()
+        data_dict = {k: v for k, v in data_dict.items() if v}
+
+        if agent and data_dict["agent"] is None:
+            data_dict["agent"] = agent
+
+        if metadata_dict:
+            self.storage.save(data=data_dict, metadata=metadata_dict)
+        else:
+            self.storage.save(data=data_dict)
 
 
     def search(self, query: str, limit: int = 3, score_threshold: float = 0.35) -> List[Any]:
         return self.storage.search(query=query, limit=limit, score_threshold=score_threshold)
 
 
+class MemoryItem:
+    """
+    A class to store item to be saved in either long term memory or short term memory.
+    """
 
-class ShortTermMemoryItem:
-    def __init__(
-        self,
-        data: Any,
-        agent: Optional[str] = None,
-        metadata: Optional[Dict[str, Any]] = None,
-    ):
+    def __init__(self, data: MemoryData = None, metadata: Optional[MemoryMetadata] = None):
         self.data = data
-        self.agent = agent
         self.metadata = metadata if metadata is not None else {}
 
 
@@ -70,12 +134,22 @@ class ShortTermMemory(Memory):
         super().__init__(storage)
 
 
-    def save(self, value: Any, metadata: Optional[Dict[str, Any]] = None, agent: Optional[str] = None) -> None:
-        item = ShortTermMemoryItem(data=value, metadata=metadata, agent=agent)
+    def save(
+            self,
+            task_description: str = None,
+            task_output: str = None,
+            agent: Optional[str] = None,
+            data: Optional[MemoryData] = None,
+            metadata: Optional[MemoryMetadata] = None
+        ) -> None:
+
+        data = data if data else MemoryData(task_description=task_description, task_output=task_output, agent=agent)
+        item = MemoryItem(data=data, metadata=metadata)
+
         if self.memory_provider == "mem0":
-            item.data = f"Remember the following insights from Agent run: {item.data}"
+            item.data.task_output = f"Remember the following insights from Agent run: {item.data.task_output}"
 
-        super().save(value=item.data, metadata=item.metadata, agent=item.agent)
+        super().save(data=item.data.__dict__, metadata=item.metadata.__dict__ if item.metadata else {})
 
 
     def search(self, query: str, limit: int = 3, score_threshold: float = 0.35,):
@@ -89,29 +163,11 @@ class ShortTermMemory(Memory):
             raise Exception(f"An error occurred while resetting the short-term memory: {str(e)}")
 
 
-
-class LongTermMemoryItem:
-    def __init__(
-        self,
-        agent: str,
-        task: str,
-        datetime: str,
-        quality: Optional[int | float] = None,
-        metadata: Optional[Dict[str, Any]] = None,
-    ):
-        self.task = task
-        self.agent = agent
-        self.quality = quality
-        self.datetime = datetime
-        self.metadata = metadata if metadata is not None else {}
-
-
-
 class LongTermMemory(Memory):
     """
     A class for managing cross runs data related to overall task executions.
     - Type: ltm
-    - Storage: LTMSQLiteStorage
+    - Storage: LTMSQLiteStorage | RAGStorage
     """
 
     def __init__(self, storage=None, path=None):
@@ -121,19 +177,25 @@ class LongTermMemory(Memory):
         super().__init__(storage)
 
 
-    def save(self, item: LongTermMemoryItem) -> None:
-        metadata = item.metadata
-        metadata.update({ "agent": item.agent })
-        self.storage.save(
-            task_description=item.task,
-            score=metadata["quality"],
-            metadata=metadata,
-            datetime=item.datetime,
-        )
+    def save(
+            self,
+            task_description: str = None,
+            task_output: str = None,
+            agent: Optional[str] = None,
+            data: Optional[MemoryData] = None,
+            metadata: Optional[MemoryMetadata] = None
+        ) -> None:
+
+        data = data if data else MemoryData(task_description=task_description, task_output=task_output, agent=agent)
+        item = MemoryItem(data=data, metadata=metadata)
+        super().save(data=item.data, metadata=item.metadata)
 
 
-    def search(self, task: str, latest_n: int = 3) -> List[Dict[str, Any]]:
-        return self.storage.load(task, latest_n)
+    def search(self, query: str, latest_n: int = 3) -> List[Dict[str, Any]]:
+        """
+        Query the storage and return the results up to latest_n.
+        """
+        return self.storage.load(query=query, latest_n=latest_n)
 
 
     def reset(self) -> None: