swarms 7.6.0__py3-none-any.whl → 7.6.2__py3-none-any.whl

{swarms-7.6.0.dist-info → swarms-7.6.2.dist-info}/RECORD

@@ -1,13 +1,16 @@
 swarms/__init__.py,sha256=DDfir3E4H8z1PlQqpc2HzovJQtNvIAzye4Fs165J0o8,512
-swarms/agents/__init__.py,sha256=7MJORblMZSbJMNhcKfmVeYC_9EfK6OBW9wdZkFqUwzc,1035
+swarms/agents/__init__.py,sha256=a0vqrSGmY93Z2Zi5uBg-B5WQJnEX0Wr0C2nfTcQAWCI,1225
+swarms/agents/agent_judge.py,sha256=xT242CX5mV64cq2B-3RGkuEHiV5aD04P_Zq8_s64iMQ,3967
 swarms/agents/agent_print.py,sha256=SXqWA2ZzXwRFdv8hkuYwOPMTasvaGTG6U29413qRCAA,918
 swarms/agents/ape_agent.py,sha256=1kz_65LJgjLlY1yv2WLBeVMs7sP9BgEVWk0w1f67YLc,1563
 swarms/agents/auto_generate_swarm_config.py,sha256=7eJ873xS7PJmyreMaa5Uub8qFu-qIinuyMuogB2Ehjc,8474
 swarms/agents/consistency_agent.py,sha256=41h0yvnjzmKsE8-q4UsN0ckHP7WWmB5E_z64ec9QaJM,7414
 swarms/agents/create_agents_from_yaml.py,sha256=PgFIpuYZehxEl79BAK6TolSZwydDQzvGMAKhLsHuBbc,13008
+swarms/agents/flexion_agent.py,sha256=gVkQi5DDMLnYdyp2hkpgBnagDlMlKbnfnlWEaA8YrCo,21467
+swarms/agents/gkp_agent.py,sha256=5Jms3zHQ2qwJ6-PHDh9X-cFtAlH4dSUoDgRqN-xZzog,21067
 swarms/agents/i_agent.py,sha256=_eKUcgPfiVqQpF5Q-Sv1tT-JZxIeNl9Fp7OrnjVUtz8,12276
 swarms/agents/openai_assistant.py,sha256=mTSEtj26J0mc5pCeWrmMY0EXzTRYQfyfw_BtOqtcCHc,11044
-swarms/agents/reasoning_agents.py,sha256=ZVSKs92wdsEKyfyPZNNUvAfEOXWKyRFYqUC-M02Tews,4573
+swarms/agents/reasoning_agents.py,sha256=tF7K710lj-VZQVc2VjqxMWGLNGMsEWRarXcJnXz1wkc,5741
 swarms/agents/reasoning_duo.py,sha256=s9SnXoproQaBrShLtiruGJituy8sJlZJATc5Vy_FdwI,3875
 swarms/agents/tool_agent.py,sha256=G7rhBACsHsGUMT4H9eF5aY7e3Gx-5jOmJkhCF1jm9mU,5087
 swarms/artifacts/__init__.py,sha256=M111xTw7IVVt8gLDwW7Kau5n1YdwkL3vbCJPVeEWktI,83
@@ -20,6 +23,7 @@ swarms/prompts/__init__.py,sha256=ZFcghAs4b0Rsjcc_DIFssiztZub5tJ66rxmJD2_tXpQ,74
 swarms/prompts/accountant_swarm_prompts.py,sha256=swceN1B0fZCOedd3-y3gjNOHDmR-3H5YK17ytp7tTDM,11320
 swarms/prompts/ag_prompt.py,sha256=_mydaflPh3kPpogqfObJ9UK5D8ySAGgtcO4BoNBN3LE,2658
 swarms/prompts/aga.py,sha256=4rRG7r9LF9JBrGOLN4Hy_6cSe0z5YOeKFbGnOmRfFJ0,9156
