plato-sdk-v2 2.0.64__py3-none-any.whl → 2.3.4__py3-none-any.whl

{plato_sdk_v2-2.0.64.dist-info → plato_sdk_v2-2.3.4.dist-info}/RECORD

@@ -1,4 +1,4 @@
-plato/__init__.py,sha256=5e5V0xR4tbIT6NfGo5DNilmd2TSLe0mBRjFR1anuFP4,2014
+plato/__init__.py,sha256=a9E0KS1602GWHHStnf7wDEuvPCvh2GpPh0Sf8oKZx5Q,1795
 plato/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 plato/_generated/__init__.py,sha256=8BNQD2tT98Nrl8jP97bIzUxFTkb1681a1kbJExyZYuo,738
 plato/_generated/client.py,sha256=_oMKXyAShQVddCaIKnfB2zPkRsDlCwLp-N3RFoKq_v8,5489
@@ -283,10 +283,13 @@ plato/_generated/api/v2/user/get_current_user.py,sha256=tvamtbWTEkeeNUBLSPqZIcCG
 plato/_generated/api/version/__init__.py,sha256=dQXTYrXjD1RZcvWwnlqXWAZ-eAV-V-6JSNuY7uaca7o,70
 plato/_generated/api/version/check.py,sha256=HTVNw0oi9gbvX4pOVoH4y4JywCxdl1pJTCk2PjJFwJ4,778
 plato/_generated/models/__init__.py,sha256=VGsTMYUCCqa7r_tl5m--dwjy7viYAN20nU2HhVAibcU,154130
-plato/_sims_generator/__init__.py,sha256=Cg8atlEoWsd5nf0qLz3_EbHxFVFtBwPDDUZbx3jxRrc,1125
+plato/_sims_generator/__init__.py,sha256=Km4QOl9wxjQ5dgpdhk9QnBFJFFc9eq3rPbMWIQRjIn0,1602
 plato/_sims_generator/cli.py,sha256=mzolN-dxfMkVAdA-vC0esnai-cGg-i4ozOw8dACefV4,2709
+plato/_sims_generator/instruction.py,sha256=Na9M-jIdBPhp_fLuBPTicoFnWriRyi8YiZ-eQBj64HI,6644
 plato/_sims_generator/parser.py,sha256=BbgRYllqYf7H76JyMfe7LYo1we-kh7YEOxUwrYT3shc,38347
 plato/_sims_generator/python.py,sha256=ZYZJeOhGhYV95iB2_G-stQHaIv5sQj0jo3GXSLb6SdA,18295
+plato/_sims_generator/templates/instruction/helpers.py.jinja,sha256=kJ8lhVLPygjMh8l-In9rLtFci6Bxg3zVoz5BZFfLKms,4937
+plato/_sims_generator/templates/instruction/init.py.jinja,sha256=oabz8xKdkz9SjRBYwg_GoHZdz1ZrhHXOOEazw2pOczQ,879
 plato/_sims_generator/templates/python/api_init.py.jinja,sha256=HYNtVsiGU91IzhBtBA3qlLv8b3U5kZbmW-FqdZ9F7mQ,188
 plato/_sims_generator/templates/python/client.py.jinja,sha256=H99L4kgRfTwJs3DOC9P-DcGtHNYBDNxcOvtedWEgWYM,19755
 plato/_sims_generator/templates/python/endpoint.py.jinja,sha256=1ToZiS8w_aZydVBhNt5KxLIbCo3toWCi4NPjLAmhVy4,12478
@@ -294,8 +297,13 @@ plato/_sims_generator/templates/python/errors.py.jinja,sha256=8L_FbHczBNLXJrbSlN
 plato/_sims_generator/templates/python/package_init.py.jinja,sha256=sOcJxUT0LuOWu5jOMGGKYxfCEjcYQv1hGF3n0iOA4hQ,986
 plato/_sims_generator/templates/python/tag_init.py.jinja,sha256=WB_9cv0JKIVg5TOXeSolET3tAfVg7sExjboh5jbCXz4,170
 plato/_sims_generator/templates/python/version_init.py.jinja,sha256=sGvFcYVfzXFyQDAe0PSOrg9yys93KE0XInFQNb1TvCY,179
