chore: enable pyupgrade fixes (#1806)

# What does this PR do?

The goal of this PR is code base modernization.

Schema reflection code needed a minor adjustment to handle UnionTypes
and collections.abc.AsyncIterator. (Both are preferred for latest Python
releases.)

Note to reviewers: almost all changes here are automatically generated
by pyupgrade. Some additional unused imports were cleaned up. The only
change worth of note can be found under `docs/openapi_generator` and
`llama_stack/strong_typing/schema.py` where reflection code was updated
to deal with "newer" types.

Signed-off-by: Ihar Hrachyshka <ihar.hrachyshka@gmail.com>

llama_stack/providers/inline/tool_runtime/code_interpreter/__init__.py

@@ -4,14 +4,14 @@
 # This source code is licensed under the terms described in the LICENSE file in
 # the root directory of this source tree.
 
-from typing import Any, Dict
+from typing import Any
 
 from .config import CodeInterpreterToolConfig
 
 __all__ = ["CodeInterpreterToolConfig", "CodeInterpreterToolRuntimeImpl"]
 
 
-async def get_provider_impl(config: CodeInterpreterToolConfig, _deps: Dict[str, Any]):
+async def get_provider_impl(config: CodeInterpreterToolConfig, _deps: dict[str, Any]):
     from .code_interpreter import CodeInterpreterToolRuntimeImpl
 
     impl = CodeInterpreterToolRuntimeImpl(config)

llama_stack/providers/inline/tool_runtime/code_interpreter/code_execution.py

@@ -18,7 +18,6 @@ from dataclasses import dataclass
 from datetime import datetime
 from io import BytesIO
 from pathlib import Path
-from typing import List
 
 from PIL import Image
 
@@ -45,7 +44,7 @@ except:
 """
 
 
-def generate_bwrap_command(bind_dirs: List[str]) -> str:
+def generate_bwrap_command(bind_dirs: list[str]) -> str:
     """
     Generate the bwrap command string for binding all
     directories in the current directory read-only.
@@ -71,7 +70,7 @@ class CodeExecutionContext:
 
 @dataclass
 class CodeExecutionRequest:
-    scripts: List[str]
+    scripts: list[str]
     only_last_cell_stdouterr: bool = True
     only_last_cell_fail: bool = True
     seed: int = 0

llama_stack/providers/inline/tool_runtime/code_interpreter/code_interpreter.py

@@ -9,7 +9,7 @@ import asyncio
 import logging
 import os
 import tempfile
-from typing import Any, Dict, Optional
+from typing import Any
 
 from llama_stack.apis.common.content_types import URL
 from llama_stack.apis.tools import (
@@ -46,7 +46,7 @@ class CodeInterpreterToolRuntimeImpl(ToolsProtocolPrivate, ToolRuntime):
         return
 
     async def list_runtime_tools(
-        self, tool_group_id: Optional[str] = None, mcp_endpoint: Optional[URL] = None
+        self, tool_group_id: str | None = None, mcp_endpoint: URL | None = None
     ) -> ListToolDefsResponse:
         return ListToolDefsResponse(
             data=[
@@ -64,7 +64,7 @@ class CodeInterpreterToolRuntimeImpl(ToolsProtocolPrivate, ToolRuntime):
             ]
         )
 
-    async def invoke_tool(self, tool_name: str, kwargs: Dict[str, Any]) -> ToolInvocationResult:
+    async def invoke_tool(self, tool_name: str, kwargs: dict[str, Any]) -> ToolInvocationResult:
         script = kwargs["code"]
         # Use environment variable to control bwrap usage
         force_disable_bwrap = os.environ.get("DISABLE_CODE_SANDBOX", "").lower() in ("1", "true", "yes")

llama_stack/providers/inline/tool_runtime/code_interpreter/config.py

@@ -4,12 +4,12 @@
 # This source code is licensed under the terms described in the LICENSE file in
 # the root directory of this source tree.
 
-from typing import Any, Dict
+from typing import Any
 
 from pydantic import BaseModel
 
 
 class CodeInterpreterToolConfig(BaseModel):
     @classmethod
-    def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> Dict[str, Any]:
+    def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict[str, Any]:
         return {}

llama_stack/providers/inline/tool_runtime/code_interpreter/utils.py

@@ -15,7 +15,7 @@ def get_code_env_prefix() -> str:
     global CODE_ENV_PREFIX
 
     if CODE_ENV_PREFIX is None:
-        with open(CODE_ENV_PREFIX_FILE, "r") as f:
+        with open(CODE_ENV_PREFIX_FILE) as f:
             CODE_ENV_PREFIX = f.read()
 
     return CODE_ENV_PREFIX