+swarms/prompts/agent_judge_prompt.py,sha256=-2q5mWLRyO7pj7j512ZNYRm-uEZttPgDGaxZo3oM_QQ,2235
 swarms/prompts/agent_prompt.py,sha256=6SXkMWcFIIAY02N9P6kMdzo4wtPwzg1Bl9R2qsIdicM,2705
 swarms/prompts/agent_prompts.py,sha256=63ZqAO1jfH3TKtl9ibLLagSG5kAPlbq4xtCGgGEuuoA,6886
 swarms/prompts/agent_system_prompts.py,sha256=DXHqZ1ELY4lg4LC37fNq52P2lRkNPjgOCfoan7r9vzY,7113
@@ -74,15 +78,13 @@ swarms/schemas/__init__.py,sha256=EqqtVcpoptF1kfy19Wykp22ut4AA0z-yMQ5H9WB7ptA,18
 swarms/schemas/agent_input_schema.py,sha256=qhPyThMx2on91yG9mzNdP_08GpMh1IRDHDwFna29jPs,6345
 swarms/schemas/agent_step_schemas.py,sha256=a14gb58vR0xOwB_fwSJQbN6yb9HddEaT30E6hUrzEQA,2573
 swarms/schemas/base_schemas.py,sha256=UvBLVWg2qRen4tK5GJz50v42SiX95EQ5qK7hfyAHTEU,3267
-swarms/structs/__init__.py,sha256=iPQmbvhyBMobHP1SXIxncVdpsMDnPWwcAEZz_TEORt0,4354
-swarms/structs/agent.py,sha256=tcGFYvQGdv-2VD9kaQvIKcdBxhv4H9WhWmEMzNcFeYc,96232
-swarms/structs/agent_builder.py,sha256=D4X-rYW1GZ_vG5k5nutd1cW7adBH69CfUtxx5OrgeUY,12180
+swarms/structs/__init__.py,sha256=S-I4b0CT6uZoUJht66JDFuKwi8yZ50TSW7rGjszEBdY,4578
+swarms/structs/agent.py,sha256=X1IG7pB1PkHOdqOiUh9NbPhU-oZ6TuGqX8FwFoHOKVQ,97286
+swarms/structs/agent_builder.py,sha256=tYNpfO4_8cgfMHfgA5DAOWffHnt70p6CLt59esqfVCY,12133
 swarms/structs/agent_registry.py,sha256=il507cO1NF-d4ChyANVLuWrN8bXsEAi8_7bLJ_sTU6A,12112
 swarms/structs/agent_roles.py,sha256=8XEw6RjOOZelaZaWt4gXaYQm5WMLEhSO7W6Z8sQjmFg,582
 swarms/structs/agent_router.py,sha256=YZw5AaK2yTvxkOA7ouED_4MoYgn0XZggvo1wrglp-4E,13017
-swarms/structs/agent_security.py,sha256=uDQ6zoCm6mY2eMGSpjo-vidZrnLSdd-LTD75OpXDpKw,10618
 swarms/structs/agents_available.py,sha256=SedxDim-0IWgGsNwJZxRIUMfKyAFFXdvXSYeBNu0zGw,2804
-swarms/structs/airflow_swarm.py,sha256=3wekDOYx4eLLO68eVvSO89_sv5G-4-uMpXTmrhsHPuE,14250
 swarms/structs/async_workflow.py,sha256=7YWsLPyGY-1-mMxoIXWQ0FnYH6F227nxsS9PFAJoF9Q,26214
 swarms/structs/auto_swarm.py,sha256=AHWswlEWDL_i3V8IP362tx6pi_B2arlZhALykrkI5OA,8215
 swarms/structs/auto_swarm_builder.py,sha256=vPM5Kq59D_FvuWJB8hxgHuEvTXsxDxovlBnHGVQsM4o,10938
@@ -90,33 +92,35 @@ swarms/structs/base_structure.py,sha256=GDu4QJQQmhU7IyuFJHIh9UVThACCva-L7uoMbVD9
 swarms/structs/base_swarm.py,sha256=_pggyl32BoEq-W2Cg-03h0z9xpPT_wXFjss4Yvysoxc,24033
 swarms/structs/base_workflow.py,sha256=DTfFwX3AdFYxACDYwUDqhsbcDZnITlg5TeEYyxmJBCc,11414
 swarms/structs/concat.py,sha256=utezSxNyh1mIwXgdf8-dJ803NDPyEy79WE8zJHuooGk,732
