airtrain 0.1.53__py3-none-any.whl → 0.1.58__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 (31) hide show
  1. airtrain/__init__.py +61 -2
  2. airtrain/__pycache__/__init__.cpython-313.pyc +0 -0
  3. airtrain/agents/__init__.py +45 -0
  4. airtrain/agents/example_agent.py +348 -0
  5. airtrain/agents/groq_agent.py +289 -0
  6. airtrain/agents/memory.py +663 -0
  7. airtrain/agents/registry.py +465 -0
  8. airtrain/core/__pycache__/__init__.cpython-313.pyc +0 -0
  9. airtrain/core/__pycache__/schemas.cpython-313.pyc +0 -0
  10. airtrain/core/__pycache__/skills.cpython-313.pyc +0 -0
  11. airtrain/core/skills.py +102 -0
  12. airtrain/integrations/combined/list_models_factory.py +9 -3
  13. airtrain/integrations/groq/__init__.py +18 -1
  14. airtrain/integrations/groq/models_config.py +162 -0
  15. airtrain/integrations/groq/skills.py +93 -17
  16. airtrain/integrations/together/__init__.py +15 -1
  17. airtrain/integrations/together/models_config.py +123 -1
  18. airtrain/integrations/together/skills.py +117 -20
  19. airtrain/telemetry/__init__.py +38 -0
  20. airtrain/telemetry/service.py +167 -0
  21. airtrain/telemetry/views.py +237 -0
  22. airtrain/tools/__init__.py +41 -0
  23. airtrain/tools/command.py +211 -0
  24. airtrain/tools/filesystem.py +166 -0
  25. airtrain/tools/network.py +111 -0
  26. airtrain/tools/registry.py +320 -0
  27. {airtrain-0.1.53.dist-info → airtrain-0.1.58.dist-info}/METADATA +37 -1
  28. {airtrain-0.1.53.dist-info → airtrain-0.1.58.dist-info}/RECORD +31 -13
  29. {airtrain-0.1.53.dist-info → airtrain-0.1.58.dist-info}/WHEEL +1 -1
  30. {airtrain-0.1.53.dist-info → airtrain-0.1.58.dist-info}/entry_points.txt +0 -0
  31. {airtrain-0.1.53.dist-info → airtrain-0.1.58.dist-info}/top_level.txt +0 -0
@@ -1,5 +1,11 @@
1
- airtrain/__init__.py,sha256=i6mjHhZX92DcxUqX11yuGwkelJmvE0gUK3QwQvYg1kU,2099
1
+ airtrain/__init__.py,sha256=8g5IGtnihwqBfltBMs9vMLsTNsnhwiYSXlWBkIOpqC0,3312
2
2
  airtrain/__main__.py,sha256=EU8ffFmCdC1G-UcHHt0Oo3lB1PGqfC6kwzH39CnYSwU,72
3
+ airtrain/__pycache__/__init__.cpython-313.pyc,sha256=b5Xwx91d-BMvzIS-P4uZK8btsx2_mlxfUxvtoT4pVeA,3000
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
3
9
  airtrain/builder/__init__.py,sha256=D33sr0k_WAe6FAJkk8rUaivEzFaeVqLXkQgyFWEhfPU,110
4
10
  airtrain/builder/agent_builder.py,sha256=3XnGUAcK_6lWoUDtL0TanliQZuh7u0unhNbnrz1z2-I,5018
5
11
  airtrain/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -12,7 +18,10 @@ airtrain/contrib/travel/models.py,sha256=E4Mtds5TINmXLXu65aTYqv6wwOh2KclJHZ2eXRr
12
18
  airtrain/core/__init__.py,sha256=9h7iKwTzZocCPc9bU6j8bA02BokteWIOcO1uaqGMcrk,254
13
19
  airtrain/core/credentials.py,sha256=PgQotrQc46J5djidKnkK1znUv3fyNkUFDO-m2Kn_Gzo,4006
