hyperforge-generate 1.0.0.post24__py3-none-any.whl

hyperforge_generate/agent.py

@@ -0,0 +1,101 @@
+from time import time
+from uuid import uuid4
+
+from hyperforge.agent import Agent
+from hyperforge.configure import agent
+from hyperforge.manager import Manager
+from hyperforge.memory import QuestionMemory
+from hyperforge.trace import trace_agent
+from nuclia.lib.nua_responses import ChatModel, UserPrompt
+
+from hyperforge import PROMPT_ENVIRONMENT
+from hyperforge_generate.config import GenerateAgentConfig
+
+GENERATE_PROMPT = """
+Generate the following request using **only** the provided context. Refrain from incorporating any outside knowledge. If the context is insufficient to answer the question comprehensively, respond with: "Not enough data to generate this."
+
+
+{% if rules -%}
+# Generation Rules
+{% for rule in rules -%}
+- {{rule}}
+{% endfor -%}
+{% endif -%}
+
+{{prompt}}
+
+{{context}}
+
+MAIN QUESTION: {{question}}
+
+# Notes
+- Use the context provided without being overly selective.
+- Please try to generate if possible, even if it requires to make a bit of a deduction.
+- If they are images attached, look through them carefully.
+- For rules related to charts or images, pay extra attention to extracting details and interpreting data presented visually. Think about it carefully before giving inaccurate interpretations
+
+"""
+
+GENERATE_PROMPT_TEMPLATE = PROMPT_ENVIRONMENT.from_string(GENERATE_PROMPT)
+
+
+@agent(
+    id="generate",
+    agent_type="generation",
+    title="Generate Answer",
+    description="Generate answers based on provided context.",
+    config_schema=GenerateAgentConfig,
+)
+class GenerateAgent(Agent[GenerateAgentConfig]):
+    __root_agent__ = True
+
+    @trace_agent
+    async def __call__(
+        self,
+        memory: QuestionMemory,
+        manager: Manager,
+    ):
+        # For each context in memory add context query, summary and answer onto a text and the initial question
+        prompt = GENERATE_PROMPT_TEMPLATE.render(
+            question=memory.original_question,
+            context=memory.contexts_minimal(),
+            prompt=self.config.prompt,
+            rules=memory.generation_rules,
+        )
+        t0 = time()
+
+        chat_model = ChatModel(
+            user_id="generate",
+            question="",
+            user_prompt=UserPrompt(prompt=prompt),
+            format_prompt=False,
+            generative_model=self.config.model,
+            max_tokens=2000,
+            tracking=memory.get_tracking_info(),
+        )
+
+        agent_path = f"/generation/{self.config.id if self.config.id else 'default'}"
+        resp, input_tokens, output_tokens = await manager.execute_raw(
+            chat_model, memory=memory, module="generate", agent_path=agent_path
+        )
+
+        generated_text = resp.answer or ""
+
+        memory.generated_texts[uuid4().hex] = generated_text
+
+        await memory.add_generated_text(
+            self.config.id if self.config.id else "default", generated_text
+        )
+        # We assume that JSON generation always produces an answer
+        memory.is_answered = True
+
+        await memory.add_step(
+            step_module=self.config.module,
+            step_title=self.step_title("Generate"),
+            step_value="",
+            step_reason="",
+            step_agent_path=agent_path,
+            timeit=time() - t0,
+            input_nuclia_tokens=input_tokens,
+            output_nuclia_tokens=output_tokens,
+        )

hyperforge_generate/config.py

@@ -0,0 +1,26 @@
+from typing import Literal, Optional
+
+from hyperforge.agent import AgentConfig
+from hyperforge.utils import WidgetType
+from pydantic import Field
+from pydantic.config import ConfigDict
+
+
+class GenerateAgentConfig(AgentConfig):
+    model_config = ConfigDict(title="Generate")
+    module: Literal["generate"] = "generate"
+    prompt: Optional[str] = Field(
+        None,
+        json_schema_extra={
+            "show_in_node": True,
+            "widget": WidgetType.EXPANDABLE_TEXTAREA,
+        },
+    )
+    model: str = Field(
+        default="chatgpt-azure-4o-mini",
+        title="Generative model",
+        description="Model used to generate the response",
+        json_schema_extra={"widget": WidgetType.MODEL_SELECT},
+    )
+    images: bool = False
+    generate_image: bool = False

hyperforge_generate-1.0.0.post24.dist-info/METADATA

@@ -0,0 +1,19 @@
+Metadata-Version: 2.4
+Name: hyperforge_generate
+Version: 1.0.0.post24
+Summary: External 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
+
+# Generate Hyperforge agents

hyperforge_generate-1.0.0.post24.dist-info/RECORD

@@ -0,0 +1,7 @@
+hyperforge_generate/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+hyperforge_generate/agent.py,sha256=OuMnJ1BpB5JO-KsP0dLgxuZvFe2wIMn8bJow7QW6sic,3317
+hyperforge_generate/config.py,sha256=rREyYti3JU4wqur8fRz5vvJHgDIAYdRlkI1VVpyI9ng,787
+hyperforge_generate-1.0.0.post24.dist-info/METADATA,sha256=Z_XuDciZ7POQZC_UJa42jskRcnwfmUshxwxUTTp4jFI,728
+hyperforge_generate-1.0.0.post24.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+hyperforge_generate-1.0.0.post24.dist-info/top_level.txt,sha256=D83iCWoSgtfse_skPmV0EYOvF384aM1z_ti317LBiic,20
+hyperforge_generate-1.0.0.post24.dist-info/RECORD,,

hyperforge_generate-1.0.0.post24.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_generate-1.0.0.post24.dist-info/top_level.txt

@@ -0,0 +1 @@
+hyperforge_generate