hyperforge-static 1.0.0.post20__tar.gz

hyperforge_static-1.0.0.post20/PKG-INFO

@@ -0,0 +1,19 @@
+Metadata-Version: 2.4
+Name: hyperforge_static
+Version: 1.0.0.post20
+Summary: Static Context Hyperforge agent
+Author-email: Nuclia <nucliadb@nuclia.com>
+License-Expression: Apache-2.0
+Project-URL: Homepage, https://progress.com
+Project-URL: Repository, https://github.com/nuclia/forge
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: <4,>=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: hyperforge
+
+# Advanced Generation Hyperforge agents

hyperforge_static-1.0.0.post20/README.md

@@ -0,0 +1 @@
+# Advanced Generation Hyperforge agents

hyperforge_static-1.0.0.post20/pyproject.toml

@@ -0,0 +1,38 @@
+[build-system]
+requires = ["setuptools"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "hyperforge_static"
+version = "1.0.0.post20"
+license = "Apache-2.0"
+description = "Static Context Hyperforge agent"
+authors = [{ name = "Nuclia", email = "nucliadb@nuclia.com" }]
+readme = "README.md"
+classifiers = [
+    "Programming Language :: Python",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3 :: Only",
+    "Topic :: Software Development :: Libraries :: Python Modules",
+]
+requires-python = ">=3.10, <4"
+dependencies = ["hyperforge"]
+
+[dependency-groups]
+# these dependencies are actually in the src/ folder (under
+# src/nucliadb_utils/tests/), but only used when the module is imported
+dev = [
+    "pytest",
+    "pytest-benchmark",
+    "pytest-docker-fixtures>=1.4.2",
+    "pytest-lazy-fixtures",
+]
+
+[project.urls]
+Homepage = "https://progress.com"
+Repository = "https://github.com/nuclia/forge"
+
+[tool.pytest.ini_options]
+asyncio_mode = "auto"

hyperforge_static-1.0.0.post20/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

hyperforge_static-1.0.0.post20/src/hyperforge_static/__init__.py

@@ -0,0 +1,3 @@
+from .agent import StaticAgent
+
+__all__ = ["StaticAgent"]

hyperforge_static-1.0.0.post20/src/hyperforge_static/agent.py

@@ -0,0 +1,96 @@
+from time import time
+from typing import Any, ClassVar, Dict, List, Optional, Tuple
+from uuid import uuid4
+
+from hyperforge.agent import Agent
+from hyperforge.configure import agent
+from hyperforge.context.agent import ContextAgent
+from hyperforge.definition import FunctionDefinition
+from hyperforge.manager import Manager
+from hyperforge.memory.memory import Chunk, Context, QuestionMemory
+
+from hyperforge_static.config import StaticAgentConfig
+
+
+@agent(
+    id="static",
+    agent_type="context",
+    title="Static Context",
+    description="Provide static context to answer questions.",
+    config_schema=StaticAgentConfig,
+)
+class StaticAgent(ContextAgent, Agent[StaticAgentConfig]):
+    __published_functions__: ClassVar[Dict[str, FunctionDefinition]] = {
+        "static_context": FunctionDefinition(
+            name="static_context",
+            description="Provide static context to answer questions.",
+            parameters={},
+        )
+    }
+
+    async def static_context(
+        self,
+        memory: QuestionMemory,
+        manager: Manager,
+        question: Optional[str] = "",
+        question_uuid: Optional[str] = None,
+    ) -> Context:
+        if question_uuid is None:
+            question_uuid = uuid4().hex
+
+        context = Context(
+            agent_id=self.config.id if self.config.id else "static_context",
+            original_question_uuid=memory.original_question_uuid,
+            actual_question_uuid=question_uuid,
+            question=question,
+            source="static_context",
+            agent="static_context",
+            title=self.config.title if self.config.title else "Static Context",
+        )
+        if self.config.context:
+            context.chunks.append(
+                Chunk(
+                    chunk_id=uuid4().hex,
+                    text=self.config.context,
+                    origin_agent=self.config.module,
+                )
+            )
+        if self.config.structured:
+            context.structured.append(self.config.structured)
+        return context
+
+    async def _get_question_context(
+        self,
+        memory: QuestionMemory,
+        manager: Manager,
+        question_uuid: str,
+        question: str,
+        flow_id: str,
+        extra_context: Optional[Dict[str, Any]] = None,
+    ) -> List[Tuple[str, str]]:
+        t0 = time()
+        error = None
+
+        context = await self.static_context(
+            memory=memory,
+            manager=manager,
+        )
+
+        await memory.add_step(
+            step_module=self.config.module,
+            step_title=self.step_title("Static context"),
+            step_agent_path=f"/context/{self.config.id if self.config.id else 'default'}",
+            step_value=" Static context retrieval",
+            timeit=time() - t0,
+            input_nuclia_tokens=0,
+            output_nuclia_tokens=0,
+            error=error,
+        )
+        missing = await self.save_ctx_and_return_missing(
+            context=context,
+            question=question,
+            memory=memory,
+            manager=manager,
+            flow_id=flow_id,
+        )
+        return [missing] if missing is not None else []

hyperforge_static-1.0.0.post20/src/hyperforge_static/config.py

@@ -0,0 +1,33 @@
+from typing import Literal, Optional, Tuple
+
+from hyperforge.context.config import ContextAgentConfig
+from hyperforge.utils import WidgetType
+from pydantic import Field
+from pydantic.config import ConfigDict
+
+
+class StaticAgentConfig(ContextAgentConfig):
+    model_config = ConfigDict(title="Static data")
+    module: Literal["static"] = "static"
+    published_functions: Optional[Tuple[str, ...]] = Field(
+        default=("static_context",),
+        title="Published functions",
+        description="List of functions published by this agent to be used by other agents in the chain",
+        json_schema_extra={
+            "widget": WidgetType.NOT_SHOWN,
+        },
+    )
+    context: Optional[str] = Field(
+        None,
+        description="Static context to be used by the agent",
+        json_schema_extra={
+            "widget": WidgetType.EXPANDABLE_TEXTAREA,
+        },
+    )
+    structured: Optional[str] = Field(
+        None,
+        description="Structured data in JSON format to be used by the agent",
+        json_schema_extra={
+            "widget": WidgetType.EXPANDABLE_TEXTAREA,
+        },
+    )

hyperforge_static-1.0.0.post20/src/hyperforge_static.egg-info/PKG-INFO

@@ -0,0 +1,19 @@
+Metadata-Version: 2.4
+Name: hyperforge_static
+Version: 1.0.0.post20
+Summary: Static Context Hyperforge agent
+Author-email: Nuclia <nucliadb@nuclia.com>
+License-Expression: Apache-2.0
+Project-URL: Homepage, https://progress.com
+Project-URL: Repository, https://github.com/nuclia/forge
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: <4,>=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: hyperforge
+
+# Advanced Generation Hyperforge agents

hyperforge_static-1.0.0.post20/src/hyperforge_static.egg-info/SOURCES.txt

@@ -0,0 +1,11 @@
+README.md
+pyproject.toml
+src/hyperforge_static/__init__.py
+src/hyperforge_static/agent.py
+src/hyperforge_static/config.py
+src/hyperforge_static.egg-info/PKG-INFO
+src/hyperforge_static.egg-info/SOURCES.txt
+src/hyperforge_static.egg-info/dependency_links.txt
+src/hyperforge_static.egg-info/requires.txt
+src/hyperforge_static.egg-info/top_level.txt
+tests/test_static.py

hyperforge_static-1.0.0.post20/src/hyperforge_static.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

hyperforge_static-1.0.0.post20/src/hyperforge_static.egg-info/requires.txt

@@ -0,0 +1 @@
+hyperforge

hyperforge_static-1.0.0.post20/src/hyperforge_static.egg-info/top_level.txt

@@ -0,0 +1 @@
+hyperforge_static

hyperforge_static-1.0.0.post20/tests/test_static.py

@@ -0,0 +1,79 @@
+from unittest.mock import MagicMock
+from uuid import uuid4
+
+import pytest
+from hyperforge.manager import Manager
+from hyperforge.memory.memory import EphemeralSessionMemory
+from hyperforge.models import MemoryConfig, Rules
+from hyperforge_static.agent import StaticAgent
+from hyperforge_static.config import StaticAgentConfig
+
+pytestmark = pytest.mark.asyncio
+
+
+async def test_static_agent():
+    # 1. Create a mocked manager
+    manager = MagicMock(spec=Manager)
+
+    # 2. Create the configuration for StaticAgent
+    config = StaticAgentConfig(
+        id="test-static",
+        title="Static Context",
+        context="This is a test static context text.",
+        structured="{'key': 'value'}",
+        prune_context=False,
+    )
+
+    # 3. Create the StaticAgent instance
+    agent = StaticAgent(config=config)
+
+    # 4. Set up EphemeralSessionMemory and QuestionMemory
+    session = EphemeralSessionMemory.from_config(
+        config=MemoryConfig(), agent_id="test", workflow_id="test", rules=Rules()
+    )
+    session.init("test-session")
+    memory = session.start_question("What is the static context?")
+    flow_id = uuid4().hex
+
+    # 5. Call static_context directly
+    context = await agent.static_context(
+        memory=memory,
+        manager=manager,
+        question="What is the static context?",
+    )
+
+    # 6. Assert static_context returns Context with correct values
+    assert context is not None
+    assert context.agent_id == "test-static"
+    assert context.title == "Static Context"
+    assert len(context.chunks) == 1
+    assert context.chunks[0].text == "This is a test static context text."
+    assert len(context.structured) == 1
+    assert context.structured[0] == "{'key': 'value'}"
+
+    # 7. Call _get_question_context
+    missing = await agent._get_question_context(
+        memory=memory,
+        manager=manager,
+        question_uuid=memory.original_question_uuid,
+        question="What is the static context?",
+        flow_id=flow_id,
+    )
+
+    # 8. Assertions on the memory state and missing questions
+    assert (
+        missing == []
+    )  # Since no validation/fallback model is used, should return empty list
+
+    # Check that context is saved to memory under the given flow_id
+    saved_contexts = memory.get_agent_contexts(flow_id=flow_id, agent_id="test-static")
+    assert len(saved_contexts) == 1
+    assert saved_contexts[0].chunks[0].text == "This is a test static context text."
+    assert saved_contexts[0].structured[0] == "{'key': 'value'}"
+
+    # Verify step was added to memory
+    assert len(memory.steps) == 1
+    step = memory.steps[0]
+    assert step.module == "static"
+    assert "Static context" in step.title
+    assert step.value == " Static context retrieval"