chore: remove dependency on termcolor types (#967)

Signed-off-by: Louis Mandel <[email protected]>

Author: mandel

src/pdl/pdl_scheduler.py

@@ -1,14 +1,13 @@
 from typing import Any, Optional
 
 from termcolor import colored
-from termcolor._types import Color
 
 from .pdl_ast import BlockKind
 from .pdl_utils import stringify
 
 
 def color_of(kind: BlockKind):
-    color: Optional[Color]
+    color: Optional[str]
     match kind:
         case BlockKind.FUNCTION:
             color = None
@@ -52,7 +51,7 @@ def color_of(kind: BlockKind):
 
 
 def color_of_role(role: str):
-    color: Optional[Color] = None
+    color: Optional[str] = None
     match role:
         case "assistant":
             color = "green"