14
20
  airtrain/core/schemas.py,sha256=MMXrDviC4gRea_QaPpbjgO--B_UKxnD7YrxqZOLJZZU,7003
15
- airtrain/core/skills.py,sha256=LljalzeSHK5eQPTAOEAYc5D8Qn1kVSfiz9WgziTD5UM,4688
21
+ airtrain/core/skills.py,sha256=kIkI1MwIzuAwyoxdnZ5MGOb70BOB-njbVb_csJEdVvc,9244
22
+ airtrain/core/__pycache__/__init__.cpython-313.pyc,sha256=DEcKhvQGoYLFbdtCs9Fqp8sKXh8xM_viOh0drOix1cY,534
23
+ airtrain/core/__pycache__/schemas.cpython-313.pyc,sha256=DX72FLcg5BWaj-xv6yjPFZvrhYhQY7jte89gIrNGEw0,8327
24
+ airtrain/core/__pycache__/skills.cpython-313.pyc,sha256=lKR7C-bY2uS4Ad12UNSO8MLVxU-52OLJPC-nzV4u79w,11839
16
25
  airtrain/integrations/__init__.py,sha256=Y0yJt6c6uKOkMXdGPELSc6NYiLrEYr_38acViNPXWNk,2046
17
26
  airtrain/integrations/anthropic/__init__.py,sha256=K741w3v7fWsCknTo38ARqDL0D3HPlwDIvDuuBao9Tto,800
18
27
  airtrain/integrations/anthropic/credentials.py,sha256=hlTSw9HX66kYNaeQUtn0JjdZQBMNkzzFOJOoLOOzvcY,1246
@@ -27,7 +36,7 @@ airtrain/integrations/cerebras/credentials.py,sha256=KDEH4r8FGT68L9p34MLZWK65wq_
27
36
  airtrain/integrations/cerebras/skills.py,sha256=BJEb_7TglCYAukD3kcx37R8ibnJWdxVrBrwf3ZTYP-4,4924
28
37
  airtrain/integrations/combined/__init__.py,sha256=EL_uZs8436abeZA6NbM-gLdur7kJr4YfGGOYybKOhjc,467
29
38
  airtrain/integrations/combined/groq_fireworks_skills.py,sha256=Kz8UDU4-Rl71znz3ml9qVMRz669iWx4sUl37iafW0NU,4612
30
- airtrain/integrations/combined/list_models_factory.py,sha256=DA-lVRYOLlnG3J8q-Vx25s2Y3klOAYrwoOs2UuDaKRw,6582
39
+ airtrain/integrations/combined/list_models_factory.py,sha256=aLYEpJdbWVmvrrxa4e3_N0Lqq4_ZOFkByWKIqFCmJfI,7195
31
40
  airtrain/integrations/fireworks/__init__.py,sha256=GstUg0rYC-7Pg0DVbDXwL5eO1hp3WCSfroWazbGpfi0,545
32
41
  airtrain/integrations/fireworks/completion_skills.py,sha256=zxx7aNlum9scQMri5Ek0qN8VfAomhyUp3u8JJo_AFWM,5615
33
42
  airtrain/integrations/fireworks/conversation_manager.py,sha256=ifscKHYKWM_NDElin-oTzpRhyoh6pzBnklmMuH5geOY,3706
@@ -42,9 +51,10 @@ airtrain/integrations/fireworks/structured_skills.py,sha256=1wZ_7QDUhKWCSv_1lSEF
42
51
  airtrain/integrations/google/__init__.py,sha256=ElwgcXfbg_gGMm6zbkMXCQPFKZUb-yTJk986o19A7Cs,214
43
52
  airtrain/integrations/google/credentials.py,sha256=KSvWNqW8Mjr4MkysRvUqlrOSGdShNIe5u2OPO6vRrWY,2047
44
53
  airtrain/integrations/google/skills.py,sha256=ytsoksCY4qbfRO9Brnxhc2694fAj0ytnHX20SXS_FOM,4547
