vertai 0.1.0__py3-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.
- ai_sdk/__init__.py +120 -0
- ai_sdk/core/__init__.py +36 -0
- ai_sdk/core/llm.py +1010 -0
- ai_sdk/core/memory.py +374 -0
- ai_sdk/core/vector.py +645 -0
- ai_sdk/data/__init__.py +5 -0
- ai_sdk/data/parser.py +396 -0
- ai_sdk/local/__init__.py +22 -0
- ai_sdk/local/models.py +792 -0
- ai_sdk/output/__init__.py +15 -0
- ai_sdk/output/docgen.py +404 -0
- ai_sdk/output/structured.py +525 -0
- ai_sdk/scenarios/__init__.py +21 -0
- ai_sdk/scenarios/knowledge_qa.py +539 -0
- ai_sdk/scenarios/reviewer.py +262 -0
- ai_sdk/viz/__init__.py +19 -0
- ai_sdk/viz/dashboard.py +820 -0
- ai_sdk/workflow/__init__.py +29 -0
- ai_sdk/workflow/workflow.py +962 -0
- vertai-0.1.0.dist-info/METADATA +289 -0
- vertai-0.1.0.dist-info/RECORD +22 -0
- vertai-0.1.0.dist-info/WHEEL +4 -0
ai_sdk/__init__.py
ADDED
|
@@ -0,0 +1,120 @@
|
|
|
1
|
+
"""VertAI - 垂直领域 AI 智能体开发 SDK"""
|
|
2
|
+
|
|
3
|
+
__version__ = "0.1.0"
|
|
4
|
+
|
|
5
|
+
from ai_sdk.core.llm import (
|
|
6
|
+
LLMEngine,
|
|
7
|
+
LLMConfig,
|
|
8
|
+
ModelProvider,
|
|
9
|
+
ChatMessage,
|
|
10
|
+
GenerateResult,
|
|
11
|
+
)
|
|
12
|
+
from ai_sdk.core.memory import (
|
|
13
|
+
Message,
|
|
14
|
+
SessionConfig,
|
|
15
|
+
SessionMemory,
|
|
16
|
+
)
|
|
17
|
+
from ai_sdk.core.vector import (
|
|
18
|
+
Document,
|
|
19
|
+
VectorEngine,
|
|
20
|
+
VectorConfig,
|
|
21
|
+
SearchResult,
|
|
22
|
+
)
|
|
23
|
+
from ai_sdk.data.parser import DocParser
|
|
24
|
+
from ai_sdk.output.structured import StructuredOutput
|
|
25
|
+
from ai_sdk.output.docgen import DocGen
|
|
26
|
+
from ai_sdk.scenarios.reviewer import Reviewer, ReviewerConfig, ReviewResult
|
|
27
|
+
from ai_sdk.scenarios.knowledge_qa import (
|
|
28
|
+
KnowledgeQA,
|
|
29
|
+
KnowledgeQAConfig,
|
|
30
|
+
AnswerResult,
|
|
31
|
+
SourceReference,
|
|
32
|
+
)
|
|
33
|
+
from ai_sdk.viz.dashboard import (
|
|
34
|
+
Dashboard,
|
|
35
|
+
DashboardTheme,
|
|
36
|
+
Metric,
|
|
37
|
+
Chart,
|
|
38
|
+
ChartType,
|
|
39
|
+
ChartConfig,
|
|
40
|
+
)
|
|
41
|
+
from ai_sdk.workflow import (
|
|
42
|
+
Workflow,
|
|
43
|
+
WorkflowConfig,
|
|
44
|
+
WorkflowContext,
|
|
45
|
+
WorkflowResult,
|
|
46
|
+
StepResult,
|
|
47
|
+
StepStatus,
|
|
48
|
+
Step,
|
|
49
|
+
StepType,
|
|
50
|
+
ParallelConfig,
|
|
51
|
+
LoopConfig,
|
|
52
|
+
LoopType,
|
|
53
|
+
)
|
|
54
|
+
from ai_sdk.local import (
|
|
55
|
+
LocalModelManager,
|
|
56
|
+
LocalModelConfig,
|
|
57
|
+
ModelCategory,
|
|
58
|
+
ModelInfo,
|
|
59
|
+
WhisperModel,
|
|
60
|
+
EmbeddingModel,
|
|
61
|
+
)
|
|
62
|
+
from ai_sdk.local.models import check_hardware_requirements
|
|
63
|
+
|
|
64
|
+
__all__ = [
|
|
65
|
+
# LLM
|
|
66
|
+
"LLMEngine",
|
|
67
|
+
"LLMConfig",
|
|
68
|
+
"ModelProvider",
|
|
69
|
+
"ChatMessage",
|
|
70
|
+
"GenerateResult",
|
|
71
|
+
# Memory
|
|
72
|
+
"Message",
|
|
73
|
+
"SessionConfig",
|
|
74
|
+
"SessionMemory",
|
|
75
|
+
# Vector
|
|
76
|
+
"Document",
|
|
77
|
+
"VectorEngine",
|
|
78
|
+
"VectorConfig",
|
|
79
|
+
"SearchResult",
|
|
80
|
+
# Data
|
|
81
|
+
"DocParser",
|
|
82
|
+
# Output
|
|
83
|
+
"StructuredOutput",
|
|
84
|
+
"DocGen",
|
|
85
|
+
# Scenarios
|
|
86
|
+
"Reviewer",
|
|
87
|
+
"ReviewerConfig",
|
|
88
|
+
"ReviewResult",
|
|
89
|
+
"KnowledgeQA",
|
|
90
|
+
"KnowledgeQAConfig",
|
|
91
|
+
"AnswerResult",
|
|
92
|
+
"SourceReference",
|
|
93
|
+
# Viz
|
|
94
|
+
"Dashboard",
|
|
95
|
+
"DashboardTheme",
|
|
96
|
+
"Metric",
|
|
97
|
+
"Chart",
|
|
98
|
+
"ChartType",
|
|
99
|
+
"ChartConfig",
|
|
100
|
+
# Workflow
|
|
101
|
+
"Workflow",
|
|
102
|
+
"WorkflowConfig",
|
|
103
|
+
"WorkflowContext",
|
|
104
|
+
"WorkflowResult",
|
|
105
|
+
"StepResult",
|
|
106
|
+
"StepStatus",
|
|
107
|
+
"Step",
|
|
108
|
+
"StepType",
|
|
109
|
+
"ParallelConfig",
|
|
110
|
+
"LoopConfig",
|
|
111
|
+
"LoopType",
|
|
112
|
+
# Local Models
|
|
113
|
+
"LocalModelManager",
|
|
114
|
+
"LocalModelConfig",
|
|
115
|
+
"ModelCategory",
|
|
116
|
+
"ModelInfo",
|
|
117
|
+
"WhisperModel",
|
|
118
|
+
"EmbeddingModel",
|
|
119
|
+
"check_hardware_requirements",
|
|
120
|
+
]
|
ai_sdk/core/__init__.py
ADDED
|
@@ -0,0 +1,36 @@
|
|
|
1
|
+
"""核心模块 - LLM引擎、向量引擎、记忆引擎、工具引擎"""
|
|
2
|
+
|
|
3
|
+
from ai_sdk.core.llm import LLMEngine, LLMConfig
|
|
4
|
+
from ai_sdk.core.memory import (
|
|
5
|
+
Message,
|
|
6
|
+
SessionConfig,
|
|
7
|
+
SessionMemory,
|
|
8
|
+
)
|
|
9
|
+
from ai_sdk.core.vector import (
|
|
10
|
+
Document,
|
|
11
|
+
SearchResult,
|
|
12
|
+
VectorConfig,
|
|
13
|
+
VectorEngine,
|
|
14
|
+
VectorStore,
|
|
15
|
+
InMemoryVectorStore,
|
|
16
|
+
ChromaVectorStore,
|
|
17
|
+
FAISSVectorStore,
|
|
18
|
+
EmbeddingEngine,
|
|
19
|
+
)
|
|
20
|
+
|
|
21
|
+
__all__ = [
|
|
22
|
+
"LLMEngine",
|
|
23
|
+
"LLMConfig",
|
|
24
|
+
"Message",
|
|
25
|
+
"SessionConfig",
|
|
26
|
+
"SessionMemory",
|
|
27
|
+
"Document",
|
|
28
|
+
"SearchResult",
|
|
29
|
+
"VectorConfig",
|
|
30
|
+
"VectorEngine",
|
|
31
|
+
"VectorStore",
|
|
32
|
+
"InMemoryVectorStore",
|
|
33
|
+
"ChromaVectorStore",
|
|
34
|
+
"FAISSVectorStore",
|
|
35
|
+
"EmbeddingEngine",
|
|
36
|
+
]
|