From f6e0b703727fd36c2d70844aee69109ed37bb6ba Mon Sep 17 00:00:00 2001 From: Varad Meru Date: Thu, 20 Mar 2025 18:13:47 -0700 Subject: [PATCH 1/3] Adding Starlette MCP PyWorker Extension --- .../CHANGELOG.md | 0 azurefunctions-extensions-mcp-fastapi/LICENSE | 21 ++ .../MANIFEST.in | 3 + .../README.md | 74 ++++++ .../azurefunctions/__init__.py | 1 + .../azurefunctions/extensions/__init__.py | 1 + .../extensions/mcp_server/__init__.py | 29 +++ .../mcp_server/decorators/__init__.py | 6 + .../mcp_server/decorators/mcp_app.py | 78 ++++++ .../mcp_server/starlette/__init__.py | 29 +++ .../extensions/mcp_server/starlette/web.py | 149 +++++++++++ .../pyproject.toml | 53 ++++ .../function_app.py | 26 ++ .../host.json | 15 ++ .../local.settings.json | 8 + .../requirements.txt | 6 + .../function_app.py | 25 ++ .../host.json | 15 ++ .../local.settings.json | 8 + .../requirements.txt | 6 + .../tests/__init__.py | 20 ++ .../tests/test_web.py | 232 ++++++++++++++++++ 22 files changed, 805 insertions(+) create mode 100644 azurefunctions-extensions-mcp-fastapi/CHANGELOG.md create mode 100644 azurefunctions-extensions-mcp-fastapi/LICENSE create mode 100644 azurefunctions-extensions-mcp-fastapi/MANIFEST.in create mode 100644 azurefunctions-extensions-mcp-fastapi/README.md create mode 100644 azurefunctions-extensions-mcp-fastapi/azurefunctions/__init__.py create mode 100644 azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/__init__.py create mode 100644 azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/__init__.py create mode 100644 azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/decorators/__init__.py create mode 100644 azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/decorators/mcp_app.py create mode 100644 azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/starlette/__init__.py create mode 100644 azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/starlette/web.py create mode 100644 azurefunctions-extensions-mcp-fastapi/pyproject.toml create mode 100644 azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/function_app.py create mode 100644 azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/host.json create mode 100644 azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/local.settings.json create mode 100644 azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/requirements.txt create mode 100644 azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/function_app.py create mode 100644 azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/host.json create mode 100644 azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/local.settings.json create mode 100644 azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/requirements.txt create mode 100644 azurefunctions-extensions-mcp-fastapi/tests/__init__.py create mode 100644 azurefunctions-extensions-mcp-fastapi/tests/test_web.py diff --git a/azurefunctions-extensions-mcp-fastapi/CHANGELOG.md b/azurefunctions-extensions-mcp-fastapi/CHANGELOG.md new file mode 100644 index 0000000..e69de29 diff --git a/azurefunctions-extensions-mcp-fastapi/LICENSE b/azurefunctions-extensions-mcp-fastapi/LICENSE new file mode 100644 index 0000000..63447fd --- /dev/null +++ b/azurefunctions-extensions-mcp-fastapi/LICENSE @@ -0,0 +1,21 @@ +Copyright (c) Microsoft Corporation. + +MIT License + +Permission is hereby granted, free of charge, to any person obtaining a copy +of this software and associated documentation files (the "Software"), to deal +in the Software without restriction, including without limitation the rights +to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +copies of the Software, and to permit persons to whom the Software is +furnished to do so, subject to the following conditions: + +The above copyright notice and this permission notice shall be included in all +copies or substantial portions of the Software. + +THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE +SOFTWARE. \ No newline at end of file diff --git a/azurefunctions-extensions-mcp-fastapi/MANIFEST.in b/azurefunctions-extensions-mcp-fastapi/MANIFEST.in new file mode 100644 index 0000000..e1ae5ad --- /dev/null +++ b/azurefunctions-extensions-mcp-fastapi/MANIFEST.in @@ -0,0 +1,3 @@ +recursive-include azure *.py *.pyi +recursive-include tests *.py +include LICENSE README.md \ No newline at end of file diff --git a/azurefunctions-extensions-mcp-fastapi/README.md b/azurefunctions-extensions-mcp-fastapi/README.md new file mode 100644 index 0000000..4c07dd3 --- /dev/null +++ b/azurefunctions-extensions-mcp-fastapi/README.md @@ -0,0 +1,74 @@ +# Azure Functions Extensions Http FastApi library for Python +This library contains HttpV2 extensions for FastApi Request/Response types to use in your function app code. + +[Source code](https://github.com/Azure/azure-functions-python-extensions/tree/main/azurefunctions-extensions-http-fastapi) +| [Package (PyPi)](https://pypi.org/project/azurefunctions-extensions-http-fastapi/) +| API reference documentation +| Product documentation +| [Samples](hhttps://github.com/Azure/azure-functions-python-extensions/tree/main/azurefunctions-extensions-http-fastapi/samples) + + +## Getting started + +### Prerequisites +* Python 3.8 or later is required to use this package. For more details, please read our page on [Python Functions version support policy](https://learn.microsoft.com/en-us/azure/azure-functions/functions-versions?tabs=isolated-process%2Cv4&pivots=programming-language-python#languages). + + +### Instructions +1. Follow the guide to [create an app](https://learn.microsoft.com/en-us/azure/azure-functions/create-first-function-cli-python?tabs=windows%2Cbash%2Cazure-cli%2Cbrowser). If you're testing the function locally, make sure to update to the latest version of Core Tools. +2. Ensure your app is using programming model v2 and contains a http trigger function. +3. Add ```azurefunctions-extensions-http-fastapi``` to your requirements.txt +4. Import Request and different types of responses from ```azurefunctions.extensions.http.fastapi``` in your HttpTrigger functions. +5. Change the request and response types to ones imported from ```azurefunctions.extensions.http.fastapi```. +6. Add ```"PYTHON_ENABLE_INIT_INDEXING": "1"``` to your local.settings.json file or App Settings. +6. Run your function app and try it out! + +### Bind to the FastApi-type +The Azure Functions Extensions Http FastApi library for Python allows you to create a function app with FastApi Request or Response types. When your function runs, you will receive the request of FastApi Request type and you can return a FastApi response type instance. FastApi is one of top popular python web framework which provides elegant and powerful request/response types and functionalities to users. With this integration, you are empowered to use request/response the same way as using them in native FastApi. A good example is you can do http streaming upload and streaming download now! Feel free to check out [Fastapi doc](https://fastapi.tiangolo.com/reference/responses/?h=custom) for further reference. + +```python +# This Azure Function receives streaming data from a client and processes it in real-time. +# It demonstrates streaming upload capabilities for scenarios such as uploading large files, +# processing continuous data streams, or handling IoT device data. + +import azure.functions as func +from azurefunctions.extensions.mcp_server.starlette import Request, JSONResponse + +app = func.FunctionApp(http_auth_level=func.AuthLevel.ANONYMOUS) + + +@app.route(route="streaming_upload", methods=[func.HttpMethod.POST]) +async def streaming_upload(req: Request) -> JSONResponse: + """Handle streaming upload requests.""" + # Process each chunk of data as it arrives + async for chunk in req.stream(): + process_data_chunk(chunk) + + # Once all data is received, return a JSON response indicating successful processing + return JSONResponse({"status": "Data uploaded and processed successfully"}) + + +def process_data_chunk(chunk: bytes): + """Process each data chunk.""" + # Add custom processing logic here + pass +``` + +## Next steps + +### More sample code + +Get started with our [FastApi samples](hhttps://github.com/Azure/azure-functions-python-extensions/tree/main/azurefunctions-extensions-http-fastapi/samples). + +Several samples are available in this GitHub repository. These samples provide example code for additional scenarios commonly encountered while working with FastApi: + +* [fastapi_samples_streaming_upload](https://github.com/Azure/azure-functions-python-extensions/tree/main/azurefunctions-extensions-http-fastapi/samples/fastapi_samples_streaming_upload) - An example on how to send and receiving a streaming request within your function. + +* [fastapi_samples_streaming_download](https://github.com/Azure/azure-functions-python-extensions/tree/main/azurefunctions-extensions-http-fastapi/samples/fastapi_samples_streaming_download) - An example on how to send your http response via streaming to the caller.t + +## Contributing +This project welcomes contributions and suggestions. Most contributions require you to agree to a Contributor License Agreement (CLA) declaring that you have the right to, and actually do, grant us the rights to use your contribution. For details, visit https://cla.microsoft.com. + +When you submit a pull request, a CLA-bot will automatically determine whether you need to provide a CLA and decorate the PR appropriately (e.g., label, comment). Simply follow the instructions provided by the bot. You will only need to do this once across all repos using our CLA. + +This project has adopted the [Microsoft Open Source Code of Conduct](https://opensource.microsoft.com/codeofconduct/). For more information see the [Code of Conduct FAQ](https://opensource.microsoft.com/codeofconduct/faq/) or contact [opencode@microsoft.com](mailto:opencode@microsoft.com) with any additional questions or comments. diff --git a/azurefunctions-extensions-mcp-fastapi/azurefunctions/__init__.py b/azurefunctions-extensions-mcp-fastapi/azurefunctions/__init__.py new file mode 100644 index 0000000..8db66d3 --- /dev/null +++ b/azurefunctions-extensions-mcp-fastapi/azurefunctions/__init__.py @@ -0,0 +1 @@ +__path__ = __import__("pkgutil").extend_path(__path__, __name__) diff --git a/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/__init__.py b/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/__init__.py new file mode 100644 index 0000000..8db66d3 --- /dev/null +++ b/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/__init__.py @@ -0,0 +1 @@ +__path__ = __import__("pkgutil").extend_path(__path__, __name__) diff --git a/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/__init__.py b/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/__init__.py new file mode 100644 index 0000000..5b48f05 --- /dev/null +++ b/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/__init__.py @@ -0,0 +1,29 @@ +__path__ = __import__("pkgutil").extend_path(__path__, __name__) + +from .starlette import (Request, Response, + FileResponse, + HTMLResponse, + JSONResponse, + PlainTextResponse, + RedirectResponse, + StreamingResponse, RequestSynchronizer, WebApp, WebServer + ) +from .decorators import (MCPFunctionApp, FunctionRegister) + +__all__ = [ + "WebServer", + "WebApp", + "Request", + "Response", + "RequestSynchronizer", + "StreamingResponse", + "HTMLResponse", + "PlainTextResponse", + "RedirectResponse", + "JSONResponse", + "FileResponse", + "MCPFunctionApp", + "FunctionRegister" +] + +__version__ = "1.0.0b1" diff --git a/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/decorators/__init__.py b/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/decorators/__init__.py new file mode 100644 index 0000000..e8d5333 --- /dev/null +++ b/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/decorators/__init__.py @@ -0,0 +1,6 @@ +from .mcp_app import MCPFunctionApp, FunctionRegister + +__all__ = [ + "MCPFunctionApp", + "FunctionRegister" +] \ No newline at end of file diff --git a/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/decorators/mcp_app.py b/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/decorators/mcp_app.py new file mode 100644 index 0000000..8a76f4e --- /dev/null +++ b/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/decorators/mcp_app.py @@ -0,0 +1,78 @@ +from typing import Union, Any +from azure.functions import (FunctionRegister, TriggerApi, BindingApi, +AuthLevel, HttpMethod, ExternalHttpFunctionApp) +from azure.functions.decorators.function_app import HttpFunctionsAuthLevelMixin +from azurefunctions.extensions.mcp_server.starlette import Request, Response + +from mcp.server.fastmcp.server import FastMCP +from mcp.server.sse import SseServerTransport + +try: + from azure.functions import SettingsApi +except ImportError: # backwards compatibility path + class SettingsApi: + """Backwards compatibility mock of SettingsApi.""" + + pass + + +class MCPFunctionApp(FastMCP, TriggerApi, FunctionRegister, HttpFunctionsAuthLevelMixin): + """MCP Functions (DF) Blueprint container. + + It allows functions to be declared via trigger and binding decorators, + but does not automatically index/register these functions. + """ + + def __init__(self, + auth_level: Union[AuthLevel, str] = AuthLevel.FUNCTION, + name: str | None = None, instructions: str | None = None, *args, **settings: Any): + """Instantiate a Durable Functions app with which to register Functions. + + Parameters + ---------- + http_auth_level: Union[AuthLevel, str] + Authorization level required for Function invocation. + Defaults to AuthLevel.Function. + + Returns + ------- + DFApp + New instance of a Durable Functions app + """ + FunctionRegister.__init__(self, auth_level=auth_level, *args, **settings) + FastMCP.__init__(self, name, instructions, **settings) + self._add_http_app(auth_level) + + + def _add_http_app(self, auth_level) -> None: + """Add an Asgi app integrated mcp_server function. + + :param http_middleware: :class:`WsgiMiddleware` + or class:`AsgiMiddleware` instance. + + :return: None + """ + + sse = SseServerTransport("/messages/") + + @self.function_name(name="sse") + @self.route(trigger_arg_name="request" , methods=(method for method in HttpMethod), + auth_level=auth_level, + route="sse") + async def http_app_func(request: Request): + async with sse.connect_sse( + request.scope, request.receive, request._send + ) as streams: + await self._mcp_server.run( + streams[0], + streams[1], + self._mcp_server.create_initialization_options(), + ) + + @self.function_name(name="messages") + @self.route(trigger_arg_name="request", methods=(method for method in HttpMethod), + auth_level=auth_level, + route="messages") + async def http_messages_func(request: Request): + return await sse.handle_post_message(request.scope, request.receive, request._send) + diff --git a/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/starlette/__init__.py b/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/starlette/__init__.py new file mode 100644 index 0000000..65174da --- /dev/null +++ b/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/starlette/__init__.py @@ -0,0 +1,29 @@ +from starlette.requests import (Request) +from starlette.responses import (Response) + +from starlette.responses import ( + FileResponse, + HTMLResponse, + JSONResponse, + PlainTextResponse, + RedirectResponse, + StreamingResponse, +) + +from .web import RequestSynchronizer, WebApp, WebServer + +__all__ = [ + "WebServer", + "WebApp", + "Request", + "Response", + "RequestSynchronizer", + "StreamingResponse", + "HTMLResponse", + "PlainTextResponse", + "RedirectResponse", + "JSONResponse", + "FileResponse", +] + +__version__ = "1.0.0b1" diff --git a/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/starlette/web.py b/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/starlette/web.py new file mode 100644 index 0000000..0ed4e6a --- /dev/null +++ b/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/starlette/web.py @@ -0,0 +1,149 @@ +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. + +from typing import Callable + +import uvicorn +from azurefunctions.extensions.base import ( + RequestSynchronizer, + RequestTrackerMeta, + ResponseLabels, + ResponseTrackerMeta, + WebApp, + WebServer, +) + +from starlette.applications import Starlette + +from starlette.requests import Request as StarletteRequest +from starlette.responses import Response as StarletteResponse +from starlette.responses import FileResponse as StarletteFileResponse +from starlette.responses import HTMLResponse as StarletteHTMLResponse +from starlette.responses import JSONResponse as StarletteJSONResponse +from starlette.responses import PlainTextResponse as StarlettePlainTextResponse +from starlette.responses import RedirectResponse as StarletteRedirectResponse +from starlette.responses import StreamingResponse as StarletteStreamingResponse + +from pydantic import BaseModel + +import logging + + +class RequestSynchronizer(RequestSynchronizer): + def sync_route_params(self, request, path_params): + # add null checks for request and path_params + if request is None: + raise TypeError("Request object is None") + if path_params is None: + raise TypeError("Path parameters are None") + + request.path_params.clear() + request.path_params.update(path_params) + + +class Request(metaclass=RequestTrackerMeta): + request_type = StarletteRequest + synchronizer = RequestSynchronizer() + + +class Response(metaclass=ResponseTrackerMeta): + label = ResponseLabels.STANDARD + response_type = StarletteResponse + + +class StreamingResponse(metaclass=ResponseTrackerMeta): + label = ResponseLabels.STREAMING + response_type = StarletteStreamingResponse + + +class HTMLResponse(metaclass=ResponseTrackerMeta): + label = ResponseLabels.HTML + response_type = StarletteHTMLResponse + + +class PlainTextResponse(metaclass=ResponseTrackerMeta): + label = ResponseLabels.PLAIN_TEXT + response_type = StarlettePlainTextResponse + + +class RedirectResponse(metaclass=ResponseTrackerMeta): + label = ResponseLabels.REDIRECT + response_type = StarletteRedirectResponse + + +class JSONResponse(metaclass=ResponseTrackerMeta): + label = ResponseLabels.JSON + response_type = StarletteJSONResponse + + +class FileResponse(metaclass=ResponseTrackerMeta): + label = ResponseLabels.FILE + response_type = StarletteFileResponse + + +class StrResponse(metaclass=ResponseTrackerMeta): + label = ResponseLabels.STR + response_type = str + + +class DictResponse(metaclass=ResponseTrackerMeta): + label = ResponseLabels.DICT + response_type = dict + + +class BoolResponse(metaclass=ResponseTrackerMeta): + label = ResponseLabels.BOOL + response_type = bool + + +class PydanticResponse(metaclass=ResponseTrackerMeta): + label = ResponseLabels.PYDANTIC + response_type = BaseModel + + +class IntResponse(metaclass=ResponseTrackerMeta): + label = ResponseLabels.INT + response_type = int + + +class FloatResponse(metaclass=ResponseTrackerMeta): + label = ResponseLabels.FLOAT + response_type = float + + +class ListResponse(metaclass=ResponseTrackerMeta): + label = ResponseLabels.LIST + response_type = list + + +class WebApp(WebApp): + def __init__(self): + self.web_app = Starlette(debug=True) + + def route(self, func: Callable): + # Apply the api_route decorator + self.web_app.add_route(route=func, path="/{path:path}", methods=[ + "GET", + "POST", + "PUT", + "DELETE", + "OPTIONS", + "HEAD", + "PATCH", + "TRACE", + ]) + + def get_app(self): + return self.web_app + + +class WebServer(WebServer): + async def serve(self): + uvicorn_config = uvicorn.Config( + self.web_app, host=self.hostname, port=self.port, loop="asyncio", log_level="debug", lifespan="on", use_colors=True + ) + logging.info(f"Starting server on {self.hostname}:{self.port}") + # Create a Uvicorn server instance + server = uvicorn.Server(uvicorn_config) + + return await server.serve() diff --git a/azurefunctions-extensions-mcp-fastapi/pyproject.toml b/azurefunctions-extensions-mcp-fastapi/pyproject.toml new file mode 100644 index 0000000..b6a37f3 --- /dev/null +++ b/azurefunctions-extensions-mcp-fastapi/pyproject.toml @@ -0,0 +1,53 @@ +[build-system] +requires = ["setuptools >= 61.0", "wheel"] +build-backend = "setuptools.build_meta" + +[project] +name = "azurefunctions-extensions-mcp-fastapi" +dynamic = ["version"] +requires-python = ">=3.8" +authors = [{ name = "Azure Functions team at Microsoft Corp.", email = "azurefunctions@microsoft.com"}] +description = "MCP Python worker extension for Azure Functions." +readme = "README.md" +license = {text = "MIT License"} +classifiers= [ + 'License :: OSI Approved :: MIT License', + 'Intended Audience :: Developers', + 'Programming Language :: Python :: 3', + 'Programming Language :: Python :: 3.8', + 'Programming Language :: Python :: 3.9', + 'Programming Language :: Python :: 3.10', + 'Programming Language :: Python :: 3.11', + 'Operating System :: Microsoft :: Windows', + 'Operating System :: POSIX', + 'Operating System :: MacOS :: MacOS X', + 'Environment :: Web Environment', + 'Development Status :: 5 - Production/Stable', + ] +dependencies = [ + 'azurefunctions-extensions-base', + 'azure-functions', + 'starlette~=0.46.1', + 'uvicorn~=0.32.0', + 'pydantic~=2.10.0', + 'mcp~=1.3.0' + ] + +[project.optional-dependencies] +dev = [ + 'pytest', + 'pytest-cov', + 'coverage', + 'pytest-instafail', + 'pre-commit' + ] + +[tool.setuptools.dynamic] +version = {attr = "azurefunctions.extensions.mcp_server.__version__"} + +[tool.setuptools.packages.find] +exclude = [ + 'azurefunctions.extensions.http','azurefunctions.extensions', + 'azurefunctions', 'tests', 'samples' + ] + diff --git a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/function_app.py b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/function_app.py new file mode 100644 index 0000000..21ed45a --- /dev/null +++ b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/function_app.py @@ -0,0 +1,26 @@ +# This Azure Function streams real-time sensor data using Server-Sent Events (SSE). +# It simulates a sensor network transmitting temperature and humidity readings, +# which can be consumed by IoT dashboards or analytics pipelines. + +import time + +import azure.functions as func +from azurefunctions.extensions.mcp_server.starlette import Request, StreamingResponse + +app = func.FunctionApp(http_auth_level=func.AuthLevel.ANONYMOUS) + + +def generate_sensor_data(): + """Generate real-time sensor data.""" + for i in range(10): + # Simulate temperature and humidity readings + temperature = 20 + i + humidity = 50 + i + yield f"data: {{'temperature': {temperature}, 'humidity': {humidity}}}\n\n" + time.sleep(1) + + +@app.route(route="stream", methods=[func.HttpMethod.GET]) +async def stream_sensor_data(req: Request) -> StreamingResponse: + """Endpoint to stream real-time sensor data.""" + return StreamingResponse(generate_sensor_data(), media_type="text/event-stream") diff --git a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/host.json b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/host.json new file mode 100644 index 0000000..9df9136 --- /dev/null +++ b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/host.json @@ -0,0 +1,15 @@ +{ + "version": "2.0", + "logging": { + "applicationInsights": { + "samplingSettings": { + "isEnabled": true, + "excludedTypes": "Request" + } + } + }, + "extensionBundle": { + "id": "Microsoft.Azure.Functions.ExtensionBundle", + "version": "[4.*, 5.0.0)" + } +} \ No newline at end of file diff --git a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/local.settings.json b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/local.settings.json new file mode 100644 index 0000000..6cce4f6 --- /dev/null +++ b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/local.settings.json @@ -0,0 +1,8 @@ +{ + "IsEncrypted": false, + "Values": { + "FUNCTIONS_WORKER_RUNTIME": "python", + "AzureWebJobsStorage": "UseDevelopmentStorage=true", + "PYTHON_ENABLE_INIT_INDEXING": "1" + } +} \ No newline at end of file diff --git a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/requirements.txt b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/requirements.txt new file mode 100644 index 0000000..0d4eb9e --- /dev/null +++ b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/requirements.txt @@ -0,0 +1,6 @@ +# DO NOT include azure-functions-worker in this file +# The Python Worker is managed by Azure Functions platform +# Manually managing azure-functions-worker may cause unexpected issues + +azure-functions +azurefunctions-extensions-http-fastapi \ No newline at end of file diff --git a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/function_app.py b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/function_app.py new file mode 100644 index 0000000..ab097d7 --- /dev/null +++ b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/function_app.py @@ -0,0 +1,25 @@ +# This Azure Function receives streaming data from a client and processes it in real-time. +# It demonstrates streaming upload capabilities for scenarios such as uploading large files, +# processing continuous data streams, or handling IoT device data. + +import azure.functions as func +from azurefunctions.extensions.mcp_server.starlette import JSONResponse, Request + +app = func.FunctionApp(http_auth_level=func.AuthLevel.ANONYMOUS) + + +@app.route(route="streaming_upload", methods=[func.HttpMethod.POST]) +async def streaming_upload(req: Request) -> JSONResponse: + """Handle streaming upload requests.""" + # Process each chunk of data as it arrives + async for chunk in req.stream(): + process_data_chunk(chunk) + + # Once all data is received, return a JSON response indicating successful processing + return JSONResponse({"status": "Data uploaded and processed successfully"}) + + +def process_data_chunk(chunk: bytes): + """Process each data chunk.""" + # Add custom processing logic here + pass diff --git a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/host.json b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/host.json new file mode 100644 index 0000000..9df9136 --- /dev/null +++ b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/host.json @@ -0,0 +1,15 @@ +{ + "version": "2.0", + "logging": { + "applicationInsights": { + "samplingSettings": { + "isEnabled": true, + "excludedTypes": "Request" + } + } + }, + "extensionBundle": { + "id": "Microsoft.Azure.Functions.ExtensionBundle", + "version": "[4.*, 5.0.0)" + } +} \ No newline at end of file diff --git a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/local.settings.json b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/local.settings.json new file mode 100644 index 0000000..6cce4f6 --- /dev/null +++ b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/local.settings.json @@ -0,0 +1,8 @@ +{ + "IsEncrypted": false, + "Values": { + "FUNCTIONS_WORKER_RUNTIME": "python", + "AzureWebJobsStorage": "UseDevelopmentStorage=true", + "PYTHON_ENABLE_INIT_INDEXING": "1" + } +} \ No newline at end of file diff --git a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/requirements.txt b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/requirements.txt new file mode 100644 index 0000000..0d4eb9e --- /dev/null +++ b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/requirements.txt @@ -0,0 +1,6 @@ +# DO NOT include azure-functions-worker in this file +# The Python Worker is managed by Azure Functions platform +# Manually managing azure-functions-worker may cause unexpected issues + +azure-functions +azurefunctions-extensions-http-fastapi \ No newline at end of file diff --git a/azurefunctions-extensions-mcp-fastapi/tests/__init__.py b/azurefunctions-extensions-mcp-fastapi/tests/__init__.py new file mode 100644 index 0000000..528a01b --- /dev/null +++ b/azurefunctions-extensions-mcp-fastapi/tests/__init__.py @@ -0,0 +1,20 @@ +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. + +"""Bootstrap for '$ python setup.py test' command.""" + +import os.path +import sys +import unittest +import unittest.runner + + +def suite(): + test_loader = unittest.TestLoader() + return test_loader.discover(os.path.dirname(__file__), pattern="test_*.py") + + +if __name__ == "__main__": + runner = unittest.runner.TextTestRunner() + result = runner.run(suite()) + sys.exit(not result.wasSuccessful()) diff --git a/azurefunctions-extensions-mcp-fastapi/tests/test_web.py b/azurefunctions-extensions-mcp-fastapi/tests/test_web.py new file mode 100644 index 0000000..385b887 --- /dev/null +++ b/azurefunctions-extensions-mcp-fastapi/tests/test_web.py @@ -0,0 +1,232 @@ +import asyncio +import unittest +from unittest.mock import MagicMock, patch + +from azurefunctions.extensions.base import ( + RequestTrackerMeta, + ResponseLabels, + ResponseTrackerMeta, +) +from azurefunctions.extensions.mcp_server.starlette import ( + RequestSynchronizer, + WebApp, + WebServer, +) +from starlette.applications import Starlette + +from starlette.requests import Request as StarletteRequest +from starlette.responses import Response as StarletteResponse +from starlette.responses import FileResponse as StarletteFileResponse +from starlette.responses import HTMLResponse as StarletteHTMLResponse +from starlette.responses import JSONResponse as StarletteJSONResponse +from starlette.responses import PlainTextResponse as StarlettePlainTextResponse +from starlette.responses import RedirectResponse as StarletteRedirectResponse +from starlette.responses import StreamingResponse as StarletteStreamingResponse + + +class TestRequestTrackerMeta(unittest.TestCase): + def test_request_type_defined(self): + class Request(metaclass=RequestTrackerMeta): + request_type = StarletteRequest + synchronizer = RequestSynchronizer() + + self.assertTrue(hasattr(Request, "request_type")) + self.assertEqual(Request.request_type, StarletteRequest) + + def test_request_type_undefined(self): + with self.assertRaises(Exception) as context: + + class Request(metaclass=RequestTrackerMeta): + pass + + self.assertTrue("Request type not provided" in str(context.exception)) + + +class TestResponseTrackerMeta(unittest.TestCase): + def test_response_labels_defined(self): + class Response(metaclass=ResponseTrackerMeta): + label = ResponseLabels.STANDARD + response_type = StarletteResponse + + self.assertTrue(hasattr(Response, "label")) + self.assertTrue(hasattr(Response, "response_type")) + self.assertEqual(Response.label, ResponseLabels.STANDARD) + self.assertEqual(Response.response_type, StarletteResponse) + + def test_response_labels_undefined(self): + with self.assertRaises(Exception) as context: + + class Response(metaclass=ResponseTrackerMeta): + pass + + self.assertTrue("Response label not provided" in str(context.exception)) + + def test_multiple_response_labels(self): + with self.assertRaises(Exception) as context: + + class Response1(metaclass=ResponseTrackerMeta): + label = ResponseLabels.STANDARD + response_type = StarletteResponse + + class Response2(metaclass=ResponseTrackerMeta): + label = ResponseLabels.STANDARD + response_type = StarletteJSONResponse + + self.assertTrue( + "Only one response type shall be recorded" in str(context.exception) + ) + + +class TestWebApp(unittest.TestCase): + def setUp(self): + self.web_app = WebApp() + + def test_route(self): + @self.web_app.route + def test_route(request: StarletteRequest): + return {"message": "Hello"} + + self.assertTrue( + "/{path:path}" + in [endpoint.path for endpoint in self.web_app.web_app.router.routes] + ) + route = [ + endpoint + for endpoint in self.web_app.web_app.router.routes + if endpoint.path == "/{path:path}" + ][0] + self.assertEqual( + route.methods, + {"GET", "POST", "PUT", "DELETE", "OPTIONS", "HEAD", "PATCH", "TRACE"}, + ) + + def test_get_app(self): + self.assertIsInstance(self.web_app.get_app(), Starlette) + + +class TestWebServer(unittest.TestCase): + def setUp(self): + class TestApp: + pass + + self.test_app = TestApp() + self.web_app_mock = MagicMock().get_app() + self.web_app_mock.get_app.return_value = self.test_app + self.hostname = "localhost" + self.port = 8000 + self.web_server = WebServer(self.hostname, self.port, self.web_app_mock) + + @patch("uvicorn.Config") + @patch("uvicorn.Server") + def test_serve(self, server_mock, config_mock): + async def serve(): + await asyncio.sleep(0) + + config_instance_mock = config_mock.return_value + server_instance_mock = server_mock.return_value + server_instance_mock.serve = ( + serve # Mock the serve method to return a CoroutineMock + ) + + asyncio.get_event_loop().run_until_complete(self.web_server.serve()) + + config_mock.assert_called_once_with( + self.test_app, host=self.hostname, port=self.port + ) + server_mock.assert_called_once_with(config_instance_mock) + + async def run_serve(self): + await self.web_server.serve() + + +class TestRequestSynchronizer(unittest.TestCase): + def test_sync_route_params(self): + # Create a mock request object + mock_request = MagicMock() + + # Define some path parameters + path_params = {"param1": "value1", "param2": "value2"} + + # Create an instance of the ConcreteRequestSynchronizer + synchronizer = RequestSynchronizer() + + # Call the sync_route_params method with the mock request and path parameters + synchronizer.sync_route_params(mock_request, path_params) + + # Assert that the request's path_params have been updated with the provided path parameters + mock_request.path_params.clear.assert_called_once() + mock_request.path_params.update.assert_called_once_with(path_params) + + def test_sync_route_params_missing_request(self): + # Create an instance of the ConcreteRequestSynchronizer + synchronizer = RequestSynchronizer() + + # Define some path parameters + path_params = {"param1": "value1", "param2": "value2"} + + # Call the sync_route_params method with a None request and path parameters + with self.assertRaises(TypeError): + synchronizer.sync_route_params(None, path_params) + + def test_sync_route_params_missing_path_params(self): + # Create a mock request object + mock_request = MagicMock() + + # Create an instance of the ConcreteRequestSynchronizer + synchronizer = RequestSynchronizer() + + # Call the sync_route_params method with the mock request and None path parameters + with self.assertRaises(TypeError): + synchronizer.sync_route_params(mock_request, None) + + +class TestExtensionClasses(unittest.TestCase): + def test_request(self): + from azurefunctions.extensions.mcp_server.starlette.web import Request + + self.assertEqual(RequestTrackerMeta.get_request_type(), StarletteRequest) + self.assertTrue( + isinstance(RequestTrackerMeta.get_synchronizer(), RequestSynchronizer) + ) + + def test_response(self): + self.assertEqual( + ResponseTrackerMeta.get_response_type(ResponseLabels.STANDARD), + StarletteResponse, + ) + + def test_streaming_response(self): + self.assertEqual( + ResponseTrackerMeta.get_response_type(ResponseLabels.STREAMING), + StarletteStreamingResponse, + ) + + def test_html_response(self): + self.assertEqual( + ResponseTrackerMeta.get_response_type(ResponseLabels.HTML), + StarletteHTMLResponse, + ) + + def test_plain_text_response(self): + self.assertEqual( + ResponseTrackerMeta.get_response_type(ResponseLabels.PLAIN_TEXT), + StarlettePlainTextResponse, + ) + + def test_redirect_response(self): + self.assertEqual( + ResponseTrackerMeta.get_response_type(ResponseLabels.REDIRECT), + StarletteRedirectResponse, + ) + + def test_json_response(self): + self.assertEqual( + ResponseTrackerMeta.get_response_type(ResponseLabels.JSON), + StarletteJSONResponse, + ) + + def test_file_response(self): + self.assertEqual( + ResponseTrackerMeta.get_response_type(ResponseLabels.FILE), + StarletteFileResponse, + ) From 621118e0be095a3a2c59b4df57e4a39d2fcb9de1 Mon Sep 17 00:00:00 2001 From: Varad Meru Date: Thu, 20 Mar 2025 23:06:29 -0700 Subject: [PATCH 2/3] Renamed to MCP server --- .../function_app.py | 26 ------------------- .../host.json | 15 ----------- .../local.settings.json | 8 ------ .../requirements.txt | 6 ----- .../function_app.py | 25 ------------------ .../host.json | 15 ----------- .../local.settings.json | 8 ------ .../requirements.txt | 6 ----- .../CHANGELOG.md | 0 .../LICENSE | 0 .../MANIFEST.in | 0 .../README.md | 0 .../azurefunctions/__init__.py | 0 .../azurefunctions/extensions/__init__.py | 0 .../extensions/mcp_server/__init__.py | 2 +- .../mcp_server/decorators/__init__.py | 0 .../mcp_server/decorators/mcp_app.py | 0 .../mcp_server/starlette/__init__.py | 0 .../extensions/mcp_server/starlette/web.py | 0 .../pyproject.toml | 12 +++++---- .../tests/__init__.py | 0 .../tests/test_web.py | 0 22 files changed, 8 insertions(+), 115 deletions(-) delete mode 100644 azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/function_app.py delete mode 100644 azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/host.json delete mode 100644 azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/local.settings.json delete mode 100644 azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/requirements.txt delete mode 100644 azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/function_app.py delete mode 100644 azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/host.json delete mode 100644 azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/local.settings.json delete mode 100644 azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/requirements.txt rename {azurefunctions-extensions-mcp-fastapi => azurefunctions-extensions-mcp-server}/CHANGELOG.md (100%) rename {azurefunctions-extensions-mcp-fastapi => azurefunctions-extensions-mcp-server}/LICENSE (100%) rename {azurefunctions-extensions-mcp-fastapi => azurefunctions-extensions-mcp-server}/MANIFEST.in (100%) rename {azurefunctions-extensions-mcp-fastapi => azurefunctions-extensions-mcp-server}/README.md (100%) rename {azurefunctions-extensions-mcp-fastapi => azurefunctions-extensions-mcp-server}/azurefunctions/__init__.py (100%) rename {azurefunctions-extensions-mcp-fastapi => azurefunctions-extensions-mcp-server}/azurefunctions/extensions/__init__.py (100%) rename {azurefunctions-extensions-mcp-fastapi => azurefunctions-extensions-mcp-server}/azurefunctions/extensions/mcp_server/__init__.py (96%) rename {azurefunctions-extensions-mcp-fastapi => azurefunctions-extensions-mcp-server}/azurefunctions/extensions/mcp_server/decorators/__init__.py (100%) rename {azurefunctions-extensions-mcp-fastapi => azurefunctions-extensions-mcp-server}/azurefunctions/extensions/mcp_server/decorators/mcp_app.py (100%) rename {azurefunctions-extensions-mcp-fastapi => azurefunctions-extensions-mcp-server}/azurefunctions/extensions/mcp_server/starlette/__init__.py (100%) rename {azurefunctions-extensions-mcp-fastapi => azurefunctions-extensions-mcp-server}/azurefunctions/extensions/mcp_server/starlette/web.py (100%) rename {azurefunctions-extensions-mcp-fastapi => azurefunctions-extensions-mcp-server}/pyproject.toml (86%) rename {azurefunctions-extensions-mcp-fastapi => azurefunctions-extensions-mcp-server}/tests/__init__.py (100%) rename {azurefunctions-extensions-mcp-fastapi => azurefunctions-extensions-mcp-server}/tests/test_web.py (100%) diff --git a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/function_app.py b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/function_app.py deleted file mode 100644 index 21ed45a..0000000 --- a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/function_app.py +++ /dev/null @@ -1,26 +0,0 @@ -# This Azure Function streams real-time sensor data using Server-Sent Events (SSE). -# It simulates a sensor network transmitting temperature and humidity readings, -# which can be consumed by IoT dashboards or analytics pipelines. - -import time - -import azure.functions as func -from azurefunctions.extensions.mcp_server.starlette import Request, StreamingResponse - -app = func.FunctionApp(http_auth_level=func.AuthLevel.ANONYMOUS) - - -def generate_sensor_data(): - """Generate real-time sensor data.""" - for i in range(10): - # Simulate temperature and humidity readings - temperature = 20 + i - humidity = 50 + i - yield f"data: {{'temperature': {temperature}, 'humidity': {humidity}}}\n\n" - time.sleep(1) - - -@app.route(route="stream", methods=[func.HttpMethod.GET]) -async def stream_sensor_data(req: Request) -> StreamingResponse: - """Endpoint to stream real-time sensor data.""" - return StreamingResponse(generate_sensor_data(), media_type="text/event-stream") diff --git a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/host.json b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/host.json deleted file mode 100644 index 9df9136..0000000 --- a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/host.json +++ /dev/null @@ -1,15 +0,0 @@ -{ - "version": "2.0", - "logging": { - "applicationInsights": { - "samplingSettings": { - "isEnabled": true, - "excludedTypes": "Request" - } - } - }, - "extensionBundle": { - "id": "Microsoft.Azure.Functions.ExtensionBundle", - "version": "[4.*, 5.0.0)" - } -} \ No newline at end of file diff --git a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/local.settings.json b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/local.settings.json deleted file mode 100644 index 6cce4f6..0000000 --- a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/local.settings.json +++ /dev/null @@ -1,8 +0,0 @@ -{ - "IsEncrypted": false, - "Values": { - "FUNCTIONS_WORKER_RUNTIME": "python", - "AzureWebJobsStorage": "UseDevelopmentStorage=true", - "PYTHON_ENABLE_INIT_INDEXING": "1" - } -} \ No newline at end of file diff --git a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/requirements.txt b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/requirements.txt deleted file mode 100644 index 0d4eb9e..0000000 --- a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_download/requirements.txt +++ /dev/null @@ -1,6 +0,0 @@ -# DO NOT include azure-functions-worker in this file -# The Python Worker is managed by Azure Functions platform -# Manually managing azure-functions-worker may cause unexpected issues - -azure-functions -azurefunctions-extensions-http-fastapi \ No newline at end of file diff --git a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/function_app.py b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/function_app.py deleted file mode 100644 index ab097d7..0000000 --- a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/function_app.py +++ /dev/null @@ -1,25 +0,0 @@ -# This Azure Function receives streaming data from a client and processes it in real-time. -# It demonstrates streaming upload capabilities for scenarios such as uploading large files, -# processing continuous data streams, or handling IoT device data. - -import azure.functions as func -from azurefunctions.extensions.mcp_server.starlette import JSONResponse, Request - -app = func.FunctionApp(http_auth_level=func.AuthLevel.ANONYMOUS) - - -@app.route(route="streaming_upload", methods=[func.HttpMethod.POST]) -async def streaming_upload(req: Request) -> JSONResponse: - """Handle streaming upload requests.""" - # Process each chunk of data as it arrives - async for chunk in req.stream(): - process_data_chunk(chunk) - - # Once all data is received, return a JSON response indicating successful processing - return JSONResponse({"status": "Data uploaded and processed successfully"}) - - -def process_data_chunk(chunk: bytes): - """Process each data chunk.""" - # Add custom processing logic here - pass diff --git a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/host.json b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/host.json deleted file mode 100644 index 9df9136..0000000 --- a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/host.json +++ /dev/null @@ -1,15 +0,0 @@ -{ - "version": "2.0", - "logging": { - "applicationInsights": { - "samplingSettings": { - "isEnabled": true, - "excludedTypes": "Request" - } - } - }, - "extensionBundle": { - "id": "Microsoft.Azure.Functions.ExtensionBundle", - "version": "[4.*, 5.0.0)" - } -} \ No newline at end of file diff --git a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/local.settings.json b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/local.settings.json deleted file mode 100644 index 6cce4f6..0000000 --- a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/local.settings.json +++ /dev/null @@ -1,8 +0,0 @@ -{ - "IsEncrypted": false, - "Values": { - "FUNCTIONS_WORKER_RUNTIME": "python", - "AzureWebJobsStorage": "UseDevelopmentStorage=true", - "PYTHON_ENABLE_INIT_INDEXING": "1" - } -} \ No newline at end of file diff --git a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/requirements.txt b/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/requirements.txt deleted file mode 100644 index 0d4eb9e..0000000 --- a/azurefunctions-extensions-mcp-fastapi/samples/fastapi_samples_streaming_upload/requirements.txt +++ /dev/null @@ -1,6 +0,0 @@ -# DO NOT include azure-functions-worker in this file -# The Python Worker is managed by Azure Functions platform -# Manually managing azure-functions-worker may cause unexpected issues - -azure-functions -azurefunctions-extensions-http-fastapi \ No newline at end of file diff --git a/azurefunctions-extensions-mcp-fastapi/CHANGELOG.md b/azurefunctions-extensions-mcp-server/CHANGELOG.md similarity index 100% rename from azurefunctions-extensions-mcp-fastapi/CHANGELOG.md rename to azurefunctions-extensions-mcp-server/CHANGELOG.md diff --git a/azurefunctions-extensions-mcp-fastapi/LICENSE b/azurefunctions-extensions-mcp-server/LICENSE similarity index 100% rename from azurefunctions-extensions-mcp-fastapi/LICENSE rename to azurefunctions-extensions-mcp-server/LICENSE diff --git a/azurefunctions-extensions-mcp-fastapi/MANIFEST.in b/azurefunctions-extensions-mcp-server/MANIFEST.in similarity index 100% rename from azurefunctions-extensions-mcp-fastapi/MANIFEST.in rename to azurefunctions-extensions-mcp-server/MANIFEST.in diff --git a/azurefunctions-extensions-mcp-fastapi/README.md b/azurefunctions-extensions-mcp-server/README.md similarity index 100% rename from azurefunctions-extensions-mcp-fastapi/README.md rename to azurefunctions-extensions-mcp-server/README.md diff --git a/azurefunctions-extensions-mcp-fastapi/azurefunctions/__init__.py b/azurefunctions-extensions-mcp-server/azurefunctions/__init__.py similarity index 100% rename from azurefunctions-extensions-mcp-fastapi/azurefunctions/__init__.py rename to azurefunctions-extensions-mcp-server/azurefunctions/__init__.py diff --git a/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/__init__.py b/azurefunctions-extensions-mcp-server/azurefunctions/extensions/__init__.py similarity index 100% rename from azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/__init__.py rename to azurefunctions-extensions-mcp-server/azurefunctions/extensions/__init__.py diff --git a/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/__init__.py b/azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/__init__.py similarity index 96% rename from azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/__init__.py rename to azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/__init__.py index 5b48f05..1ad427c 100644 --- a/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/__init__.py +++ b/azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/__init__.py @@ -26,4 +26,4 @@ "FunctionRegister" ] -__version__ = "1.0.0b1" +__version__ = "0.0.1b1" diff --git a/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/decorators/__init__.py b/azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/decorators/__init__.py similarity index 100% rename from azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/decorators/__init__.py rename to azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/decorators/__init__.py diff --git a/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/decorators/mcp_app.py b/azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/decorators/mcp_app.py similarity index 100% rename from azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/decorators/mcp_app.py rename to azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/decorators/mcp_app.py diff --git a/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/starlette/__init__.py b/azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/starlette/__init__.py similarity index 100% rename from azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/starlette/__init__.py rename to azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/starlette/__init__.py diff --git a/azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/starlette/web.py b/azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/starlette/web.py similarity index 100% rename from azurefunctions-extensions-mcp-fastapi/azurefunctions/extensions/mcp_server/starlette/web.py rename to azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/starlette/web.py diff --git a/azurefunctions-extensions-mcp-fastapi/pyproject.toml b/azurefunctions-extensions-mcp-server/pyproject.toml similarity index 86% rename from azurefunctions-extensions-mcp-fastapi/pyproject.toml rename to azurefunctions-extensions-mcp-server/pyproject.toml index b6a37f3..53631db 100644 --- a/azurefunctions-extensions-mcp-fastapi/pyproject.toml +++ b/azurefunctions-extensions-mcp-server/pyproject.toml @@ -3,9 +3,9 @@ requires = ["setuptools >= 61.0", "wheel"] build-backend = "setuptools.build_meta" [project] -name = "azurefunctions-extensions-mcp-fastapi" +name = "azurefunctions-extensions-mcp-server" dynamic = ["version"] -requires-python = ">=3.8" +requires-python = ">=3.10" authors = [{ name = "Azure Functions team at Microsoft Corp.", email = "azurefunctions@microsoft.com"}] description = "MCP Python worker extension for Azure Functions." readme = "README.md" @@ -14,10 +14,10 @@ classifiers= [ 'License :: OSI Approved :: MIT License', 'Intended Audience :: Developers', 'Programming Language :: Python :: 3', - 'Programming Language :: Python :: 3.8', - 'Programming Language :: Python :: 3.9', 'Programming Language :: Python :: 3.10', 'Programming Language :: Python :: 3.11', + 'Programming Language :: Python :: 3.12', + 'Programming Language :: Python :: 3.13', 'Operating System :: Microsoft :: Windows', 'Operating System :: POSIX', 'Operating System :: MacOS :: MacOS X', @@ -39,7 +39,9 @@ dev = [ 'pytest-cov', 'coverage', 'pytest-instafail', - 'pre-commit' + 'pre-commit', + 'isort', + 'black' ] [tool.setuptools.dynamic] diff --git a/azurefunctions-extensions-mcp-fastapi/tests/__init__.py b/azurefunctions-extensions-mcp-server/tests/__init__.py similarity index 100% rename from azurefunctions-extensions-mcp-fastapi/tests/__init__.py rename to azurefunctions-extensions-mcp-server/tests/__init__.py diff --git a/azurefunctions-extensions-mcp-fastapi/tests/test_web.py b/azurefunctions-extensions-mcp-server/tests/test_web.py similarity index 100% rename from azurefunctions-extensions-mcp-fastapi/tests/test_web.py rename to azurefunctions-extensions-mcp-server/tests/test_web.py From ad8218540ff8eb4ee26e169887f91d77766a617a Mon Sep 17 00:00:00 2001 From: Varad Meru Date: Fri, 21 Mar 2025 08:06:27 -0700 Subject: [PATCH 3/3] Renamed files and cleaned up --- .../CHANGELOG.md | 0 .../LICENSE | 0 .../MANIFEST.in | 0 .../README.md | 2 +- .../azurefunctions/__init__.py | 0 .../azurefunctions/extensions/__init__.py | 0 .../extensions/http/__init__.py | 1 + .../extensions/http/starlette/__init__.py | 24 ++++++ .../extensions/http}/starlette/web.py | 22 ++---- .../pyproject.toml | 9 +-- .../tests/__init__.py | 0 .../tests/test_web.py | 21 ++--- .../extensions/mcp_server/__init__.py | 29 ------- .../mcp_server/decorators/__init__.py | 6 -- .../mcp_server/decorators/mcp_app.py | 78 ------------------- .../mcp_server/starlette/__init__.py | 29 ------- 16 files changed, 44 insertions(+), 177 deletions(-) rename {azurefunctions-extensions-mcp-server => azurefunctions-extensions-http-starlette}/CHANGELOG.md (100%) rename {azurefunctions-extensions-mcp-server => azurefunctions-extensions-http-starlette}/LICENSE (100%) rename {azurefunctions-extensions-mcp-server => azurefunctions-extensions-http-starlette}/MANIFEST.in (100%) rename {azurefunctions-extensions-mcp-server => azurefunctions-extensions-http-starlette}/README.md (98%) rename {azurefunctions-extensions-mcp-server => azurefunctions-extensions-http-starlette}/azurefunctions/__init__.py (100%) rename {azurefunctions-extensions-mcp-server => azurefunctions-extensions-http-starlette}/azurefunctions/extensions/__init__.py (100%) create mode 100644 azurefunctions-extensions-http-starlette/azurefunctions/extensions/http/__init__.py create mode 100644 azurefunctions-extensions-http-starlette/azurefunctions/extensions/http/starlette/__init__.py rename {azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server => azurefunctions-extensions-http-starlette/azurefunctions/extensions/http}/starlette/web.py (93%) rename {azurefunctions-extensions-mcp-server => azurefunctions-extensions-http-starlette}/pyproject.toml (85%) rename {azurefunctions-extensions-mcp-server => azurefunctions-extensions-http-starlette}/tests/__init__.py (100%) rename {azurefunctions-extensions-mcp-server => azurefunctions-extensions-http-starlette}/tests/test_web.py (95%) delete mode 100644 azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/__init__.py delete mode 100644 azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/decorators/__init__.py delete mode 100644 azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/decorators/mcp_app.py delete mode 100644 azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/starlette/__init__.py diff --git a/azurefunctions-extensions-mcp-server/CHANGELOG.md b/azurefunctions-extensions-http-starlette/CHANGELOG.md similarity index 100% rename from azurefunctions-extensions-mcp-server/CHANGELOG.md rename to azurefunctions-extensions-http-starlette/CHANGELOG.md diff --git a/azurefunctions-extensions-mcp-server/LICENSE b/azurefunctions-extensions-http-starlette/LICENSE similarity index 100% rename from azurefunctions-extensions-mcp-server/LICENSE rename to azurefunctions-extensions-http-starlette/LICENSE diff --git a/azurefunctions-extensions-mcp-server/MANIFEST.in b/azurefunctions-extensions-http-starlette/MANIFEST.in similarity index 100% rename from azurefunctions-extensions-mcp-server/MANIFEST.in rename to azurefunctions-extensions-http-starlette/MANIFEST.in diff --git a/azurefunctions-extensions-mcp-server/README.md b/azurefunctions-extensions-http-starlette/README.md similarity index 98% rename from azurefunctions-extensions-mcp-server/README.md rename to azurefunctions-extensions-http-starlette/README.md index 4c07dd3..1a02b74 100644 --- a/azurefunctions-extensions-mcp-server/README.md +++ b/azurefunctions-extensions-http-starlette/README.md @@ -32,7 +32,7 @@ The Azure Functions Extensions Http FastApi library for Python allows you to cre # processing continuous data streams, or handling IoT device data. import azure.functions as func -from azurefunctions.extensions.mcp_server.starlette import Request, JSONResponse +from azurefunctions.extensions.http.starlette import Request, JSONResponse app = func.FunctionApp(http_auth_level=func.AuthLevel.ANONYMOUS) diff --git a/azurefunctions-extensions-mcp-server/azurefunctions/__init__.py b/azurefunctions-extensions-http-starlette/azurefunctions/__init__.py similarity index 100% rename from azurefunctions-extensions-mcp-server/azurefunctions/__init__.py rename to azurefunctions-extensions-http-starlette/azurefunctions/__init__.py diff --git a/azurefunctions-extensions-mcp-server/azurefunctions/extensions/__init__.py b/azurefunctions-extensions-http-starlette/azurefunctions/extensions/__init__.py similarity index 100% rename from azurefunctions-extensions-mcp-server/azurefunctions/extensions/__init__.py rename to azurefunctions-extensions-http-starlette/azurefunctions/extensions/__init__.py diff --git a/azurefunctions-extensions-http-starlette/azurefunctions/extensions/http/__init__.py b/azurefunctions-extensions-http-starlette/azurefunctions/extensions/http/__init__.py new file mode 100644 index 0000000..8db66d3 --- /dev/null +++ b/azurefunctions-extensions-http-starlette/azurefunctions/extensions/http/__init__.py @@ -0,0 +1 @@ +__path__ = __import__("pkgutil").extend_path(__path__, __name__) diff --git a/azurefunctions-extensions-http-starlette/azurefunctions/extensions/http/starlette/__init__.py b/azurefunctions-extensions-http-starlette/azurefunctions/extensions/http/starlette/__init__.py new file mode 100644 index 0000000..3e4b01e --- /dev/null +++ b/azurefunctions-extensions-http-starlette/azurefunctions/extensions/http/starlette/__init__.py @@ -0,0 +1,24 @@ +__path__ = __import__("pkgutil").extend_path(__path__, __name__) + +from starlette.requests import Request +from starlette.responses import (FileResponse, HTMLResponse, JSONResponse, + PlainTextResponse, RedirectResponse, Response, + StreamingResponse) + +from .web import RequestSynchronizer, WebApp, WebServer + +__all__ = [ + "WebServer", + "WebApp", + "Request", + "Response", + "RequestSynchronizer", + "StreamingResponse", + "HTMLResponse", + "PlainTextResponse", + "RedirectResponse", + "JSONResponse", + "FileResponse" +] + +__version__ = "0.0.1b1" diff --git a/azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/starlette/web.py b/azurefunctions-extensions-http-starlette/azurefunctions/extensions/http/starlette/web.py similarity index 93% rename from azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/starlette/web.py rename to azurefunctions-extensions-http-starlette/azurefunctions/extensions/http/starlette/web.py index 0ed4e6a..e65ad4c 100644 --- a/azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/starlette/web.py +++ b/azurefunctions-extensions-http-starlette/azurefunctions/extensions/http/starlette/web.py @@ -1,33 +1,25 @@ # Copyright (c) Microsoft Corporation. All rights reserved. # Licensed under the MIT License. +import logging from typing import Callable import uvicorn -from azurefunctions.extensions.base import ( - RequestSynchronizer, - RequestTrackerMeta, - ResponseLabels, - ResponseTrackerMeta, - WebApp, - WebServer, -) - +from azurefunctions.extensions.base import (RequestSynchronizer, + RequestTrackerMeta, ResponseLabels, + ResponseTrackerMeta, WebApp, + WebServer) +from pydantic import BaseModel from starlette.applications import Starlette - from starlette.requests import Request as StarletteRequest -from starlette.responses import Response as StarletteResponse from starlette.responses import FileResponse as StarletteFileResponse from starlette.responses import HTMLResponse as StarletteHTMLResponse from starlette.responses import JSONResponse as StarletteJSONResponse from starlette.responses import PlainTextResponse as StarlettePlainTextResponse from starlette.responses import RedirectResponse as StarletteRedirectResponse +from starlette.responses import Response as StarletteResponse from starlette.responses import StreamingResponse as StarletteStreamingResponse -from pydantic import BaseModel - -import logging - class RequestSynchronizer(RequestSynchronizer): def sync_route_params(self, request, path_params): diff --git a/azurefunctions-extensions-mcp-server/pyproject.toml b/azurefunctions-extensions-http-starlette/pyproject.toml similarity index 85% rename from azurefunctions-extensions-mcp-server/pyproject.toml rename to azurefunctions-extensions-http-starlette/pyproject.toml index 53631db..368dffd 100644 --- a/azurefunctions-extensions-mcp-server/pyproject.toml +++ b/azurefunctions-extensions-http-starlette/pyproject.toml @@ -3,11 +3,11 @@ requires = ["setuptools >= 61.0", "wheel"] build-backend = "setuptools.build_meta" [project] -name = "azurefunctions-extensions-mcp-server" +name = "azurefunctions-extensions-http-starlette" dynamic = ["version"] requires-python = ">=3.10" authors = [{ name = "Azure Functions team at Microsoft Corp.", email = "azurefunctions@microsoft.com"}] -description = "MCP Python worker extension for Azure Functions." +description = "Starlette HTTP Python worker extension for Azure Functions." readme = "README.md" license = {text = "MIT License"} classifiers= [ @@ -29,8 +29,7 @@ dependencies = [ 'azure-functions', 'starlette~=0.46.1', 'uvicorn~=0.32.0', - 'pydantic~=2.10.0', - 'mcp~=1.3.0' + 'pydantic~=2.10.0' ] [project.optional-dependencies] @@ -45,7 +44,7 @@ dev = [ ] [tool.setuptools.dynamic] -version = {attr = "azurefunctions.extensions.mcp_server.__version__"} +version = {attr = "azurefunctions.extensions.http.starlette.__version__"} [tool.setuptools.packages.find] exclude = [ diff --git a/azurefunctions-extensions-mcp-server/tests/__init__.py b/azurefunctions-extensions-http-starlette/tests/__init__.py similarity index 100% rename from azurefunctions-extensions-mcp-server/tests/__init__.py rename to azurefunctions-extensions-http-starlette/tests/__init__.py diff --git a/azurefunctions-extensions-mcp-server/tests/test_web.py b/azurefunctions-extensions-http-starlette/tests/test_web.py similarity index 95% rename from azurefunctions-extensions-mcp-server/tests/test_web.py rename to azurefunctions-extensions-http-starlette/tests/test_web.py index 385b887..01ae0c5 100644 --- a/azurefunctions-extensions-mcp-server/tests/test_web.py +++ b/azurefunctions-extensions-http-starlette/tests/test_web.py @@ -2,25 +2,18 @@ import unittest from unittest.mock import MagicMock, patch -from azurefunctions.extensions.base import ( - RequestTrackerMeta, - ResponseLabels, - ResponseTrackerMeta, -) -from azurefunctions.extensions.mcp_server.starlette import ( - RequestSynchronizer, - WebApp, - WebServer, -) +from azurefunctions.extensions.base import (RequestTrackerMeta, ResponseLabels, + ResponseTrackerMeta) +from azurefunctions.extensions.http.starlette import ( + RequestSynchronizer, WebApp, WebServer) from starlette.applications import Starlette - from starlette.requests import Request as StarletteRequest -from starlette.responses import Response as StarletteResponse from starlette.responses import FileResponse as StarletteFileResponse from starlette.responses import HTMLResponse as StarletteHTMLResponse from starlette.responses import JSONResponse as StarletteJSONResponse from starlette.responses import PlainTextResponse as StarlettePlainTextResponse from starlette.responses import RedirectResponse as StarletteRedirectResponse +from starlette.responses import Response as StarletteResponse from starlette.responses import StreamingResponse as StarletteStreamingResponse @@ -182,9 +175,9 @@ def test_sync_route_params_missing_path_params(self): class TestExtensionClasses(unittest.TestCase): def test_request(self): - from azurefunctions.extensions.mcp_server.starlette.web import Request + from azurefunctions.extensions.http.starlette.web import Request - self.assertEqual(RequestTrackerMeta.get_request_type(), StarletteRequest) + self.assertEqual(RequestTrackerMeta.get_request_type(), Request) self.assertTrue( isinstance(RequestTrackerMeta.get_synchronizer(), RequestSynchronizer) ) diff --git a/azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/__init__.py b/azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/__init__.py deleted file mode 100644 index 1ad427c..0000000 --- a/azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/__init__.py +++ /dev/null @@ -1,29 +0,0 @@ -__path__ = __import__("pkgutil").extend_path(__path__, __name__) - -from .starlette import (Request, Response, - FileResponse, - HTMLResponse, - JSONResponse, - PlainTextResponse, - RedirectResponse, - StreamingResponse, RequestSynchronizer, WebApp, WebServer - ) -from .decorators import (MCPFunctionApp, FunctionRegister) - -__all__ = [ - "WebServer", - "WebApp", - "Request", - "Response", - "RequestSynchronizer", - "StreamingResponse", - "HTMLResponse", - "PlainTextResponse", - "RedirectResponse", - "JSONResponse", - "FileResponse", - "MCPFunctionApp", - "FunctionRegister" -] - -__version__ = "0.0.1b1" diff --git a/azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/decorators/__init__.py b/azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/decorators/__init__.py deleted file mode 100644 index e8d5333..0000000 --- a/azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/decorators/__init__.py +++ /dev/null @@ -1,6 +0,0 @@ -from .mcp_app import MCPFunctionApp, FunctionRegister - -__all__ = [ - "MCPFunctionApp", - "FunctionRegister" -] \ No newline at end of file diff --git a/azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/decorators/mcp_app.py b/azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/decorators/mcp_app.py deleted file mode 100644 index 8a76f4e..0000000 --- a/azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/decorators/mcp_app.py +++ /dev/null @@ -1,78 +0,0 @@ -from typing import Union, Any -from azure.functions import (FunctionRegister, TriggerApi, BindingApi, -AuthLevel, HttpMethod, ExternalHttpFunctionApp) -from azure.functions.decorators.function_app import HttpFunctionsAuthLevelMixin -from azurefunctions.extensions.mcp_server.starlette import Request, Response - -from mcp.server.fastmcp.server import FastMCP -from mcp.server.sse import SseServerTransport - -try: - from azure.functions import SettingsApi -except ImportError: # backwards compatibility path - class SettingsApi: - """Backwards compatibility mock of SettingsApi.""" - - pass - - -class MCPFunctionApp(FastMCP, TriggerApi, FunctionRegister, HttpFunctionsAuthLevelMixin): - """MCP Functions (DF) Blueprint container. - - It allows functions to be declared via trigger and binding decorators, - but does not automatically index/register these functions. - """ - - def __init__(self, - auth_level: Union[AuthLevel, str] = AuthLevel.FUNCTION, - name: str | None = None, instructions: str | None = None, *args, **settings: Any): - """Instantiate a Durable Functions app with which to register Functions. - - Parameters - ---------- - http_auth_level: Union[AuthLevel, str] - Authorization level required for Function invocation. - Defaults to AuthLevel.Function. - - Returns - ------- - DFApp - New instance of a Durable Functions app - """ - FunctionRegister.__init__(self, auth_level=auth_level, *args, **settings) - FastMCP.__init__(self, name, instructions, **settings) - self._add_http_app(auth_level) - - - def _add_http_app(self, auth_level) -> None: - """Add an Asgi app integrated mcp_server function. - - :param http_middleware: :class:`WsgiMiddleware` - or class:`AsgiMiddleware` instance. - - :return: None - """ - - sse = SseServerTransport("/messages/") - - @self.function_name(name="sse") - @self.route(trigger_arg_name="request" , methods=(method for method in HttpMethod), - auth_level=auth_level, - route="sse") - async def http_app_func(request: Request): - async with sse.connect_sse( - request.scope, request.receive, request._send - ) as streams: - await self._mcp_server.run( - streams[0], - streams[1], - self._mcp_server.create_initialization_options(), - ) - - @self.function_name(name="messages") - @self.route(trigger_arg_name="request", methods=(method for method in HttpMethod), - auth_level=auth_level, - route="messages") - async def http_messages_func(request: Request): - return await sse.handle_post_message(request.scope, request.receive, request._send) - diff --git a/azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/starlette/__init__.py b/azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/starlette/__init__.py deleted file mode 100644 index 65174da..0000000 --- a/azurefunctions-extensions-mcp-server/azurefunctions/extensions/mcp_server/starlette/__init__.py +++ /dev/null @@ -1,29 +0,0 @@ -from starlette.requests import (Request) -from starlette.responses import (Response) - -from starlette.responses import ( - FileResponse, - HTMLResponse, - JSONResponse, - PlainTextResponse, - RedirectResponse, - StreamingResponse, -) - -from .web import RequestSynchronizer, WebApp, WebServer - -__all__ = [ - "WebServer", - "WebApp", - "Request", - "Response", - "RequestSynchronizer", - "StreamingResponse", - "HTMLResponse", - "PlainTextResponse", - "RedirectResponse", - "JSONResponse", - "FileResponse", -] - -__version__ = "1.0.0b1"