-plato/agents/__init__.py,sha256=x_mdGT686769OPSBMN70VRIcfVdUmduLEsj325jAyPA,14578
-plato/agents/callback.py,sha256=9ziAH72HX6CDvMy_IrzzvSOlFDnhdz2TDG4hCmbc8IU,8188
+plato/agents/__init__.py,sha256=1l-Vw5SFydmPsl1toXdc2oH3eLhFwz6jtI_DKUx4vmo,2847
+plato/agents/base.py,sha256=vUbPQuNSo6Ka2lIB_ZOXgi4EoAjtAD7GIj9LnNotam0,4577
+plato/agents/build.py,sha256=CNMbVQFs2_pYit1dA29Davve28Yi4c7TNK9wBB7odrE,1621
+plato/agents/config.py,sha256=VZVMdCmEQnoR0VkrGdScG8p6zSKVFe7BZPd2h8lKNjI,5460
+plato/agents/logging.py,sha256=bL3Q14YeFxWCQydGKrQzQPgKQzBoCCXy_AAQPZpBylU,16307
+plato/agents/runner.py,sha256=UQWg3gpYhWLnGLB6Bs595qH7nZV-ch-u5fPlkltT_kc,6812
+plato/agents/trajectory.py,sha256=WdiBmua0KvCrNaM3qgPI7-7B4xmSkfbP4oZ_9_8qHzU,10529
 plato/chronos/__init__.py,sha256=RHMvSrQS_-vkKOyTRuAkp2gKDP1HEuBLDnw8jcZs1Jg,739
 plato/chronos/client.py,sha256=YcOGtHWERyOD9z8LKt8bRMVL0cEwL2hiAP4qQgdZlUI,5495
 plato/chronos/errors.py,sha256=xqQIQB43nAL5urF8qc_1KUJql7KCnspULOFHLNnf83M,4199
@@ -364,32 +372,33 @@ plato/chronos/api/worlds/create_world.py,sha256=H6yl5QIazNXgryOR5rvscSIMf8Y9kjc6
 plato/chronos/api/worlds/delete_world.py,sha256=UETu3Zk0e2VkDdAyMilv1ev-0g_j-oujH1Dc8DBqQOc,1239
 plato/chronos/api/worlds/get_world.py,sha256=eHTM1U5JiNTaZwYLh7x4QVBoRQeI5kaJ9o6xSi4-nos,1356
 plato/chronos/api/worlds/list_worlds.py,sha256=hBAuGb69tlasyn-kV_LNr9x6Rr7SHhST5hXJn1uqMf8,1253
-plato/chronos/models/__init__.py,sha256=Iyh5srlL4oqsviCnjk9xHW7g0BjgJR4lAthFkZoXafs,21008
+plato/chronos/models/__init__.py,sha256=5Hil8v_jFX1YU6LpOfqyJM4WV867Ckv6CX052Q4SCso,20996
 plato/sims/README.md,sha256=FIbJhNVNAV-SO6dq_cXX3Rg0C7HdQCfEY9YxGlkCmsM,6902
 plato/sims/__init__.py,sha256=tnoCGKZwNx6h22tEWLujdpLv6K4PpFU2RnDOhL1o-Uc,1494
 plato/sims/agent_helpers.py,sha256=kITvQywoTCS8mGhro3jZWuPJHDlje-UZujhjoahqhd0,10291
-plato/sims/cli.py,sha256=ySG5TNh9L2E-bgP0Jv4sbewReaoPw5jDYFAQk_jubuw,46093
+plato/sims/cli.py,sha256=lvdc_fSyNVmXvblFaE0saiVPdTpoR8Tlh9NN2LCBrd8,52704
 plato/sims/generate_clients.py,sha256=nEe39v3UOcks-ggv5jomcwN33R5U9n8MDNCpHoZ2lDg,5958
