fix imports

Author: nikhilc-microsoft

libraries/microsoft-agents-a365-observability-extensions-langchain/microsoft_agents_a365/observability/extensions/langchain/tracer.py

@@ -14,6 +14,13 @@
 
 from langchain_core.tracers import BaseTracer, LangChainTracer
 from langchain_core.tracers.schemas import Run
+from microsoft_agents_a365.observability.core.inference_operation_type import InferenceOperationType
+from microsoft_agents_a365.observability.core.utils import (
+    DictWithLock,
+    as_utc_nano,
+    flatten,
+    record_exception,
+)
 from opentelemetry import context as context_api
 from opentelemetry import trace as trace_api
 from opentelemetry.context import (
@@ -23,7 +30,6 @@
 from opentelemetry.trace import Span
 from opentelemetry.util.types import AttributeValue
 
-from microsoft_agents_a365.observability.core.inference_operation_type import InferenceOperationType
 from microsoft_agents_a365.observability.extensions.langchain.utils import (
     IGNORED_EXCEPTION_PATTERNS,
     add_operation_type,
@@ -38,12 +44,6 @@
     token_counts,
     tools,
 )
-from microsoft_agents_a365.observability.core.wrappers.utils import (
-    DictWithLock,
-    as_utc_nano,
-    flatten,
-    record_exception,
-)
 
 logger = logging.getLogger(__name__)
 logger.addHandler(logging.NullHandler())

libraries/microsoft-agents-a365-observability-extensions-langchain/microsoft_agents_a365/observability/extensions/langchain/utils.py

@@ -27,7 +27,7 @@
     SESSION_ID_KEY,
 )
 from microsoft_agents_a365.observability.core.inference_operation_type import InferenceOperationType
-from microsoft_agents_a365.observability.core.wrappers.utils import (
+from microsoft_agents_a365.observability.core.utils import (
     get_first_value,
     safe_json_dumps,
     stop_on_exception,

libraries/microsoft-agents-a365-observability-extensions-openai/microsoft_agents_a365/observability/extensions/openai/trace_processor.py

@@ -28,7 +28,7 @@
     GEN_AI_SYSTEM_KEY,
     INVOKE_AGENT_OPERATION_NAME,
 )
-from microsoft_agents_a365.observability.core.wrappers.utils import as_utc_nano, safe_json_dumps
+from microsoft_agents_a365.observability.core.utils import as_utc_nano, safe_json_dumps
 from opentelemetry import trace as ot_trace
 from opentelemetry.context import attach, detach
 from opentelemetry.trace import Span as OtelSpan

libraries/microsoft-agents-a365-observability-extensions-openai/microsoft_agents_a365/observability/extensions/openai/utils.py

@@ -37,7 +37,7 @@
     GEN_AI_USAGE_INPUT_TOKENS_KEY,
     GEN_AI_USAGE_OUTPUT_TOKENS_KEY,
 )
-from microsoft_agents_a365.observability.core.wrappers.utils import safe_json_dumps
+from microsoft_agents_a365.observability.core.utils import safe_json_dumps
 from opentelemetry.trace import (
     Status,
     StatusCode,

libraries/microsoft-agents-a365-observability-extensions-semantickernel/microsoft_agents_a365/observability/extensions/semantickernel/span_processor.py

@@ -2,11 +2,10 @@
 
 # Custom Span Processor
 
-from opentelemetry.sdk.trace.export import SpanProcessor
-
 from microsoft_agents_a365.observability.core.constants import GEN_AI_OPERATION_NAME_KEY
 from microsoft_agents_a365.observability.core.inference_operation_type import InferenceOperationType
-from microsoft_agents_a365.observability.core.wrappers.utils import extract_model_name
+from microsoft_agents_a365.observability.core.utils import extract_model_name
+from opentelemetry.sdk.trace.export import SpanProcessor
 
 
 class SemanticKernelSpanProcessor(SpanProcessor):