Skip to content

Commit e4af13c

Browse files
committed
fix parameters
1 parent 5a4ff8b commit e4af13c

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

veadk/integrations/ve_tls/utils.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
from typing import Any, Literal
1818

1919
import httpx
20+
from veadk.utils.logger import get_logger
2021
from volcengine.ApiInfo import ApiInfo
2122
from volcengine.auth.SignerV4 import SignerV4
2223
from volcengine.tls.const import (
@@ -28,8 +29,6 @@
2829
)
2930
from volcengine.tls.TLSService import TLSService
3031

31-
from veadk.utils.logger import get_logger
32-
3332
logger = get_logger(__name__)
3433

3534
HEADER_API_VERSION = "x-tls-apiversion"

veadk/integrations/ve_tls/ve_tls.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,9 +28,8 @@ def __init__(
2828
region: str = "cn-beijing",
2929
):
3030
try:
31-
from volcengine.tls.TLSService import TLSService
32-
3331
from veadk.integrations.ve_tls.utils import ve_tls_request
32+
from volcengine.tls.TLSService import TLSService
3433
except ImportError:
3534
raise ImportError(
3635
"Please install volcengine SDK before init VeTLS: pip install volcengine"
@@ -177,10 +176,14 @@ def create_tracing_instance(self, log_project_id: str, trace_instance_name: str)
177176
res = self._ve_tls_request(
178177
client=self._client,
179178
api="DescribeTraceInstance",
180-
body={"TraceInstanceID": res["TraceInstanceID"]},
179+
body={"TraceInstanceId": res["TraceInstanceId"]},
181180
method="GET",
182181
)
183182

183+
logger.info(
184+
f"Create tracing instance finished, tracing instance ID: {res['TraceInstanceId']}"
185+
)
186+
184187
return res
185188
except KeyError:
186189
raise ValueError(f"Failed to create tracing instance: {res}")

0 commit comments

Comments
 (0)