-swarms/structs/concurrent_workflow.py,sha256=8myerrLcOfEDGPlfp4xYbf79qBBAzxh80zrw1Dv7OXY,15525
+swarms/structs/concurrent_workflow.py,sha256=MgK6xPjwCUrLHC1UzM4O45BELnVBoomgaw6CPa_9SgQ,15526
 swarms/structs/conversation.py,sha256=7oh3x5XFyscnG1TAIMyL1BKJ4jC-unDEv-RFRVmxJO8,17966
 swarms/structs/csv_to_agent.py,sha256=ug9JqQFPguXeU9JQpSUXuVtOpHYdJhlpKJUJBovo694,9443
 swarms/structs/de_hallucination_swarm.py,sha256=Aa1kIJLmyXXrFl6qpmAq6mfkSaItQYD0MPgQ_e-nPvc,10305
+swarms/structs/deep_research_swarm.py,sha256=h4jtrcuiAKMWMYo8I7oaq6eFn8cJpqHhml58EveNbZ4,16756
+swarms/structs/dynamic_conversational_swarm.py,sha256=n_d1jDCzBwiGb0QjJpW_MlXxqEkhGEhC1ttaebH7f3Q,8098
 swarms/structs/graph_swarm.py,sha256=HPHlLWwdSPSe4o-I06ZOIgtBg72a06llEnv8-aglf3Q,20962
 swarms/structs/graph_workflow.py,sha256=TAaUG_J3898hhghPOp0WEAV3Zf0in6s48ZSVbSTX-vQ,8629
 swarms/structs/groupchat.py,sha256=oLaWLuZ0ObHVXN9k3K8CpLSDSZwnCHBgJHsdGmHqAS8,15906
-swarms/structs/hiearchical_swarm.py,sha256=-qepZLY7RLxHlT4dU-dMEWSKnfBHEVrFr6HRn8lGh1Q,35273
-swarms/structs/majority_voting.py,sha256=1-FQGMxDwg5mYeoP5rXTgiXEXF8Yc4vTc93ByGoInK4,10121
+swarms/structs/hiearchical_swarm.py,sha256=XDEdy5kAISmWKraMR26VX47eCT4YgGTI2FNcBQzIvLE,35274
+swarms/structs/hybrid_hiearchical_peer_swarm.py,sha256=s6ievLmSo93feHB9xwHAQJIohPgLqco04dngvAFRxQw,9680
+swarms/structs/majority_voting.py,sha256=F_t_MOC3YCRyMw5N6qKdFThpaXZxwixRw592Ku5Uhto,10122
 swarms/structs/malt.py,sha256=0ZOuLfOzaJ4vAVOM6J1aZ3yWAiKxfMkNIBNp8pjsEqE,19392
 swarms/structs/matrix_swarm.py,sha256=qHuhOYrTyOv6ujHMe8PrQT-h-WmaCPCfX4ghv5L8UFI,9765
 swarms/structs/meme_agent_persona_generator.py,sha256=b3kKlumhsV4KV88-GS3CUnGO1UfKTU3fT8SAMj0ZlwQ,10645
-swarms/structs/mixture_of_agents.py,sha256=OGCTvyRQjHk9os4ZDYixdbDIIMeLMxfseeznS9rzdko,10574
+swarms/structs/mixture_of_agents.py,sha256=G8_MVMrDd0-ZD_gJ5YZgtTCUjl7COha9Me-vOYMXsAE,10575
 swarms/structs/model_router.py,sha256=V5pZHYlxSmCvAA2Gsns7LaCz8dXtRi7pCvb-oLGHYIY,12739
 swarms/structs/multi_agent_collab.py,sha256=odh2NQRR23LidsphCxUfAke369lDdgL__w3Xovu9jkA,7731
 swarms/structs/multi_agent_exec.py,sha256=NbR-CnxxBuCa0Ll3Os9wfKdkIsQiCpqsTgFX9hWRefw,17421
