airia 0.1.13__py3-none-any.whl → 0.1.14__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.
- airia/client/_request_handler/__init__.py +4 -0
- airia/client/_request_handler/async_request_handler.py +272 -0
- airia/client/_request_handler/base_request_handler.py +108 -0
- airia/client/_request_handler/sync_request_handler.py +255 -0
- airia/client/async_client.py +25 -678
- airia/client/base_client.py +2 -368
- airia/client/conversations/__init__.py +4 -0
- airia/client/conversations/async_conversations.py +187 -0
- airia/client/conversations/base_conversations.py +135 -0
- airia/client/conversations/sync_conversations.py +182 -0
- airia/client/pipeline_execution/__init__.py +4 -0
- airia/client/pipeline_execution/async_pipeline_execution.py +178 -0
- airia/client/pipeline_execution/base_pipeline_execution.py +96 -0
- airia/client/pipeline_execution/sync_pipeline_execution.py +178 -0
- airia/client/pipelines_config/__init__.py +4 -0
- airia/client/pipelines_config/async_pipelines_config.py +127 -0
- airia/client/pipelines_config/base_pipelines_config.py +76 -0
- airia/client/pipelines_config/sync_pipelines_config.py +127 -0
- airia/client/project/__init__.py +4 -0
- airia/client/project/async_project.py +122 -0
- airia/client/project/base_project.py +74 -0
- airia/client/project/sync_project.py +120 -0
- airia/client/store/__init__.py +4 -0
- airia/client/store/async_store.py +377 -0
- airia/client/store/base_store.py +243 -0
- airia/client/store/sync_store.py +352 -0
- airia/client/sync_client.py +25 -656
- airia/constants.py +1 -1
- airia/exceptions.py +8 -8
- airia/logs.py +9 -9
- airia/types/_request_data.py +11 -4
- airia/types/api/__init__.py +0 -27
- airia/types/api/conversations/__init__.py +3 -0
- airia/types/api/{conversations.py → conversations/_conversations.py} +49 -12
- airia/types/api/pipeline_execution/__init__.py +13 -0
- airia/types/api/{pipeline_execution.py → pipeline_execution/_pipeline_execution.py} +30 -13
- airia/types/api/pipelines_config/__init__.py +3 -0
- airia/types/api/pipelines_config/get_pipeline_config.py +401 -0
- airia/types/api/project/__init__.py +3 -0
- airia/types/api/{get_projects.py → project/get_projects.py} +16 -4
- airia/types/api/store/__init__.py +4 -0
- airia/types/api/store/get_file.py +145 -0
- airia/types/api/store/get_files.py +21 -0
- airia/types/sse/__init__.py +1 -0
- airia/types/sse/sse_messages.py +55 -21
- airia/utils/sse_parser.py +5 -4
- {airia-0.1.13.dist-info → airia-0.1.14.dist-info}/METADATA +4 -2
- airia-0.1.14.dist-info/RECORD +55 -0
- airia/types/api/get_pipeline_config.py +0 -214
- airia-0.1.13.dist-info/RECORD +0 -24
- {airia-0.1.13.dist-info → airia-0.1.14.dist-info}/WHEEL +0 -0
- {airia-0.1.13.dist-info → airia-0.1.14.dist-info}/licenses/LICENSE +0 -0
- {airia-0.1.13.dist-info → airia-0.1.14.dist-info}/top_level.txt +0 -0
|
@@ -1,214 +0,0 @@
|
|
|
1
|
-
"""
|
|
2
|
-
Pydantic models for pipeline configuration API responses.
|
|
3
|
-
|
|
4
|
-
This module defines comprehensive data structures for pipeline configuration exports,
|
|
5
|
-
including all components like agents, models, tools, data sources, and deployment settings.
|
|
6
|
-
"""
|
|
7
|
-
from typing import Any, Dict, List, Optional
|
|
8
|
-
|
|
9
|
-
from pydantic import BaseModel, Field
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
class Metadata(BaseModel):
|
|
13
|
-
"""
|
|
14
|
-
Pipeline metadata and export configuration.
|
|
15
|
-
|
|
16
|
-
Contains version information, export settings, and descriptive metadata
|
|
17
|
-
about the pipeline configuration.
|
|
18
|
-
"""
|
|
19
|
-
id: str
|
|
20
|
-
export_version: str = Field(alias="exportVersion")
|
|
21
|
-
tagline: Optional[str] = None
|
|
22
|
-
agent_description: Optional[str] = Field(alias="agentDescription", default=None)
|
|
23
|
-
industry: Optional[str] = None
|
|
24
|
-
tasks: Optional[str] = None
|
|
25
|
-
credential_export_option: str = Field(alias="credentialExportOption")
|
|
26
|
-
data_source_export_option: str = Field(alias="dataSourceExportOption")
|
|
27
|
-
version_information: str = Field(alias="versionInformation")
|
|
28
|
-
state: str
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
class Agent(BaseModel):
|
|
32
|
-
"""
|
|
33
|
-
AI agent configuration and workflow definition.
|
|
34
|
-
|
|
35
|
-
Represents the core agent that executes the pipeline, including its
|
|
36
|
-
identity, industry specialization, and step-by-step workflow configuration.
|
|
37
|
-
"""
|
|
38
|
-
name: str
|
|
39
|
-
execution_name: str = Field(alias="executionName")
|
|
40
|
-
agent_description: Optional[str] = Field(alias="agentDescription", default=None)
|
|
41
|
-
video_link: Optional[str] = Field(alias="videoLink", default=None)
|
|
42
|
-
industry: Optional[str] = None
|
|
43
|
-
sub_industries: List[str] = Field(alias="subIndustries", default_factory=list)
|
|
44
|
-
agent_details: Dict[str, Any] = Field(alias="agentDetails", default_factory=dict)
|
|
45
|
-
id: str
|
|
46
|
-
agent_icon: Optional[str] = Field(alias="agentIcon", default=None)
|
|
47
|
-
steps: List[Dict[str, Any]]
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
class PromptMessage(BaseModel):
|
|
51
|
-
text: str
|
|
52
|
-
order: int
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
class Prompt(BaseModel):
|
|
56
|
-
name: str
|
|
57
|
-
version_change_description: str = Field(alias="versionChangeDescription")
|
|
58
|
-
prompt_message_list: List[PromptMessage] = Field(alias="promptMessageList")
|
|
59
|
-
id: str
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
class CredentialData(BaseModel):
|
|
63
|
-
key: str
|
|
64
|
-
value: str
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
class CredentialsDefinition(BaseModel):
|
|
68
|
-
name: str
|
|
69
|
-
credential_type: str = Field(alias="credentialType")
|
|
70
|
-
source_type: str = Field(alias="sourceType")
|
|
71
|
-
credential_data_list: List[CredentialData] = Field(alias="credentialDataList")
|
|
72
|
-
id: str
|
|
73
|
-
|
|
74
|
-
|
|
75
|
-
class HeaderDefinition(BaseModel):
|
|
76
|
-
key: str
|
|
77
|
-
value: str
|
|
78
|
-
|
|
79
|
-
|
|
80
|
-
class ParameterDefinition(BaseModel):
|
|
81
|
-
name: str
|
|
82
|
-
parameter_type: str = Field(alias="parameterType")
|
|
83
|
-
parameter_description: str = Field(alias="parameterDescription")
|
|
84
|
-
default: str
|
|
85
|
-
valid_options: List[str] = Field(alias="validOptions", default_factory=list)
|
|
86
|
-
id: str
|
|
87
|
-
|
|
88
|
-
|
|
89
|
-
class Tool(BaseModel):
|
|
90
|
-
tool_type: str = Field(alias="toolType")
|
|
91
|
-
name: str
|
|
92
|
-
standardized_name: str = Field(alias="standardizedName")
|
|
93
|
-
tool_description: str = Field(alias="toolDescription")
|
|
94
|
-
purpose: str
|
|
95
|
-
api_endpoint: str = Field(alias="apiEndpoint")
|
|
96
|
-
credentials_definition: Optional[CredentialsDefinition] = Field(
|
|
97
|
-
alias="credentialsDefinition"
|
|
98
|
-
)
|
|
99
|
-
headers_definition: List[HeaderDefinition] = Field(alias="headersDefinition")
|
|
100
|
-
body: str
|
|
101
|
-
parameters_definition: List[ParameterDefinition] = Field(
|
|
102
|
-
alias="parametersDefinition"
|
|
103
|
-
)
|
|
104
|
-
method_type: str = Field(alias="methodType")
|
|
105
|
-
route_through_acc: bool = Field(alias="routeThroughACC")
|
|
106
|
-
use_user_credentials: bool = Field(alias="useUserCredentials")
|
|
107
|
-
use_user_credentials_type: str = Field(alias="useUserCredentialsType")
|
|
108
|
-
id: str
|
|
109
|
-
|
|
110
|
-
|
|
111
|
-
class Model(BaseModel):
|
|
112
|
-
"""
|
|
113
|
-
Language model configuration and deployment settings.
|
|
114
|
-
|
|
115
|
-
Defines an AI model used in the pipeline, including its deployment details,
|
|
116
|
-
pricing configuration, authentication settings, and capabilities.
|
|
117
|
-
"""
|
|
118
|
-
id: str
|
|
119
|
-
display_name: str = Field(alias="displayName")
|
|
120
|
-
model_name: str = Field(alias="modelName")
|
|
121
|
-
prompt_id: Optional[str] = Field(alias="promptId", default=None)
|
|
122
|
-
system_prompt_definition: Optional[Any] = Field(
|
|
123
|
-
alias="systemPromptDefinition", default=None
|
|
124
|
-
)
|
|
125
|
-
url: str
|
|
126
|
-
input_type: str = Field(alias="inputType")
|
|
127
|
-
provider: str
|
|
128
|
-
credentials_definition: Optional[CredentialsDefinition] = Field(
|
|
129
|
-
alias="credentialsDefinition"
|
|
130
|
-
)
|
|
131
|
-
deployment_type: str = Field(alias="deploymentType")
|
|
132
|
-
source_type: str = Field(alias="sourceType")
|
|
133
|
-
connection_string: Optional[str] = Field(alias="connectionString", default=None)
|
|
134
|
-
container_name: Optional[str] = Field(alias="containerName", default=None)
|
|
135
|
-
deployed_key: Optional[str] = Field(alias="deployedKey", default=None)
|
|
136
|
-
deployed_url: Optional[str] = Field(alias="deployedUrl", default=None)
|
|
137
|
-
state: Optional[str] = None
|
|
138
|
-
uploaded_container_id: Optional[str] = Field(
|
|
139
|
-
alias="uploadedContainerId", default=None
|
|
140
|
-
)
|
|
141
|
-
library_model_id: Optional[str] = Field(alias="libraryModelId")
|
|
142
|
-
input_token_price: str = Field(alias="inputTokenPrice")
|
|
143
|
-
output_token_price: str = Field(alias="outputTokenPrice")
|
|
144
|
-
token_units: int = Field(alias="tokenUnits")
|
|
145
|
-
has_tool_support: bool = Field(alias="hasToolSupport")
|
|
146
|
-
allow_airia_credentials: bool = Field(alias="allowAiriaCredentials")
|
|
147
|
-
allow_byok_credentials: bool = Field(alias="allowBYOKCredentials")
|
|
148
|
-
author: Optional[str]
|
|
149
|
-
price_type: str = Field(alias="priceType")
|
|
150
|
-
|
|
151
|
-
|
|
152
|
-
class PythonCodeBlock(BaseModel):
|
|
153
|
-
id: str
|
|
154
|
-
code: str
|
|
155
|
-
|
|
156
|
-
|
|
157
|
-
class Router(BaseModel):
|
|
158
|
-
id: str
|
|
159
|
-
model_id: str = Field(alias="modelId")
|
|
160
|
-
model: Optional[Any] = None
|
|
161
|
-
router_config: Dict[str, Dict[str, Any]] = Field(alias="routerConfig")
|
|
162
|
-
|
|
163
|
-
|
|
164
|
-
class ChunkingConfig(BaseModel):
|
|
165
|
-
id: str
|
|
166
|
-
chunk_size: int = Field(alias="chunkSize")
|
|
167
|
-
chunk_overlap: int = Field(alias="chunkOverlap")
|
|
168
|
-
strategy_type: str = Field(alias="strategyType")
|
|
169
|
-
|
|
170
|
-
|
|
171
|
-
class DataSourceFile(BaseModel):
|
|
172
|
-
data_source_id: str = Field(alias="dataSourceId")
|
|
173
|
-
file_path: Optional[str] = Field(None, alias="filePath")
|
|
174
|
-
input_token: Optional[str] = Field(None, alias="inputToken")
|
|
175
|
-
file_count: Optional[int] = Field(None, alias="fileCount")
|
|
176
|
-
|
|
177
|
-
|
|
178
|
-
class DataSource(BaseModel):
|
|
179
|
-
id: str = Field(alias="id")
|
|
180
|
-
name: Optional[str] = None
|
|
181
|
-
execution_name: Optional[str] = Field(None, alias="executionName")
|
|
182
|
-
chunking_config: ChunkingConfig = Field(alias="chunkingConfig")
|
|
183
|
-
data_source_type: str = Field(alias="dataSourceType")
|
|
184
|
-
database_type: str = Field(alias="databaseType")
|
|
185
|
-
embedding_provider: str = Field(alias="embeddingProvider")
|
|
186
|
-
is_user_specific: bool = Field(alias="isUserSpecific")
|
|
187
|
-
files: Optional[List[DataSourceFile]] = None
|
|
188
|
-
configuration_json: Optional[str] = Field(None, alias="configurationJson")
|
|
189
|
-
credentials: Optional[CredentialsDefinition]
|
|
190
|
-
is_image_processing_enabled: bool = Field(alias="isImageProcessingEnabled")
|
|
191
|
-
|
|
192
|
-
|
|
193
|
-
class GetPipelineConfigResponse(BaseModel):
|
|
194
|
-
"""
|
|
195
|
-
Complete pipeline configuration export response.
|
|
196
|
-
|
|
197
|
-
This is the root response model containing all components of a pipeline
|
|
198
|
-
configuration, including the agent definition, associated resources,
|
|
199
|
-
and deployment settings.
|
|
200
|
-
"""
|
|
201
|
-
metadata: Metadata
|
|
202
|
-
agent: Agent
|
|
203
|
-
data_sources: Optional[List[DataSource]] = Field(
|
|
204
|
-
alias="dataSources", default_factory=list
|
|
205
|
-
)
|
|
206
|
-
prompts: Optional[List[Prompt]] = Field(default_factory=list)
|
|
207
|
-
tools: Optional[List[Tool]] = Field(default_factory=list)
|
|
208
|
-
models: Optional[List[Model]] = Field(default_factory=list)
|
|
209
|
-
memories: Optional[Any] = None
|
|
210
|
-
python_code_blocks: Optional[List[PythonCodeBlock]] = Field(
|
|
211
|
-
alias="pythonCodeBlocks", default_factory=list
|
|
212
|
-
)
|
|
213
|
-
routers: Optional[List[Router]] = Field(default_factory=list)
|
|
214
|
-
deployment: Optional[Any] = None
|
airia-0.1.13.dist-info/RECORD
DELETED
|
@@ -1,24 +0,0 @@
|
|
|
1
|
-
airia/__init__.py,sha256=T39gO8E5T5zxlw-JP78ruxOu7-LeKOJCJzz6t40kdQo,231
|
|
2
|
-
airia/constants.py,sha256=QDCxP-j3QQp5JGftYIYoqMu1fwXd56KZYAPD6_5YmPU,646
|
|
3
|
-
airia/exceptions.py,sha256=4Z55n-cRJrtTa5-pZBIK2oZD4-Z99aUtKx_kfTFYY5o,1146
|
|
4
|
-
airia/logs.py,sha256=_JO55-TcCtS1aVmmWUWEFKd2bMW1oYsz1j_8ajgPIdA,4570
|
|
5
|
-
airia/client/__init__.py,sha256=6gSQ9bl7j79q1HPE0o5py3IRdkwWWuU_7J4h05Dd2o8,127
|
|
6
|
-
airia/client/async_client.py,sha256=smYbJj90gBer24eLi4vwl_5z5jgPU6KA0wAWSv8GDxw,33290
|
|
7
|
-
airia/client/base_client.py,sha256=LdAcTU1URuq6b1XXStd-7Sa-LiYm8H2TgzDyJSYegls,17456
|
|
8
|
-
airia/client/sync_client.py,sha256=gH0VISV1ZGUjP53ignIrn8yN_iUOPgFV0EFZ8TsafVk,32118
|
|
9
|
-
airia/types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
10
|
-
airia/types/_api_version.py,sha256=Uzom6O2ZG92HN_Z2h-lTydmO2XYX9RVs4Yi4DJmXytE,255
|
|
11
|
-
airia/types/_request_data.py,sha256=EUYMWQzKlX8Isf_VYP1bcb3uau6fjzRm9guXSaXPiwE,1041
|
|
12
|
-
airia/types/api/__init__.py,sha256=gP4vOYxOG8Gfq8E2EEe1K_KW73nbJ--HUt3n7BUzu4A,913
|
|
13
|
-
airia/types/api/conversations.py,sha256=Kkb3ybdBCQBCBVHe-gRUjGV2wqGA4Y5d5EGTitMuJ1Q,2981
|
|
14
|
-
airia/types/api/get_pipeline_config.py,sha256=GTmY2IoCmXNCCmwOI9gQ5pkEJTyhQdaJxzhWaQhEmsM,7781
|
|
15
|
-
airia/types/api/get_projects.py,sha256=krGV2gAxH81E_5OtUzUNdWZN7S-2jkAV5TiqJmW8A48,3479
|
|
16
|
-
airia/types/api/pipeline_execution.py,sha256=-lcae-UqNrXqqy8DuyXEbMFBm3dVXQj0cl2jpeFeb78,1831
|
|
17
|
-
airia/types/sse/__init__.py,sha256=DOFbw7DASl96g_UQwSS8iHY1mX4T6CLX7It5QWxtwJ0,1862
|
|
18
|
-
airia/types/sse/sse_messages.py,sha256=q4DOqcYbYnsS3mBYO_-V7rZScGIChrCuh15jrcO0aII,14320
|
|
19
|
-
airia/utils/sse_parser.py,sha256=YCnlXHqkAFUeKoy60RH-IUgQ9ZkrzK_UQ4VsnQldCUU,4274
|
|
20
|
-
airia-0.1.13.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
|
|
21
|
-
airia-0.1.13.dist-info/METADATA,sha256=coFoSzERDhm84fg0oaXgYalPIBAWWWOiJ3uexckVLRI,4364
|
|
22
|
-
airia-0.1.13.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
|
23
|
-
airia-0.1.13.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
|
|
24
|
-
airia-0.1.13.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|