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.
Files changed (110) hide show
  1. airtrain/__init__.py +148 -2
  2. airtrain/__main__.py +4 -0
  3. airtrain/__pycache__/__init__.cpython-313.pyc +0 -0
  4. airtrain/agents/__init__.py +45 -0
  5. airtrain/agents/example_agent.py +348 -0
  6. airtrain/agents/groq_agent.py +289 -0
  7. airtrain/agents/memory.py +663 -0
  8. airtrain/agents/registry.py +465 -0
  9. airtrain/builder/__init__.py +3 -0
  10. airtrain/builder/agent_builder.py +122 -0
  11. airtrain/cli/__init__.py +0 -0
  12. airtrain/cli/builder.py +23 -0
  13. airtrain/cli/main.py +120 -0
  14. airtrain/contrib/__init__.py +29 -0
  15. airtrain/contrib/travel/__init__.py +35 -0
  16. airtrain/contrib/travel/agents.py +243 -0
  17. airtrain/contrib/travel/models.py +59 -0
  18. airtrain/core/__init__.py +7 -0
  19. airtrain/core/__pycache__/__init__.cpython-313.pyc +0 -0
  20. airtrain/core/__pycache__/schemas.cpython-313.pyc +0 -0
  21. airtrain/core/__pycache__/skills.cpython-313.pyc +0 -0
  22. airtrain/core/credentials.py +171 -0
  23. airtrain/core/schemas.py +237 -0
  24. airtrain/core/skills.py +269 -0
  25. airtrain/integrations/__init__.py +74 -0
  26. airtrain/integrations/anthropic/__init__.py +33 -0
  27. airtrain/integrations/anthropic/credentials.py +32 -0
  28. airtrain/integrations/anthropic/list_models.py +110 -0
  29. airtrain/integrations/anthropic/models_config.py +100 -0
  30. airtrain/integrations/anthropic/skills.py +155 -0
  31. airtrain/integrations/aws/__init__.py +6 -0
  32. airtrain/integrations/aws/credentials.py +36 -0
  33. airtrain/integrations/aws/skills.py +98 -0
  34. airtrain/integrations/cerebras/__init__.py +6 -0
  35. airtrain/integrations/cerebras/credentials.py +19 -0
  36. airtrain/integrations/cerebras/skills.py +127 -0
  37. airtrain/integrations/combined/__init__.py +21 -0
  38. airtrain/integrations/combined/groq_fireworks_skills.py +126 -0
  39. airtrain/integrations/combined/list_models_factory.py +210 -0
  40. airtrain/integrations/fireworks/__init__.py +21 -0
  41. airtrain/integrations/fireworks/completion_skills.py +147 -0
  42. airtrain/integrations/fireworks/conversation_manager.py +109 -0
  43. airtrain/integrations/fireworks/credentials.py +26 -0
  44. airtrain/integrations/fireworks/list_models.py +128 -0
  45. airtrain/integrations/fireworks/models.py +139 -0
  46. airtrain/integrations/fireworks/requests_skills.py +207 -0
  47. airtrain/integrations/fireworks/skills.py +181 -0
  48. airtrain/integrations/fireworks/structured_completion_skills.py +175 -0
  49. airtrain/integrations/fireworks/structured_requests_skills.py +291 -0
  50. airtrain/integrations/fireworks/structured_skills.py +102 -0
  51. airtrain/integrations/google/__init__.py +7 -0
  52. airtrain/integrations/google/credentials.py +58 -0
  53. airtrain/integrations/google/skills.py +122 -0
  54. airtrain/integrations/groq/__init__.py +23 -0
  55. airtrain/integrations/groq/credentials.py +24 -0
  56. airtrain/integrations/groq/models_config.py +162 -0
  57. airtrain/integrations/groq/skills.py +201 -0
  58. airtrain/integrations/ollama/__init__.py +6 -0
  59. airtrain/integrations/ollama/credentials.py +26 -0
  60. airtrain/integrations/ollama/skills.py +41 -0
  61. airtrain/integrations/openai/__init__.py +37 -0
  62. airtrain/integrations/openai/chinese_assistant.py +42 -0
  63. airtrain/integrations/openai/credentials.py +39 -0
  64. airtrain/integrations/openai/list_models.py +112 -0
  65. airtrain/integrations/openai/models_config.py +224 -0
  66. airtrain/integrations/openai/skills.py +342 -0
  67. airtrain/integrations/perplexity/__init__.py +49 -0
  68. airtrain/integrations/perplexity/credentials.py +43 -0
  69. airtrain/integrations/perplexity/list_models.py +112 -0
  70. airtrain/integrations/perplexity/models_config.py +128 -0
  71. airtrain/integrations/perplexity/skills.py +279 -0
  72. airtrain/integrations/sambanova/__init__.py +6 -0
  73. airtrain/integrations/sambanova/credentials.py +20 -0
  74. airtrain/integrations/sambanova/skills.py +129 -0
  75. airtrain/integrations/search/__init__.py +21 -0
  76. airtrain/integrations/search/exa/__init__.py +23 -0
  77. airtrain/integrations/search/exa/credentials.py +30 -0
  78. airtrain/integrations/search/exa/schemas.py +114 -0
  79. airtrain/integrations/search/exa/skills.py +115 -0
  80. airtrain/integrations/together/__init__.py +33 -0
  81. airtrain/integrations/together/audio_models_config.py +34 -0
  82. airtrain/integrations/together/credentials.py +22 -0
  83. airtrain/integrations/together/embedding_models_config.py +92 -0
  84. airtrain/integrations/together/image_models_config.py +69 -0
  85. airtrain/integrations/together/image_skill.py +143 -0
  86. airtrain/integrations/together/list_models.py +76 -0
  87. airtrain/integrations/together/models.py +95 -0
  88. airtrain/integrations/together/models_config.py +399 -0
  89. airtrain/integrations/together/rerank_models_config.py +43 -0
  90. airtrain/integrations/together/rerank_skill.py +49 -0
  91. airtrain/integrations/together/schemas.py +33 -0
  92. airtrain/integrations/together/skills.py +305 -0
  93. airtrain/integrations/together/vision_models_config.py +49 -0
  94. airtrain/telemetry/__init__.py +38 -0
  95. airtrain/telemetry/service.py +167 -0
  96. airtrain/telemetry/views.py +237 -0
  97. airtrain/tools/__init__.py +45 -0
  98. airtrain/tools/command.py +398 -0
  99. airtrain/tools/filesystem.py +166 -0
  100. airtrain/tools/network.py +111 -0
  101. airtrain/tools/registry.py +320 -0
  102. airtrain/tools/search.py +450 -0
  103. airtrain/tools/testing.py +135 -0
  104. airtrain-0.1.4.dist-info/METADATA +222 -0
  105. airtrain-0.1.4.dist-info/RECORD +108 -0
  106. {airtrain-0.1.2.dist-info → airtrain-0.1.4.dist-info}/WHEEL +1 -1
  107. airtrain-0.1.4.dist-info/entry_points.txt +2 -0
  108. airtrain-0.1.2.dist-info/METADATA +0 -106
  109. airtrain-0.1.2.dist-info/RECORD +0 -5
  110. {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,5 +1,5 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (75.8.0)
2
+ Generator: setuptools (78.0.2)
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
5
 
@@ -0,0 +1,2 @@
1
+ [console_scripts]
2
+ airtrain = airtrain.cli.main:main
@@ -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.
@@ -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,,