-swarms/structs/multi_agent_orchestrator.py,sha256=MsCqBGrirhbdx4aZqY_1MRa9hRqt_ZF3HLmt7JCsINI,13399
+swarms/structs/multi_agent_orchestrator.py,sha256=_trjXCW31ZeVR7N2hURLUPDZhYa-Wa3ADMk1wnNJdcQ,13400
 swarms/structs/octotools.py,sha256=GZo0qtFM69W7vvewk6_k09vicgw0c0_v7MiPvEZCagE,31406
 swarms/structs/omni_agent_types.py,sha256=RdKLfZ-lXDJrEa0aJT_Rfx9TypJQo8SISqKz4fnLkAk,230
-swarms/structs/output_type.py,sha256=56bALerdISngZ86Q-7LLlmHZw6edAFkea2bPlKdM_oE,279
-swarms/structs/output_types.py,sha256=i56WfkzUR6YgP1qINJ8t-By8OVS_sdbO64IvCmV-pbM,213
+swarms/structs/output_types.py,sha256=F1jNbDLJrubRIUyheMGMahJfGikbWZ_yNmbE9QVIz9A,280
 swarms/structs/pulsar_swarm.py,sha256=4_L0GqPBgnA3AJajpNLgO4IAG6U36nIntFK9WNJScv8,15968
 swarms/structs/queue_swarm.py,sha256=8vcA-rh280midcdgfA5IwJzBmMgdn71nRH6KndWu-DA,6770
-swarms/structs/rearrange.py,sha256=KVBZVvoTd5vA7-kzpFHHpoQWzss0NZurCI1K-QR-MJc,28638
+swarms/structs/rearrange.py,sha256=_WiWNab5NFRhpeXrlO_57o2Q0j194A-6k405Gwr4hQU,28639
 swarms/structs/round_robin.py,sha256=MGk623KiN9uSxTMG6MY_BIAkvEDh1RPwyl5Min7GLOU,7573
 swarms/structs/safe_loading.py,sha256=gmYX8G9TsvAIp6OCvREBZt5mwSFc-p-t1rSnDBfhEmE,7124
-swarms/structs/sequential_workflow.py,sha256=B0obj_tawZjeHN6i6GWAbLFpLdJMemmpz7NutmHjOpY,8684
+swarms/structs/sequential_workflow.py,sha256=KrIfR9SToNTdyBTDd3DYvXbYa3SNELxUbeYDdjVhUZo,8685
 swarms/structs/spreadsheet_swarm.py,sha256=ToX56QJjlm_nnC3MYppvKC_NTr9Zy_orkBzfxNLdwbA,14845
 swarms/structs/stopping_conditions.py,sha256=Z0Jx0U2feAfQfuVV_IJGgal62DoVsGPN8K6HkkqB_bM,484
 swarms/structs/swarm_arange.py,sha256=6fexCPsXRgdLbpY0p9rp_REipeXzsbv1_GOtD9B4HaI,15179
@@ -127,9 +131,9 @@ swarms/structs/swarm_load_balancer.py,sha256=pUCc5FEBcuJ_GmOFeTWBPfXlUdiTOjYcJqV
 swarms/structs/swarm_matcher.py,sha256=E2KwHHEJxmW-UfTeMPWZ6VCmYdQ_I9_fwrfJbxD02GY,23322
 swarms/structs/swarm_output_type.py,sha256=tW8Iqar1Jaf2Lzw66nAPc6MDk7-srQl5_XUKFvzoam4,755
 swarms/structs/swarm_registry.py,sha256=P0XRrqp1qBNyt0BycqPQljUzKv9jClaQMhtaBMinhYg,5578
-swarms/structs/swarm_router.py,sha256=V8qEojE0XkqZQKcrm_9w7NCou0uiesSAIgeuPbahRlY,26404
+swarms/structs/swarm_router.py,sha256=A7efNKGkyj_Tqjv8v-bJJVkl22zyiNBLAlJqWNFtWjM,26803
 swarms/structs/swarming_architectures.py,sha256=VvkSA9nQnF91V2C5-ALwSY1peZckeM1G4pPeQS7IVsE,13109
