MindsDB 25.5.4.0__py3-none-any.whl → 25.5.4.1__py3-none-any.whl

{mindsdb-25.5.4.0.dist-info → mindsdb-25.5.4.1.dist-info}/RECORD

@@ -1,19 +1,15 @@
-mindsdb/__about__.py,sha256=qdyLrPJthOack70IzRcKOJ9is1w28JD60cFxanFX8sQ,456
+mindsdb/__about__.py,sha256=QDcsJNqSqjZ8PJ0cUrDKDkRlBeCubeeLB5R6sxAlm1U,456
 mindsdb/__init__.py,sha256=fZopLiAYa9MzMZ0d48JgHc_LddfFKDzh7n_8icsjrVs,54
 mindsdb/__main__.py,sha256=bkLqiYLsx1KyCyi6tJjyMBkpMvO3g37wOC0PymMbiHU,24037
 mindsdb/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mindsdb/api/a2a/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mindsdb/api/a2a/__main__.py,sha256=x8-7l-S3j6DpxQUrg6Y9yOGG2vC77DqYNi2gPNy25EE,3592
-mindsdb/api/a2a/a2a_client.py,sha256=Lyw7CFj031pZNK_wvY4MAR1ptZoDRaitbCi0jmN4wQA,13634
+mindsdb/api/a2a/__main__.py,sha256=zrzTzt9zgN38NImInrmxCsdr66urJNS0JOq0xJaKHac,4712
 mindsdb/api/a2a/agent.py,sha256=DN3VCsdFuoFBXyQDeH4C9pNGnHSEM-KnHSCjL-HY4ls,13007
 mindsdb/api/a2a/constants.py,sha256=1BmvDneWzNG6YJFUc-gZ9CNCO4wq2_12vFe6aw23yX4,2292
-mindsdb/api/a2a/run_a2a.py,sha256=7uvTtc6Te4qBiF9I0_48dVBZmY4YmKrMT0jbBwnGGhE,4347
-mindsdb/api/a2a/task_manager.py,sha256=7zYsAizpZlE4WlsEotpRSos742dZbzpVyeKvuTaUXLs,24620
+mindsdb/api/a2a/run_a2a.py,sha256=IQa7pSvgDoh7u5V2Ec5flzoeAJsNuSMnkRehEMPrI3Q,2878
+mindsdb/api/a2a/task_manager.py,sha256=SwTwDC4_4pLIm8471Bi10WFvpI7uQ7qFLTDMcpg9mgI,23970
 mindsdb/api/a2a/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mindsdb/api/a2a/common/types.py,sha256=EYBpqEj-nZapSKe2jI8BcNDfSlTXXTbw8RcSA1DII9Q,8787
-mindsdb/api/a2a/common/client/__init__.py,sha256=V8lbSEpOWNM2XjG5ky7NIUM0Q2HZTexzL3I8WqZfo-w,117
-mindsdb/api/a2a/common/client/card_resolver.py,sha256=yCDxwNAgAz-6lDWGr8uhj8V0DGs9kSiybPsmhK3vh8Y,671
-mindsdb/api/a2a/common/client/client.py,sha256=VLZaBTf1vhbLq_4Wr5Z4o1PMMW2NsCdTTVQDlGJklA8,3346
 mindsdb/api/a2a/common/server/__init__.py,sha256=Bl7Qa3Pk4QWY4yDPRNyHeTzNJpAe9HgUJd8-oTnMPv8,152
 mindsdb/api/a2a/common/server/server.py,sha256=ymnM0HblF9iH1Htay-LfFu-_CQjXieN_ZEgyHYkHPfk,6352
 mindsdb/api/a2a/common/server/task_manager.py,sha256=NnKO2ZcI-KBWiX4wNc8FbLmnVsb5KeIm62B4jdX73IU,10189
@@ -21,7 +17,7 @@ mindsdb/api/a2a/common/server/utils.py,sha256=1k_fBfECT0JQFlFcAJhpNlTYYOruxDlV0S
 mindsdb/api/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mindsdb/api/common/check_auth.py,sha256=cQEZqsnCbrRtUf8j4H6uPF98cDPu79t8TdtwBi5g30w,1345
 mindsdb/api/executor/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-mindsdb/api/executor/command_executor.py,sha256=uaKdegKhR4cswXE0uN3zuIFlJD8MnohRQ1HSGfFIwag,80825
