airtrain 0.1.35__py3-none-any.whl → 0.1.37__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 CHANGED
@@ -1,6 +1,6 @@
1
1
  """Airtrain - A platform for building and deploying AI agents with structured skills"""
2
2
 
3
- __version__ = "0.1.35"
3
+ __version__ = "0.1.37"
4
4
 
5
5
  # Core imports
6
6
  from .core.skills import Skill, ProcessingError
airtrain/__main__.py ADDED
@@ -0,0 +1,4 @@
1
+ from airtrain.cli.main import cli
2
+
3
+ if __name__ == "__main__":
4
+ cli()
@@ -0,0 +1,23 @@
1
+ import click
2
+ from airtrain.builder.agent_builder import AgentBuilder
3
+ import json
4
+
5
+
6
+ @click.command()
7
+ def build():
8
+ """Build a custom AI agent through an interactive process"""
9
+ try:
10
+ builder = AgentBuilder()
11
+ specification = builder.build_agent()
12
+
13
+ # Display the final specification
14
+ click.echo("\n=== Agent Specification ===")
15
+ click.echo(json.dumps(specification.model_dump(), indent=2))
16
+
17
+ click.echo(
18
+ "\nAgent specification complete! You can now use this specification to initialize your agent."
19
+ )
20
+
21
+ except Exception as e:
22
+ click.echo(f"\nError building agent: {str(e)}")
23
+ return 1
airtrain/cli/main.py CHANGED
@@ -5,6 +5,7 @@ from airtrain.integrations.anthropic.skills import AnthropicChatSkill, Anthropic
5
5
  import os
6
6
  from dotenv import load_dotenv
7
7
  import sys
8
+ from .builder import build
8
9
 
9
10
  # Load environment variables
10
11
  load_dotenv()
@@ -106,5 +107,5 @@ def chat(provider: str, temperature: float, system_prompt: str):
106
107
  sys.exit(1)
107
108
 
108
109
 
109
- if __name__ == "__main__":
110
- cli()
110
+ # Add to existing cli group
111
+ cli.add_command(build)
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.2
2
2
  Name: airtrain
3
- Version: 0.1.35
3
+ Version: 0.1.37
4
4
  Summary: A platform for building and deploying AI agents with structured skills
5
5
  Home-page: https://github.com/rosaboyle/airtrain.dev
6
6
  Author: Dheeraj Pai
@@ -1,6 +1,8 @@
1
- airtrain/__init__.py,sha256=HxQoDsFRceRrgV-8HMhQ9NQYkwqFREBJoxpqHBVcE40,2099
1
+ airtrain/__init__.py,sha256=WbNEBdO9_ZcI0yNOBlO_zo7g8oZ_piJcTBuG6doFpNo,2099
2
+ airtrain/__main__.py,sha256=EU8ffFmCdC1G-UcHHt0Oo3lB1PGqfC6kwzH39CnYSwU,72
2
3
  airtrain/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
3
- airtrain/cli/main.py,sha256=Y6QxIcoqctJNn7_HLRvOb3WhF9MhBeK3KacpEV-Xwto,3415
4
+ airtrain/cli/builder.py,sha256=cI0FZCfRrgXPmjt8lOnHZwCWKrOB2doaOn49kmxVxHs,669
5
+ airtrain/cli/main.py,sha256=Sf4-I4ENlZZ_AehPEaf1rdebZEMCv5K41sSZTnrH9y4,3456
4
6
  airtrain/contrib/__init__.py,sha256=pG-7mJ0pBMqp3Q86mIF9bo1PqoBOVSGlnEK1yY1U1ok,641
5
7
  airtrain/contrib/travel/__init__.py,sha256=clmBodw4nkTA-DsgjVGcXfJGPaWxIpCZDtdO-8RzL0M,811
6
8
  airtrain/contrib/travel/agents.py,sha256=tpQtZ0WUiXBuhvZtc2JlEam5TuR5l-Tndi14YyImDBM,8975
@@ -59,8 +61,8 @@ airtrain/integrations/together/rerank_skill.py,sha256=gjH24hLWCweWKPyyfKZMG3K_g9
59
61
  airtrain/integrations/together/schemas.py,sha256=pBMrbX67oxPCr-sg4K8_Xqu1DWbaC4uLCloVSascROg,1210
60
62
  airtrain/integrations/together/skills.py,sha256=8DwkexMJu1Gm6QmNDfNasYStQ31QsXBbFP99zR-YCf0,7598
61
63
  airtrain/integrations/together/vision_models_config.py,sha256=m28HwYDk2Kup_J-a1FtynIa2ZVcbl37kltfoHnK8zxs,1544
62
- airtrain-0.1.35.dist-info/METADATA,sha256=puapNvVJkJ9KZ5BBsnrkQ5aFEetk52M79ePuZd6EOCo,5375
63
- airtrain-0.1.35.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
64
- airtrain-0.1.35.dist-info/entry_points.txt,sha256=rrJ36IUsyq6n1dSfTWXqVAgpQLPRWDfCqwd6_3B-G0U,52
65
- airtrain-0.1.35.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
66
- airtrain-0.1.35.dist-info/RECORD,,
64
+ airtrain-0.1.37.dist-info/METADATA,sha256=bnCDFNSAXHCwH7K8Gdnile1QREhrOB19MIdua51Ktr8,5375
65
+ airtrain-0.1.37.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
66
+ airtrain-0.1.37.dist-info/entry_points.txt,sha256=rrJ36IUsyq6n1dSfTWXqVAgpQLPRWDfCqwd6_3B-G0U,52
67
+ airtrain-0.1.37.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
68
+ airtrain-0.1.37.dist-info/RECORD,,