fast-agent-mcp 0.2.13__py3-none-any.whl → 0.2.16__py3-none-any.whl

{fast_agent_mcp-0.2.13.dist-info → fast_agent_mcp-0.2.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fast-agent-mcp
-Version: 0.2.13
+Version: 0.2.16
 Summary: Define, Prompt and Test MCP enabled Agents and Workflows
 Author-email: Shaun Smith <fastagent@llmindset.co.uk>, Sarmad Qadri <sarmad@lastmileai.dev>
 License:                                  Apache License

{fast_agent_mcp-0.2.13.dist-info → fast_agent_mcp-0.2.16.dist-info}/RECORD

@@ -1,70 +1,70 @@
 mcp_agent/__init__.py,sha256=-AIoeL4c9UAp_P4U0z-uIWTTmQWdihOis5nbQ5L_eao,1664
 mcp_agent/app.py,sha256=jBmzYM_o50g8vhlTgkkf5TGiBWNbXWViYnd0WANbpzo,10276
-mcp_agent/config.py,sha256=z3EtltnwdEM03WevxSHusymMvHB02BN4brDsLYQ-M68,12268
+mcp_agent/config.py,sha256=Fr29xaR6nkEPv6dt5XFMNg4vjh6IeyaFmRC1T0cxk-Y,12088
 mcp_agent/console.py,sha256=Gjf2QLFumwG1Lav__c07X_kZxxEUSkzV-1_-YbAwcwo,813
 mcp_agent/context.py,sha256=pp_F1Q1jgAxGrRccSZJutn1JUxYfVue-St3S8tUyptM,7903
 mcp_agent/context_dependent.py,sha256=QXfhw3RaQCKfscEEBRGuZ3sdMWqkgShz2jJ1ivGGX1I,1455
-mcp_agent/event_progress.py,sha256=25iz0yyg-O4glMmtijcYpDdUmtUIKsCmR_8A52GgeC4,2716
+mcp_agent/event_progress.py,sha256=3dqk5Pn1tAG_m_wn4IPNwLWLyzm7CyKIidqHN-4l-JY,2736
 mcp_agent/mcp_server_registry.py,sha256=pSD3euU-Oc2LAVenqkLU7UmutAzk6A9liYVLjCj4J70,10068
 mcp_agent/progress_display.py,sha256=GeJU9VUt6qKsFVymG688hCMVCsAygG9ifiiEb5IcbN4,361
 mcp_agent/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mcp_agent/agents/agent.py,sha256=Tn2YKw_ytx9b8jC-65WYQmrnD43kYiZsLa4sVHxn9d4,3854
-mcp_agent/agents/base_agent.py,sha256=dzyy4tDHJfRC4Sp-JqXeDwykk5SH55k89dUXQROIdQ4,23488
+mcp_agent/agents/agent.py,sha256=GgaUHoilgqzh9PQYr5k2WiPj4pagwicf9-ZLFsHkNNo,3848
+mcp_agent/agents/base_agent.py,sha256=nY93atndy6OdOZi5LAbJTJ05F5PHvvfk1RKq0NFeKWw,23498
 mcp_agent/agents/workflow/__init__.py,sha256=HloteEW6kalvgR0XewpiFAqaQlMPlPJYg5p3K33IUzI,25
-mcp_agent/agents/workflow/chain_agent.py,sha256=efftXdHc5F-XY8jnz5npHbKHhqnzHh28WbU5yQ4yUn0,6105
-mcp_agent/agents/workflow/evaluator_optimizer.py,sha256=VWdzVIy_qSiVsDJO22ta3RB3drkvBfXk9HxBYMpsC5U,13300
-mcp_agent/agents/workflow/orchestrator_agent.py,sha256=30hFQyAmtjQTX6Li_zWWIHCpdNpfZkDo57YXXW5xIsI,21561
+mcp_agent/agents/workflow/chain_agent.py,sha256=eIlImirrSXkqBJmPuAJgOKis81Cl6lZEGM0-6IyaUV8,6105
+mcp_agent/agents/workflow/evaluator_optimizer.py,sha256=ysUMGM2NzeCIutgr_vXH6kUPpZMw0cX4J_Wl1r8eT84,13296
+mcp_agent/agents/workflow/orchestrator_agent.py,sha256=bNjOLTsPvUAZEvdd2Nd2YcV8GTFctuImBuS-RQsLC5A,21557
 mcp_agent/agents/workflow/orchestrator_models.py,sha256=5P_aXADVT4Et8qT4e1cb9RelmHX5dCRrzu8j8T41Kdg,7230
 mcp_agent/agents/workflow/orchestrator_prompts.py,sha256=EXKEI174sshkZyPPEnWbwwNafzSPuA39MXL7iqG9cWc,9106
-mcp_agent/agents/workflow/parallel_agent.py,sha256=SgIXJx2X_MSlLOv6WXYRezwjDYjU9f95eKQzTm5Y_lk,7087
-mcp_agent/agents/workflow/router_agent.py,sha256=c4MU55U6q1DRayP0sDoyxdlnKX-N0LPbRv-MFlwbwrY,11165
+mcp_agent/agents/workflow/parallel_agent.py,sha256=GQTxAqwrPEdle-rPWMvoLOuzE_X69_HYEgYSm98eXdM,7087
+mcp_agent/agents/workflow/router_agent.py,sha256=Jkib-STvzk7jQc2JoWct5I3sdeUHXEuY7vQIPyCsA-I,11165
 mcp_agent/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mcp_agent/cli/__main__.py,sha256=AVZ7tQFhU_sDOGuUGJq8ujgKtcxsYJBJwHbVaaiRDlI,166
-mcp_agent/cli/main.py,sha256=_yMv39nP5faJ8rsp-4xKofX1Mnj7rTKnhBJnC2taNJU,3241
+mcp_agent/cli/main.py,sha256=m0Ndb8y5qCfYyTDBDaJ4g1TuUiX-xtTKNJJDPsltw6o,2884
 mcp_agent/cli/terminal.py,sha256=GRwD-RGW7saIz2IOWZn5vD6JjiArscELBThm1GTFkuI,1065
-mcp_agent/cli/commands/config.py,sha256=jU2gl4d5YESrdUboh3u6mxf7CxVT-_DT_sK8Vuh3ajw,231
+mcp_agent/cli/commands/check_config.py,sha256=9Ryxo_fLInm3YKdYv46yLrAJgnQtMisGreu6Kkriw2g,16677
 mcp_agent/cli/commands/quickstart.py,sha256=SM3CHMzDgvTxIpKjFuX9BrS_N1vRoXNBDaO90aWx1Rk,14586
-mcp_agent/cli/commands/setup.py,sha256=YoZM5DiYj-4CaB5cOPistH8UHCTzZGnOsFhaFcqosB0,6470
+mcp_agent/cli/commands/setup.py,sha256=iEMwhc1mBIsX2PH2yeBnr0GPE5rrI8QXXcwcovJleGc,6470
 mcp_agent/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mcp_agent/core/agent_app.py,sha256=5nQJNo8DocIRWiX4pVKAHUZF8s6HWpc-hJnfzl_1v1c,9697
-mcp_agent/core/agent_types.py,sha256=LuWslu9YI6JRnAWwh_A1ZejK72-e839wH7tf2MHxSIU,1389
-mcp_agent/core/direct_decorators.py,sha256=Q6t3VpRPLCRzqJycPZIkKbbEJMVocxdScp5o2xn4gLU,14460
-mcp_agent/core/direct_factory.py,sha256=hYFCucZVAQ2wrfqIe9Qameoa-cCRaQ53R97EMHvUZAM,17572
-mcp_agent/core/enhanced_prompt.py,sha256=A0FJ_-dr1RLq3uzmFSxiOBxw5synW2BhA4QntQyYVwg,18792
+mcp_agent/core/agent_types.py,sha256=jAYYpcya_Gvc_PtEXmJh7_HgvgIUQx4so_5M-pkpBoM,1459
+mcp_agent/core/direct_decorators.py,sha256=4e_C4_UGnqSnu8y4Wjb6px_9XMktBxebzr91v57MKb4,14450
+mcp_agent/core/direct_factory.py,sha256=rsnxnjvslur7fuptCQfCkUTFbwguVM-FU1Yh6AJOsjg,17842
+mcp_agent/core/enhanced_prompt.py,sha256=kEgeD7F1s8LRZDc3Xr6CIxQxewxXy9z-0xZsxfIblsY,18851
 mcp_agent/core/error_handling.py,sha256=xoyS2kLe0eG0bj2eSJCJ2odIhGUve2SbDR7jP-A-uRw,624
 mcp_agent/core/exceptions.py,sha256=ENAD_qGG67foxy6vDkIvc-lgopIUQy6O7zvNPpPXaQg,2289
 mcp_agent/core/fastagent.py,sha256=T2kyq32wBJCOj13Zy1G_XJjQZb1S4HVdx3OBzmEMHBg,18644
-mcp_agent/core/interactive_prompt.py,sha256=y56K2ZIvj5hZZwtEDHezJCOlduRwAcj2fc4GqhKq9ZY,23357
+mcp_agent/core/interactive_prompt.py,sha256=9s5c-XXGAKAqYh1SUVQIMRGFIcxfFjkaPmke1tyInaA,23854
 mcp_agent/core/mcp_content.py,sha256=2D7KHY9mG_vxoDwFLKvsPQV9VRIzHItM7V-jcEnACh8,8878
 mcp_agent/core/prompt.py,sha256=qnintOUGEoDPYLI9bu9G2OlgVMCe5ZPUZilgMzydXhc,7919
 mcp_agent/core/request_params.py,sha256=bEjWo86fqxdiWm2U5nPDd1uCUpcIQO9oiCinhB8lQN0,1185
-mcp_agent/core/validation.py,sha256=euK9FD7rn0px9mBMCBecWHChnZTNVIdc4452ruOTB0Q,11207
+mcp_agent/core/validation.py,sha256=Kmio1Xx-xNyCVd03RLfTxofEAWmVDPxUnQoyOSUMjR0,11445
 mcp_agent/executor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mcp_agent/executor/decorator_registry.py,sha256=CozMdIzE1LyyBfE3_R_kuxb5yi-mi--u_czKOZRp6Dg,3832
 mcp_agent/executor/executor.py,sha256=MzLSnW9nHrLHYChR3oQa5B8dajQGX26q6-S2BJCxv0o,9507
 mcp_agent/executor/task_registry.py,sha256=PCALFeYtkQrPBg4RBJnlA0aDI8nHclrNkHGUS4kV3W8,1242
-mcp_agent/executor/temporal.py,sha256=cxLUW8WWFerHlmkDlu5E241h0L-ZU9SeenYHoGQ6c-U,14964
-mcp_agent/executor/workflow.py,sha256=T2RzVZ2zD2024zPgQgXERW8zZ3XUv-3dwEeBDZCUkBg,6721
 mcp_agent/executor/workflow_signal.py,sha256=_X8Bli9XSl3QX05uZ2fc88QhLj5aDjvmsI6Bo-48oO8,11248
 mcp_agent/human_input/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mcp_agent/human_input/handler.py,sha256=s712Z5ssTCwjL9-VKoIdP5CtgMh43YvepynYisiWTTA,3144
 mcp_agent/human_input/types.py,sha256=RtWBOVzy8vnYoQrc36jRLn8z8N3C4pDPMBN5vF6qM5Y,1476
 mcp_agent/llm/__init__.py,sha256=d8zgwG-bRFuwiMNMYkywg_qytk4P8lawyld_meuUmHI,68
-mcp_agent/llm/augmented_llm.py,sha256=YIB3I_taoglo_vSmZLQ50cv1qCSctaQlWVwjI-7WTkk,18304
-mcp_agent/llm/augmented_llm_passthrough.py,sha256=U0LssNWNVuZRuD9I7Wuvpo7vdDW4xtoPLirnYCgBGTY,6128
-mcp_agent/llm/augmented_llm_playback.py,sha256=YVR2adzjMf9Q5WfYBytryWMRqJ87a3kNBnjxhApsMcU,3413
+mcp_agent/llm/augmented_llm.py,sha256=ibswlM9DdcE0yC35GqzVw3fMVWllNmKHxryrK6rFsYc,18636
+mcp_agent/llm/augmented_llm_passthrough.py,sha256=PxakaEYbXL2IeHI18yg8Vr21DT_y5FO6bPeh3XcQvZk,6204
+mcp_agent/llm/augmented_llm_playback.py,sha256=RwXtxux9Z6Y3myZbX28xoqyFyi5ZqM3FGBfohR8fgtE,3493
 mcp_agent/llm/memory.py,sha256=UakoBCJBf59JBtB6uyZM0OZjlxDW_VHtSfDs08ibVEc,3312
-mcp_agent/llm/model_factory.py,sha256=19kTu_H-c6704A50-JyZF-S5M7nNJMhVN3pwhaoRXdw,7962
+mcp_agent/llm/model_factory.py,sha256=EmsBPWphFON7VHyh3NWKhaeGJy4spfaNJ5cSqd4SHoo,7516
 mcp_agent/llm/prompt_utils.py,sha256=yWQHykoK13QRF7evHUKxVF0SpVLN-Bsft0Yixzvn0g0,4825
+mcp_agent/llm/provider_key_manager.py,sha256=rlSB1_2KYRKkESc94Ilh16Nm9THVsfBiDBbsCFOsr34,2770
+mcp_agent/llm/provider_types.py,sha256=2d36ZrqdO4E6BXpRgdzg_OJaszpb5BEgQKVRwpUYbVc,285
 mcp_agent/llm/sampling_converter.py,sha256=C7wPBlmT0eD90XWabC22zkxsrVHKCrjwIwg6cG628cI,2926
 mcp_agent/llm/sampling_format_converter.py,sha256=xGz4odHpOcP7--eFaJaFtUR8eR9jxZS7MnLH6J7n0EU,1263
 mcp_agent/llm/providers/__init__.py,sha256=heVxtmuqFJOnjjxHz4bWSqTAxXoN1E8twC_gQ_yJpHk,265
 mcp_agent/llm/providers/anthropic_utils.py,sha256=vYDN5G5jKMhD2CQg8veJYab7tvvzYkDMq8M1g_hUAQg,3275
-mcp_agent/llm/providers/augmented_llm_anthropic.py,sha256=CNKpTEvWqjOteACUx_Vha0uFpPt32C17JrkSXg_allM,14445
-mcp_agent/llm/providers/augmented_llm_deepseek.py,sha256=SdYDqZZ9hM9sBvW1FSItNn_ENEKQXGNKwVHGnjqjyAA,1927
-mcp_agent/llm/providers/augmented_llm_generic.py,sha256=v6r8mbDO69fpQnvFoy62QyQTnq2qXR2DBbn5o3pyhvg,1533
-mcp_agent/llm/providers/augmented_llm_openai.py,sha256=Wso9GVgsq8y3sqlOzTk_iQqrkCOL3LyuG07nA1PWDng,17913
-mcp_agent/llm/providers/augmented_llm_openrouter.py,sha256=jcnbjMO2NHETOOwc3BPZUcfv8gObczpnYDRZeKlV6eY,3552
+mcp_agent/llm/providers/augmented_llm_anthropic.py,sha256=8AyO8iaSjhLpU3uAaqlN2lJyo8KduUmVUlqtUU_GBuE,13752
+mcp_agent/llm/providers/augmented_llm_deepseek.py,sha256=S6_sSV07RhpcqWXi5M8uUQLNhw_Lc0nePtgVwrxCAho,1287
+mcp_agent/llm/providers/augmented_llm_generic.py,sha256=5Uq8ZBhcFuQTt7koP_5ykolREh2iWu8zKhNbh3pM9lQ,1210
+mcp_agent/llm/providers/augmented_llm_openai.py,sha256=jwV-_GmGMg-nNGVcG_P0i5J5PIj30sOoGxrWT6RUbZI,17244
+mcp_agent/llm/providers/augmented_llm_openrouter.py,sha256=V_TlVKm92GHBxYIo6gpvH_6cAaIdppS25Tz6x5T7LW0,2341
 mcp_agent/llm/providers/multipart_converter_anthropic.py,sha256=t5lHYGfFUacJldnrVtMNW-8gEMoto8Y7hJkDrnyZR-Y,16650
 mcp_agent/llm/providers/multipart_converter_openai.py,sha256=zCj0LBgd9FDG8aL_GeTrPo2ssloYnmC_Uj3ENWVUJAg,16753
 mcp_agent/llm/providers/openai_multipart.py,sha256=qKBn7d3jSabnJmVgWweVzqh8q9mBqr09fsPmP92niAQ,6899
@@ -81,7 +81,7 @@ mcp_agent/logging/tracing.py,sha256=d5lSXakzzi5PtQpUkVkOnYaGX8NduGPq__S7vx-Ln8U,
 mcp_agent/logging/transport.py,sha256=m8YsLLu5T8eof_ndpLQs4gHOzqqEL98xsVwBwDsBfxI,17335
 mcp_agent/mcp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mcp_agent/mcp/gen_client.py,sha256=fAVwFVCgSamw4PwoWOV4wrK9TABx1S_zZv8BctRyF2k,3030
-mcp_agent/mcp/interfaces.py,sha256=vma7bbWbY3zp1RM6hMYxVO4aV6Vfaygm-nLwzK2jFKI,6748
+mcp_agent/mcp/interfaces.py,sha256=zc4GM6Q97yqucHlNWWuODjQAIECxdwWXqFTxJ-G_AoY,6803
 mcp_agent/mcp/logger_textio.py,sha256=vljC1BtNTCxBAda9ExqNB-FwVNUZIuJT3h1nWmCjMws,3172
 mcp_agent/mcp/mcp_agent_client_session.py,sha256=RMYNltc2pDIzxwEJSS5589RbvPO0KWV4Y3jSyAmhKf0,4181
 mcp_agent/mcp/mcp_aggregator.py,sha256=jaWbOvb3wioECohZ47CubyxfJ5QkfNSshu1hwhZksG4,40486
@@ -102,7 +102,7 @@ mcp_agent/mcp/prompts/prompt_load.py,sha256=Zo0FogqWFEG5FtF1d9ZH-RWsCSSMsi5FIEQH
 mcp_agent/mcp/prompts/prompt_server.py,sha256=SiUR2xYfd3vEpghnYRdzz2rFEMtAbCKx2xzUXgvz1g8,18501
 mcp_agent/mcp/prompts/prompt_template.py,sha256=EejiqGkau8OizORNyKTUwUjrPof5V-hH1H_MBQoQfXw,15732
 mcp_agent/mcp_server/__init__.py,sha256=zBU51ITHIEPScd9nRafnhEddsWqXRPAAvHhkrbRI2_4,155
-mcp_agent/mcp_server/agent_server.py,sha256=eMPUHcHgXNCldwOAh648D0TLNMPC_zSQtqShsT8s6Eg,16010
+mcp_agent/mcp_server/agent_server.py,sha256=yne6wdc2sHwzOGuLFikLxUnHEuPEr3l8cfPlT_jbzBI,19811
 mcp_agent/resources/examples/data-analysis/analysis-campaign.py,sha256=QdNdo0-7LR4Uzw61hEU_jVKmWyk6A9YpGo81kMwVobM,7267
 mcp_agent/resources/examples/data-analysis/analysis.py,sha256=M9z8Q4YC5OGuqSa5uefYmmfmctqMn-WqCSfg5LI407o,2609
 mcp_agent/resources/examples/data-analysis/fastagent.config.yaml,sha256=ini94PHyJCfgpjcjHKMMbGuHs6LIj46F1NwY0ll5HVk,1609
@@ -143,8 +143,8 @@ mcp_agent/resources/examples/workflows/parallel.py,sha256=n0dFN26QvYd2wjgohcaUBf
 mcp_agent/resources/examples/workflows/router.py,sha256=E4x_-c3l4YW9w1i4ARcDtkdeqIdbWEGfsMzwLYpdbVc,1677
 mcp_agent/resources/examples/workflows/short_story.txt,sha256=X3y_1AyhLFN2AKzCKvucJtDgAFIJfnlbsbGZO5bBWu0,1187
 mcp_agent/ui/console_display.py,sha256=TVGDtJ37hc6UG0ei9g7ZPZZfFNeS1MYozt-Mx8HsPCk,9752
-fast_agent_mcp-0.2.13.dist-info/METADATA,sha256=aPMXpckeF8JsbXqQ8gGFWXpN208nlwUqkXmeG4yKd-w,29940
-fast_agent_mcp-0.2.13.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-fast_agent_mcp-0.2.13.dist-info/entry_points.txt,sha256=bRniFM5zk3Kix5z7scX0gf9VnmGQ2Cz_Q1Gh7Ir4W00,186
-fast_agent_mcp-0.2.13.dist-info/licenses/LICENSE,sha256=cN3FxDURL9XuzE5mhK9L2paZo82LTfjwCYVT7e3j0e4,10939
-fast_agent_mcp-0.2.13.dist-info/RECORD,,
+fast_agent_mcp-0.2.16.dist-info/METADATA,sha256=8Ed0qKTHlBEYhMa5HleFhAtZjrmV-Kq0Sir2RMO9yW0,29940
+fast_agent_mcp-0.2.16.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+fast_agent_mcp-0.2.16.dist-info/entry_points.txt,sha256=bRniFM5zk3Kix5z7scX0gf9VnmGQ2Cz_Q1Gh7Ir4W00,186
+fast_agent_mcp-0.2.16.dist-info/licenses/LICENSE,sha256=cN3FxDURL9XuzE5mhK9L2paZo82LTfjwCYVT7e3j0e4,10939
+fast_agent_mcp-0.2.16.dist-info/RECORD,,

mcp_agent/agents/agent.py

@@ -66,7 +66,7 @@ class Agent(BaseAgent):
         agent_name_str = str(self.name)
 
         # Create agent_types dictionary with just this agent
-        agent_types = {agent_name_str: self.agent_type}
+        agent_types = {agent_name_str: self.agent_type.value}
 
         # Create the interactive prompt
         prompt = InteractivePrompt(agent_types=agent_types)
@@ -84,7 +84,7 @@ class Agent(BaseAgent):
         async def list_prompts_wrapper(agent_name):
             # Always call list_prompts on this agent regardless of agent_name
             return await self.list_prompts()
-            
+
         # Define wrapper for list_resources function
         async def list_resources_wrapper(agent_name):
             # Always call list_resources on this agent regardless of agent_name

mcp_agent/agents/base_agent.py

@@ -32,7 +32,7 @@ from mcp.types import (
 )
 from pydantic import BaseModel
 
-from mcp_agent.core.agent_types import AgentConfig
+from mcp_agent.core.agent_types import AgentConfig, AgentType
 from mcp_agent.core.exceptions import PromptExitError
 from mcp_agent.core.prompt import Prompt
 from mcp_agent.core.request_params import RequestParams
@@ -623,7 +623,7 @@ class BaseAgent(MCPAggregator, AgentProtocol):
         return response.first_text()
 
     @property
-    def agent_type(self) -> str:
+    def agent_type(self) -> AgentType:
         """
         Return the type of this agent.
 
@@ -632,7 +632,7 @@ class BaseAgent(MCPAggregator, AgentProtocol):
         Returns:
             String representing the agent type
         """
-        return self.config.agent_type
+        return AgentType.BASIC
 
     @property
     def message_history(self) -> List[PromptMessageMultipart]:

mcp_agent/agents/workflow/chain_agent.py

@@ -26,9 +26,9 @@ class ChainAgent(BaseAgent):
 
     # TODO -- consider adding "repeat" mode
     @property
-    def agent_type(self) -> str:
+    def agent_type(self) -> AgentType:
         """Return the type of this agent."""
-        return AgentType.CHAIN.value
+        return AgentType.CHAIN
 
     def __init__(
         self,

mcp_agent/agents/workflow/evaluator_optimizer.py

@@ -64,11 +64,11 @@ class EvaluatorOptimizerAgent(BaseAgent):
     for refinement, continuing until a quality threshold is reached or a maximum
     number of refinement cycles is completed.
     """
-    
+
     @property
-    def agent_type(self) -> str:
+    def agent_type(self) -> AgentType:
         """Return the type of this agent."""
-        return AgentType.EVALUATOR_OPTIMIZER.value
+        return AgentType.EVALUATOR_OPTIMIZER
 
     def __init__(
         self,

mcp_agent/agents/workflow/orchestrator_agent.py

@@ -46,11 +46,11 @@ class OrchestratorAgent(BaseAgent):
     to specialized worker agents, synthesizing their results into a cohesive output.
     Supports both full planning and iterative planning modes.
     """
-    
+
     @property
-    def agent_type(self) -> str:
+    def agent_type(self) -> AgentType:
         """Return the type of this agent."""
-        return AgentType.ORCHESTRATOR.value
+        return AgentType.ORCHESTRATOR
 
     def __init__(
         self,

mcp_agent/agents/workflow/parallel_agent.py

@@ -20,9 +20,9 @@ class ParallelAgent(BaseAgent):
     """
     
     @property
-    def agent_type(self) -> str:
+    def agent_type(self) -> AgentType:
         """Return the type of this agent."""
-        return AgentType.PARALLEL.value
+        return AgentType.PARALLEL
 
     def __init__(
         self,

mcp_agent/agents/workflow/router_agent.py

@@ -106,9 +106,9 @@ class RouterAgent(BaseAgent):
     """
 
     @property
-    def agent_type(self) -> str:
+    def agent_type(self) -> AgentType:
         """Return the type of this agent."""
-        return AgentType.ROUTER.value
+        return AgentType.ROUTER
 
     def __init__(
         self,