+mindsdb/api/executor/command_executor.py,sha256=mXpbe1KI83mvJCWK-RVWErr6iAaaq8XQ_eROe98s1I8,78515
 mindsdb/api/executor/exceptions.py,sha256=SoxhetIKIVOAwbVP_NatfoKnwt-Xj2yFCiAIqSDxpIE,566
 mindsdb/api/executor/controllers/__init__.py,sha256=px47lPVKqfpqgcoEBHyWoax-ad01rNOTJQCgQmG0Flo,50
 mindsdb/api/executor/controllers/session_controller.py,sha256=2Jf-V0nj7k0aB4scujNVyx91h54odkDrdK1ydsCo46g,3072
@@ -83,7 +79,7 @@ mindsdb/api/http/namespaces/agents.py,sha256=H3C4MmNCsrk3Dv8Cw1kx0j7EvNOia5PfyfQ
 mindsdb/api/http/namespaces/analysis.py,sha256=maYxI-TP9RswWHgKX3KAZQJK_XSBQfpZMqv0ZH1ZC0k,4295
 mindsdb/api/http/namespaces/auth.py,sha256=Qm1ZUBdbv_nTDzSQHdzXEosdpYCRzIa17k1yYErOeuk,5483
 mindsdb/api/http/namespaces/chatbots.py,sha256=pLSaTzm2GtqJM-RW8MrEQTIJybLRCMbLv99BNABxMjw,11877
-mindsdb/api/http/namespaces/config.py,sha256=tC7MItEt_hjpG1O8-y85sEqPtwX_iPj7mNdTQ6zSg-0,8785
+mindsdb/api/http/namespaces/config.py,sha256=eHZybqUA-HryK7zTvEteMqcDw39J-yKtEs3nRdvrtz8,8454
 mindsdb/api/http/namespaces/databases.py,sha256=QuHD5zSIftfb5qRdPJS3UtZEnfQnf1fp0wVKe98h7Rg,18543
 mindsdb/api/http/namespaces/default.py,sha256=r8PXn00Um2eyKB5e_Kj7fzk4e4LYH-JCzXCpxgJA2vY,4729
 mindsdb/api/http/namespaces/file.py,sha256=ONMsQDDC2Ug-O38cbPKSL8UpQ78soQKsPHOfzUj-Bfo,6933
@@ -359,7 +355,7 @@ mindsdb/integrations/handlers/byom_handler/byom_handler.py,sha256=uI-UN2XLx7jAx7
 mindsdb/integrations/handlers/byom_handler/connection_args.py,sha256=chqPtpiEBtko8kxS15ImkZ6Q3kuhv7vwOzV3JeyU4cU,522
 mindsdb/integrations/handlers/byom_handler/icon.svg,sha256=q8G2Qws-_E79MdxpSDHtNCqLwK0RWauNmRwGszIs_FE,2542
 mindsdb/integrations/handlers/byom_handler/proc_wrapper.py,sha256=GqOB8Zw4-SYsn4lu3tkYORaNfgM4bm0HMDPyOJtErKk,5644
-mindsdb/integrations/handlers/byom_handler/requirements.txt,sha256=WIPqhfnSGN4zXV7N9gor-Mjk5xsua3vH9ENBltRP9dY,27
+mindsdb/integrations/handlers/byom_handler/requirements.txt,sha256=VSOkZhUmawzRmgjd4WHhL7PW6yP63v3EBL7JmC-WWR4,10
 mindsdb/integrations/handlers/cassandra_handler/__about__.py,sha256=rMC_DPK11JoHKA9UceTm_jkUOggj04XKkAUqw5P8wXg,347
 mindsdb/integrations/handlers/cassandra_handler/__init__.py,sha256=US2TGuaKp48HJX_ayeNypltbagOASrnKSSUsCMNjsl4,564
 mindsdb/integrations/handlers/cassandra_handler/cassandra_handler.py,sha256=HoQPp0Ox20BDCNB8rtrwAEoGLuh7gWDiP4YQC4_pQhU,787
@@ -873,7 +869,7 @@ mindsdb/integrations/handlers/lancedb_handler/__init__.py,sha256=nwShTdkEcKKhiDL
 mindsdb/integrations/handlers/lancedb_handler/connection_args.py,sha256=KtHvSBLPS9P7LKlXULwRlUTUCvpG69bES2DUL-e2xME,956
 mindsdb/integrations/handlers/lancedb_handler/icon.svg,sha256=UHDcgJDgSbA74qyN6dLXKquPHXjYHd93g-TyQlIR84s,3475
 mindsdb/integrations/handlers/lancedb_handler/lancedb_handler.py,sha256=zQik_f_QjArzSrla_uxwYPQIdiKIwnYZIIABQmI__Zc,12399
