genkit-plugin-flask 0.0.1.dev1__tar.gz → 0.3.0.dev1__tar.gz

{genkit_plugin_flask-0.0.1.dev1 → genkit_plugin_flask-0.3.0.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: genkit-plugin-flask
-Version: 0.0.1.dev1
+Version: 0.3.0.dev1
 Summary: Genkit Firebase Plugin
 Author: Google
 License: Apache-2.0
@@ -13,13 +13,15 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries
-Requires-Python: >=3.12
+Requires-Python: >=3.10
 Requires-Dist: flask
-Requires-Dist: genkit-ai
+Requires-Dist: genkit
 Requires-Dist: genkit-plugin-flask
 Requires-Dist: genkit-plugin-google-genai
 Requires-Dist: pydantic>=2.10.5

{genkit_plugin_flask-0.0.1.dev1 → genkit_plugin_flask-0.3.0.dev1}/pyproject.toml

@@ -8,14 +8,16 @@ classifiers = [
   "Operating System :: OS Independent",
   "License :: OSI Approved :: Apache Software License",
   "Programming Language :: Python",
+  "Programming Language :: Python :: 3 :: Only",
+  "Programming Language :: Python :: 3.10",
+  "Programming Language :: Python :: 3.11",
   "Programming Language :: Python :: 3.12",
   "Programming Language :: Python :: 3.13",
-  "Programming Language :: Python :: 3 :: Only",
   "Topic :: Scientific/Engineering :: Artificial Intelligence",
   "Topic :: Software Development :: Libraries",
 ]
 dependencies = [
-  "genkit-ai",
+  "genkit",
   "genkit-plugin-google-genai",
   "genkit-plugin-flask",
   "pydantic>=2.10.5",
@@ -25,8 +27,8 @@ description = "Genkit Firebase Plugin"
 license = { text = "Apache-2.0" }
 name = "genkit-plugin-flask"
 readme = "README.md"
-requires-python = ">=3.12"
-version = "0.0.1-dev.1"
+requires-python = ">=3.10"
+version = "0.3.0.dev1"
 
 [build-system]
 build-backend = "hatchling.build"

{genkit_plugin_flask-0.0.1.dev1 → genkit_plugin_flask-0.3.0.dev1}/src/genkit/plugins/flask/handler.py

@@ -16,6 +16,7 @@
 
 """Genkit Flask plugin."""
 
+import asyncio
 from collections.abc import Callable
 
 from flask import Response, request
@@ -23,10 +24,26 @@ from genkit.ai import Genkit
 from genkit.ai.registry import FlowWrapper
 from genkit.aio.loop import create_loop, iter_over_async
 from genkit.codec import dump_dict, dump_json
+from genkit.core.context import ContextProvider, RequestData
 from genkit.core.error import GenkitError, get_callable_json
 
 
-def genkit_flask_handler(ai: Genkit) -> Callable:
+class _FlaskRequestData(RequestData):
+    def __init__(self):
+        self.method = request.method
+
+        self.headers = {}
+        for key, value in request.headers:
+            self.headers[key.lower()] = value
+
+        input_data = request.get_json()
+        if 'data' not in input_data:
+            return Response(status=400, response='flow request must be wrapped in {"data": data} object')
+
+        self.input = input_data.get('data')
+
+
+def genkit_flask_handler(ai: Genkit, context_provider: ContextProvider | None = None) -> Callable:
     """A decorator for serving Genkit flows via a flask sever.
 
     ```python
@@ -57,12 +74,19 @@ def genkit_flask_handler(ai: Genkit) -> Callable:
             if 'data' not in input_data:
                 return Response(status=400, response='flow request must be wrapped in {"data": data} object')
 
-            stream = request.headers.get('Accept') == 'text/event-stream' or request.args.get('stream') == 'true'
+            request_data = _FlaskRequestData()
+            context = None
+            if context_provider:
+                context = context_provider(request_data)
+                if asyncio.iscoroutine(context):
+                    context = await context
+
+            stream = request_data.headers.get('accept') == 'text/event-stream' or request.args.get('stream') == 'true'
             if stream:
 
                 async def async_gen():
                     try:
-                        stream, response = flow._action.stream(input_data.get('data'))
+                        stream, response = flow._action.stream(input_data.get('data'), context=context)
                         async for chunk in stream:
                             yield f'data: {dump_json({"message": dump_dict(chunk)})}\n\n'
 
@@ -77,7 +101,7 @@ def genkit_flask_handler(ai: Genkit) -> Callable:
                 return iter
             else:
                 try:
-                    response = await flow._action.arun_raw(input_data.get('data'))
+                    response = await flow._action.arun_raw(input_data.get('data'), context=context)
                     return {'result': dump_dict(response.response)}
                 except Exception as e:
                     ex = e