-plato/sims/registry.py,sha256=_xtjcI4HHGBsJ4c0kzxK90U2B_h6FK-5FAoAwyWZokQ,8777
+plato/sims/registry.py,sha256=1LHW1SMJJNYIvn0zY9BPkHo4yHpdDAk7BxTYw1HpweY,11963
 plato/v1/__init__.py,sha256=t1Ejb7YCFOVSKZL9hJ0UsmEE1mULDvO__f75dz1UueQ,197
 plato/v1/audit_ui.py,sha256=zYYufJKn7291uCNb_59ItmDTYsPr7YLshBFwcAwl1LQ,10990
 plato/v1/cli.py,sha256=iEt58vvW7ab9YH0CLcBHvf4653fk1gcEdij4HZc10YY,269
 plato/v1/config.py,sha256=Zc5qGBcR8UfLuZrfQhkZ5cUmV_9pPKHPyKpU0SNysGY,667
 plato/v1/exceptions.py,sha256=BS5A0NLZ2FgL6YVaIL-c8FXASTsOpuKrF3_Pb0tad6w,109
-plato/v1/flow_executor.py,sha256=DrpcWECW120P6WlS2Ol7AVcGHgKiJ-BFmYQi1_fdENw,19010
+plato/v1/flow_executor.py,sha256=rHYPPG4XPdq00Ty7_c4r7dZ-LixtUrEV3KEg1i-Owto,19720
 plato/v1/sandbox_sdk.py,sha256=5-ESB3spe4BFGn1dVfgSWwKyfQeeUokPWsmbti3obOY,41854
-plato/v1/sdk.py,sha256=b-mDCwVtxWbVsHEZ1x5m1kJ4Hxs57Y7l-UBR2aUP0dY,33323
-plato/v1/sync_env.py,sha256=qClrMAe5CNj6KX8lDhHJi7GIToDkm00o8KNAYfJMf_Q,24793
-plato/v1/sync_flow_executor.py,sha256=NTRKtlAeuFct6kKWRQ4LsRwnGC1Yo58ICN91dDzOjBg,18523
-plato/v1/sync_sdk.py,sha256=ztAvqR5yW_ALj8g97K6U-jSQzwzVNVKWlGZ-tdMUP_M,25620
+plato/v1/sdk.py,sha256=zNcYgANeMZG8r9AH3POJhJ11UQcOvdsLfUFABoB1cwg,33382
+plato/v1/sync_env.py,sha256=UIfDpx3nPHBNWzahSddvol0SvfK9vU4mr3MOIOPCGr8,24878
+plato/v1/sync_flow_executor.py,sha256=kgvNYOtA9FHeNfP7qb8ZPUIlTsfIss_Z98W8uX5veck,19233
+plato/v1/sync_sdk.py,sha256=2sedg1QJiSxr1I3kCyfaLAnlAgHlbblc3QQP_47O30k,25697
 plato/v1/cli/__init__.py,sha256=om4b7PxgsoI7rEwuQelmQkqPdhMVn53_5qEN8kvksYw,105
-plato/v1/cli/agent.py,sha256=mdQrzEoFuMQnb_kMcIAkE7DbSak5Zv7A5RhPtXq-5rQ,44028
+plato/v1/cli/agent.py,sha256=G6TV3blG_BqMDBWS-CG7GwzqoqcJTMsIKQ88jvLXb4k,43745
 plato/v1/cli/main.py,sha256=ktPtBvMwykR7AjXmTQ6bmZkHdzpAjhX5Fq66cDbGSzA,6844