-mindsdb/integrations/handlers/lancedb_handler/requirements.txt,sha256=yBw6h3zm01B_3qmewTg3ymJ1U_nRAenRda4t0i5wvTo,37
+mindsdb/integrations/handlers/lancedb_handler/requirements.txt,sha256=syp4bJ3vJNjPo7QEMRnpjQcNsUTpwiWZeCzg_OVVWnk,21
 mindsdb/integrations/handlers/lancedb_handler/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mindsdb/integrations/handlers/lancedb_handler/tests/test_lancedb_handler.py,sha256=dHFnAMplZCfXk0DwpDM1gKIsG42vEypx9-51nRoCQZc,3351
 mindsdb/integrations/handlers/langchain_embedding_handler/__about__.py,sha256=CqrMPm5gho1vYeXuFNSisjwl32JpHR3IoKv3xw_7fx8,388
@@ -935,7 +931,7 @@ mindsdb/integrations/handlers/lindorm_handler/tests/test_lindorm_handler.py,sha2
 mindsdb/integrations/handlers/litellm_handler/__about__.py,sha256=qKuzAOGHI5gYpkNeGTTulSTMuQdHqmWFPTtJhLil4jw,341
 mindsdb/integrations/handlers/litellm_handler/__init__.py,sha256=phvu1-lcM1HovXh4GHunND_ScTON4Yb3s4VXNVd_epU,550
 mindsdb/integrations/handlers/litellm_handler/icon.png,sha256=mfPLH1-n3D1q3U57hADzl7s2G-W-EDevUHzmvnLTjy8,37122
-mindsdb/integrations/handlers/litellm_handler/litellm_handler.py,sha256=-C82eH--GVPD-4P7_lPk3GbcTGmyqYLjuGZwW9iQ4pk,4911
+mindsdb/integrations/handlers/litellm_handler/litellm_handler.py,sha256=1lE0R1Z1S4Nwtv5PVfR3uohfxNsO_WNjDe3QJLRYdhM,5625
 mindsdb/integrations/handlers/litellm_handler/requirements.txt,sha256=dy-zvg7q-XWnmLu6dh1DSphrzDsLZHec2oSRuF-jWmw,18
 mindsdb/integrations/handlers/litellm_handler/settings.py,sha256=LUnndIS2pqpQB84fLe2AhBWzSzpeHbeoeVNz3T6-Rlk,2301
 mindsdb/integrations/handlers/llama_index_handler/__about__.py,sha256=BtxlhE6R27NMF-AQRn5hhTK3cSdkyUDU6f9venjZ0dU,359
@@ -1719,8 +1715,8 @@ mindsdb/integrations/libs/response.py,sha256=YMRiPLoQ6UR2jj9XD3yuqniTgyL5xT_p9F8
 mindsdb/integrations/libs/storage_handler.py,sha256=g4rcAD4TzmxWmEtS00235_NAnrdulIir4If6E4y_OUo,3512
 mindsdb/integrations/libs/vectordatabase_handler.py,sha256=p4wWZIl1fq6PnCiHSjTPW5Jlrmdw9MzU7RzKvpDUXVA,18445
 mindsdb/integrations/libs/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mindsdb/integrations/libs/llm/config.py,sha256=FH5ODdNmiqdwTLHpXmEla0zwEPlitNactTm_i2UUmC0,4337
-mindsdb/integrations/libs/llm/utils.py,sha256=K5jc9NyME4MZWyVyirhgQA_Dq6HRvP4lhJqnlrcjWy0,25101
+mindsdb/integrations/libs/llm/config.py,sha256=p_N1E3b-KB2g-OT2gC66QP1ahh7VftjXublf4k_uLig,4957
+mindsdb/integrations/libs/llm/utils.py,sha256=Wz2aptwngVyPX8tVrNx1yUHffpgnMDHQG-e01LoXFwc,25140
 mindsdb/integrations/libs/ml_handler_process/__init__.py,sha256=TW1DslwZTaD8oeGaxPx99uYkPAeDIeHWAturBcnDBFU,756
 mindsdb/integrations/libs/ml_handler_process/create_engine_process.py,sha256=YHbk1x7GEIZXcM0IazB2z2tvu5Yd24Fb4aJPpUg321A,788
 mindsdb/integrations/libs/ml_handler_process/create_validation_process.py,sha256=-69NRhBGnNG0hgNi9O_W5LffIxPBUmuxy4bbQY2Z-_8,444