-swarms/structs/swarms_api.py,sha256=Pv-X-GkQ05gGPvTy8c_LES_IC8CBO9d2RPIssIjuqB8,13759
+swarms/structs/swarms_api.py,sha256=f-RpvfKfRK2AaapkyPN2ihXJvIGN4JWB_A7pJu4WyiU,13735
 swarms/structs/talk_hier.py,sha256=npyEuL52SCgQmMynIvGjfatNqOz4toq0EyhEtSNmQhQ,25649
 swarms/structs/tree_swarm.py,sha256=AnIxrt0KhWxAQN8uGjfCcOq-XCmsuTJiH8Ex4mXy8V8,12500
 swarms/structs/utils.py,sha256=Mo6wHQYOB8baWZUKnAJN5Dsgubpo81umNwJIEDitb2A,1873
@@ -137,7 +141,7 @@ swarms/structs/various_alt_swarms.py,sha256=qdBuOF31UjatlKRu-9bxwyRQzIjohRhTv_63
 swarms/structs/workspace_manager.py,sha256=t0OgUP9dDU7xS6N3mAT2PbXvjHTsUK3nf2mxppcfZ70,5473
 swarms/telemetry/__init__.py,sha256=yibtkHEbQRPUv6ir1FhDHlAO_3nwKJPQH4LjzBC2AuQ,661
 swarms/telemetry/bootup.py,sha256=tBuhOb3m7o2q7nMjb2j9v2ib5Hx2E9Oya8yff5z4zIA,1655
-swarms/telemetry/main.py,sha256=xVdcKlfNONtNkOkcW7lLDCr3PseT_CaJc25sFCPSFKo,8287
+swarms/telemetry/main.py,sha256=ZoVvf4XInh0Q5CcfwBFj9KmpxrQjOm2Aje70CElgGZA,8420
 swarms/tools/__init__.py,sha256=pqIMcRQr4gtoNdbyI1N5k4upkYSBMxACJbxfB9yrV4c,1493
 swarms/tools/base_tool.py,sha256=BiBCFHin8AyZO3FYOGA-n3M2o-F36xUeIBUiybnZYjI,15179
 swarms/tools/cohere_func_call_schema.py,sha256=XJ6_yBMXCrV9KjN7v9Bk1iFj69TRlGIWYKsUTA1oGiQ,600
@@ -147,12 +151,14 @@ swarms/tools/function_util.py,sha256=DAnAPO0Ik__TAqL7IJzFmkukHnhpsW_QtALl3yj837g
 swarms/tools/json_former.py,sha256=4ugLQ_EZpghhuhFsVKsy-ehin9K64pqVE2gLU7BTO_M,14376
 swarms/tools/json_utils.py,sha256=WKMZjcJ0Vt6lgIjiTBenslcfjgRSLX4UWs4uDkKFMQI,1316
 swarms/tools/logits_processor.py,sha256=NifZZ5w9yemWGJAJ5nHFrphtZVX1XlyesgvYZTxK1GM,2965
+swarms/tools/mcp_integration.py,sha256=Aqt2gLTIinJdHM_ZnIICqCXp2Fz_AbDPfaJVljDuIAI,17529
 swarms/tools/openai_func_calling_schema_pydantic.py,sha256=6BAH9kuaVTvJIbjgSSJ5XvHhWvWszPxgarkfUuE5Ads,978
 swarms/tools/openai_tool_creator_decorator.py,sha256=SYZjHnARjWvnH9cBdj7Kc_Yy1muvNxMT3RQz8KkA2SE,2578
 swarms/tools/py_func_to_openai_func_str.py,sha256=W112Gu0CmAiHrNWnRMcnoGiVZEy2FxAU4xMvnG9XP4g,15710
 swarms/tools/pydantic_to_json.py,sha256=-tjUBwKVnKUWWtEGfYGLEfJbIOCrSVF-p1e6kT4_Geg,3850
 swarms/tools/tool_parse_exec.py,sha256=FW5XzkuNEs2YrroybjKChbCzDvaCs7ypknSDpYhfkd4,8717
 swarms/tools/tool_registry.py,sha256=ULZmIKBTx9XRCJRD9hwXfY3iQw9v94arw-VV6jcuftY,7992
