[Enhancement]: Display SDK Version in Deployed Applications (openapi.json)

agenta-cli/agenta/cli/main.py

@@ -47,9 +47,9 @@ def check_latest_version() -> Union[str, None]:
 
 
 def notify_update(available_version: str):
-    import pkg_resources
+    import importlib.metadata
 
-    installed_version = pkg_resources.get_distribution("agenta").version
+    installed_version = importlib.metadata.version("agenta")
     if available_version > installed_version:
         click.echo(
             click.style(

agenta-cli/agenta/sdk/decorators/routing.py

@@ -18,6 +18,7 @@
 from agenta.sdk.context.routing import routing_context_manager, routing_context
 from agenta.sdk.context.tracing import tracing_context
 from agenta.sdk.router import router
+from agenta.sdk.utils import helpers
 from agenta.sdk.utils.exceptions import suppress
 from agenta.sdk.utils.logging import log
 from agenta.sdk.types import (
@@ -287,6 +288,9 @@ async def wrapper_deployed(*args, **kwargs) -> Any:
         app.openapi_schema = None  # Forces FastAPI to re-generate the schema
         openapi_schema = app.openapi()
 
+        # Inject the current version of the SDK into the openapi_schema
+        openapi_schema["agenta_sdk"] = {"version": helpers.get_current_version()}
+
         for route in entrypoint.routes:
             self.override_schema(
                 openapi_schema=openapi_schema,

agenta-cli/agenta/sdk/utils/helpers.py

@@ -0,0 +1,8 @@
+import importlib.metadata
+
+
+def get_current_version():
+    """Returns the current version of Agenta's SDK."""
+
+    version = importlib.metadata.version("agenta")
+    return version

agenta-cli/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "agenta"
-version = "0.28.0"
+version = "0.28.2a1"
 description = "The SDK for agenta is an open-source LLMOps platform."
 readme = "README.md"
 authors = ["Mahmoud Mabrouk <[email protected]>"]