@@ -1785,7 +1781,7 @@ mindsdb/integrations/utilities/rag/loaders/vector_store_loader/vector_store_load
 mindsdb/integrations/utilities/rag/pipelines/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mindsdb/integrations/utilities/rag/pipelines/rag.py,sha256=Ij91MJ0QpqocHs2vLfLO43PFZPFCyt1mvFjwvDecVU4,15551
 mindsdb/integrations/utilities/rag/rerankers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mindsdb/integrations/utilities/rag/rerankers/base_reranker.py,sha256=dtXKri1RHsjhbfkgsKDTvrx0MedhFxc6iwlARpJDQ0c,17998
+mindsdb/integrations/utilities/rag/rerankers/base_reranker.py,sha256=6ZcyEDeiykNKeTH_jH_vGBYA3yAsiIzmumvXVuVVqYc,16005
 mindsdb/integrations/utilities/rag/rerankers/reranker_compressor.py,sha256=IN7R2dtqEtqdhPQh5r62oZOII_4m_ZU-XCpnbvSqjw0,3729
 mindsdb/integrations/utilities/rag/retrievers/__init__.py,sha256=Kuo3AJxzHVXMxPFxGqz2AXNPzjBzyMuk2yQj9pFpOsI,128
 mindsdb/integrations/utilities/rag/retrievers/auto_retriever.py,sha256=ODNXqeBuDfatGQLvKvogO0aA-A5v3Z4xbCbvO5ICvt4,3923
@@ -1800,9 +1796,9 @@ mindsdb/interfaces/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVG
 mindsdb/interfaces/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mindsdb/interfaces/agents/agents_controller.py,sha256=l_lC08HnE7F_Iki-hm9VznEY8Afmv1-diO7YJIyqfSo,27106
 mindsdb/interfaces/agents/callback_handlers.py,sha256=_iOUz-g5Hglf2wtfpx6ENKNCu2Cba-ZbpQtXyR2xElY,6700
-mindsdb/interfaces/agents/constants.py,sha256=7drGeBdnwJOBnPKgjd8lKhIYu1yYlYrDUxMvStIuEsA,5822
+mindsdb/interfaces/agents/constants.py,sha256=ms_XhYhxzGcQ27fPKZxZgWvG9H5TdyVOSbsesyCzz4I,5927
 mindsdb/interfaces/agents/event_dispatch_callback_handler.py,sha256=-76yTtxTHO5AkFTtr_RvYfkdUROJHcKZx6KJDZvj_-M,1331
-mindsdb/interfaces/agents/langchain_agent.py,sha256=GUaxPkT7a5qXCKo2j3p-M8qpiJkwLfZv0C3epxbrMAU,28313
+mindsdb/interfaces/agents/langchain_agent.py,sha256=p_gSmIvIkEkfOlm_vdForcJMBRlqoQeIu3QHrkJU1Ps,28543
 mindsdb/interfaces/agents/langfuse_callback_handler.py,sha256=-51IWB5U2_m71xx00IwSOAK6gJ2n-HD_CzbtbmEfbBQ,11598
 mindsdb/interfaces/agents/litellm_server.py,sha256=j33LqHlUt9XApMozP4mHy9iFGT1Lx7WRqKHwGfJfT8w,11366
 mindsdb/interfaces/agents/mcp_client_agent.py,sha256=9oki-GDsN1H6fYsNLBQWTMYSHGpWliYF9I5AZSuXS8U,9814
@@ -1833,7 +1829,7 @@ mindsdb/interfaces/jobs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG
 mindsdb/interfaces/jobs/jobs_controller.py,sha256=2-L61saLbpaQ4hAT_HepIq0DUqdUxsPuNNMkj3ZfAJk,18299
 mindsdb/interfaces/jobs/scheduler.py,sha256=eHBWTpZozI_AC-hKmg4p0qRoGZW8O3gEt5Y9nDhHWpw,3696
 mindsdb/interfaces/knowledge_base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mindsdb/interfaces/knowledge_base/controller.py,sha256=UZww7HeADl03JC6GkfjVI8SuoZiylf2B_qqeX_pLiOQ,46617