+swarms/tools/tool_schema_base_model.py,sha256=0biTGIoibsPPP3fOrkC6WvNU5vXaalyccVKC1fpO_eg,1409
 swarms/tools/tool_utils.py,sha256=yXzzqG7Ytd8ybB8bsjNUNLaXIuIp9JbbpUKCiHxQqo8,2816
 swarms/utils/__init__.py,sha256=9qKE_11pxom74j3qExSm6Z_LwR5lrpC5YG17v22eLlo,975
 swarms/utils/agent_ops_check.py,sha256=08UomeSv1uw_oEDlum0yG-5SsKkxqPRbeIWeKC75b08,685
@@ -177,8 +183,8 @@ swarms/utils/swarm_reliability_checks.py,sha256=MsgUULt3HYg72D0HifZNmtCyJYpLA2UD
 swarms/utils/try_except_wrapper.py,sha256=appEGu9Afy3TmdkNNXUgQ9yU9lj2j0uNkIoW0JhVzzY,3917
 swarms/utils/visualizer.py,sha256=0ylohEk62MAS6iPRaDOV03m9qo2k5J56tWlKJk_46p4,16927
 swarms/utils/wrapper_clusterop.py,sha256=PMSCVM7ZT1vgj1D_MYAe835RR3SMLYxA-si2JS02yNQ,4220
-swarms-7.6.0.dist-info/LICENSE,sha256=jwRtEmTWjLrEsvFB6QFdYs2cEeZPRMdj-UMOFkPF8_0,11363
-swarms-7.6.0.dist-info/METADATA,sha256=u9Sd0TLcj3E3QPSiEOQHBNUKdN_kjHPB1nhjr-NqrGw,103019
-swarms-7.6.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-swarms-7.6.0.dist-info/entry_points.txt,sha256=2K0rTtfO1X1WaO-waJlXIKw5Voa_EpAL_yU0HXE2Jgc,47
-swarms-7.6.0.dist-info/RECORD,,
+swarms-7.6.2.dist-info/LICENSE,sha256=jwRtEmTWjLrEsvFB6QFdYs2cEeZPRMdj-UMOFkPF8_0,11363
+swarms-7.6.2.dist-info/METADATA,sha256=WGZDZCWtB66TM0b9VrFeXvKaD1fKdAkQi2itc4u7mkA,105115
+swarms-7.6.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+swarms-7.6.2.dist-info/entry_points.txt,sha256=2K0rTtfO1X1WaO-waJlXIKw5Voa_EpAL_yU0HXE2Jgc,47
+swarms-7.6.2.dist-info/RECORD,,

swarms/structs/agent_security.py

