Merge pull request #69 from nforro/tools

Add `set_zstream_release` tool

Author: nforro

beeai/agents/tests/unit/test_tools.py

@@ -13,6 +13,8 @@
     AddChangelogEntryToolInput,
     BumpReleaseTool,
     BumpReleaseToolInput,
+    SetZStreamReleaseTool,
+    SetZStreamReleaseToolInput,
 )
 
 
@@ -111,3 +113,39 @@ async def test_bump_release(minimal_spec):
     result = output.result
     assert not result
     assert minimal_spec.read_text().splitlines()[3] == "Release:        3%{?dist}"
+
+
+@pytest.fixture
+def autorelease_spec(tmp_path):
+    spec = tmp_path / "test.spec"
+    spec.write_text(
+        dedent(
+            """
+            Name:           test
+            Version:        0.1
+            Release:        %autorelease
+            Summary:        Test package
+
+            License:        MIT
+
+            %description
+            Test package
+
+            %changelog
+            %autochangelog
+            """
+        )
+    )
+    return spec
+
+
+@pytest.mark.asyncio
+async def test_set_zstream_release(autorelease_spec):
+    latest_ystream_evr = "0.1-4.el10"
+    tool = SetZStreamReleaseTool()
+    output = await tool.run(
+        input=SetZStreamReleaseToolInput(spec=autorelease_spec, latest_ystream_evr=latest_ystream_evr)
+    ).middleware(GlobalTrajectoryMiddleware(pretty=True))
+    result = output.result
+    assert not result
+    assert autorelease_spec.read_text().splitlines()[3] == "Release:        4%{?dist}.%{autorelease -n}"

beeai/agents/tools/specfile.py

@@ -4,6 +4,7 @@
 
 from pydantic import BaseModel, Field
 from specfile import Specfile
+from specfile.utils import EVR
 
 from beeai_framework.context import RunContext
 from beeai_framework.emitter import Emitter
@@ -77,3 +78,38 @@ async def _run(
         except Exception as e:
             return StringToolOutput(result=f"Failed to bump release: {e}")
         return StringToolOutput()
+
+
+class SetZStreamReleaseToolInput(BaseModel):
+    spec: Path = Field(description="Absolute path to a spec file")
+    latest_ystream_evr: str = Field(description="EVR (Epoch-Version-Release) of the latest Y-Stream build")
+
+
+class SetZStreamReleaseTool(Tool[SetZStreamReleaseToolInput, ToolRunOptions, StringToolOutput]):
+    name = "set_zstream_release"
+    description = """
+    Sets the value of the `Release` field in the specified spec file to a Z-Stream release
+    based on the release of the latest Y-Stream build.
+    Returns error message on failure.
+    """
+    input_schema = SetZStreamReleaseToolInput
+
+    def _create_emitter(self) -> Emitter:
+        return Emitter.root().child(
+            namespace=["tool", "specfile", self.name],
+            creator=self,
+        )
+
+    async def _run(
+        self, tool_input: SetZStreamReleaseToolInput, options: ToolRunOptions | None, context: RunContext
+    ) -> StringToolOutput:
+        try:
+            with Specfile(tool_input.spec) as spec:
+                if not spec.has_autorelease:
+                    return StringToolOutput(result="The specified spec file doesn't use %autorelease")
+                base_release = EVR.from_string(tool_input.latest_ystream_evr).release
+                base_raw_release = base_release.rsplit(".el", maxsplit=1)[0]
+                spec.raw_release = base_raw_release + "%{?dist}.%{autorelease -n}"
+        except Exception as e:
+            return StringToolOutput(result=f"Failed to set Z-Stream release: {e}")
+        return StringToolOutput()