+mindsdb/interfaces/knowledge_base/controller.py,sha256=DOiZpyN5iLukWL96cA0sFVrdL36Lkfd9q3KFsrXMsDE,48218
 mindsdb/interfaces/knowledge_base/utils.py,sha256=gRWJkHVic5mOy1rnjd7eON5mo7rdAyxVYfEbg0iJxmk,855
 mindsdb/interfaces/knowledge_base/preprocessing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mindsdb/interfaces/knowledge_base/preprocessing/constants.py,sha256=iW5q65albIfTT1ZuxNceJ8o7yrOcttCH1Kx4Vdo-iPY,296
@@ -1945,7 +1941,7 @@ mindsdb/migrations/versions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMp
 mindsdb/utilities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mindsdb/utilities/auth.py,sha256=nfC8oqvaN3GAATc_LeHJ34Kg3PYfyfJ-KI6TN_fOB48,2568
 mindsdb/utilities/cache.py,sha256=nkfEt8Pw5H_8sPNXMaS2SZCC3NrEMi486K8m61zqu-s,7590
-mindsdb/utilities/config.py,sha256=qF1Tr1YPz0c2O5GqUbZZkWKci_7kstCkuL-5OXAFJgI,26578
+mindsdb/utilities/config.py,sha256=lEQH2rkaf3PwoSSdnHecc0Yhskm0Zp1niFtW6KSlePw,26360
 mindsdb/utilities/context.py,sha256=IdH0bXIIBHuJ_HzVQIRAZhOs4GD15AwDpXTlNnTBHek,1846
 mindsdb/utilities/context_executor.py,sha256=OcJu-FgHZUmtAa_jOfXtwr7LPH3Vw2FPPJlx_9cWi7w,1945
 mindsdb/utilities/exception.py,sha256=q-9cwMLmQvuPpwdjRG0xNZ23z9cxHSfyT2295Rk6waA,1034
@@ -1981,8 +1977,8 @@ mindsdb/utilities/profiler/__init__.py,sha256=d4VXl80uSm1IotR-WwbBInPmLmACiK0Azx
 mindsdb/utilities/profiler/profiler.py,sha256=KCUtOupkbM_nCoof9MtiuhUzDGezx4a4NsBX6vGWbPA,3936
 mindsdb/utilities/render/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mindsdb/utilities/render/sqlalchemy_render.py,sha256=6NqOcfz19dhd0PEyNfOtNHsBt1fGjq2E3nFNUad5HOk,31444
-mindsdb-25.5.4.0.dist-info/licenses/LICENSE,sha256=ziqdjujs6WDn-9g3t0SISjHCBc2pLRht3gnRbQoXmIs,5804
-mindsdb-25.5.4.0.dist-info/METADATA,sha256=fildVyD-98sD9p-kaIUtYeQlu3EQhWdw5hW8VhWFXPk,44945
-mindsdb-25.5.4.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mindsdb-25.5.4.0.dist-info/top_level.txt,sha256=10wPR96JDf3hM8aMP7Fz0lDlmClEP480zgXISJKr5jE,8
-mindsdb-25.5.4.0.dist-info/RECORD,,
+mindsdb-25.5.4.1.dist-info/licenses/LICENSE,sha256=ziqdjujs6WDn-9g3t0SISjHCBc2pLRht3gnRbQoXmIs,5804
+mindsdb-25.5.4.1.dist-info/METADATA,sha256=lt4b5XplYOvrc1hkuBhDoh0TTt-xF7OTX69ce9Yoko8,44798
+mindsdb-25.5.4.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mindsdb-25.5.4.1.dist-info/top_level.txt,sha256=10wPR96JDf3hM8aMP7Fz0lDlmClEP480zgXISJKr5jE,8
+mindsdb-25.5.4.1.dist-info/RECORD,,

mindsdb/api/a2a/a2a_client.py

