hyperforge-static 1.0.0.post20__py3-none-any.whl

hyperforge_static/__init__.py

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

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/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.dist-info/METADATA

@@ -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.dist-info/RECORD

@@ -0,0 +1,7 @@
+hyperforge_static/__init__.py,sha256=2-rDONgRc7X_lQHVpnwTFZ-rzeLwMdstR-0EEBlUhiM,58
+hyperforge_static/agent.py,sha256=TikdRAovoIeRu-UYNLViKYXGNuFCQefuF7cS4ZJ6QSc,3140
+hyperforge_static/config.py,sha256=9T6f2LXQK6qJ_wAUYmyV9GP3eHV7ze-IiS4rOdL8pNE,1116
+hyperforge_static-1.0.0.post20.dist-info/METADATA,sha256=1JzZlw9LoUtb027U_1xU7nzGoVML8aWgDKiL5DzPXy4,743
+hyperforge_static-1.0.0.post20.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+hyperforge_static-1.0.0.post20.dist-info/top_level.txt,sha256=hYurTUYO8QayDbG8LU9ymAgLSYz8dZMsuyhrr31DI8c,18
+hyperforge_static-1.0.0.post20.dist-info/RECORD,,

hyperforge_static-1.0.0.post20.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

hyperforge_static-1.0.0.post20.dist-info/top_level.txt

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