-plato/v1/cli/pm.py,sha256=sIrh7qVyL-GOI8cgCUDFO2StXCxaKkCpGoN2waMqiyw,47579
-plato/v1/cli/sandbox.py,sha256=x0QE7ojZNI9WlKWGGxBVbLzwmkDfv80LUCz_HIWLNW8,87759
-plato/v1/cli/ssh.py,sha256=pePHD0lFPwSkATZYSannpFtHfJWKImAdLyS2463aRRw,6287
+plato/v1/cli/pm.py,sha256=uLM6WszKqxq9Czg1FraDyWb9_INUuHZq63imvRYfRLw,49734
+plato/v1/cli/sandbox.py,sha256=5rth_jL73L72GC0VJ0meXRgZo2EpsJ_qI3ipFjfXzJY,95185
+plato/v1/cli/ssh.py,sha256=10ag6S1sxMcAmvcg24qy-yYwXb1miWfqxkXOs4QX8u0,6623
 plato/v1/cli/utils.py,sha256=be-llK6T6NHnIQl_Kfs-8EPu9JhIuZ_k9tJ3Ts-AKt4,3887
-plato/v1/cli/world.py,sha256=f-65n70-V2EBcYs9hbYqCXVT8n2xm7opIgmB5Z6yi3Y,9580
+plato/v1/cli/verify.py,sha256=7QmQwfOOkr8a51f8xfVIr2zif7wGl2E8HOZTbOaIoV0,20671
+plato/v1/cli/world.py,sha256=yBUadOJs1QYm6Jmx_ACDzogybRq5x4B-BnTvGO_ulQk,9757
 plato/v1/examples/doordash_tasks.py,sha256=8Sz9qx-vTmiOAiCAbrDRvZGsA1qQQBr1KHbxXdjr7OI,23233
 plato/v1/examples/loadtest.py,sha256=ZsQYNN_fZjE7CbrbVJb4KDc0OLaH7b66iPrEHDhuw0U,5609
 plato/v1/examples/test_env.py,sha256=8kUISbZyMi0Xh9HK7Il1okKQyz0Iq-vAKWgzC8kqUfU,4513
@@ -428,41 +437,37 @@ plato/v1/extensions/envgen-recorder/sessions.js,sha256=IrW0elMYYh4cUEmNZiHC-XhFL
 plato/v1/extensions/envgen-recorder/styles.css,sha256=NbpKw4NSbYw6RVDOn4zpNwwVnjFQtKIene2MZBz9kb0,5359
 plato/v1/models/__init__.py,sha256=rnv5oO0RQ_mbnv_0FxvCWpCnSKmBePoYoIQ0h1LuKE8,813
 plato/v1/models/build_models.py,sha256=zcRwg2pzU5GFDp_Qi7g0UCVNGkOtXTlHNvq_6gm5mJc,10201
-plato/v1/models/env.py,sha256=UClRO8wgpZutcwGsAc1Sk3tnZDucQkkT-qH_WluX77o,30958
+plato/v1/models/env.py,sha256=UZgdjZrIkODeuTLTHaIyUqUJM83S8YYY_RbYqDeYjDk,31043
 plato/v1/models/flow.py,sha256=bkleb7-OR6V5tzPtggf0ZJhHZQwnktCYr1C-ONpRzwE,6920
 plato/v1/models/sandbox.py,sha256=yRN036G91tFAXGxU1ni7zCU1o7U1N8nI0mjLlBRDoSk,3478
 plato/v1/models/task.py,sha256=QXwdFpDM_NLjRpQSK6duibXJXFAPZ8-PpyuLWZC5o4I,4897
 plato/v1/utils/proxytunnel.py,sha256=N1MshFY3Pu3d19cTU619-e-gP2_j-89CEcld7lCoiGk,5706
