Skip to content

Commit 200309b

Browse files
committed
fix(tls): remove strong deps
1 parent b682bb2 commit 200309b

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

veadk/integrations/ve_tls/ve_tls.py

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@
1515
import os
1616

1717
from veadk.consts import DEFAULT_TLS_LOG_PROJECT_NAME, DEFAULT_TLS_TRACING_INSTANCE_NAME
18-
from veadk.integrations.ve_tls.utils import ve_tls_request
1918
from veadk.utils.logger import get_logger
2019

2120
logger = get_logger(__name__)
@@ -30,11 +29,15 @@ def __init__(
3029
):
3130
try:
3231
from volcengine.tls.TLSService import TLSService
32+
33+
from veadk.integrations.ve_tls.utils import ve_tls_request
3334
except ImportError:
3435
raise ImportError(
3536
"Please install volcengine SDK before init VeTLS: pip install volcengine"
3637
)
3738

39+
self._ve_tls_request = ve_tls_request
40+
3841
self.access_key = (
3942
access_key if access_key else os.getenv("VOLCENGINE_ACCESS_KEY", "")
4043
)
@@ -68,7 +71,7 @@ def get_project_id_by_name(self, project_name: str) -> str:
6871

6972
try:
7073
res = None
71-
res = ve_tls_request(
74+
res = self._ve_tls_request(
7275
client=self._client,
7376
api="DescribeProjects",
7477
body=request_body,
@@ -100,7 +103,7 @@ def create_log_project(self, project_name: str) -> str:
100103
"Tags": [{"Key": "provider", "Value": "VeADK"}],
101104
}
102105
try:
103-
res = ve_tls_request(
106+
res = self._ve_tls_request(
104107
client=self._client, api="CreateProject", body=request_body
105108
)
106109

@@ -123,7 +126,7 @@ def get_trace_instance_by_name(self, log_project_id: str, trace_instance_name: s
123126
"TraceInstanceName": trace_instance_name,
124127
}
125128
try:
126-
res = ve_tls_request(
129+
res = self._ve_tls_request(
127130
client=self._client,
128131
api="DescribeTraceInstances",
129132
body=request_body,
@@ -155,7 +158,7 @@ def create_tracing_instance(self, log_project_id: str, trace_instance_name: str)
155158

156159
try:
157160
res = None
158-
res = ve_tls_request(
161+
res = self._ve_tls_request(
159162
client=self._client,
160163
api="CreateTraceInstance",
161164
body=request_body,
@@ -170,7 +173,7 @@ def create_tracing_instance(self, log_project_id: str, trace_instance_name: str)
170173
)
171174

172175
# after creation, get the trace instance details
173-
res = ve_tls_request(
176+
res = self._ve_tls_request(
174177
client=self._client,
175178
api="DescribeTraceInstance",
176179
body={"TraceInstanceID": res["TraceInstanceID"]},

0 commit comments

Comments
 (0)