feat(core): APP use new SDK component (#1050)

dbgpt/util/tracer/__init__.py

@@ -1,23 +1,23 @@
 from dbgpt.util.tracer.base import (
-    SpanType,
     Span,
-    SpanTypeRunName,
-    Tracer,
     SpanStorage,
     SpanStorageType,
+    SpanType,
+    SpanTypeRunName,
+    Tracer,
     TracerContext,
 )
 from dbgpt.util.tracer.span_storage import (
-    MemorySpanStorage,
     FileSpanStorage,
+    MemorySpanStorage,
     SpanStorageContainer,
 )
 from dbgpt.util.tracer.tracer_impl import (
-    root_tracer,
-    trace,
-    initialize_tracer,
     DefaultTracer,
     TracerManager,
+    initialize_tracer,
+    root_tracer,
+    trace,
 )
 
 __all__ = [

dbgpt/util/tracer/base.py

@@ -1,13 +1,13 @@
 from __future__ import annotations
 
-from typing import Dict, Callable, Optional, List
-from dataclasses import dataclass
-from abc import ABC, abstractmethod
-from enum import Enum
 import uuid
+from abc import ABC, abstractmethod
+from dataclasses import dataclass
 from datetime import datetime
+from enum import Enum
+from typing import Callable, Dict, List, Optional
 
-from dbgpt.component import BaseComponent, SystemApp, ComponentType
+from dbgpt.component import BaseComponent, ComponentType, SystemApp
 
 
 class SpanType(str, Enum):

dbgpt/util/tracer/span_storage.py

@@ -1,12 +1,12 @@
-import os
-import json
-import time
 import datetime
-import threading
-import queue
+import json
 import logging
-from typing import Optional, List
+import os
+import queue
+import threading
+import time
 from concurrent.futures import Executor, ThreadPoolExecutor
+from typing import List, Optional
 
 from dbgpt.component import SystemApp
 from dbgpt.util.tracer.base import Span, SpanStorage

dbgpt/util/tracer/tests/test_base.py

@@ -1,8 +1,7 @@
 from typing import Dict
+
 from dbgpt.component import SystemApp
-
-from dbgpt.util.tracer import Span, SpanType, SpanStorage, Tracer
-
+from dbgpt.util.tracer import Span, SpanStorage, SpanType, Tracer
 
 # Mock implementations
 

dbgpt/util/tracer/tests/test_span_storage.py

@@ -1,18 +1,19 @@
-import os
-import pytest
 import asyncio
 import json
+import os
 import tempfile
 import time
-from unittest.mock import patch
 from datetime import datetime, timedelta
+from unittest.mock import patch
+
+import pytest
 
 from dbgpt.util.tracer import (
-    SpanStorage,
     FileSpanStorage,
     Span,
-    SpanType,
+    SpanStorage,
     SpanStorageContainer,
+    SpanType,
 )
 
 

dbgpt/util/tracer/tests/test_tracer_impl.py

@@ -1,14 +1,15 @@
 import pytest
-from dbgpt.util.tracer import (
-    Span,
-    SpanStorageType,
-    SpanStorage,
-    DefaultTracer,
-    TracerManager,
-    Tracer,
-    MemorySpanStorage,
-)
+
 from dbgpt.component import SystemApp
+from dbgpt.util.tracer import (
+    DefaultTracer,
+    MemorySpanStorage,
+    Span,
+    SpanStorage,
+    SpanStorageType,
+    Tracer,
+    TracerManager,
+)
 
 
 @pytest.fixture

dbgpt/util/tracer/tracer_cli.py

@@ -1,10 +1,12 @@
-import os
-import click
-import logging
 import glob
 import json
+import logging
+import os
 from datetime import datetime
-from typing import Iterable, Dict, Callable
+from typing import Callable, Dict, Iterable
+
+import click
+
 from dbgpt.configs.model_config import LOGDIR
 from dbgpt.util.tracer import SpanType, SpanTypeRunName
 

dbgpt/util/tracer/tracer_impl.py

@@ -1,22 +1,21 @@
-from typing import Dict, Optional
-from contextvars import ContextVar
-from functools import wraps
 import asyncio
 import inspect
 import logging
+from contextvars import ContextVar
+from functools import wraps
+from typing import Dict, Optional
 
-
-from dbgpt.component import SystemApp, ComponentType
+from dbgpt.component import ComponentType, SystemApp
+from dbgpt.util.module_utils import import_from_checked_string
 from dbgpt.util.tracer.base import (
-    SpanType,
     Span,
-    Tracer,
     SpanStorage,
     SpanStorageType,
+    SpanType,
+    Tracer,
     TracerContext,
 )
 from dbgpt.util.tracer.span_storage import MemorySpanStorage
-from dbgpt.util.module_utils import import_from_checked_string
 
 logger = logging.getLogger(__name__)
 

dbgpt/util/tracer/tracer_middleware.py

@@ -4,8 +4,8 @@ from contextvars import ContextVar
 from starlette.middleware.base import BaseHTTPMiddleware
 from starlette.requests import Request
 from starlette.types import ASGIApp
-from dbgpt.util.tracer import TracerContext, Tracer
 
+from dbgpt.util.tracer import Tracer, TracerContext
 
 _DEFAULT_EXCLUDE_PATHS = ["/api/controller/heartbeat"]