-plato/v2/__init__.py,sha256=lWegWEfRZu2ofNgkyQLH77Rt2iodtN6U0yVObllGwEY,1583
+plato/v2/__init__.py,sha256=YUzdRnXsL61ey-LOoR-HZGYgfGKBxnuMtFIu-D0VB8E,1662
 plato/v2/models.py,sha256=41EJPdbAsW-gARVND0cg4jqpzQuYlrbAW9hF8WI8kX4,2369
 plato/v2/types.py,sha256=MrHiE8AobOegeTfoKKHZ5JTccaFfS3-EW0arV7q8S5c,3533
 plato/v2/async_/__init__.py,sha256=zcNwuX2XD-WkUCFLvXtdRqFmCgeuY3-D4zGTRVwsAFI,484
 plato/v2/async_/artifact.py,sha256=JBWVQeVaZhkU2qn_knyzyA7wd5iQ8qxfLQ_l9GPhgYs,1217
 plato/v2/async_/client.py,sha256=GVgAgNN5gsDME8iV0zxqnwbsVS93J6cknOcq_VXwYN8,4209
-plato/v2/async_/environment.py,sha256=Dv_4QuQx5vLVp1m1mJR2fXybvbhz-Q4gLtZecCdyFT8,4622
+plato/v2/async_/environment.py,sha256=M5IeWYLwREOIyuS2zqgBSqHE_x66_OZXrevA9Rkc8Is,5825
 plato/v2/async_/flow_executor.py,sha256=Tl4nRu1ZPWJFNNxyTGy-PxvebZEUD18ZDaz8T2chtzU,14188
-plato/v2/async_/session.py,sha256=ZMqT0Bjyr9lmSV58XpnUw_MddD3-XItzSl7MM73phss,34086
+plato/v2/async_/session.py,sha256=ssTEFgfpDQElQEmeKulLWKLoW-wA3m_nI3UvEWb9lB0,36749
 plato/v2/sync/__init__.py,sha256=_WigxuehCC8A2yRy4mSaMQmIZhS0Gch4hhZC1VHyYXs,310
 plato/v2/sync/artifact.py,sha256=wTLC-tugG128wLvh-JqNPb0zsw5FXEJlZNahurSWink,1169
 plato/v2/sync/client.py,sha256=Q9fS1BF4KxTMMnceMwCMlb5dNFZ6LA4gsXWNLgsL2eE,3870
-plato/v2/sync/environment.py,sha256=EJS_MBXHQgdOYztmnvcDPyQV1Z0dV6BA-7UEJ_Z-bgE,3878
+plato/v2/sync/environment.py,sha256=WnDzbyEHpwCSEP8XnfNSjIYS7rt7lYR4HGJjzprZmTQ,5066
 plato/v2/sync/flow_executor.py,sha256=N41-WCWIJVcCR2UmPUEiK7roNacYoeONkRXpR7lUgT8,13941
-plato/v2/sync/session.py,sha256=yVSrUlViN-ge-KPdvfcbZ-lJWT0CFZJt083ETrvhf38,25775
+plato/v2/sync/session.py,sha256=jc71bZFb-1m4QTtU4V00OkNH8fHac3jyWTwq-uOK-BE,28409
 plato/v2/utils/__init__.py,sha256=XLeFFsjXkm9g2raMmo7Wt4QN4hhCrNZDJKnpffJ4LtM,38
 plato/v2/utils/db_cleanup.py,sha256=lnI5lsMHNHpG85Y99MaE4Rzc3618piuzhvH-uXO1zIc,8702
 plato/v2/utils/models.py,sha256=PwehSSnIRG-tM3tWL1PzZEH77ZHhIAZ9R0UPs6YknbM,1441
 plato/v2/utils/proxy_tunnel.py,sha256=8ZTd0jCGSfIHMvSv1fgEyacuISWnGPHLPbDglWroTzY,10463