45
- airtrain/integrations/groq/__init__.py,sha256=B_X2fXbsJfFD6GquKeVCsEJjwd9Ygbq1uEHlV4Jy7YE,154
54
+ airtrain/integrations/groq/__init__.py,sha256=qwafHN-TIzTPqNo7bGHt-W8csYmOsrbO_3WsJKbTpHY,505
46
55
  airtrain/integrations/groq/credentials.py,sha256=bdTHykcIeaQ7td8KZlQBPfEFAkvJuxk2f_cbTLPD_I4,844
47
- airtrain/integrations/groq/skills.py,sha256=XNwGE2fjb9MDth3NI5uqSiEQYLsLBuCFF7YTu_xoTug,4823
56
+ airtrain/integrations/groq/models_config.py,sha256=Vec4mOopVLKE6uCibV6cHxna0HN-WsZ22gF9TIbu9Vs,4853
57
+ airtrain/integrations/groq/skills.py,sha256=2_hTHhXRZmJJdcNLYwb6UpGZqWeUp-xo46Pk8BsVeFY,7403
48
58
  airtrain/integrations/ollama/__init__.py,sha256=zMHBsGzViVrvxAeJmfq6r-ZfSE6Dy5QcKLhe4d5fEcM,164
49
59
  airtrain/integrations/ollama/credentials.py,sha256=D7O4kUAb_VHs5s1ncUN9Ezhu5PvLfgj3RifAkB9sEZk,940
50
60
  airtrain/integrations/ollama/skills.py,sha256=QHEvIrFmuwFuC3ZAmy6xL2hNNGZWii1z9Y884JuOvnI,1558
@@ -57,7 +67,7 @@ airtrain/integrations/openai/skills.py,sha256=kEDe5q0Zlv_X-JGOYtb552ktb3aQQYVUYc
57
67
  airtrain/integrations/sambanova/__init__.py,sha256=dp_263iOckM_J9pOEvyqpf3FrejD6-_x33r0edMCTe0,179
58
68
  airtrain/integrations/sambanova/credentials.py,sha256=JyN8sbMCoXuXAjim46aI3LTicBijoemS7Ao0rn4yBJU,824
59
69
  airtrain/integrations/sambanova/skills.py,sha256=SZ_GAimMiOCILiNkzyhNflyRR6bdC5r0Tnog19K8geU,4997
60
- airtrain/integrations/together/__init__.py,sha256=P6AH0kWgYmXGOupbLqBXCsY-KV1tcCV9UtvmJ6BGqFE,463
70
+ airtrain/integrations/together/__init__.py,sha256=5p2v5kvryl-Z8eSMxdrMQtgzynPo6zQ6vZqpxBADA1I,878
61
71
  airtrain/integrations/together/audio_models_config.py,sha256=GtqfmKR1vJ5x4B3kScvEO3x4exvzwNP78vcGVTk_fBE,1004
62
72
  airtrain/integrations/together/credentials.py,sha256=cYNhyIwgsxm8LfiFfT-omBvgV3mUP6SZeRSukyzzDlI,747
63
73
  airtrain/integrations/together/embedding_models_config.py,sha256=F0ISAXCG_Pcnf-ojkvZwIXacXD8LaU8hQmGHCFzmlds,2927
@@ -65,14 +75,22 @@ airtrain/integrations/together/image_models_config.py,sha256=JlCozrphI9zE4uYpGfj
65
75
  airtrain/integrations/together/image_skill.py,sha256=wQ8wSzfL-QHpM_esYGLNXf8ciOPPsz-QJw6zSrxZT68,5214
66
76
  airtrain/integrations/together/list_models.py,sha256=_QLGqweBiK6saz3n4xTQRmXSSs-qGFnV9kma-MSaE9o,2520
67
77
  airtrain/integrations/together/models.py,sha256=q5KsouOK7IvyzGZ7nhSjTpZw-CcLfPghJr6o_UU9uMo,3652
