hyperforge-http 1.0.0.post19__py3-none-any.whl

hyperforge_http/__init__.py

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

hyperforge_http/agent.py

@@ -0,0 +1,104 @@
+from time import time
+from typing import Any, 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.manager import Manager
+from hyperforge.memory import Chunk, Context, QuestionMemory
+from hyperforge.utils import check_dns
+from hyperforge.utils.http import safe_http_client
+
+from hyperforge_http.config import HTTPStaticAgentConfig
+
+
+@agent(
+    id="http",
+    agent_type="context",
+    title="HTTP Static",
+    description="Use HTTP Static to get information from the internet to answer questions.",
+    config_schema=HTTPStaticAgentConfig,
+)
+class HTTPStaticAgent(ContextAgent, Agent[HTTPStaticAgentConfig]):
+    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
+
+        url = await check_dns(self.config.url)
+
+        async with safe_http_client() as client:
+            if self.config.method == "GET":
+                response = await client.get(
+                    url,
+                    headers=self.config.headers,
+                    timeout=self.config.timeout,
+                    params={self.config.question_query_param: question}
+                    if self.config.question_query_param
+                    else None,
+                )
+
+            elif self.config.method == "POST":
+                response = await client.post(
+                    url,
+                    headers=self.config.headers,
+                    timeout=self.config.timeout,
+                    params={self.config.question_query_param: question}
+                    if self.config.question_query_param
+                    else None,
+                    data={self.config.question_post_field: question}
+                    if self.config.question_post_field
+                    else None,
+                )
+            else:
+                raise ValueError(f"Unsupported HTTP method: {self.config.method}")
+
+        if response.status_code != 200:
+            error = f"HTTP request failed with status code {response.status_code}"
+            context_text = ""
+        else:
+            context_text = response.content.decode("utf-8")
+
+        context = Context(
+            agent_id=self.config.id if self.config.id else "http",
+            original_question_uuid=memory.original_question_uuid,
+            actual_question_uuid=question_uuid,
+            question=question,
+            source="http",
+            agent="http",
+            title=self.config.title if self.config.title else "HTTP Call",
+        )
+        context.chunks.append(
+            Chunk(
+                chunk_id=uuid4().hex,
+                text=context_text,
+                origin_agent=self.config.module,
+            )
+        )
+
+        await memory.add_step(
+            step_module=self.config.module,
+            step_title=self.step_title("HTTP request"),
+            step_agent_path=f"/context/{self.config.id if self.config.id else 'default'}",
+            step_value=f" HTTP {self.config.method} to {url}",
+            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_http/config.py

@@ -0,0 +1,36 @@
+from typing import Literal, Optional
+
+from hyperforge.context.agent import ContextAgentConfig
+from hyperforge.utils import WidgetType, sync_dns_validation
+from pydantic import Field, field_validator
+from pydantic.config import ConfigDict
+
+
+class HTTPStaticAgentConfig(ContextAgentConfig):
+    model_config = ConfigDict(title="HTTP call")
+    module: Literal["http"] = "http"
+    method: Literal["GET", "POST"] = Field(
+        default="GET", description="HTTP method to use for the request"
+    )
+    url: str = Field(..., description="The URL to fetch context from")
+
+    headers: dict[str, str] = Field(
+        default_factory=dict,
+        description="Optional headers to include in the HTTP request",
+        json_schema_extra={"widget": WidgetType.KEY_VALUE_FIELD},
+    )
+    question_query_param: Optional[str] = Field(
+        None,
+        description="Optional query parameter to include the question in the request",
+    )
+    question_post_field: Optional[str] = Field(
+        None,
+        description="Optional POST field id to include the question in the request body",
+    )
+    timeout: int = Field(
+        default=10, description="Timeout for the HTTP request in seconds"
+    )
+
+    @field_validator("url")
+    def validate_url(cls, v):
+        return sync_dns_validation(v)

hyperforge_http-1.0.0.post19.dist-info/METADATA

@@ -0,0 +1,19 @@
+Metadata-Version: 2.4
+Name: hyperforge_http
+Version: 1.0.0.post19
+Summary: HTTP 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
+
+# HTTP Static Hyperforge agents

hyperforge_http-1.0.0.post19.dist-info/RECORD

@@ -0,0 +1,7 @@
+hyperforge_http/__init__.py,sha256=vWIsEjujY8mKH3jhBJz6UKjHM3aHVyq1TBkn_iQxlCM,66
+hyperforge_http/agent.py,sha256=TiTCN6znoXlwycjc5VWyCzBQIChhzcSZ3w4wKQxk4q4,3692
+hyperforge_http/config.py,sha256=1vidNJAJGR_xVgk32Jv48jGX_zAvSlXNdhtnQXgfH0o,1296
+hyperforge_http-1.0.0.post19.dist-info/METADATA,sha256=TAabWU3-9hIgHfZk0StaYUNHf87nx2gmka_31-_MZCE,723
+hyperforge_http-1.0.0.post19.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+hyperforge_http-1.0.0.post19.dist-info/top_level.txt,sha256=L2wPsat-N3TFvF5N5ePV5_OlYXEs9yjT76DemW67wCE,16
+hyperforge_http-1.0.0.post19.dist-info/RECORD,,

hyperforge_http-1.0.0.post19.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_http-1.0.0.post19.dist-info/top_level.txt

@@ -0,0 +1 @@
+hyperforge_http