-plato/world/__init__.py,sha256=v-vAQSMgROw3exUWmXc8PPUmaiYItbUdEoHPiP182wo,1312
-plato/world/base.py,sha256=pBAoxQEalNW8EB66Co_SshtaqtncWg7rNfZVagdN1xg,8769
-plato/world/config.py,sha256=7P8yMQU2LulxuK6KnK9IxRouO5gXQ9bl7xsBdcgtkn4,4077
-plato/world/types.py,sha256=KelUwCpFefurzh-Vu-IooTgThW6LDq4kvo2_M3mRv4k,1270
-plato/worlds/README.md,sha256=UkzsR6dgUK9oMQW1TXheHCabW0pEodOmqw5tGYh-XZU,5253
-plato/worlds/__init__.py,sha256=-a2oRiN0ECurYWG3o313C8rqyowKjiFz4__GYmVMrA0,1001
-plato/worlds/base.py,sha256=OSPs4wW-G6K1qoWEFcRAWPDFZ0-N1XTYp4wyzHHIF8A,6750
+plato/worlds/README.md,sha256=TgG4aidude0ouJSCfY81Ev45hsUxPkO85HUIiWNqkcc,5463
+plato/worlds/__init__.py,sha256=ALoou3l5lXvs_YZc5eH6HdMHpvhnpzKWqz__aSC1jFc,2152
+plato/worlds/base.py,sha256=KsDq1TVVd4BKEFo42WodF4DPgodr00Gd1NrfzZEPqyw,25632
 plato/worlds/build_hook.py,sha256=KSoW0kqa5b7NyZ7MYOw2qsZ_2FkWuz0M3Ru7AKOP7Qw,3486
-plato/worlds/config.py,sha256=ClrHX-yu3OVqbWfGs5Lzkta1PvNJq-fTyD6W92YX-KI,4016
-plato/worlds/runner.py,sha256=k2mJG99VgvaJfnPhOpuMEiwJ8Pl06XnELkK82ULvp-c,4613
-plato_sdk_v2-2.0.64.dist-info/METADATA,sha256=jfXtCJnP5IYqYpJS7ee7JBwTbRsAAcQOu2n3eu6qaHY,8612
-plato_sdk_v2-2.0.64.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-plato_sdk_v2-2.0.64.dist-info/entry_points.txt,sha256=wP8m2stltweMpm5iEaxPy-4u3y-0fonKBWsbKUE-5Gs,90
-plato_sdk_v2-2.0.64.dist-info/RECORD,,
+plato/worlds/config.py,sha256=fkWoetCawIPf5xgj5EyJCvopmPPzZQAqt0Ke9P4hq5c,10845
+plato/worlds/runner.py,sha256=gwTBuDPoiP3aGF4UgqNKHrfv1wohr0tsex0xlcuaw8c,18202
+plato_sdk_v2-2.3.4.dist-info/METADATA,sha256=k70jP_81ZXfk_e5_nZQNO8Huplz6Wm3JoazP2-loCmg,8510
+plato_sdk_v2-2.3.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+plato_sdk_v2-2.3.4.dist-info/entry_points.txt,sha256=upGMbJCx6YWUTKrPoYvYUYfFCqYr75nHDwhA-45m6p8,136
+plato_sdk_v2-2.3.4.dist-info/RECORD,,

{plato_sdk_v2-2.0.64.dist-info → plato_sdk_v2-2.3.4.dist-info}/entry_points.txt

@@ -1,3 +1,4 @@
 [console_scripts]
 plato = plato.v1.cli:main
+plato-agent-runner = plato.agents.runner:main
 plato-world-runner = plato.worlds.runner:main

plato/agents/callback.py