68
- airtrain/integrations/together/models_config.py,sha256=lCPouXjKa49lebGbMaaL2SU-CMYxOc-dJviUOir2e_w,8831
78
+ airtrain/integrations/together/models_config.py,sha256=sF4V4E0q-QgiIbghxzwVxZC7Q4RRjP_OVtm7GeDA1xw,12638
69
79
  airtrain/integrations/together/rerank_models_config.py,sha256=coCg0IOG2tU4L2uc2uPtPdoBwGjSc_zQwxENwdDuwHE,1188
70
80
  airtrain/integrations/together/rerank_skill.py,sha256=gjH24hLWCweWKPyyfKZMG3K_g9gWzm80WgiJNjkA9eg,1894
71
81
  airtrain/integrations/together/schemas.py,sha256=pBMrbX67oxPCr-sg4K8_Xqu1DWbaC4uLCloVSascROg,1210
72
- airtrain/integrations/together/skills.py,sha256=8DwkexMJu1Gm6QmNDfNasYStQ31QsXBbFP99zR-YCf0,7598
82
+ airtrain/integrations/together/skills.py,sha256=PU2z9YtiavZ8-ODOIXCi0stymaCsvyackTT1g2TOxZ0,11565
73
83
  airtrain/integrations/together/vision_models_config.py,sha256=m28HwYDk2Kup_J-a1FtynIa2ZVcbl37kltfoHnK8zxs,1544
74
- airtrain-0.1.53.dist-info/METADATA,sha256=EBQxAMR1cCJzu1rY7EYlmyHZ3tPY_mcMnxvH996objw,5375
75
- airtrain-0.1.53.dist-info/WHEEL,sha256=tTnHoFhvKQHCh4jz3yCn0WPTYIy7wXx3CJtJ7SJGV7c,91
76
- airtrain-0.1.53.dist-info/entry_points.txt,sha256=rrJ36IUsyq6n1dSfTWXqVAgpQLPRWDfCqwd6_3B-G0U,52
77
- airtrain-0.1.53.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
78
- airtrain-0.1.53.dist-info/RECORD,,
84
+ airtrain/telemetry/__init__.py,sha256=_xDHzSmQvRCqihT0QTmF7bS9yKEl2LaZ3Zq05hXaI3k,1108
85
+ airtrain/telemetry/service.py,sha256=6IB2-FL3fGsYYgMIVQ9MNCp9UlSE5cVhLlB3VBYkAnY,5592
86
+ airtrain/telemetry/views.py,sha256=qob1swyLNEk_UIpDi5VTwMDsEsGZhJheFQrGbP8T5hw,8115
87
+ airtrain/tools/__init__.py,sha256=RRqbnUqKorql5D1FDQArxpdwC5d8kUaeNzN9iZKY8V8,827
88
+ airtrain/tools/command.py,sha256=-Qmto_PqivHQBasW8PtSYoTk_NR5Wl4zz4n7u71egMQ,6983
89
+ airtrain/tools/filesystem.py,sha256=-YYdHj_KeSWPYXeRhWhIX9s_KujVA1R5tF3r93zRVTU,6324
90
+ airtrain/tools/network.py,sha256=YR0AtMXDXkhCsXcx7_t2d12ItnKY8XXTmyP1kdj2M4c,3883
91
+ airtrain/tools/registry.py,sha256=K-1H5EipYcDNDx2jdpsEY9gjfV4aNCGI1pY2UsgSpC0,10246
92
+ airtrain-0.1.58.dist-info/METADATA,sha256=3FCm8ujh-Lu1bEttg5i2GhsONwI5WwjQYAhSl2R10P0,6503
93
+ airtrain-0.1.58.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
94
+ airtrain-0.1.58.dist-info/entry_points.txt,sha256=rrJ36IUsyq6n1dSfTWXqVAgpQLPRWDfCqwd6_3B-G0U,52
95
+ airtrain-0.1.58.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
96
+ airtrain-0.1.58.dist-info/RECORD,,
@@ -1,5 +1,5 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (77.0.1)
2
+ Generator: setuptools (77.0.3)
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
5