feat(py): add support for ai.define_resource

Author: yesudeep

py/packages/genkit/pyproject.toml

@@ -50,6 +50,7 @@ dependencies = [
   "dotpromptz>=0.1.4",
   "uvicorn>=0.34.0",
   "anyio>=4.9.0",
+  "uritemplate>=4.1.1",
 ]
 description = "Genkit AI Framework"
 license = { text = "Apache-2.0" }

py/packages/genkit/src/genkit/ai/_registry.py

@@ -47,6 +47,7 @@
 import structlog
 from pydantic import BaseModel
 
+from genkit.ai.resource import ResourceFn, ResourceOptions, define_resource
 from genkit.blocks.embedding import EmbedderFn, EmbedderOptions
 from genkit.blocks.evaluator import BatchEvaluatorFn, EvaluatorFn
 from genkit.blocks.formats.types import FormatDef
@@ -675,6 +676,21 @@ async def prompt(
             variant=variant,
         )
 
+    def define_resource(self, opts: ResourceOptions, fn: ResourceFn) -> Action:
+        """Defines a resource and registers it with the registry.
+
+        This creates a resource action that can handle requests for a specific URI
+        or URI template.
+
+        Args:
+            opts: Options defining the resource (name, uri, template, etc.).
+            fn: The function that implements resource content retrieval.
+
+        Returns:
+            The registered `Action` for the resource.
+        """
+        return define_resource(self.registry, opts, fn)
+
 
 class FlowWrapper:
     """A wapper for flow functions to add `stream` method."""