airtrain 0.1.2__py3-none-any.whl → 0.1.4__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.
- airtrain/__init__.py +148 -2
- airtrain/__main__.py +4 -0
- airtrain/__pycache__/__init__.cpython-313.pyc +0 -0
- airtrain/agents/__init__.py +45 -0
- airtrain/agents/example_agent.py +348 -0
- airtrain/agents/groq_agent.py +289 -0
- airtrain/agents/memory.py +663 -0
- airtrain/agents/registry.py +465 -0
- airtrain/builder/__init__.py +3 -0
- airtrain/builder/agent_builder.py +122 -0
- airtrain/cli/__init__.py +0 -0
- airtrain/cli/builder.py +23 -0
- airtrain/cli/main.py +120 -0
- airtrain/contrib/__init__.py +29 -0
- airtrain/contrib/travel/__init__.py +35 -0
- airtrain/contrib/travel/agents.py +243 -0
- airtrain/contrib/travel/models.py +59 -0
- airtrain/core/__init__.py +7 -0
- airtrain/core/__pycache__/__init__.cpython-313.pyc +0 -0
- airtrain/core/__pycache__/schemas.cpython-313.pyc +0 -0
- airtrain/core/__pycache__/skills.cpython-313.pyc +0 -0
- airtrain/core/credentials.py +171 -0
- airtrain/core/schemas.py +237 -0
- airtrain/core/skills.py +269 -0
- airtrain/integrations/__init__.py +74 -0
- airtrain/integrations/anthropic/__init__.py +33 -0
- airtrain/integrations/anthropic/credentials.py +32 -0
- airtrain/integrations/anthropic/list_models.py +110 -0
- airtrain/integrations/anthropic/models_config.py +100 -0
- airtrain/integrations/anthropic/skills.py +155 -0
- airtrain/integrations/aws/__init__.py +6 -0
- airtrain/integrations/aws/credentials.py +36 -0
- airtrain/integrations/aws/skills.py +98 -0
- airtrain/integrations/cerebras/__init__.py +6 -0
- airtrain/integrations/cerebras/credentials.py +19 -0
- airtrain/integrations/cerebras/skills.py +127 -0
- airtrain/integrations/combined/__init__.py +21 -0
- airtrain/integrations/combined/groq_fireworks_skills.py +126 -0
- airtrain/integrations/combined/list_models_factory.py +210 -0
- airtrain/integrations/fireworks/__init__.py +21 -0
- airtrain/integrations/fireworks/completion_skills.py +147 -0
- airtrain/integrations/fireworks/conversation_manager.py +109 -0
- airtrain/integrations/fireworks/credentials.py +26 -0
- airtrain/integrations/fireworks/list_models.py +128 -0
- airtrain/integrations/fireworks/models.py +139 -0
- airtrain/integrations/fireworks/requests_skills.py +207 -0
- airtrain/integrations/fireworks/skills.py +181 -0
- airtrain/integrations/fireworks/structured_completion_skills.py +175 -0
- airtrain/integrations/fireworks/structured_requests_skills.py +291 -0
- airtrain/integrations/fireworks/structured_skills.py +102 -0
- airtrain/integrations/google/__init__.py +7 -0
- airtrain/integrations/google/credentials.py +58 -0
- airtrain/integrations/google/skills.py +122 -0
- airtrain/integrations/groq/__init__.py +23 -0
- airtrain/integrations/groq/credentials.py +24 -0
- airtrain/integrations/groq/models_config.py +162 -0
- airtrain/integrations/groq/skills.py +201 -0
- airtrain/integrations/ollama/__init__.py +6 -0
- airtrain/integrations/ollama/credentials.py +26 -0
- airtrain/integrations/ollama/skills.py +41 -0
- airtrain/integrations/openai/__init__.py +37 -0
- airtrain/integrations/openai/chinese_assistant.py +42 -0
- airtrain/integrations/openai/credentials.py +39 -0
- airtrain/integrations/openai/list_models.py +112 -0
- airtrain/integrations/openai/models_config.py +224 -0
- airtrain/integrations/openai/skills.py +342 -0
- airtrain/integrations/perplexity/__init__.py +49 -0
- airtrain/integrations/perplexity/credentials.py +43 -0
- airtrain/integrations/perplexity/list_models.py +112 -0
- airtrain/integrations/perplexity/models_config.py +128 -0
- airtrain/integrations/perplexity/skills.py +279 -0
- airtrain/integrations/sambanova/__init__.py +6 -0
- airtrain/integrations/sambanova/credentials.py +20 -0
- airtrain/integrations/sambanova/skills.py +129 -0
- airtrain/integrations/search/__init__.py +21 -0
- airtrain/integrations/search/exa/__init__.py +23 -0
- airtrain/integrations/search/exa/credentials.py +30 -0
- airtrain/integrations/search/exa/schemas.py +114 -0
- airtrain/integrations/search/exa/skills.py +115 -0
- airtrain/integrations/together/__init__.py +33 -0
- airtrain/integrations/together/audio_models_config.py +34 -0
- airtrain/integrations/together/credentials.py +22 -0
- airtrain/integrations/together/embedding_models_config.py +92 -0
- airtrain/integrations/together/image_models_config.py +69 -0
- airtrain/integrations/together/image_skill.py +143 -0
- airtrain/integrations/together/list_models.py +76 -0
- airtrain/integrations/together/models.py +95 -0
- airtrain/integrations/together/models_config.py +399 -0
- airtrain/integrations/together/rerank_models_config.py +43 -0
- airtrain/integrations/together/rerank_skill.py +49 -0
- airtrain/integrations/together/schemas.py +33 -0
- airtrain/integrations/together/skills.py +305 -0
- airtrain/integrations/together/vision_models_config.py +49 -0
- airtrain/telemetry/__init__.py +38 -0
- airtrain/telemetry/service.py +167 -0
- airtrain/telemetry/views.py +237 -0
- airtrain/tools/__init__.py +45 -0
- airtrain/tools/command.py +398 -0
- airtrain/tools/filesystem.py +166 -0
- airtrain/tools/network.py +111 -0
- airtrain/tools/registry.py +320 -0
- airtrain/tools/search.py +450 -0
- airtrain/tools/testing.py +135 -0
- airtrain-0.1.4.dist-info/METADATA +222 -0
- airtrain-0.1.4.dist-info/RECORD +108 -0
- {airtrain-0.1.2.dist-info → airtrain-0.1.4.dist-info}/WHEEL +1 -1
- airtrain-0.1.4.dist-info/entry_points.txt +2 -0
- airtrain-0.1.2.dist-info/METADATA +0 -106
- airtrain-0.1.2.dist-info/RECORD +0 -5
- {airtrain-0.1.2.dist-info → airtrain-0.1.4.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,222 @@
|
|
1
|
+
Metadata-Version: 2.4
|
2
|
+
Name: airtrain
|
3
|
+
Version: 0.1.4
|
4
|
+
Summary: A platform for building and deploying AI agents with structured skills
|
5
|
+
Home-page: https://github.com/rosaboyle/airtrain.dev
|
6
|
+
Author: Dheeraj Pai
|
7
|
+
Author-email: Dheeraj Pai <helloworldcmu@gmail.com>
|
8
|
+
Project-URL: Homepage, https://github.com/rosaboyle/airtrain.dev
|
9
|
+
Project-URL: Documentation, https://docs.airtrain.dev/
|
10
|
+
Classifier: Development Status :: 3 - Alpha
|
11
|
+
Classifier: Intended Audience :: Developers
|
12
|
+
Classifier: License :: OSI Approved :: MIT License
|
13
|
+
Classifier: Operating System :: OS Independent
|
14
|
+
Classifier: Programming Language :: Python :: 3
|
15
|
+
Classifier: Programming Language :: Python :: 3.8
|
16
|
+
Classifier: Programming Language :: Python :: 3.9
|
17
|
+
Classifier: Programming Language :: Python :: 3.10
|
18
|
+
Requires-Python: >=3.8
|
19
|
+
Description-Content-Type: text/markdown
|
20
|
+
Requires-Dist: pydantic>=2.10.6
|
21
|
+
Requires-Dist: openai>=1.60.1
|
22
|
+
Requires-Dist: python-dotenv>=1.0.1
|
23
|
+
Requires-Dist: PyYAML>=6.0.2
|
24
|
+
Requires-Dist: firebase-admin>=6.6.0
|
25
|
+
Requires-Dist: loguru>=0.7.3
|
26
|
+
Requires-Dist: requests>=2.32.3
|
27
|
+
Requires-Dist: boto3>=1.36.6
|
28
|
+
Requires-Dist: together>=1.3.13
|
29
|
+
Requires-Dist: anthropic>=0.45.0
|
30
|
+
Requires-Dist: groq>=0.15.0
|
31
|
+
Requires-Dist: cerebras-cloud-sdk>=1.19.0
|
32
|
+
Requires-Dist: google-genai>=1.0.0
|
33
|
+
Requires-Dist: fireworks-ai>=0.15.12
|
34
|
+
Requires-Dist: google-generativeai>=0.8.4
|
35
|
+
Requires-Dist: click>=8.0.0
|
36
|
+
Requires-Dist: rich>=13.3.1
|
37
|
+
Requires-Dist: prompt-toolkit>=3.0.36
|
38
|
+
Requires-Dist: colorama>=0.4.6
|
39
|
+
Requires-Dist: typer>=0.9.0
|
40
|
+
Requires-Dist: posthog>=3.7.0
|
41
|
+
Provides-Extra: dev
|
42
|
+
Requires-Dist: black>=24.10.0; extra == "dev"
|
43
|
+
Requires-Dist: flake8>=7.1.1; extra == "dev"
|
44
|
+
Requires-Dist: isort>=5.13.0; extra == "dev"
|
45
|
+
Requires-Dist: mypy>=1.9.0; extra == "dev"
|
46
|
+
Requires-Dist: pytest>=7.0.0; extra == "dev"
|
47
|
+
Requires-Dist: twine>=4.0.0; extra == "dev"
|
48
|
+
Requires-Dist: build>=0.10.0; extra == "dev"
|
49
|
+
Requires-Dist: types-PyYAML>=6.0; extra == "dev"
|
50
|
+
Requires-Dist: types-requests>=2.31.0; extra == "dev"
|
51
|
+
Requires-Dist: types-Markdown>=3.5.0; extra == "dev"
|
52
|
+
Requires-Dist: toml>=0.10.2; extra == "dev"
|
53
|
+
Dynamic: author
|
54
|
+
Dynamic: home-page
|
55
|
+
Dynamic: requires-python
|
56
|
+
|
57
|
+
# Airtrain
|
58
|
+
|
59
|
+
A powerful platform for building and deploying AI agents with structured skills and capabilities.
|
60
|
+
|
61
|
+
## Features
|
62
|
+
|
63
|
+
- **Structured Skills**: Build modular AI skills with defined input/output schemas
|
64
|
+
- **Multiple LLM Integrations**: Built-in support for OpenAI and Anthropic models
|
65
|
+
- **Structured Outputs**: Parse LLM responses into structured Pydantic models
|
66
|
+
- **Credential Management**: Secure handling of API keys and credentials
|
67
|
+
- **Type Safety**: Full type hints and Pydantic model support
|
68
|
+
- **Image Support**: Handle image inputs for multimodal models
|
69
|
+
- **Error Handling**: Robust error handling and logging
|
70
|
+
|
71
|
+
## Installation
|
72
|
+
|
73
|
+
```bash
|
74
|
+
pip install airtrain
|
75
|
+
```
|
76
|
+
|
77
|
+
## Quick Start
|
78
|
+
|
79
|
+
### 1. Basic OpenAI Chat
|
80
|
+
|
81
|
+
```python
|
82
|
+
from airtrain.integrations.openai.skills import OpenAIChatSkill, OpenAIInput
|
83
|
+
|
84
|
+
# Initialize the skill
|
85
|
+
skill = OpenAIChatSkill()
|
86
|
+
|
87
|
+
# Create input
|
88
|
+
input_data = OpenAIInput(
|
89
|
+
user_input="Explain quantum computing in simple terms.",
|
90
|
+
system_prompt="You are a helpful teacher.",
|
91
|
+
max_tokens=500,
|
92
|
+
temperature=0.7
|
93
|
+
)
|
94
|
+
|
95
|
+
# Get response
|
96
|
+
result = skill.process(input_data)
|
97
|
+
print(result.response)
|
98
|
+
print(f"Tokens Used: {result.usage['total_tokens']}")
|
99
|
+
```
|
100
|
+
|
101
|
+
### 2. Anthropic Claude Integration
|
102
|
+
|
103
|
+
```python
|
104
|
+
from airtrain.integrations.anthropic.skills import AnthropicChatSkill, AnthropicInput
|
105
|
+
|
106
|
+
# Initialize the skill
|
107
|
+
skill = AnthropicChatSkill()
|
108
|
+
|
109
|
+
# Create input
|
110
|
+
input_data = AnthropicInput(
|
111
|
+
user_input="Explain the theory of relativity.",
|
112
|
+
system_prompt="You are a physics expert.",
|
113
|
+
model="claude-3-opus-20240229",
|
114
|
+
temperature=0.3
|
115
|
+
)
|
116
|
+
|
117
|
+
# Get response
|
118
|
+
result = skill.process(input_data)
|
119
|
+
print(result.response)
|
120
|
+
print(f"Usage: {result.usage}")
|
121
|
+
```
|
122
|
+
|
123
|
+
### 3. Structured Output with OpenAI
|
124
|
+
|
125
|
+
```python
|
126
|
+
from pydantic import BaseModel
|
127
|
+
from typing import List
|
128
|
+
from airtrain.integrations.openai.skills import OpenAIParserSkill, OpenAIParserInput
|
129
|
+
|
130
|
+
# Define your response model
|
131
|
+
class PersonInfo(BaseModel):
|
132
|
+
name: str
|
133
|
+
age: int
|
134
|
+
occupation: str
|
135
|
+
skills: List[str]
|
136
|
+
|
137
|
+
# Initialize the parser skill
|
138
|
+
parser_skill = OpenAIParserSkill()
|
139
|
+
|
140
|
+
# Create input with response model
|
141
|
+
input_data = OpenAIParserInput(
|
142
|
+
user_input="Tell me about John Doe, a 30-year-old software engineer who specializes in Python and AI",
|
143
|
+
system_prompt="Extract structured information about the person.",
|
144
|
+
response_model=PersonInfo
|
145
|
+
)
|
146
|
+
|
147
|
+
# Get structured response
|
148
|
+
result = parser_skill.process(input_data)
|
149
|
+
person_info = result.parsed_response
|
150
|
+
print(f"Name: {person_info.name}")
|
151
|
+
print(f"Skills: {', '.join(person_info.skills)}")
|
152
|
+
```
|
153
|
+
|
154
|
+
## Error Handling
|
155
|
+
|
156
|
+
All skills include built-in error handling:
|
157
|
+
|
158
|
+
```python
|
159
|
+
from airtrain.core.skills import ProcessingError
|
160
|
+
|
161
|
+
try:
|
162
|
+
result = skill.process(input_data)
|
163
|
+
except ProcessingError as e:
|
164
|
+
print(f"Processing failed: {e}")
|
165
|
+
```
|
166
|
+
|
167
|
+
## Advanced Features
|
168
|
+
|
169
|
+
- Image Analysis Support
|
170
|
+
- Function Calling
|
171
|
+
- Custom Validators
|
172
|
+
- Async Processing
|
173
|
+
- Token Usage Tracking
|
174
|
+
|
175
|
+
For more examples and detailed documentation, visit our [documentation](https://airtrain.readthedocs.io/).
|
176
|
+
|
177
|
+
## Documentation
|
178
|
+
|
179
|
+
For detailed documentation, visit [our documentation site](https://docs.airtrain.dev/).
|
180
|
+
|
181
|
+
## Telemetry
|
182
|
+
|
183
|
+
Airtrain collects telemetry data to help improve the library. The data collected includes:
|
184
|
+
|
185
|
+
- Agent run information (model used, task description, environment settings)
|
186
|
+
- Agent steps and actions (full action details and reasoning)
|
187
|
+
- Performance metrics (token usage, execution time, CPU and memory usage)
|
188
|
+
- System information (OS, Python version, machine details)
|
189
|
+
- Error information (complete stack traces and context)
|
190
|
+
- Model usage details (prompts, responses, parameters)
|
191
|
+
|
192
|
+
The telemetry helps us identify usage patterns, troubleshoot issues, and improve the library based on real-world usage. The user ID is stored at `~/.cache/airtrain/telemetry_user_id`.
|
193
|
+
|
194
|
+
### Disabling Telemetry
|
195
|
+
|
196
|
+
Telemetry is enabled by default, but you can disable it if needed:
|
197
|
+
|
198
|
+
1. Set an environment variable:
|
199
|
+
```bash
|
200
|
+
export AIRTRAIN_TELEMETRY_ENABLED=false
|
201
|
+
```
|
202
|
+
|
203
|
+
2. In your Python code:
|
204
|
+
```python
|
205
|
+
import os
|
206
|
+
os.environ["AIRTRAIN_TELEMETRY_ENABLED"] = "false"
|
207
|
+
```
|
208
|
+
|
209
|
+
### Viewing Telemetry Debug Information
|
210
|
+
|
211
|
+
To see what telemetry data is being sent:
|
212
|
+
```python
|
213
|
+
os.environ["AIRTRAIN_LOGGING_LEVEL"] = "debug"
|
214
|
+
```
|
215
|
+
|
216
|
+
## Contributing
|
217
|
+
|
218
|
+
Contributions are welcome! Please feel free to submit a Pull Request.
|
219
|
+
|
220
|
+
## License
|
221
|
+
|
222
|
+
This project is licensed under the MIT License - see the LICENSE file for details.
|
@@ -0,0 +1,108 @@
|
|
1
|
+
airtrain/__init__.py,sha256=ZuBTmZytnwuAL8TOZICaOUJN_oUnklyejtnROu92OfA,3408
|
2
|
+
airtrain/__main__.py,sha256=EU8ffFmCdC1G-UcHHt0Oo3lB1PGqfC6kwzH39CnYSwU,72
|
3
|
+
airtrain/__pycache__/__init__.cpython-313.pyc,sha256=dpEF7frJyo9d8oGBrMXnU_0RHPVdDcv8OjYAaeD-cRY,2814
|
4
|
+
airtrain/agents/__init__.py,sha256=r6v5_bblxamRgiaCT8CVhyzaDdWohGM7sSjLgIUpA5s,795
|
5
|
+
airtrain/agents/example_agent.py,sha256=0dCS8QXIvUYYkxwyOEMLMdlQ4KgMAerQ56r7NcYGqTw,11681
|
6
|
+
airtrain/agents/groq_agent.py,sha256=s-f-cGSrgsR4Jlrv6xzeg2Z0LAEHByhFMYVbTvu-Bkg,10558
|
7
|
+
airtrain/agents/memory.py,sha256=tJEDYTjQa7WNVS1gOvVVeOzQxsXbzp21SEDWiMm1o0g,20869
|
8
|
+
airtrain/agents/registry.py,sha256=epd7xtFXvGel3d64IZJh69rWzgT45s4PIV-Teh6O_E8,13560
|
9
|
+
airtrain/builder/__init__.py,sha256=D33sr0k_WAe6FAJkk8rUaivEzFaeVqLXkQgyFWEhfPU,110
|
10
|
+
airtrain/builder/agent_builder.py,sha256=3XnGUAcK_6lWoUDtL0TanliQZuh7u0unhNbnrz1z2-I,5018
|
11
|
+
airtrain/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
12
|
+
airtrain/cli/builder.py,sha256=cI0FZCfRrgXPmjt8lOnHZwCWKrOB2doaOn49kmxVxHs,669
|
13
|
+
airtrain/cli/main.py,sha256=WGt0WXhfRl7D_UGNtCMRDWiBTBwbXcRbkEZOh9StXOo,3559
|
14
|
+
airtrain/contrib/__init__.py,sha256=pG-7mJ0pBMqp3Q86mIF9bo1PqoBOVSGlnEK1yY1U1ok,641
|
15
|
+
airtrain/contrib/travel/__init__.py,sha256=clmBodw4nkTA-DsgjVGcXfJGPaWxIpCZDtdO-8RzL0M,811
|
16
|
+
airtrain/contrib/travel/agents.py,sha256=tpQtZ0WUiXBuhvZtc2JlEam5TuR5l-Tndi14YyImDBM,8975
|
17
|
+
airtrain/contrib/travel/models.py,sha256=E4Mtds5TINmXLXu65aTYqv6wwOh2KclJHZ2eXRrBqiY,1547
|
18
|
+
airtrain/core/__init__.py,sha256=9h7iKwTzZocCPc9bU6j8bA02BokteWIOcO1uaqGMcrk,254
|
19
|
+
airtrain/core/credentials.py,sha256=J1jd8vLrOfet0GhLI1J44d35o7skjriBsMgpODmXwfo,5592
|
20
|
+
airtrain/core/schemas.py,sha256=MMXrDviC4gRea_QaPpbjgO--B_UKxnD7YrxqZOLJZZU,7003
|
21
|
+
airtrain/core/skills.py,sha256=kIkI1MwIzuAwyoxdnZ5MGOb70BOB-njbVb_csJEdVvc,9244
|
22
|
+
airtrain/core/__pycache__/__init__.cpython-313.pyc,sha256=F-LNXmzAUVNRVIygdklhVzj2xHqGxCHH_5FHRWu0XHQ,533
|
23
|
+
airtrain/core/__pycache__/schemas.cpython-313.pyc,sha256=AmWsvCsDUIZETyDD7g42gjEXoNi8lclcZ5oePV5OJ98,8326
|
24
|
+
airtrain/core/__pycache__/skills.cpython-313.pyc,sha256=B6rXeaS2OxrjrvE83CE_9ZuDmi0RI2st4Glb5WlV4B0,11838
|
25
|
+
airtrain/integrations/__init__.py,sha256=Fvr8keOQBe4PrxK7Is2B2jGodxY0J4zkcO6wT1T4e6A,2373
|
26
|
+
airtrain/integrations/anthropic/__init__.py,sha256=K741w3v7fWsCknTo38ARqDL0D3HPlwDIvDuuBao9Tto,800
|
27
|
+
airtrain/integrations/anthropic/credentials.py,sha256=hlTSw9HX66kYNaeQUtn0JjdZQBMNkzzFOJOoLOOzvcY,1246
|
28
|
+
airtrain/integrations/anthropic/list_models.py,sha256=o7FABp0Cq3gs76zOF-CM9ohmYslWT6vK9qtQabV9XzI,3973
|
29
|
+
airtrain/integrations/anthropic/models_config.py,sha256=TZt31hLcT-9YK-NxqiarMyOwvUWMgXAzAcPfSwzDSiQ,3347
|
30
|
+
airtrain/integrations/anthropic/skills.py,sha256=bs9OYHeY1ETsvXt60vq6xNj2PQK_sBULnaVGvSQ9vXQ,5872
|
31
|
+
airtrain/integrations/aws/__init__.py,sha256=3x7v2NxpAfI-U-YgwQeH5PtsmUrNLPMfLyUGFLiBjbs,155
|
32
|
+
airtrain/integrations/aws/credentials.py,sha256=nN-daKAl7qOb_VdRpsThG8gN5GeSUkx-ji5E_gF_vYw,1444
|
33
|
+
airtrain/integrations/aws/skills.py,sha256=2l16Y5zYeNd9trrPca6Rbhvl6a-GJBuCQMu7RqX9txo,3872
|
34
|
+
airtrain/integrations/cerebras/__init__.py,sha256=zAD-qV38OzHhMCz1z-NvjjqcYEhURbm8RWTOKHNqbew,174
|
35
|
+
airtrain/integrations/cerebras/credentials.py,sha256=KDEH4r8FGT68L9p34MLZWK65wq_a703pqIF3ODaSbts,694
|
36
|
+
airtrain/integrations/cerebras/skills.py,sha256=BJEb_7TglCYAukD3kcx37R8ibnJWdxVrBrwf3ZTYP-4,4924
|
37
|
+
airtrain/integrations/combined/__init__.py,sha256=EL_uZs8436abeZA6NbM-gLdur7kJr4YfGGOYybKOhjc,467
|
38
|
+
airtrain/integrations/combined/groq_fireworks_skills.py,sha256=Kz8UDU4-Rl71znz3ml9qVMRz669iWx4sUl37iafW0NU,4612
|
39
|
+
airtrain/integrations/combined/list_models_factory.py,sha256=3sQWQw3UwyzOzlNKDpRNzZiA-qN7zOIgyom7CDI0OiM,7724
|
40
|
+
airtrain/integrations/fireworks/__init__.py,sha256=GstUg0rYC-7Pg0DVbDXwL5eO1hp3WCSfroWazbGpfi0,545
|
41
|
+
airtrain/integrations/fireworks/completion_skills.py,sha256=zxx7aNlum9scQMri5Ek0qN8VfAomhyUp3u8JJo_AFWM,5615
|
42
|
+
airtrain/integrations/fireworks/conversation_manager.py,sha256=ifscKHYKWM_NDElin-oTzpRhyoh6pzBnklmMuH5geOY,3706
|
43
|
+
airtrain/integrations/fireworks/credentials.py,sha256=eeV9y_4pTe8LZX02I7kfA_YNY2D7MSbFl7JEZVn22zQ,864
|
44
|
+
airtrain/integrations/fireworks/list_models.py,sha256=o4fP0K3qstBopO7va2LysLp4_KUf5Iz_YROrYkaNtVs,4686
|
45
|
+
airtrain/integrations/fireworks/models.py,sha256=yo4xtweSi4qQftg04r4naRddx3KjU9Jluzqf5C7V9f4,4626
|
46
|
+
airtrain/integrations/fireworks/requests_skills.py,sha256=h6HRV5dGvV7t3zyjD-awW47RyeDbu8onNevhcgSSy94,8235
|
47
|
+
airtrain/integrations/fireworks/skills.py,sha256=Ns1tXXTVtTeeVYadzm4dnmmOboo430WTMu2o56oWTDc,7156
|
48
|
+
airtrain/integrations/fireworks/structured_completion_skills.py,sha256=airYakYWXzYRS9nfNfrH90N3eeN8YW7GaY3ygLSiBO8,6622
|
49
|
+
airtrain/integrations/fireworks/structured_requests_skills.py,sha256=uQR-nygtWmdGTwvU-aUdMNOMit_PiBVPYRa80ZloHLs,11852
|
50
|
+
airtrain/integrations/fireworks/structured_skills.py,sha256=1wZ_7QDUhKWCSv_1lSEF6VnAqEeEA3jWHq7n0fWicgw,3897
|
51
|
+
airtrain/integrations/google/__init__.py,sha256=ElwgcXfbg_gGMm6zbkMXCQPFKZUb-yTJk986o19A7Cs,214
|
52
|
+
airtrain/integrations/google/credentials.py,sha256=KSvWNqW8Mjr4MkysRvUqlrOSGdShNIe5u2OPO6vRrWY,2047
|
53
|
+
airtrain/integrations/google/skills.py,sha256=ytsoksCY4qbfRO9Brnxhc2694fAj0ytnHX20SXS_FOM,4547
|
54
|
+
airtrain/integrations/groq/__init__.py,sha256=qwafHN-TIzTPqNo7bGHt-W8csYmOsrbO_3WsJKbTpHY,505
|
55
|
+
airtrain/integrations/groq/credentials.py,sha256=bdTHykcIeaQ7td8KZlQBPfEFAkvJuxk2f_cbTLPD_I4,844
|
56
|
+
airtrain/integrations/groq/models_config.py,sha256=Vec4mOopVLKE6uCibV6cHxna0HN-WsZ22gF9TIbu9Vs,4853
|
57
|
+
airtrain/integrations/groq/skills.py,sha256=2_hTHhXRZmJJdcNLYwb6UpGZqWeUp-xo46Pk8BsVeFY,7403
|
58
|
+
airtrain/integrations/ollama/__init__.py,sha256=zMHBsGzViVrvxAeJmfq6r-ZfSE6Dy5QcKLhe4d5fEcM,164
|
59
|
+
airtrain/integrations/ollama/credentials.py,sha256=D7O4kUAb_VHs5s1ncUN9Ezhu5PvLfgj3RifAkB9sEZk,940
|
60
|
+
airtrain/integrations/ollama/skills.py,sha256=QHEvIrFmuwFuC3ZAmy6xL2hNNGZWii1z9Y884JuOvnI,1558
|
61
|
+
airtrain/integrations/openai/__init__.py,sha256=LYLxgDOAMUhw0ChBqj7XAJSTMNt9JiS2hHJDnJWS6IE,807
|
62
|
+
airtrain/integrations/openai/chinese_assistant.py,sha256=F8bMeUUDly7BYG6wO648cAEIj5S_frVE5tm1Xno63Gc,1583
|
63
|
+
airtrain/integrations/openai/credentials.py,sha256=NfRyp1QgEtgm8cxt2-BOLq-6d0X-Pcm80NnfHM8p0FY,1470
|
64
|
+
airtrain/integrations/openai/list_models.py,sha256=vg8pZwLZ3F2Fx42X18WykpJOzZD9JG-2KJi49XWgSKo,4121
|
65
|
+
airtrain/integrations/openai/models_config.py,sha256=W9mu_z9tCC4ZUKHSJ6Hk4X09TRZLqEhT7TtRY5JEk5g,8007
|
66
|
+
airtrain/integrations/openai/skills.py,sha256=kEDe5q0Zlv_X-JGOYtb552ktb3aQQYVUYczVwMH0jxA,12823
|
67
|
+
airtrain/integrations/perplexity/__init__.py,sha256=asVQs-oVXVhnLmAOZ6l9R8KoigeHmw8D5OONb_aGgnY,1189
|
68
|
+
airtrain/integrations/perplexity/credentials.py,sha256=5acl2DcF0dk7DQJWRwmFkBXPQ1zfKN4ARW31dP-4rpQ,1524
|
69
|
+
airtrain/integrations/perplexity/list_models.py,sha256=iOOxQ50BfLc226ps0gSdFl-enxa-SmStiuU2vFTUhJ0,4298
|
70
|
+
airtrain/integrations/perplexity/models_config.py,sha256=cMHk6kjy-u8KfxU0KmuJ9rLiquwSGuj_-SOtj9wHlM4,4060
|
71
|
+
airtrain/integrations/perplexity/skills.py,sha256=-RBwnv5wg0s-TwG--ZrCy2WikuMpNx15FGdsoNctRB0,10545
|
72
|
+
airtrain/integrations/sambanova/__init__.py,sha256=dp_263iOckM_J9pOEvyqpf3FrejD6-_x33r0edMCTe0,179
|
73
|
+
airtrain/integrations/sambanova/credentials.py,sha256=JyN8sbMCoXuXAjim46aI3LTicBijoemS7Ao0rn4yBJU,824
|
74
|
+
airtrain/integrations/sambanova/skills.py,sha256=SZ_GAimMiOCILiNkzyhNflyRR6bdC5r0Tnog19K8geU,4997
|
75
|
+
airtrain/integrations/search/__init__.py,sha256=r-hCYff5IdgI4IgQyMUzKzc_CVryl1oiweaGC3TwZ-8,405
|
76
|
+
airtrain/integrations/search/exa/__init__.py,sha256=YdINIFcuT4pC72obDYx7DHVdMzlEg03BRQJi57SKc3g,486
|
77
|
+
airtrain/integrations/search/exa/credentials.py,sha256=yOJIQdWs6RXX2Msk453VHH7FYQffpF946afaEWgVg4o,900
|
78
|
+
airtrain/integrations/search/exa/schemas.py,sha256=hCdg-J__3p666h_ep0GbKngbYqQG74XlhYrcFvJBet4,3797
|
79
|
+
airtrain/integrations/search/exa/skills.py,sha256=z2k3f-t_WZiBKj82pkmAxtaqm1-P9yU0H2J44vOJbb0,3786
|
80
|
+
airtrain/integrations/together/__init__.py,sha256=5p2v5kvryl-Z8eSMxdrMQtgzynPo6zQ6vZqpxBADA1I,878
|
81
|
+
airtrain/integrations/together/audio_models_config.py,sha256=GtqfmKR1vJ5x4B3kScvEO3x4exvzwNP78vcGVTk_fBE,1004
|
82
|
+
airtrain/integrations/together/credentials.py,sha256=cYNhyIwgsxm8LfiFfT-omBvgV3mUP6SZeRSukyzzDlI,747
|
83
|
+
airtrain/integrations/together/embedding_models_config.py,sha256=F0ISAXCG_Pcnf-ojkvZwIXacXD8LaU8hQmGHCFzmlds,2927
|
84
|
+
airtrain/integrations/together/image_models_config.py,sha256=JlCozrphI9zE4uYpGfj4DCWSN6GZGyr84Tb1HmjNQ28,2455
|
85
|
+
airtrain/integrations/together/image_skill.py,sha256=wQ8wSzfL-QHpM_esYGLNXf8ciOPPsz-QJw6zSrxZT68,5214
|
86
|
+
airtrain/integrations/together/list_models.py,sha256=_QLGqweBiK6saz3n4xTQRmXSSs-qGFnV9kma-MSaE9o,2520
|
87
|
+
airtrain/integrations/together/models.py,sha256=q5KsouOK7IvyzGZ7nhSjTpZw-CcLfPghJr6o_UU9uMo,3652
|
88
|
+
airtrain/integrations/together/models_config.py,sha256=sF4V4E0q-QgiIbghxzwVxZC7Q4RRjP_OVtm7GeDA1xw,12638
|
89
|
+
airtrain/integrations/together/rerank_models_config.py,sha256=coCg0IOG2tU4L2uc2uPtPdoBwGjSc_zQwxENwdDuwHE,1188
|
90
|
+
airtrain/integrations/together/rerank_skill.py,sha256=gjH24hLWCweWKPyyfKZMG3K_g9gWzm80WgiJNjkA9eg,1894
|
91
|
+
airtrain/integrations/together/schemas.py,sha256=pBMrbX67oxPCr-sg4K8_Xqu1DWbaC4uLCloVSascROg,1210
|
92
|
+
airtrain/integrations/together/skills.py,sha256=PU2z9YtiavZ8-ODOIXCi0stymaCsvyackTT1g2TOxZ0,11565
|
93
|
+
airtrain/integrations/together/vision_models_config.py,sha256=m28HwYDk2Kup_J-a1FtynIa2ZVcbl37kltfoHnK8zxs,1544
|
94
|
+
airtrain/telemetry/__init__.py,sha256=_xDHzSmQvRCqihT0QTmF7bS9yKEl2LaZ3Zq05hXaI3k,1108
|
95
|
+
airtrain/telemetry/service.py,sha256=6IB2-FL3fGsYYgMIVQ9MNCp9UlSE5cVhLlB3VBYkAnY,5592
|
96
|
+
airtrain/telemetry/views.py,sha256=qob1swyLNEk_UIpDi5VTwMDsEsGZhJheFQrGbP8T5hw,8115
|
97
|
+
airtrain/tools/__init__.py,sha256=AauO_EEBcK09mkyCuvvVK_Oz0L5DOrnuySViWXCOt6Y,1021
|
98
|
+
airtrain/tools/command.py,sha256=-2axK36InyWSHf0F2N9WafAsi6746akUs8er9TWSy6o,13333
|
99
|
+
airtrain/tools/filesystem.py,sha256=-YYdHj_KeSWPYXeRhWhIX9s_KujVA1R5tF3r93zRVTU,6324
|
100
|
+
airtrain/tools/network.py,sha256=YR0AtMXDXkhCsXcx7_t2d12ItnKY8XXTmyP1kdj2M4c,3883
|
101
|
+
airtrain/tools/registry.py,sha256=K-1H5EipYcDNDx2jdpsEY9gjfV4aNCGI1pY2UsgSpC0,10246
|
102
|
+
airtrain/tools/search.py,sha256=MJNi17g6aBPSqbF0ChV8ZgMlzz_PoKSPAIpe_dazdt8,15081
|
103
|
+
airtrain/tools/testing.py,sha256=q4ALEPRzukiadY6wFSPY7vA-T1o3XInLhXt18dsf6yY,4397
|
104
|
+
airtrain-0.1.4.dist-info/METADATA,sha256=6SYaBbP77-MgvA-xe6F_Xs2RlPqZixrrsGWagLrYvXk,6502
|
105
|
+
airtrain-0.1.4.dist-info/WHEEL,sha256=DK49LOLCYiurdXXOXwGJm6U4DkHkg4lcxjhqwRa0CP4,91
|
106
|
+
airtrain-0.1.4.dist-info/entry_points.txt,sha256=rrJ36IUsyq6n1dSfTWXqVAgpQLPRWDfCqwd6_3B-G0U,52
|
107
|
+
airtrain-0.1.4.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
|
108
|
+
airtrain-0.1.4.dist-info/RECORD,,
|
@@ -1,106 +0,0 @@
|
|
1
|
-
Metadata-Version: 2.2
|
2
|
-
Name: airtrain
|
3
|
-
Version: 0.1.2
|
4
|
-
Summary: A platform for building and deploying AI agents with structured skills
|
5
|
-
Home-page: https://github.com/rosaboyle/airtrain.dev
|
6
|
-
Author: Dheeraj Pai
|
7
|
-
Author-email: helloworldcmu@gmail.com
|
8
|
-
Classifier: Development Status :: 3 - Alpha
|
9
|
-
Classifier: Intended Audience :: Developers
|
10
|
-
Classifier: License :: OSI Approved :: MIT License
|
11
|
-
Classifier: Operating System :: OS Independent
|
12
|
-
Classifier: Programming Language :: Python :: 3
|
13
|
-
Classifier: Programming Language :: Python :: 3.8
|
14
|
-
Classifier: Programming Language :: Python :: 3.9
|
15
|
-
Classifier: Programming Language :: Python :: 3.10
|
16
|
-
Requires-Python: >=3.8
|
17
|
-
Description-Content-Type: text/markdown
|
18
|
-
Requires-Dist: pydantic>=2.0.0
|
19
|
-
Requires-Dist: openai>=1.0.0
|
20
|
-
Requires-Dist: python-dotenv>=0.19.0
|
21
|
-
Requires-Dist: PyYAML>=5.4.1
|
22
|
-
Requires-Dist: firebase-admin>=5.0.0
|
23
|
-
Requires-Dist: loguru>=0.5.3
|
24
|
-
Dynamic: author
|
25
|
-
Dynamic: author-email
|
26
|
-
Dynamic: classifier
|
27
|
-
Dynamic: description
|
28
|
-
Dynamic: description-content-type
|
29
|
-
Dynamic: home-page
|
30
|
-
Dynamic: requires-dist
|
31
|
-
Dynamic: requires-python
|
32
|
-
Dynamic: summary
|
33
|
-
|
34
|
-
# Airtrain
|
35
|
-
|
36
|
-
A powerful platform for building and deploying AI agents with structured skills and capabilities.
|
37
|
-
|
38
|
-
## Features
|
39
|
-
|
40
|
-
- **Structured Skills**: Build modular AI skills with defined input/output schemas
|
41
|
-
- **OpenAI Integration**: Built-in support for OpenAI's GPT models with structured outputs
|
42
|
-
- **Credential Management**: Secure handling of API keys and credentials
|
43
|
-
- **Type Safety**: Full type hints and Pydantic model support
|
44
|
-
- **Async Support**: Both synchronous and asynchronous API implementations
|
45
|
-
|
46
|
-
## Installation
|
47
|
-
|
48
|
-
```bash
|
49
|
-
pip install airtrain
|
50
|
-
```
|
51
|
-
|
52
|
-
## Quick Start
|
53
|
-
|
54
|
-
### Creating a Structured OpenAI Skill
|
55
|
-
|
56
|
-
```python
|
57
|
-
from airtrain.core.skills import Skill
|
58
|
-
from airtrain.core.schemas import InputSchema, OutputSchema
|
59
|
-
from pydantic import BaseModel
|
60
|
-
from typing import List
|
61
|
-
|
62
|
-
# Define your response model
|
63
|
-
class PersonInfo(BaseModel):
|
64
|
-
name: str
|
65
|
-
age: int
|
66
|
-
occupation: str
|
67
|
-
skills: List[str]
|
68
|
-
|
69
|
-
# Create a skill
|
70
|
-
class OpenAIParserSkill(Skill):
|
71
|
-
def process(self, input_data):
|
72
|
-
# Implementation
|
73
|
-
return parsed_response
|
74
|
-
|
75
|
-
# Use the skill
|
76
|
-
skill = OpenAIParserSkill()
|
77
|
-
result = skill.process(input_data)
|
78
|
-
```
|
79
|
-
|
80
|
-
### Managing Credentials
|
81
|
-
|
82
|
-
```python
|
83
|
-
from airtrain.core.credentials import OpenAICredentials
|
84
|
-
from pathlib import Path
|
85
|
-
|
86
|
-
# Load credentials
|
87
|
-
creds = OpenAICredentials(
|
88
|
-
api_key="your-api-key",
|
89
|
-
organization_id="optional-org-id"
|
90
|
-
)
|
91
|
-
|
92
|
-
# Save to environment
|
93
|
-
creds.load_to_env()
|
94
|
-
```
|
95
|
-
|
96
|
-
## Documentation
|
97
|
-
|
98
|
-
For detailed documentation, visit [our documentation site](https://docs.airtrain.dev/).
|
99
|
-
|
100
|
-
## Contributing
|
101
|
-
|
102
|
-
Contributions are welcome! Please feel free to submit a Pull Request.
|
103
|
-
|
104
|
-
## License
|
105
|
-
|
106
|
-
This project is licensed under the MIT License - see the LICENSE file for details.
|
airtrain-0.1.2.dist-info/RECORD
DELETED
@@ -1,5 +0,0 @@
|
|
1
|
-
airtrain/__init__.py,sha256=RZudrDcjRIFIJgG_RZ0QIyDGsQumx1AddVbSco5bliY,111
|
2
|
-
airtrain-0.1.2.dist-info/METADATA,sha256=ydUui_sePJzM4SXb2a9dwXDnZEn6QvJnw2irOPnX_G0,2786
|
3
|
-
airtrain-0.1.2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
|
4
|
-
airtrain-0.1.2.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
|
5
|
-
airtrain-0.1.2.dist-info/RECORD,,
|
File without changes
|