isagellm-core 0.2.2.5__cp311-none-any.whl → 0.3.0.2__cp311-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {isagellm_core-0.2.2.5.dist-info → isagellm_core-0.3.0.2.dist-info}/METADATA +6 -3
- {isagellm_core-0.2.2.5.dist-info → isagellm_core-0.3.0.2.dist-info}/RECORD +21 -17
- {isagellm_core-0.2.2.5.dist-info → isagellm_core-0.3.0.2.dist-info}/WHEEL +1 -1
- {isagellm_core-0.2.2.5.dist-info → isagellm_core-0.3.0.2.dist-info}/entry_points.txt +1 -0
- sagellm_core/__init__.py +13 -1
- sagellm_core/__init__.pyc +0 -0
- sagellm_core/__pycache__/__init__.cpython-311.pyc +0 -0
- sagellm_core/__pycache__/engine.cpython-311.pyc +0 -0
- sagellm_core/__pycache__/pd_executor.cpython-311.pyc +0 -0
- sagellm_core/__pycache__/runtime.cpython-311.pyc +0 -0
- sagellm_core/engine.pyc +0 -0
- sagellm_core/engines/__init__.py +8 -2
- sagellm_core/engines/__init__.pyc +0 -0
- sagellm_core/engines/__pycache__/__init__.cpython-311.pyc +0 -0
- sagellm_core/engines/__pycache__/cpu.cpython-311.pyc +0 -0
- sagellm_core/engines/__pycache__/hf_cuda.cpython-311.pyc +0 -0
- sagellm_core/engines/cpu.pyc +0 -0
- sagellm_core/engines/hf_cuda.pyc +0 -0
- sagellm_core/pd_executor.pyc +0 -0
- sagellm_core/runtime.pyc +0 -0
- {isagellm_core-0.2.2.5.dist-info → isagellm_core-0.3.0.2.dist-info}/top_level.txt +0 -0
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: isagellm-core
|
|
3
|
-
Version: 0.
|
|
4
|
-
Summary: sageLLM core runtime (
|
|
3
|
+
Version: 0.3.0.2
|
|
4
|
+
Summary: sageLLM core runtime with PD separation (MVP)
|
|
5
5
|
Author: IntelliStream Team
|
|
6
6
|
License: Proprietary - IntelliStream
|
|
7
7
|
Classifier: Development Status :: 3 - Alpha
|
|
@@ -13,8 +13,10 @@ Requires-Python: ==3.11.*
|
|
|
13
13
|
Description-Content-Type: text/markdown
|
|
14
14
|
Requires-Dist: pydantic>=2.0.0
|
|
15
15
|
Requires-Dist: pyyaml>=6.0.0
|
|
16
|
-
Requires-Dist: isagellm-protocol<0.2.0,>=0.1.0
|
|
16
|
+
Requires-Dist: isagellm-protocol<0.2.0,>=0.1.2.0
|
|
17
17
|
Requires-Dist: isagellm-backend<0.3.0,>=0.2.0.0
|
|
18
|
+
Requires-Dist: isagellm-comm<0.2.0,>=0.1.0
|
|
19
|
+
Requires-Dist: isagellm-kv-cache<0.2.0,>=0.1.0
|
|
18
20
|
Provides-Extra: dev
|
|
19
21
|
Requires-Dist: pytest>=7.0.0; extra == "dev"
|
|
20
22
|
Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
|
|
@@ -23,6 +25,7 @@ Requires-Dist: ruff>=0.8.0; extra == "dev"
|
|
|
23
25
|
Requires-Dist: mypy>=1.0.0; extra == "dev"
|
|
24
26
|
Requires-Dist: types-PyYAML>=6.0.0; extra == "dev"
|
|
25
27
|
Requires-Dist: pre-commit>=3.0.0; extra == "dev"
|
|
28
|
+
Requires-Dist: isage-pypi-publisher>=0.2.0; extra == "dev"
|
|
26
29
|
|
|
27
30
|
# sagellm-core
|
|
28
31
|
|
|
@@ -1,44 +1,48 @@
|
|
|
1
|
-
sagellm_core/__init__.py,sha256=
|
|
2
|
-
sagellm_core/__init__.pyc,sha256=
|
|
1
|
+
sagellm_core/__init__.py,sha256=OozZ7LilGRA4lRj-WS9O5vWeBEQerkXaH4A4megQZtw,3192
|
|
2
|
+
sagellm_core/__init__.pyc,sha256=5yAtX8HDcJbSZyIsj3xLQXbcSSRpqtmTlfgkDVNAmA4,3107
|
|
3
3
|
sagellm_core/__main__.pyc,sha256=48Ej1ycqV-z87qawGOTNBKNgY3EcbffUOOXrDecOR3g,384
|
|
4
4
|
sagellm_core/config.pyc,sha256=cCMLvMVb_s3HzPYSfqlbRWodQ4Uby8U962l1vmunT5Y,8453
|
|
5
5
|
sagellm_core/demo.pyc,sha256=0hZkAJch6ETjSDsPq8XkDLNcd8qAys_hqOH1qi0xBzg,6028
|
|
6
|
-
sagellm_core/engine.pyc,sha256=
|
|
6
|
+
sagellm_core/engine.pyc,sha256=o5yzFfLhpRx5lEYC2E75q9b04UqsDY63fC4nZdkHyeU,12258
|
|
7
7
|
sagellm_core/engine_factory.pyc,sha256=cNbEImM0MNcCgVeC0zViAT1AZkPAt2WARrVd1U__tS4,14886
|
|
8
8
|
sagellm_core/factory.pyc,sha256=B0s_JSOjVPwE95Oj2moP_vVA3vCrWsi2753UrwF8-eM,2737
|
|
9
9
|
sagellm_core/health.pyc,sha256=c4A60b1ZANbAqck59AV-kczK2b355EPi9JxoYHRlhl8,684
|
|
10
|
+
sagellm_core/pd_executor.pyc,sha256=EgSfZQsvlK0tGtvT3aO5LFYMvalf8V8EUOi-Vtxw_rQ,17835
|
|
10
11
|
sagellm_core/plugins.pyc,sha256=g6eke_yH65_7Bu6yfT7v19cZ7gjIAWbkyG7gv1JSXy0,2617
|
|
11
12
|
sagellm_core/py.typed,sha256=ixa8YukDZ3kLo0WsFJRGohLMyHzbMur1ALmmASML2cs,64
|
|
12
13
|
sagellm_core/runner.pyc,sha256=GK_t15IH2etN6flRqUui4iUPkD39zl3_rw1wnOSG5Jg,14238
|
|
14
|
+
sagellm_core/runtime.pyc,sha256=RFPMN7Vu9M5O1ZYgFQ9REz4wkMFznRPVPt2ra3u2vns,10018
|
|
13
15
|
sagellm_core/workload.pyc,sha256=G9yMm2L4ZyZpGsCxuEOQ83t78Y_lCQmbg5p6OHl5rn0,4127
|
|
14
|
-
sagellm_core/__pycache__/__init__.cpython-311.pyc,sha256=
|
|
16
|
+
sagellm_core/__pycache__/__init__.cpython-311.pyc,sha256=MEdTFvN9x22XnIHIjV6OKwne-QKulAg2a_DS1f-6xd8,3137
|
|
15
17
|
sagellm_core/__pycache__/base_engine.cpython-311.pyc,sha256=XiynMTxG3srYsUwzIoEbwEJOmpjBBB6UNEoAXDxDgWM,9563
|
|
16
18
|
sagellm_core/__pycache__/config.cpython-311.pyc,sha256=6gUdzUs3m31zGQVa2OzlPcPbX1it2Ziag0cQbCaXvIU,8483
|
|
17
19
|
sagellm_core/__pycache__/demo.cpython-311.pyc,sha256=PN6yZ3PgPWjiZ4XqpqRT-s5YYWitMwR_muYFvPmXdWA,6058
|
|
18
|
-
sagellm_core/__pycache__/engine.cpython-311.pyc,sha256=
|
|
20
|
+
sagellm_core/__pycache__/engine.cpython-311.pyc,sha256=B6ynwwETTD_kjZ2CvUao_QRjM0EluEWg60Be-nRiVeE,12288
|
|
19
21
|
sagellm_core/__pycache__/engine_factory.cpython-311.pyc,sha256=Qte3LXKSpcOWGhu-Hns7xPvTdkyYry3v_EeYQbfVp4w,14916
|
|
20
22
|
sagellm_core/__pycache__/factory.cpython-311.pyc,sha256=R4MbGyreD3U2iWmwQ0r14R347GAY9RM_I1xOiWovozo,2767
|
|
21
23
|
sagellm_core/__pycache__/health.cpython-311.pyc,sha256=Z2mo-4iGINZp0w9AmCJomhHGRSzvbwxCfmdMAjg5vc4,714
|
|
22
24
|
sagellm_core/__pycache__/mock_engine.cpython-311.pyc,sha256=zCHNncwIzEgqZ0lc7kiWkS-zOHf7CQ73UhCXztR6mFM,581
|
|
25
|
+
sagellm_core/__pycache__/pd_executor.cpython-311.pyc,sha256=mKsNTfzbGNTPFu7b3sD4zdngu-kPD2dcX2Ube3T5klw,17865
|
|
23
26
|
sagellm_core/__pycache__/plugins.cpython-311.pyc,sha256=gaoaHApylGSrmMxXVwFs2v4idiZJnisttawlXTbXBWY,2647
|
|
24
27
|
sagellm_core/__pycache__/runner.cpython-311.pyc,sha256=krW5MagsiQcX5J9fljblEshfsQB_gAIAWR7S0uD-TzU,14268
|
|
28
|
+
sagellm_core/__pycache__/runtime.cpython-311.pyc,sha256=STMbXptWWqislgsqEk4kIXmUYXdFzGDihSeXGWLy_6k,10048
|
|
25
29
|
sagellm_core/__pycache__/workload.cpython-311.pyc,sha256=Y01UUCDE85D8GY5kF_UsHyNCBbMkmsoB3vOUJV-vulA,4157
|
|
26
|
-
sagellm_core/engines/__init__.py,sha256=
|
|
27
|
-
sagellm_core/engines/__init__.pyc,sha256=
|
|
28
|
-
sagellm_core/engines/cpu.pyc,sha256=
|
|
30
|
+
sagellm_core/engines/__init__.py,sha256=M6hOhbAC_oEEnSdHrxFq804zk-F5X6xQriEU30rHnkA,1181
|
|
31
|
+
sagellm_core/engines/__init__.pyc,sha256=SfE-guU0Ssj6MLDnNNVJkcQkk7swE_j3VWDAAGODJOA,1223
|
|
32
|
+
sagellm_core/engines/cpu.pyc,sha256=IFlVXjCdNtGYam01L5G3h6YmFPux6kA7eIadBWbU4VM,27909
|
|
29
33
|
sagellm_core/engines/embedding.pyc,sha256=Be4To-qpamFuGcyFZ1MKPk6XOf_HDBKnBGoIisxNemE,11476
|
|
30
|
-
sagellm_core/engines/hf_cuda.pyc,sha256=
|
|
34
|
+
sagellm_core/engines/hf_cuda.pyc,sha256=2hS7sTGzAMhYBPDguyoFeJZU5U8Yg6uvKkQfPW0v-9A,53946
|
|
31
35
|
sagellm_core/engines/pytorch.pyc,sha256=MLcymoDnwMduWoWVDZmYuG6UoQavNO-FjddPQ8ccccI,18408
|
|
32
36
|
sagellm_core/engines/pytorch_engine.pyc,sha256=fdzf-VN2hVRWzP8PCmGV-pg9qUIJWWqEH662CRmh5_0,11645
|
|
33
|
-
sagellm_core/engines/__pycache__/__init__.cpython-311.pyc,sha256=
|
|
34
|
-
sagellm_core/engines/__pycache__/cpu.cpython-311.pyc,sha256=
|
|
37
|
+
sagellm_core/engines/__pycache__/__init__.cpython-311.pyc,sha256=DRqfKC9q3aFbxcNRrkV72yfMQchhMy6ho_YshGNHA3A,1253
|
|
38
|
+
sagellm_core/engines/__pycache__/cpu.cpython-311.pyc,sha256=JxNjjfN50UshGJMBw5OUge4jSUz_mLa_6wJu01cv2Ao,27939
|
|
35
39
|
sagellm_core/engines/__pycache__/embedding.cpython-311.pyc,sha256=ZNqFpuCsS4Wp0xg7yV3GBbaJ4G0saqGNaKMdivxEBkA,11506
|
|
36
|
-
sagellm_core/engines/__pycache__/hf_cuda.cpython-311.pyc,sha256=
|
|
40
|
+
sagellm_core/engines/__pycache__/hf_cuda.cpython-311.pyc,sha256=x5DpzAxT5SO7-nzqD0h9KOFiMkIp0cYpXyCoDd359VE,53976
|
|
37
41
|
sagellm_core/engines/__pycache__/mock.cpython-311.pyc,sha256=1g9YyKfo9yv-6VTjzzfacv-9ZICHjsVXjyJ_IajRnQo,14641
|
|
38
42
|
sagellm_core/engines/__pycache__/pytorch.cpython-311.pyc,sha256=MRSQN2ZBJmuIxFIA_yflaxarFhslfT0SG0Lw7-EFyDQ,18438
|
|
39
43
|
sagellm_core/engines/__pycache__/pytorch_engine.cpython-311.pyc,sha256=dLBcBODxEqgmH25nCBPWhwy3uu3-dqzyzOwQy-izN7A,11675
|
|
40
|
-
isagellm_core-0.
|
|
41
|
-
isagellm_core-0.
|
|
42
|
-
isagellm_core-0.
|
|
43
|
-
isagellm_core-0.
|
|
44
|
-
isagellm_core-0.
|
|
44
|
+
isagellm_core-0.3.0.2.dist-info/METADATA,sha256=0T1fEtPJkmazEt_HcNxxhkUmTi51bDqxWBkEmyCaoAs,10939
|
|
45
|
+
isagellm_core-0.3.0.2.dist-info/WHEEL,sha256=aE-1NVKXD6yEdmE9BpJA-cG2szRv8aTAmFSuenkwrqA,94
|
|
46
|
+
isagellm_core-0.3.0.2.dist-info/entry_points.txt,sha256=gGLYUgimyfJCNgP1jv3W7LjvrxLHP9ylyASec6FhcLI,189
|
|
47
|
+
isagellm_core-0.3.0.2.dist-info/top_level.txt,sha256=wcgdWrvkaoYYh_dWSFI5Toi8PZsHutVqfhTB2tb0K6g,13
|
|
48
|
+
isagellm_core-0.3.0.2.dist-info/RECORD,,
|
sagellm_core/__init__.py
CHANGED
|
@@ -6,6 +6,8 @@
|
|
|
6
6
|
- Engine 工厂函数
|
|
7
7
|
- 插件系统
|
|
8
8
|
- Demo Runner
|
|
9
|
+
- 分布式 Runtime(PD 分离 MVP)
|
|
10
|
+
- PD 分离执行器
|
|
9
11
|
"""
|
|
10
12
|
|
|
11
13
|
from __future__ import annotations
|
|
@@ -34,6 +36,10 @@ from sagellm_core.health import HealthStatus
|
|
|
34
36
|
from sagellm_core.plugins import PluginResolutionError, list_entry_points, resolve_kind
|
|
35
37
|
from sagellm_core.runner import DemoRunner, RunnerContext
|
|
36
38
|
|
|
39
|
+
# PD 分离 MVP 模块
|
|
40
|
+
from sagellm_core.runtime import DistributedConfig, DistributedRuntime, RuntimeState
|
|
41
|
+
from sagellm_core.pd_executor import PDExecutionContext, PDSeparatedExecutor
|
|
42
|
+
|
|
37
43
|
# PyTorch engine (optional, loaded lazily)
|
|
38
44
|
PyTorchEngine = None
|
|
39
45
|
PyTorchEngineConfig = None
|
|
@@ -72,7 +78,7 @@ try:
|
|
|
72
78
|
except Exception:
|
|
73
79
|
pass # torch or transformers not available
|
|
74
80
|
|
|
75
|
-
__version__ = "0.
|
|
81
|
+
__version__ = "0.3.0.2"
|
|
76
82
|
|
|
77
83
|
__all__ = [
|
|
78
84
|
# Version
|
|
@@ -111,4 +117,10 @@ __all__ = [
|
|
|
111
117
|
"demo_main",
|
|
112
118
|
"DemoRunner",
|
|
113
119
|
"RunnerContext",
|
|
120
|
+
# PD Separation MVP
|
|
121
|
+
"DistributedConfig",
|
|
122
|
+
"DistributedRuntime",
|
|
123
|
+
"RuntimeState",
|
|
124
|
+
"PDExecutionContext",
|
|
125
|
+
"PDSeparatedExecutor",
|
|
114
126
|
]
|
sagellm_core/__init__.pyc
CHANGED
|
Binary file
|
|
Binary file
|
|
Binary file
|
|
Binary file
|
|
Binary file
|
sagellm_core/engine.pyc
CHANGED
|
Binary file
|
sagellm_core/engines/__init__.py
CHANGED
|
@@ -9,9 +9,13 @@ This module provides built-in engine implementations:
|
|
|
9
9
|
|
|
10
10
|
from __future__ import annotations
|
|
11
11
|
|
|
12
|
-
from sagellm_core.engines.cpu import CPUEngine
|
|
12
|
+
from sagellm_core.engines.cpu import CPUEngine, create_cpu_engine
|
|
13
13
|
from sagellm_core.engines.embedding import EmbeddingEngine, EmbeddingEngineConfig
|
|
14
|
-
from sagellm_core.engines.hf_cuda import
|
|
14
|
+
from sagellm_core.engines.hf_cuda import (
|
|
15
|
+
HFCudaEngine,
|
|
16
|
+
HFCudaEngineInstanceConfig,
|
|
17
|
+
create_hf_cuda_engine,
|
|
18
|
+
)
|
|
15
19
|
from sagellm_core.engines.pytorch_engine import PyTorchEngine, create_pytorch_engine
|
|
16
20
|
|
|
17
21
|
# Export aliases for backward compatibility
|
|
@@ -20,10 +24,12 @@ HFCudaEngineConfig = HFCudaEngineInstanceConfig
|
|
|
20
24
|
__all__ = [
|
|
21
25
|
# CPU engine
|
|
22
26
|
"CPUEngine",
|
|
27
|
+
"create_cpu_engine",
|
|
23
28
|
# CUDA engine
|
|
24
29
|
"HFCudaEngine",
|
|
25
30
|
"HFCudaEngineConfig",
|
|
26
31
|
"HFCudaEngineInstanceConfig",
|
|
32
|
+
"create_hf_cuda_engine",
|
|
27
33
|
# PyTorch engine (unified)
|
|
28
34
|
"PyTorchEngine",
|
|
29
35
|
"create_pytorch_engine",
|
|
Binary file
|
|
Binary file
|
|
Binary file
|
|
Binary file
|
sagellm_core/engines/cpu.pyc
CHANGED
|
Binary file
|
sagellm_core/engines/hf_cuda.pyc
CHANGED
|
Binary file
|
|
Binary file
|
sagellm_core/runtime.pyc
ADDED
|
Binary file
|
|
File without changes
|