Revert removing the unused imports (#385)

python/sglang/lang/chat_template.py

@@ -1,6 +1,6 @@
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 from enum import Enum, auto
-from typing import Callable, Dict, List, Tuple
+from typing import Callable, Dict, List, Optional, Tuple
 
 
 class ChatTemplateStyle(Enum):

python/sglang/lang/compiler.py

@@ -5,7 +5,13 @@ from typing import List, Union
 
 from sglang.global_config import global_config
 from sglang.lang.interpreter import ProgramState, StreamExecutor, pin_program
-from sglang.lang.ir import SglArgument, SglExpr, SglSamplingParams, SglVariable
+from sglang.lang.ir import (
+    SglArgument,
+    SglConstantText,
+    SglExpr,
+    SglSamplingParams,
+    SglVariable,
+)
 
 
 def compile_func(function, backend):

python/sglang/lang/interpreter.py

@@ -7,7 +7,7 @@ import threading
 import uuid
 from concurrent.futures import ThreadPoolExecutor
 from contextlib import contextmanager
-from typing import Any, Callable, Dict, List, Optional
+from typing import Any, Callable, Dict, List, Optional, Union
 
 import tqdm
 
@@ -18,6 +18,7 @@ from sglang.lang.ir import (
     SglConstantText,
     SglExpr,
     SglExprList,
+    SglFunction,
     SglGen,
     SglImage,
     SglRoleBegin,

python/sglang/lang/ir.py

@@ -472,4 +472,4 @@ class SglCommitLazy(SglExpr):
         super().__init__()
 
     def __repr__(self):
-        return "CommitLazy()"
+        return f"CommitLazy()"

python/sglang/lang/tracer.py

@@ -1,16 +1,20 @@
 """Tracing a program."""
 
 import uuid
-from typing import Any, Dict, List, Optional
+from typing import Any, Callable, Dict, List, Optional, Union
 
 from sglang.backend.base_backend import BaseBackend
+from sglang.global_config import global_config
 from sglang.lang.interpreter import ProgramState, ProgramStateGroup
 from sglang.lang.ir import (
     SglArgument,
+    SglCommitLazy,
+    SglConcateAndAppend,
     SglConstantText,
     SglExpr,
     SglExprList,
     SglFork,
+    SglFunction,
     SglGen,
     SglGetForkItem,
     SglRoleBegin,