Simplify common structure, remove old sglang file

Author: Aphoh

components/backends/sglang/src/dynamo/sglang/args.py

@@ -15,6 +15,7 @@
 from sglang.srt.server_args import ServerArgs
 
 from dynamo._core import get_reasoning_parser_names, get_tool_parser_names
+from dynamo.common.config_dump import register_encoder
 from dynamo.runtime.logging import configure_dynamo_logging
 from dynamo.sglang import __version__
 
@@ -77,7 +78,7 @@ class DynamoArgs:
 
     # preprocessing options
     use_sglang_tokenizer: bool = False
-    dump_config_to: Optional[str] = False
+    dump_config_to: Optional[str] = None
 
 
 class DisaggregationMode(Enum):
@@ -101,6 +102,20 @@ def _set_serving_strategy(self):
             return DisaggregationMode.DECODE
 
 
+# Register SGLang-specific encoders with the shared system
+@register_encoder(Config)
+def _preprocess_for_encode_config(
+    config: Config,
+) -> Dict[str, Any]:  # pyright: ignore[reportUnusedFunction]
+    return {
+        "server_args": config.server_args,
+        "dynamo_args": config.dynamo_args,
+        "serving_mode": config.serving_mode.value
+        if config.serving_mode is not None
+        else "None",
+    }
+
+
 def _set_parser(
     sglang_str: Optional[str],
     dynamo_str: Optional[str],

components/backends/sglang/src/dynamo/sglang/main.py

@@ -11,7 +11,6 @@
 import uvloop
 from sglang.srt.utils import get_ip
 
-import dynamo.sglang.utils.debug_handler  # noqa: F401
 from dynamo.common.config_dump import dump_config
 from dynamo.llm import ZmqKvEventPublisher, ZmqKvEventPublisherConfig
 from dynamo.runtime import DistributedRuntime, dynamo_worker

components/backends/sglang/src/dynamo/sglang/utils/debug_handler.py

@@ -2,11 +2,15 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import argparse
+import dataclasses
+import functools
+import json
 import logging
 import pathlib
+from enum import Enum
 from typing import Any, Dict, Optional
 
-from dynamo.common.utils import canonical_json_encoder, get_dynamo_version
+from dynamo.common import __version__
 
 from .environment import get_environment_vars
 from .system_info import get_runtime_info, get_system_info
@@ -86,7 +90,7 @@ def get_config_dump(config: Any, extra_info: Dict[str, str] = {}) -> str:
             "environment": get_environment_vars(),
             "config": config,
             "runtime_info": get_runtime_info(),
-            "dynamo_version": get_dynamo_version(),
+            "dynamo_version": __version__,
         }
 
         # Add common versions
@@ -126,3 +130,57 @@ def add_config_dump_args(parser: argparse.ArgumentParser):
         default=None,
         help="Dump debug config to the specified file path. If not specified, the config will be dumped to stdout at INFO level.",
     )
+
+
+@functools.singledispatch
+def _preprocess_for_encode(obj: object) -> object:
+    """
+    Single dispatch function for preprocessing objects before JSON encoding.
+
+    This function should be extended using @register_encoder decorator
+    for backend-specific types.
+    """
+    if dataclasses.is_dataclass(obj) and not isinstance(obj, type):
+        return dataclasses.asdict(obj)
+    logger.warning(f"Unknown type {type(obj)}, using __dict__ or str(obj)")
+    if hasattr(obj, "__dict__"):
+        return obj.__dict__
+    return str(obj)
+
+
+def register_encoder(type_class):
+    """
+    Decorator to register custom encoders for specific types.
+
+    Usage:
+        @register_encoder(MyClass)
+        def encode_my_class(obj: MyClass):
+            return {"field": obj.field}
+    """
+    logger.info(f"Registering encoder for {type_class}")
+    return _preprocess_for_encode.register(type_class)
+
+
+@register_encoder(set)
+def _preprocess_for_encode_set(
+    obj: set,
+) -> list:  # pyright: ignore[reportUnusedFunction]
+    return sorted(list(obj))
+
+
+@register_encoder(Enum)
+def _preprocess_for_encode_enum(
+    obj: Enum,
+) -> str:  # pyright: ignore[reportUnusedFunction]
+    return str(obj)
+
+
+# Create a canonical JSON encoder with consistent formatting
+canonical_json_encoder = json.JSONEncoder(
+    ensure_ascii=False,
+    separators=(",", ":"),
+    allow_nan=False,
+    sort_keys=True,
+    indent=None,
+    default=_preprocess_for_encode,
+)