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.
Files changed (53) hide show
  1. airia/client/_request_handler/__init__.py +4 -0
  2. airia/client/_request_handler/async_request_handler.py +272 -0
  3. airia/client/_request_handler/base_request_handler.py +108 -0
  4. airia/client/_request_handler/sync_request_handler.py +255 -0
  5. airia/client/async_client.py +25 -678
  6. airia/client/base_client.py +2 -368
  7. airia/client/conversations/__init__.py +4 -0
  8. airia/client/conversations/async_conversations.py +187 -0
  9. airia/client/conversations/base_conversations.py +135 -0
  10. airia/client/conversations/sync_conversations.py +182 -0
  11. airia/client/pipeline_execution/__init__.py +4 -0
  12. airia/client/pipeline_execution/async_pipeline_execution.py +178 -0
  13. airia/client/pipeline_execution/base_pipeline_execution.py +96 -0
  14. airia/client/pipeline_execution/sync_pipeline_execution.py +178 -0
  15. airia/client/pipelines_config/__init__.py +4 -0
  16. airia/client/pipelines_config/async_pipelines_config.py +127 -0
  17. airia/client/pipelines_config/base_pipelines_config.py +76 -0
  18. airia/client/pipelines_config/sync_pipelines_config.py +127 -0
  19. airia/client/project/__init__.py +4 -0
  20. airia/client/project/async_project.py +122 -0
  21. airia/client/project/base_project.py +74 -0
  22. airia/client/project/sync_project.py +120 -0
  23. airia/client/store/__init__.py +4 -0
  24. airia/client/store/async_store.py +377 -0
  25. airia/client/store/base_store.py +243 -0
  26. airia/client/store/sync_store.py +352 -0
  27. airia/client/sync_client.py +25 -656
  28. airia/constants.py +1 -1
  29. airia/exceptions.py +8 -8
  30. airia/logs.py +9 -9
  31. airia/types/_request_data.py +11 -4
  32. airia/types/api/__init__.py +0 -27
  33. airia/types/api/conversations/__init__.py +3 -0
  34. airia/types/api/{conversations.py → conversations/_conversations.py} +49 -12
  35. airia/types/api/pipeline_execution/__init__.py +13 -0
  36. airia/types/api/{pipeline_execution.py → pipeline_execution/_pipeline_execution.py} +30 -13
  37. airia/types/api/pipelines_config/__init__.py +3 -0
  38. airia/types/api/pipelines_config/get_pipeline_config.py +401 -0
  39. airia/types/api/project/__init__.py +3 -0
  40. airia/types/api/{get_projects.py → project/get_projects.py} +16 -4
  41. airia/types/api/store/__init__.py +4 -0
  42. airia/types/api/store/get_file.py +145 -0
  43. airia/types/api/store/get_files.py +21 -0
  44. airia/types/sse/__init__.py +1 -0
  45. airia/types/sse/sse_messages.py +55 -21
  46. airia/utils/sse_parser.py +5 -4
  47. {airia-0.1.13.dist-info → airia-0.1.14.dist-info}/METADATA +4 -2
  48. airia-0.1.14.dist-info/RECORD +55 -0
  49. airia/types/api/get_pipeline_config.py +0 -214
  50. airia-0.1.13.dist-info/RECORD +0 -24
  51. {airia-0.1.13.dist-info → airia-0.1.14.dist-info}/WHEEL +0 -0
  52. {airia-0.1.13.dist-info → airia-0.1.14.dist-info}/licenses/LICENSE +0 -0
  53. {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
@@ -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