@@ -1,439 +0,0 @@
-#!/usr/bin/env python
-"""Simple command-line tool to interact with an A2A server.
-
-This script can:
-  • fetch the agent card (metadata) from the server
-  • send a single (non-streaming) task request
-  • send a streaming request and live-print SSE events
-
-Originally located at the repository root; restored to that location.
-"""
-
-from __future__ import annotations
-
-import argparse
-import json
-import sys
-import uuid
-from typing import Iterator, Optional
-import logging
-
-import requests
-
-DEFAULT_HOST = "localhost"
-DEFAULT_PORT = 10002
-
-###############################################################################
-# Helper functions
-###############################################################################
-
-
-def _log_json(log_func, obj: dict, *, prefix: str = "") -> None:
-    """Log JSON helper."""
-    log_func(prefix + json.dumps(obj, indent=2))
-
-
-def get_agent_info(
-    a2a_host: str = DEFAULT_HOST,
-    a2a_port: int = DEFAULT_PORT,
-) -> Optional[dict]:
-    """Retrieve the agent card from `/.well-known/agent.json` (or legacy path)."""
-
-    url = f"http://{a2a_host}:{a2a_port}/.well-known/agent.json"
-
-    logging.debug(f"Fetching agent info from {url} …")
-
-    try:
-        response = requests.get(url, timeout=10)
-        if response.status_code == 404:
-            legacy_url = f"http://{a2a_host}:{a2a_port}/agent-card"
-            logging.debug(f"Not found, trying legacy path {legacy_url}")
-            response = requests.get(legacy_url, timeout=10)
-
-        if response.ok:
-            card = response.json()
-            _log_json(logging.debug, card, prefix="Received agent card:\n")
-            return card
-
-        logging.debug(f"Failed to fetch agent card – status: {response.status_code}")
-    except requests.RequestException as exc:
-        logging.debug(f"Connection error while fetching agent card: {exc}")
-
-    return None
-
-
-###############################################################################
-# Task helpers
-###############################################################################
-
-
-def _new_ids() -> tuple[str, str, str]:
-    """Generate fresh UUID4 strings for taskId, sessionId, requestId."""
-
-    return uuid.uuid4().hex, uuid.uuid4().hex, uuid.uuid4().hex
-
-
-def _post_json(url: str, payload: dict, *, stream: bool = False) -> requests.Response:
-    headers = {"Content-Type": "application/json"}
-    if stream:
-        headers["Accept"] = "text/event-stream"
-    return requests.post(
-        url, json=payload, headers=headers, stream=stream, timeout=None
-    )
-
-
-###############################################################################
-# Non-streaming request
-###############################################################################
-
-
-def send_a2a_query(
-    query: str,
-    *,
-    a2a_host: str = DEFAULT_HOST,
-    a2a_port: int = DEFAULT_PORT,
-) -> bool:
-    """Send a *blocking* `tasks/send` request and print the result."""
-
-    task_id, session_id, request_id = _new_ids()
-
-    payload = {
-        "jsonrpc": "2.0",
-        "id": request_id,
-        "method": "tasks/send",
-        "params": {
-            "id": task_id,
-            "sessionId": session_id,
-            "message": {
-                "role": "user",
-                "parts": [
-                    {"type": "text", "text": query},
-                ],
-            },
-            "acceptedOutputModes": ["text/plain"],
-        },
-    }
-
-    url = f"http://{a2a_host}:{a2a_port}/a2a"
-    logging.debug(f"POST {url}")
-    _log_json(logging.debug, payload, prefix="Request →\n")
-    logging.info("Sending query …")
-
-    try:
-        response = _post_json(url, payload)
-    except requests.RequestException as exc:
-        logging.error(f"⚠️  Network error: {exc}")
-        return False
-
-    if not response.ok:
-        logging.error(f"⚠️  HTTP {response.status_code}\n{response.text}")
-        return False
-
-    try:
-        data = response.json()
-    except ValueError:
-        logging.error(f"⚠️  Invalid JSON response: {response.text[:200]}")
-        return False
-
-    _log_json(logging.debug, data, prefix="Full response ←\n")
-
-    if "error" in data:
-        logging.error(f"⚠️  RPC error: {data['error'].get('message')}")
-        return False
-
-    result = data.get("result")
-    if not result:
-        logging.error("⚠️  No result field in response")
-        return False
-
-    # Print status/"thinking" message
-    status = result.get("status", {})
-    msg_parts = status.get("message", {}).get("parts", [])
-    if msg_parts:
-        logging.info("\nAgent thinking:")
-        _log_parts(logging.info, msg_parts)
-
-    # Print artifacts (agent answer)
-    artifacts = result.get("artifacts") or []
-    if artifacts:
-        logging.info("\nAgent response:")
-        for artifact in artifacts:
-            _log_parts(logging.info, artifact.get("parts", []))
-    else:
-        logging.info("(No artifacts returned)")
-
-    return True
-
-
-###############################################################################
-# Streaming request helpers
-###############################################################################
-
-
-def _iter_sse_lines(resp: requests.Response) -> Iterator[str]:
-    """Yield raw Server-Sent-Event lines (decoded)."""
-    buf = ""
-    for chunk in resp.iter_content(chunk_size=1024):
-        if not chunk:
-            continue
-        buf += chunk.decode()
-        while "\n" in buf:
-            line, buf = buf.split("\n", 1)
-            yield line.rstrip("\r")
-
-
-def _parse_sse_event(lines: list[str]) -> dict | None:
-    """Parse an SSE event block into a dict."""
-    data_lines = [line_content[6:] for line_content in lines if line_content.startswith("data: ")]
-    if not data_lines:
-        return None
-    try:
-        return json.loads("\n".join(data_lines))
-    except json.JSONDecodeError:
-        return None
-
-
-def send_streaming_query(
-    query: str,
-    *,
-    a2a_host: str = DEFAULT_HOST,
-    a2a_port: int = DEFAULT_PORT,
-) -> bool:
-    """Send a `tasks/sendSubscribe` request and stream responses as they arrive."""
-
-    task_id, session_id, request_id = _new_ids()
-
-    payload = {
-        "jsonrpc": "2.0",
-        "id": request_id,
-        "method": "tasks/sendSubscribe",
-        "params": {
-            "id": task_id,
-            "sessionId": session_id,
-            "message": {
-                "role": "user",
-                "parts": [
-                    {"type": "text", "text": query},
-                ],
-            },
-            "acceptedOutputModes": ["text/plain"],
-        },
-    }
-
-    url = f"http://{a2a_host}:{a2a_port}/a2a"
-    logging.debug(f"POST (stream) {url}")
-    _log_json(logging.debug, payload, prefix="Request →\n")
-
-    try:
-        response = _post_json(url, payload, stream=True)
-    except requests.RequestException as exc:
-        logging.error(f"⚠️  Network error: {exc}")
-        return False
-
-    if not response.ok:
-        logging.error(f"⚠️  HTTP {response.status_code}\n{response.text}")
-        return False
-
-    logging.info("Streaming events (Ctrl-C to abort):")
-
-    try:
-        event_lines: list[str] = []
-        for line in _iter_sse_lines(response):
-            if line == "":  # blank = dispatch event
-                event = _parse_sse_event(event_lines)
-                event_lines.clear()
-                if event is not None:
-                    _handle_stream_event(event)
-                continue
-            event_lines.append(line)
-    except KeyboardInterrupt:
-        logging.info("\nInterrupted by user.")
-        return False
-
-    return True
-
-
-###############################################################################
-# Output helpers
-###############################################################################
-
-
-def _log_parts(log_func, parts: list[dict]) -> None:
-    for part in parts:
-        if part.get("type") == "text":
-            log_func(part.get("text", ""))
-        elif part.get("type") == "data":
-            log_func("\nStructured data:")
-            _log_json(log_func, part.get("data", {}))
-
-
-def _handle_stream_event(event: dict) -> None:
-    """Handle a single SSE event parsed as JSON.
-
-    The server sends JSON-RPC messages of the form::
-
-        {"jsonrpc":"2.0","id":"…","result":{…}}
-
-    where *result* is either a TaskStatusUpdateEvent or a TaskArtifactUpdateEvent.
-    These objects do not carry an explicit ``type`` field, so we infer it based
-    on their keys.
-    """
-
-    # If logging level is DEBUG, log the raw event and mimic the original 'return' behavior
-    if logging.getLogger().isEnabledFor(logging.DEBUG):
-        _log_json(logging.debug, event, prefix="Raw Event (DEBUG) ← ")
-        # This return mimics the original behavior where verbose mode would
-        # print the raw event and skip further processing for this event.
-        return
-
-    # If the server uses an older "typed" envelope just keep legacy handling
-    if "type" in event:
-        etype = event["type"]
-        if etype == "status":
-            message = event.get("status", {}).get("message", {})
-            parts = message.get("parts", [])
-            if parts:
-                _log_parts(logging.info, parts)
-        elif etype == "artifact":
-            artifacts = event.get("artifacts") or []
-            for artifact in artifacts:
-                _log_parts(logging.info, artifact.get("parts", []))
-        elif etype == "end":
-            logging.info("\n[stream end]")
-        return
-
-    # ---- New A2A 0.2 style messages ---------------------------------------
-    # Extract "result" or "error" from the JSON-RPC envelope.
-    if "error" in event:
-        err = event["error"]
-        logging.error(f"RPC error: {err.get('message')}")
-        return
-
-    result = event.get("result")
-    if not result:
-        # Nothing useful to display
-        return
-
-    # Status update?  (TaskStatusUpdateEvent)
-    if "status" in result:
-        message = result["status"].get("message", {})
-        parts = message.get("parts", [])
-        if parts:
-            _log_parts(logging.info, parts)
-
-        # If final flag present we can acknowledge.
-        if result.get("final"):
-            logging.info("\n[completed]")
-        return
-
-    # Artifact update? (TaskArtifactUpdateEvent)
-    if "artifact" in result:
-        artifact = result["artifact"]
-        _log_parts(logging.info, artifact.get("parts", []))
-        return
-
-
-###############################################################################
-# CLI argument parsing
-###############################################################################
-
-
-def parse_args(argv: list[str] | None = None) -> argparse.Namespace:
-    parser = argparse.ArgumentParser(description="Interact with an A2A server")
-
-    parser.add_argument(
-        "query", nargs="*", help="Query to send. If omitted, just fetch agent card."
-    )
-
-    parser.add_argument(
-        "--host", default=DEFAULT_HOST, help="A2A host (default: %(default)s)"
-    )
-    parser.add_argument(
-        "--port", type=int, default=DEFAULT_PORT, help="A2A port (default: %(default)s)"
-    )
-
-    parser.add_argument(
-        "--stream",
-        action="store_true",
-        help="Use streaming subscribe request instead of blocking send",
-    )
-
-    parser.add_argument("--verbose", action="store_true", help="Verbose output (debug)")
-
-    return parser.parse_args(argv)
-
-
-###############################################################################
-# Main entry-point
-###############################################################################
-
-
-def main(argv: list[str] | None = None) -> None:
-    args = parse_args(argv)
-
-    # Configure logging
-    log_level = logging.DEBUG if args.verbose else logging.INFO
-    logging.basicConfig(format="%(message)s", level=log_level, stream=sys.stdout)
-    if args.verbose:
-        logging.debug("Verbose mode enabled.")
-
-    # 1. Always fetch agent-card first (helps verify server is live)
-    agent_info = get_agent_info(args.host, args.port)
-    if agent_info is None:
-        logging.critical("⚠️  Could not retrieve agent info – abort")
-        sys.exit(1)
-
-    if not args.query:
-        # Enter interactive REPL style mode
-        logging.info("Interactive mode – type 'exit' or Ctrl-D to quit.")
-        try:
-            while True:
-                try:
-                    user_input = input("> ").strip()
-                except EOFError:
-                    # Ctrl-D
-                    logging.info("")
-                    break
-
-                if user_input.lower() in {"exit", "quit"}:
-                    break
-
-                if not user_input:
-                    continue
-
-                ok = (
-                    send_streaming_query(
-                        user_input,
-                        a2a_host=args.host,
-                        a2a_port=args.port,
-                    )
-                    if args.stream
-                    else send_a2a_query(
-                        user_input,
-                        a2a_host=args.host,
-                        a2a_port=args.port,
-                    )
-                )
-
-                # Add separator between queries
-                if ok:
-                    logging.info("\n" + "—" * 30)
-        except KeyboardInterrupt:
-            # Ctrl-C to exit
-            logging.info("")
-        sys.exit(0)
-
-    # Single query passed via CLI
-    query_str = " ".join(args.query)
-
-    ok = (
-        send_streaming_query(query_str, a2a_host=args.host, a2a_port=args.port)
-        if args.stream
-        else send_a2a_query(query_str, a2a_host=args.host, a2a_port=args.port)
-    )
-
-    sys.exit(0 if ok else 2)
-
-
-if __name__ == "__main__":
-    main()

mindsdb/api/a2a/common/client/__init__.py

@@ -1,4 +0,0 @@
-from .client import A2AClient
-from .card_resolver import A2ACardResolver
-
-__all__ = ["A2AClient", "A2ACardResolver"]

mindsdb/api/a2a/common/client/card_resolver.py

@@ -1,21 +0,0 @@
-import httpx
-from common.types import (
-    AgentCard,
-    A2AClientJSONError,
-)
-import json
-
-
-class A2ACardResolver:
-    def __init__(self, base_url, agent_card_path="/.well-known/agent.json"):
-        self.base_url = base_url.rstrip("/")
-        self.agent_card_path = agent_card_path.lstrip("/")
-
-    def get_agent_card(self) -> AgentCard:
-        with httpx.Client() as client:
-            response = client.get(self.base_url + "/" + self.agent_card_path)
-            response.raise_for_status()
-            try:
-                return AgentCard(**response.json())
-            except json.JSONDecodeError as e:
-                raise A2AClientJSONError(str(e)) from e