@@ -1,318 +0,0 @@
-import base64
-import json
-import uuid
-from datetime import datetime
-from dataclasses import dataclass
-from typing import Optional, Union, Dict, List
-
-from cryptography.fernet import Fernet
-from cryptography.hazmat.primitives import hashes, serialization
-from cryptography.hazmat.primitives.asymmetric import padding, rsa
-from cryptography.hazmat.primitives.kdf.pbkdf2 import PBKDF2HMAC
-
-
-@dataclass
-class EncryptedMessage:
-    """Structure for encrypted messages between agents"""
-
-    sender_id: str
-    receiver_id: str
-    encrypted_content: bytes
-    timestamp: float
-    message_id: str
-    session_id: str
-
-
-class EncryptionSession:
-    """Represents an encrypted communication session between agents"""
-
-    def __init__(
-        self,
-        session_id: str,
-        agent_ids: List[str],
-        encrypted_keys: Dict[str, bytes],
-        created_at: datetime,
-    ):
-        self.session_id = session_id
-        self.agent_ids = agent_ids
-        self.encrypted_keys = encrypted_keys
-        self.created_at = created_at
-
-
-class AgentEncryption:
-    """
-    Handles encryption for agent data both at rest and in transit.
-    Supports both symmetric (for data at rest) and asymmetric (for data in transit) encryption.
-    Also supports secure multi-agent communication.
-    """
-
-    def __init__(
-        self,
-        agent_id: Optional[str] = None,
-        encryption_key: Optional[str] = None,
-        enable_transit_encryption: bool = False,
-        enable_rest_encryption: bool = False,
-        enable_multi_agent: bool = False,
-    ):
-        self.agent_id = agent_id or str(uuid.uuid4())
-        self.enable_transit_encryption = enable_transit_encryption
-        self.enable_rest_encryption = enable_rest_encryption
-        self.enable_multi_agent = enable_multi_agent
-
-        # Multi-agent communication storage
-        self.sessions: Dict[str, EncryptionSession] = {}
-        self.known_agents: Dict[str, "AgentEncryption"] = {}
-
-        if enable_rest_encryption:
-            # Initialize encryption for data at rest
-            if encryption_key:
-                self.encryption_key = base64.urlsafe_b64encode(
-                    PBKDF2HMAC(
-                        algorithm=hashes.SHA256(),
-                        length=32,
-                        salt=f"agent_{self.agent_id}".encode(),  # Unique salt per agent
-                        iterations=100000,
-                    ).derive(encryption_key.encode())
-                )
-            else:
-                self.encryption_key = Fernet.generate_key()
-
-            self.cipher_suite = Fernet(self.encryption_key)
-
-        if enable_transit_encryption or enable_multi_agent:
-            # Generate RSA key pair for transit encryption
-            self.private_key = rsa.generate_private_key(
-                public_exponent=65537, key_size=2048
-            )
-            self.public_key = self.private_key.public_key()
-
-    def register_agent(
-        self, agent_id: str, agent_encryption: "AgentEncryption"
-    ) -> None:
-        """Register another agent for secure communication"""
-        if not self.enable_multi_agent:
-            raise ValueError("Multi-agent support is not enabled")
-        self.known_agents[agent_id] = agent_encryption
-
-    def create_session(self, agent_ids: List[str]) -> str:
-        """Create a new encrypted session between multiple agents"""
-        if not self.enable_multi_agent:
-            raise ValueError("Multi-agent support is not enabled")
-
-        session_id = str(uuid.uuid4())
-
-        # Generate a shared session key
-        session_key = Fernet.generate_key()
-
-        # Create encrypted copies of the session key for each agent
-        encrypted_keys = {}
-        for agent_id in agent_ids:
-            if (
-                agent_id not in self.known_agents
-                and agent_id != self.agent_id
-            ):
-                raise ValueError(f"Agent {agent_id} not registered")
-
-            if agent_id == self.agent_id:
-                agent_public_key = self.public_key
-            else:
-                agent_public_key = self.known_agents[
-                    agent_id
-                ].public_key
-
-            encrypted_key = agent_public_key.encrypt(
-                session_key,
-                padding.OAEP(
-                    mgf=padding.MGF1(algorithm=hashes.SHA256()),
-                    algorithm=hashes.SHA256(),
-                    label=None,
-                ),
-            )
-            encrypted_keys[agent_id] = encrypted_key
-
-        # Store session information
-        self.sessions[session_id] = EncryptionSession(
-            session_id=session_id,
-            agent_ids=agent_ids,
-            encrypted_keys=encrypted_keys,
-            created_at=datetime.now(),
-        )
-
-        return session_id
-
-    def encrypt_message(
-        self,
-        content: Union[str, dict],
-        receiver_id: str,
-        session_id: str,
-    ) -> EncryptedMessage:
-        """Encrypt a message for another agent within a session"""
-        if not self.enable_multi_agent:
-            raise ValueError("Multi-agent support is not enabled")
-
-        if session_id not in self.sessions:
-            raise ValueError("Invalid session ID")
-
-        session = self.sessions[session_id]
-        if (
-            self.agent_id not in session.agent_ids
-            or receiver_id not in session.agent_ids
-        ):
-            raise ValueError("Sender or receiver not in session")
-
-        # Serialize content if it's a dictionary
-        if isinstance(content, dict):
-            content = json.dumps(content)
-
-        # Get the session key
-        encrypted_session_key = session.encrypted_keys[self.agent_id]
-        session_key = self.decrypt_session_key(encrypted_session_key)
-
-        # Create Fernet cipher with session key
-        cipher = Fernet(session_key)
-
-        # Encrypt the message
-        encrypted_content = cipher.encrypt(content.encode())
-
-        return EncryptedMessage(
-            sender_id=self.agent_id,
-            receiver_id=receiver_id,
-            encrypted_content=encrypted_content,
-            timestamp=datetime.now().timestamp(),
-            message_id=str(uuid.uuid4()),
-            session_id=session_id,
-        )
-
-    def decrypt_message(
-        self, message: EncryptedMessage
-    ) -> Union[str, dict]:
-        """Decrypt a message from another agent"""
-        if not self.enable_multi_agent:
-            raise ValueError("Multi-agent support is not enabled")
-
-        if message.session_id not in self.sessions:
-            raise ValueError("Invalid session ID")
-
-        if self.agent_id != message.receiver_id:
-            raise ValueError("Message not intended for this agent")
-
-        session = self.sessions[message.session_id]
-
-        # Get the session key
-        encrypted_session_key = session.encrypted_keys[self.agent_id]
-        session_key = self.decrypt_session_key(encrypted_session_key)
-
-        # Create Fernet cipher with session key
-        cipher = Fernet(session_key)
-
-        # Decrypt the message
-        decrypted_content = cipher.decrypt(
-            message.encrypted_content
-        ).decode()
-
-        # Try to parse as JSON
-        try:
-            return json.loads(decrypted_content)
-        except json.JSONDecodeError:
-            return decrypted_content
-
-    def decrypt_session_key(self, encrypted_key: bytes) -> bytes:
-        """Decrypt a session key using the agent's private key"""
-        return self.private_key.decrypt(
-            encrypted_key,
-            padding.OAEP(
-                mgf=padding.MGF1(algorithm=hashes.SHA256()),
-                algorithm=hashes.SHA256(),
-                label=None,
-            ),
-        )
-
-    # Original methods preserved below
-    def encrypt_at_rest(self, data: Union[str, dict, bytes]) -> bytes:
-        """Encrypts data for storage"""
-        if not self.enable_rest_encryption:
-            return (
-                data
-                if isinstance(data, bytes)
-                else str(data).encode()
-            )
-
-        if isinstance(data, dict):
-            data = json.dumps(data)
-        if isinstance(data, str):
-            data = data.encode()
-
-        return self.cipher_suite.encrypt(data)
-
-    def decrypt_at_rest(
-        self, encrypted_data: bytes
-    ) -> Union[str, dict]:
-        """Decrypts stored data"""
-        if not self.enable_rest_encryption:
-            return encrypted_data.decode()
-
-        decrypted_data = self.cipher_suite.decrypt(encrypted_data)
-
-        try:
-            return json.loads(decrypted_data)
-        except json.JSONDecodeError:
-            return decrypted_data.decode()
-
-    def encrypt_for_transit(self, data: Union[str, dict]) -> bytes:
-        """Encrypts data for transmission"""
-        if not self.enable_transit_encryption:
-            return str(data).encode()
-
-        if isinstance(data, dict):
-            data = json.dumps(data)
-
-        return self.public_key.encrypt(
-            data.encode(),
-            padding.OAEP(
-                mgf=padding.MGF1(algorithm=hashes.SHA256()),
-                algorithm=hashes.SHA256(),
-                label=None,
-            ),
-        )
-
-    def decrypt_from_transit(
-        self, data: Union[bytes, str]
-    ) -> Union[str, dict]:
-        """Decrypts received data, handling both encrypted and unencrypted inputs"""
-        if not self.enable_transit_encryption:
-            return data.decode() if isinstance(data, bytes) else data
-
-        try:
-            if isinstance(data, bytes) and len(data) == 256:
-                decrypted_data = self.private_key.decrypt(
-                    data,
-                    padding.OAEP(
-                        mgf=padding.MGF1(algorithm=hashes.SHA256()),
-                        algorithm=hashes.SHA256(),
-                        label=None,
-                    ),
-                ).decode()
-            else:
-                return (
-                    data.decode() if isinstance(data, bytes) else data
-                )
-
-            try:
-                return json.loads(decrypted_data)
-            except json.JSONDecodeError:
-                return decrypted_data
-        except ValueError:
-            return data.decode() if isinstance(data, bytes) else data
-
-    def get_public_key_pem(self) -> bytes:
-        """Returns the public key in PEM format for sharing"""
-        if (
-            not self.enable_transit_encryption
-            and not self.enable_multi_agent
-        ):
-            return b""
-
-        return self.public_key.public_bytes(
-            encoding=serialization.Encoding.PEM,
-            format=serialization.PublicFormat.SubjectPublicKeyInfo,
-        )