airtrain 0.1.36__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.36"
3
+ __version__ = "0.1.37"
4
4
 
5
5
  # Core imports
6
6
  from .core.skills import Skill, ProcessingError
@@ -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()
@@ -104,3 +105,7 @@ def chat(provider: str, temperature: float, system_prompt: str):
104
105
  except Exception as e:
105
106
  click.echo(f"Failed to initialize chat: {str(e)}")
106
107
  sys.exit(1)
108
+
109
+
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.36
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,7 +1,8 @@
1
- airtrain/__init__.py,sha256=lpwUTZWEEI75R6-mR0i-dY5JRXYcjI5KVFJpHb9VCNg,2099
1
+ airtrain/__init__.py,sha256=WbNEBdO9_ZcI0yNOBlO_zo7g8oZ_piJcTBuG6doFpNo,2099
2
2
  airtrain/__main__.py,sha256=EU8ffFmCdC1G-UcHHt0Oo3lB1PGqfC6kwzH39CnYSwU,72
3
3
  airtrain/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
4
- airtrain/cli/main.py,sha256=ix6P4b99uuFYVjhBh_Mo76gmr5FndURYSs-Gm5xpgw0,3376
4
+ airtrain/cli/builder.py,sha256=cI0FZCfRrgXPmjt8lOnHZwCWKrOB2doaOn49kmxVxHs,669
5
+ airtrain/cli/main.py,sha256=Sf4-I4ENlZZ_AehPEaf1rdebZEMCv5K41sSZTnrH9y4,3456
5
6
  airtrain/contrib/__init__.py,sha256=pG-7mJ0pBMqp3Q86mIF9bo1PqoBOVSGlnEK1yY1U1ok,641
6
7
  airtrain/contrib/travel/__init__.py,sha256=clmBodw4nkTA-DsgjVGcXfJGPaWxIpCZDtdO-8RzL0M,811
7
8
  airtrain/contrib/travel/agents.py,sha256=tpQtZ0WUiXBuhvZtc2JlEam5TuR5l-Tndi14YyImDBM,8975
@@ -60,8 +61,8 @@ airtrain/integrations/together/rerank_skill.py,sha256=gjH24hLWCweWKPyyfKZMG3K_g9
60
61
  airtrain/integrations/together/schemas.py,sha256=pBMrbX67oxPCr-sg4K8_Xqu1DWbaC4uLCloVSascROg,1210
61
62
  airtrain/integrations/together/skills.py,sha256=8DwkexMJu1Gm6QmNDfNasYStQ31QsXBbFP99zR-YCf0,7598
62
63
  airtrain/integrations/together/vision_models_config.py,sha256=m28HwYDk2Kup_J-a1FtynIa2ZVcbl37kltfoHnK8zxs,1544
63
- airtrain-0.1.36.dist-info/METADATA,sha256=fHfGlJedMIhgMn0igC6DSS1DAzivUrblEpDxWA8_X9I,5375
64
- airtrain-0.1.36.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
65
- airtrain-0.1.36.dist-info/entry_points.txt,sha256=rrJ36IUsyq6n1dSfTWXqVAgpQLPRWDfCqwd6_3B-G0U,52
66
- airtrain-0.1.36.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
67
- airtrain-0.1.36.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,,