@@ -1,246 +0,0 @@
-"""Chronos callback utilities for agents.
-
-This module provides utilities for agents to communicate with Chronos,
-including pushing logs, uploading trajectories, and uploading zipped logs.
-"""
-
-from __future__ import annotations
-
-import base64
-import io
-import json
-import logging
-import zipfile
-from pathlib import Path
-from typing import Any
-
-import httpx
-
-logger = logging.getLogger(__name__)
-
-
-class ChronosCallback:
-    """Utility class for communicating with Chronos server.
-
-    Handles pushing logs, updating status, and uploading artifacts
-    (trajectory and zipped logs) to the Chronos callback endpoints.
-
-    Example:
-        callback = ChronosCallback(
-            callback_url="http://chronos.example.com/api/callback",
-            session_id="abc123",
-        )
-
-        # Push logs during execution
-        await callback.push_logs([
-            {"level": "info", "message": "Starting agent..."},
-        ])
-
-        # After agent completes, upload artifacts
-        await callback.upload_artifacts(logs_dir="/path/to/logs")
-    """
-
-    def __init__(self, callback_url: str, session_id: str):
-        """Initialize the callback client.
-
-        Args:
-            callback_url: Full callback base URL (e.g., http://server/api/callback)
-            session_id: The Chronos session ID for this run
-        """
-        self.callback_url = callback_url.rstrip("/")
-        self.session_id = session_id
-        self._enabled = bool(callback_url and session_id)
-
-    @property
-    def enabled(self) -> bool:
-        """Check if callbacks are enabled (both server and session_id set)."""
-        return self._enabled
-
-    async def push_logs(self, logs: list[dict[str, Any]]) -> bool:
-        """Push log entries to Chronos.
-
-        Args:
-            logs: List of log entries, each with 'level' and 'message' keys.
-                  Optional keys: 'timestamp', 'extra'
-
-        Returns:
-            True if logs were pushed successfully, False otherwise.
-        """
-        if not self._enabled:
-            return False
-
-        try:
-            async with httpx.AsyncClient(timeout=10.0) as client:
-                response = await client.post(
-                    f"{self.callback_url}/logs",
-                    json={"session_id": self.session_id, "logs": logs},
-                )
-                return response.status_code == 200
-        except Exception as e:
-            logger.warning(f"Failed to push logs to Chronos: {e}")
-            return False
-
-    async def push_log(
-        self,
-        message: str,
-        level: str = "info",
-        extra: dict[str, Any] | None = None,
-    ) -> bool:
-        """Push a single log entry to Chronos.
-
-        Args:
-            message: Log message
-            level: Log level (debug, info, warning, error)
-            extra: Optional additional structured data
-
-        Returns:
-            True if log was pushed successfully, False otherwise.
-        """
-        log_entry: dict[str, Any] = {"level": level, "message": message}
-        if extra:
-            log_entry["extra"] = extra
-        return await self.push_logs([log_entry])
-
-    async def update_status(
-        self,
-        status: str,
-        message: str | None = None,
-        extra: dict[str, Any] | None = None,
-    ) -> bool:
-        """Update the session status in Chronos.
-
-        Args:
-            status: New status (running, completed, failed)
-            message: Optional status message
-            extra: Optional additional data
-
-        Returns:
-            True if status was updated successfully, False otherwise.
-        """
-        if not self._enabled:
-            return False
-
-        try:
-            async with httpx.AsyncClient(timeout=10.0) as client:
-                response = await client.post(
-                    f"{self.callback_url}/status",
-                    json={
-                        "session_id": self.session_id,
-                        "status": status,
-                        "message": message,
-                        "extra": extra,
-                    },
-                )
-                return response.status_code == 200
-        except Exception as e:
-            logger.warning(f"Failed to update status in Chronos: {e}")
-            return False
-
-    def find_trajectory(self, logs_dir: str) -> dict[str, Any] | None:
-        """Find and load ATIF trajectory from logs directory.
-
-        Agents write ATIF trajectory to /logs/agent/trajectory.json after
-        completing their run. This method finds and loads that file.
-
-        Args:
-            logs_dir: Path to the logs directory
-
-        Returns:
-            Parsed ATIF trajectory dict if found, None otherwise.
-        """
-        trajectory_path = Path(logs_dir) / "agent" / "trajectory.json"
-
-        if not trajectory_path.exists():
-            return None
-
-        try:
-            with open(trajectory_path) as f:
-                data = json.load(f)
-                if isinstance(data, dict) and "schema_version" in data:
-                    logger.info(f"Found ATIF trajectory: {trajectory_path}")
-                    return data
-                return None
-        except Exception as e:
-            logger.warning(f"Failed to load trajectory from {trajectory_path}: {e}")
-            return None
-
-    def zip_logs_dir(self, logs_dir: str) -> bytes:
-        """Zip the entire logs directory.
-
-        Args:
-            logs_dir: Path to the logs directory
-
-        Returns:
-            Zip file contents as bytes.
-        """
-        logs_path = Path(logs_dir)
-        buffer = io.BytesIO()
-
-        with zipfile.ZipFile(buffer, "w", zipfile.ZIP_DEFLATED) as zf:
-            for file_path in logs_path.rglob("*"):
-                if file_path.is_file():
-                    arcname = file_path.relative_to(logs_path)
-                    zf.write(file_path, arcname)
-
-        buffer.seek(0)
-        return buffer.read()
-
-    async def upload_artifacts(
-        self,
-        logs_dir: str,
-        trajectory: dict[str, Any] | None = None,
-    ) -> bool:
-        """Upload trajectory and zipped logs to Chronos.
-
-        If trajectory is not provided, attempts to find it in the logs directory.
-        The logs directory is zipped and uploaded to S3 via Chronos.
-
-        Args:
-            logs_dir: Path to the logs directory
-            trajectory: Optional pre-loaded trajectory dict. If None, will
-                       attempt to find trajectory.json in logs_dir.
-
-        Returns:
-            True if artifacts were uploaded successfully, False otherwise.
-        """
-        if not self._enabled:
-            return False
-
-        # Find trajectory if not provided
-        if trajectory is None:
-            trajectory = self.find_trajectory(logs_dir)
-            if trajectory:
-                logger.info("Found ATIF trajectory in logs directory")
-            else:
-                logger.info("No ATIF trajectory found in logs directory")
-
-        # Zip logs directory
-        logs_base64: str | None = None
-        try:
-            logs_zip = self.zip_logs_dir(logs_dir)
-            logs_base64 = base64.b64encode(logs_zip).decode("utf-8")
-            logger.info(f"Zipped logs: {len(logs_zip)} bytes")
-        except Exception as e:
-            logger.warning(f"Failed to zip logs: {e}")
-
-        # Upload to Chronos
-        try:
-            async with httpx.AsyncClient(timeout=60.0) as client:
-                response = await client.post(
-                    f"{self.callback_url}/artifacts",
-                    json={
-                        "session_id": self.session_id,
-                        "trajectory": trajectory,
-                        "logs_base64": logs_base64,
-                    },
-                )
-                if response.status_code == 200:
-                    result = response.json()
-                    logger.info(f"Uploaded artifacts to Chronos: {result}")
-                    return True
-                else:
-                    logger.warning(f"Failed to upload artifacts: {response.status_code} {response.text}")
-                    return False
-        except Exception as e:
-            logger.warning(f"Failed to upload artifacts to Chronos: {e}")
-            return False

