Skip to content

Commit 182a78a

Browse files
authored
[AINode] Refactor thrift integration (apache#16323)
1 parent 64c6900 commit 182a78a

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

68 files changed

+215
-187
lines changed

iotdb-core/ainode/.gitignore

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# generated by Thrift
2-
/ainode/thrift/
2+
/iotdb/thrift/
33

44
# generated by maven
55
/ainode/conf/
File renamed without changes.
File renamed without changes.

iotdb-core/ainode/ainode/core/ai_node.py renamed to iotdb-core/ainode/iotdb/ainode/core/ai_node.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -22,19 +22,19 @@
2222

2323
import psutil
2424

25-
from ainode.core.config import AINodeDescriptor
26-
from ainode.core.constant import AINODE_SYSTEM_FILE_NAME
27-
from ainode.core.log import Logger
28-
from ainode.core.rpc.client import ClientManager
29-
from ainode.core.rpc.handler import AINodeRPCServiceHandler
30-
from ainode.core.rpc.service import AINodeRPCService
31-
from ainode.thrift.common.ttypes import (
25+
from iotdb.ainode.core.config import AINodeDescriptor
26+
from iotdb.ainode.core.constant import AINODE_SYSTEM_FILE_NAME
27+
from iotdb.ainode.core.log import Logger
28+
from iotdb.ainode.core.rpc.client import ClientManager
29+
from iotdb.ainode.core.rpc.handler import AINodeRPCServiceHandler
30+
from iotdb.ainode.core.rpc.service import AINodeRPCService
31+
from iotdb.thrift.common.ttypes import (
3232
TAINodeConfiguration,
3333
TAINodeLocation,
3434
TEndPoint,
3535
TNodeResource,
3636
)
37-
from ainode.thrift.confignode.ttypes import TNodeVersionInfo
37+
from iotdb.thrift.confignode.ttypes import TNodeVersionInfo
3838

3939
logger = Logger()
4040

iotdb-core/ainode/ainode/core/config.py renamed to iotdb-core/ainode/iotdb/ainode/core/config.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
#
1818
import os
1919

20-
from ainode.core.constant import (
20+
from iotdb.ainode.core.constant import (
2121
AINODE_BUILD_INFO,
2222
AINODE_BUILTIN_MODELS_DIR,
2323
AINODE_CLUSTER_INGRESS_ADDRESS,
@@ -47,10 +47,10 @@
4747
AINODE_THRIFT_COMPRESSION_ENABLED,
4848
AINODE_VERSION_INFO,
4949
)
50-
from ainode.core.exception import BadNodeUrlError
51-
from ainode.core.log import Logger
52-
from ainode.core.util.decorator import singleton
53-
from ainode.thrift.common.ttypes import TEndPoint
50+
from iotdb.ainode.core.exception import BadNodeUrlError
51+
from iotdb.ainode.core.log import Logger
52+
from iotdb.ainode.core.util.decorator import singleton
53+
from iotdb.thrift.common.ttypes import TEndPoint
5454

5555
logger = Logger()
5656

iotdb-core/ainode/ainode/core/constant.py renamed to iotdb-core/ainode/iotdb/ainode/core/constant.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,8 @@
2121
from enum import Enum
2222
from typing import List
2323

24-
from ainode.core.model.model_enums import BuiltInModelType
25-
from ainode.thrift.common.ttypes import TEndPoint
24+
from iotdb.ainode.core.model.model_enums import BuiltInModelType
25+
from iotdb.thrift.common.ttypes import TEndPoint
2626

2727
AINODE_VERSION_INFO = "UNKNOWN"
2828
AINODE_BUILD_INFO = "UNKNOWN"

iotdb-core/ainode/ainode/core/exception.py renamed to iotdb-core/ainode/iotdb/ainode/core/exception.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,10 @@
1717
#
1818
import re
1919

20-
from ainode.core.constant import MODEL_CONFIG_FILE_IN_YAML, MODEL_WEIGHTS_FILE_IN_PT
20+
from iotdb.ainode.core.constant import (
21+
MODEL_CONFIG_FILE_IN_YAML,
22+
MODEL_WEIGHTS_FILE_IN_PT,
23+
)
2124

2225

2326
class _BaseError(Exception):

iotdb-core/ainode/ainode/core/inference/dispatcher/__init__.py renamed to iotdb-core/ainode/iotdb/ainode/core/inference/__init__.py

File renamed without changes.

iotdb-core/ainode/ainode/core/inference/pool_scheduler/__init__.py renamed to iotdb-core/ainode/iotdb/ainode/core/inference/dispatcher/__init__.py

File renamed without changes.

0 commit comments

Comments
 (0)