plato/world/__init__.py

@@ -1,44 +0,0 @@
-"""World module - OpenAI Gym-like interface for Plato simulators.
-
-Provides an abstract World base class that manages Plato sessions and
-exposes simulator URLs for agents to interact with via plato.sims SDK.
-
-Usage:
-    from plato.world import World, WorldConfig, EnvSpec, Observation, StepResult
-
-    class MyWorld(World):
-        async def _on_reset(self) -> Observation:
-            return Observation(step=0, date=self.config.start_date, data={})
-
-        async def _on_step(self) -> StepResult:
-            return StepResult(observation=obs, reward=100.0)
-
-        def get_prompt(self) -> str:
-            return f"Day {self._current_step}: Do something..."
-
-    config = WorldConfig(
-        name="my-world",
-        envs=[EnvSpec(alias="store", simulator="spree")],
-        num_steps=30,
-    )
-
-    async with MyWorld(config) as world:
-        obs = await world.reset()
-        for _ in range(config.num_steps):
-            # Agent takes turn using world.get_env_vars()
-            result = await world.step()
-            if result.done:
-                break
-"""
-
-from plato.world.base import World
-from plato.world.config import EnvSpec, WorldConfig
-from plato.world.types import Observation, StepResult
-
-__all__ = [
-    "World",
-    "WorldConfig",
-    "EnvSpec",
-    "Observation",
-    "StepResult",
-]