kailash 0.6.1__py3-none-any.whl → 0.6.3__py3-none-any.whl

kailash/workflow/graph.py

@@ -346,18 +346,27 @@ class Workflow:
         # Validate cycle parameters and issue deprecation warning
         if cycle:
             # Issue deprecation warning for cycle usage via connect()
-            warnings.warn(
-                "Using workflow.connect() with cycle=True is deprecated and will be removed in v0.2.0. "
-                "Use the new CycleBuilder API instead:\n"
-                "  workflow.create_cycle('cycle_name')\\\n"
-                "    .connect(source_node, target_node)\\\n"
-                "    .max_iterations(N)\\\n"
-                "    .converge_when('condition')\\\n"
-                "    .build()\n"
-                "See Phase 5 API documentation for details.",
-                DeprecationWarning,
-                stacklevel=2,
-            )
+            # Skip warning if called from CycleBuilder (check stack)
+            import inspect
+
+            frame = inspect.currentframe()
+            caller_frame = frame.f_back if frame else None
+            caller_filename = caller_frame.f_code.co_filename if caller_frame else ""
+
+            # Only warn if NOT called from CycleBuilder
+            if "cycle_builder.py" not in caller_filename:
+                warnings.warn(
+                    "Using workflow.connect() with cycle=True is deprecated and will be removed in v0.2.0. "
+                    "Use the new CycleBuilder API instead:\n"
+                    "  workflow.create_cycle('cycle_name')\\\n"
+                    "    .connect(source_node, target_node)\\\n"
+                    "    .max_iterations(N)\\\n"
+                    "    .converge_when('condition')\\\n"
+                    "    .build()\n"
+                    "See Phase 5 API documentation for details.",
+                    DeprecationWarning,
+                    stacklevel=2,
+                )
 
             # Import enhanced exceptions for better error messaging
             try:
@@ -426,11 +435,20 @@ class Workflow:
             for c in self.connections
             if c.source_node == source_node and c.target_node == target_node
         ]
+        # Allow multiple connections between same nodes for different mappings
+        # Only reject if it's a duplicate mapping, not just any existing connection
         if existing_connections and not cycle:
-            raise ConnectionError(
-                f"Connection already exists between '{source_node}' and '{target_node}'. "
-                f"Existing mappings: {[c.model_dump() for c in existing_connections]}"
-            )
+            # Check if any of the new mappings already exist
+            existing_mappings = set()
+            for conn in existing_connections:
+                existing_mappings.add((conn.source_output, conn.target_input))
+
+            for source_output, target_input in mapping.items():
+                if (source_output, target_input) in existing_mappings:
+                    raise ConnectionError(
+                        f"Duplicate connection already exists: '{source_node}.{source_output}' -> '{target_node}.{target_input}'. "
+                        f"Existing mappings: {[c.model_dump() for c in existing_connections]}"
+                    )
 
         # Create connections (store in self.connections list)
         for source_output, target_input in mapping.items():
@@ -495,7 +513,58 @@ class Workflow:
                 }
             )
 
-        # Add or update the edge (NetworkX will update if edge exists)
+        # CRITICAL FIX: Merge edge data for multiple connections between same nodes
+        # Check if edge already exists and merge mappings
+        existing_edge_data = None
+        if self.graph.has_edge(source_node, target_node):
+            existing_edge_data = self.graph.get_edge_data(source_node, target_node)
+
+        if existing_edge_data and "mapping" in existing_edge_data:
+            # Merge with existing mapping
+            merged_mapping = existing_edge_data["mapping"].copy()
+            merged_mapping.update(mapping)
+            edge_data = {
+                "mapping": merged_mapping,  # Merged mapping dictionary
+            }
+
+            # Update backward compatibility fields
+            if len(merged_mapping) == 1:
+                edge_data["from_output"] = list(merged_mapping.keys())[0]
+                edge_data["to_input"] = list(merged_mapping.values())[0]
+            else:
+                edge_data["from_output"] = list(merged_mapping.keys())
+                edge_data["to_input"] = list(merged_mapping.values())
+
+            # Preserve any existing cycle metadata
+            if existing_edge_data.get("cycle"):
+                edge_data.update(
+                    {
+                        k: v
+                        for k, v in existing_edge_data.items()
+                        if k not in ["mapping", "from_output", "to_input"]
+                    }
+                )
+        else:
+            # No existing edge or no mapping, use new mapping as-is
+            # (edge_data was already set above)
+            pass
+
+        # Add cycle metadata to edge if this is a cycle connection
+        if cycle:
+            edge_data.update(
+                {
+                    "cycle": cycle,
+                    "max_iterations": max_iterations,
+                    "convergence_check": convergence_check,
+                    "cycle_id": cycle_id,
+                    "timeout": timeout,
+                    "memory_limit": memory_limit,
+                    "condition": condition,
+                    "parent_cycle": parent_cycle,
+                }
+            )
+
+        # Add or update the edge with merged data
         self.graph.add_edge(source_node, target_node, **edge_data)
 
         # Enhanced logging for cycles

kailash/workflow/templates.py

@@ -176,15 +176,10 @@ class CycleTemplates:
         # Connect processor to evaluator
         workflow.connect(processor_node, evaluator_node)
 
-        # Close the cycle with convergence condition
-        workflow.connect(
-            evaluator_node,
-            processor_node,
-            cycle=True,
-            max_iterations=max_iterations,
-            convergence_check=convergence,
-            cycle_id=cycle_id,
-        )
+        # Close the cycle with convergence condition using new API
+        workflow.create_cycle(cycle_id).connect(
+            evaluator_node, processor_node
+        ).max_iterations(max_iterations).converge_when(convergence).build()
 
         return cycle_id
 

{kailash-0.6.1.dist-info → kailash-0.6.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kailash
-Version: 0.6.1
+Version: 0.6.3
 Summary: Python SDK for the Kailash container-node architecture
 Home-page: https://github.com/integrum/kailash-python-sdk
 Author: Integrum
@@ -21,7 +21,7 @@ Requires-Dist: matplotlib>=3.5
 Requires-Dist: pyyaml>=6.0
 Requires-Dist: click>=8.0
 Requires-Dist: pytest>=8.3.5
-Requires-Dist: mcp[cli]>=1.9.2
+Requires-Dist: mcp[cli,server]>=1.9.2
 Requires-Dist: pandas>=2.2.3
 Requires-Dist: numpy>=2.2.5
 Requires-Dist: scipy>=1.15.3
@@ -64,6 +64,7 @@ Requires-Dist: twilio>=9.6.3
 Requires-Dist: qrcode>=8.2
 Requires-Dist: aiofiles>=24.1.0
 Requires-Dist: bcrypt>=4.3.0
+Requires-Dist: plotly>=6.2.0
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0; extra == "dev"
 Requires-Dist: pytest-cov>=3.0; extra == "dev"

{kailash-0.6.1.dist-info → kailash-0.6.3.dist-info}/RECORD

@@ -1,4 +1,4 @@
-kailash/__init__.py,sha256=3ymX4NjOkCaHdw8nCAxgzBLwNu5hIc9i8zWgOMBRmEg,1724
+kailash/__init__.py,sha256=8W7LrVttYlpEMHu8_N4a5CilyUWMORrc8rnCbaAJ8HU,1724
 kailash/__main__.py,sha256=vr7TVE5o16V6LsTmRFKG6RDKUXHpIWYdZ6Dok2HkHnI,198
 kailash/access_control.py,sha256=2ctdRFeSeu-d7DU04Aovxh6Rt_4t3IyQfkKEjTeQiMM,25519
 kailash/access_control_abac.py,sha256=FPfa_8PuDP3AxTjdWfiH3ntwWO8NodA0py9W8SE5dno,30263
@@ -23,7 +23,7 @@ kailash/config/__init__.py,sha256=9qNwtvXAVV-KkHbuL1ZbtC6yXDtowH4YoFiOP-Qbe-w,63
 kailash/config/database_config.py,sha256=rdlqIP9WUzC0kvAdSjno1LMpu_bEy2v5FgFdgJy-qsc,11588
 kailash/core/actors/__init__.py,sha256=o8CrwZRTPW5uB3UZiiFtz34n1Q-XBoKNwTz5NMTDQgA,433
 kailash/core/actors/adaptive_pool_controller.py,sha256=tque9heLsLwjrNlM1wZSAYi1RInv19Z3HTjbozc9XtY,22484
-kailash/core/actors/connection_actor.py,sha256=TFqlYPEnV8Hjmc706tdRzEw9pBJs2OieH8UeaOrWYIo,18949
+kailash/core/actors/connection_actor.py,sha256=M8fOX1a3jvH5PUkfQyk0eBJqCk0SD9KGZCw0TXLON_o,18979
 kailash/core/actors/supervisor.py,sha256=7_YXm9o4O_xBs5YxsDFt7RjzaMvTbKcK-jd4KPmSt1M,11968
 kailash/core/ml/__init__.py,sha256=eaD-bmoxMXtwwtKWePsoX1IkcpysX0bMAGyMR7jaAqI,64
 kailash/core/ml/query_patterns.py,sha256=4wV1pBOwUiK80UTKuMXSoRgR0pojtjg2BMNoa5qWpc0,20290
@@ -39,25 +39,24 @@ kailash/edge/compliance.py,sha256=tUiKiib7FjNtZ62ThQIRhxkehOddndTGLu-Y4He7iRQ,31
 kailash/edge/discovery.py,sha256=tEV7W2XVPfEgEOM5oaRAlWAtuXV3M4xM07Z7OcnXZxY,24311
 kailash/edge/location.py,sha256=ZrG4CqN-taZFo9VDLvZ2z1lNiTHgXcZvzBJS9meXY8k,19717
 kailash/gateway/__init__.py,sha256=zh2uCsDHi82pZ_tGJklsH_FS-Mx9V31r0hkxk48CM-g,927
-kailash/gateway/api.py,sha256=rvbqV85Av1w7b0FsOg9pulcguu1Oq0D37t029N-tqsA,9478
-kailash/gateway/enhanced_gateway.py,sha256=WYs56k31VLRbfryeRwuiahDAL59aHk99eGrmGtvuD0A,13389
+kailash/gateway/api.py,sha256=xpK8PIamsqQPpKAJwacyV7RA_Snjv2pc_0ljnnU9Oy4,9534
+kailash/gateway/enhanced_gateway.py,sha256=IlN1XV01FQrF4rGcq_z9LE4uUHAAAQoVsRNToXENen0,13399
 kailash/gateway/resource_resolver.py,sha256=IC1dceiKfjfUWToYCIBcrUapuR3LlDG6RJ4o7haLY10,7746
 kailash/gateway/security.py,sha256=kf4Quf6u7dqhs80fQQ982eHbRb4weDKG0DaYNeKntT4,7557
-kailash/mcp/__init__.py,sha256=jQHP7EVT126QXmi0TqR1mU3QNrUeEB4oIC4sD4B2a8c,1813
-kailash/mcp/ai_registry_server.py,sha256=9pOzJnJFpxJnZPfLo2QvVZ5yvAq5IRqzXPbQL1rL1ns,28104
-kailash/mcp/client.py,sha256=sTouSiuiu1nbahMXSWcP8-mr1k7cqdBCzSxm8G7le-s,16058
-kailash/mcp/client_new.py,sha256=YU671JvAM0uvuX0uhGZCIKI8co3fqz0cs6HqLZ59Xyo,10285
-kailash/mcp/server.py,sha256=aWU0DHj89FN_eEgH6aVjA05WjthugrXMHdPlgwJ6kZg,8246
-kailash/mcp/server_enhanced.py,sha256=lRIDSNs0c8urMq_SURwi7eBhTWKa-rq2FAB8xZf9CwI,14724
-kailash/mcp/servers/ai_registry.py,sha256=7k17ld0DUQYL476N5EbiNpPy8D0HqPRhrl8Wk1ajjj8,9992
-kailash/mcp/utils/__init__.py,sha256=R20N-iiKXUPxc9MOh6vPO1vIfkPmwhEQ5KNFgGd4xSs,771
-kailash/mcp/utils/cache.py,sha256=dLEseovPaXL4lRzMSw7tqd3tJHwnWfhdZ-HKGyScJXI,8414
-kailash/mcp/utils/config.py,sha256=DyZxgdy3vqI5pwhQ_E-42mhueVGNHiuOtTUOrM9HC_U,8124
-kailash/mcp/utils/formatters.py,sha256=D-2j1nvmprApiUI13HWY-L2_WPSAcJDtVdHcshAuOdo,9740
-kailash/mcp/utils/metrics.py,sha256=MNUjWGQyq1EGdeqzAKCCZJNgcWHOyaYAV8MlS2cb-4k,13754
+kailash/mcp_server/__init__.py,sha256=ETEeaRTqj2xeX07MyaRgcPiaomdLoJ24iNmRbRjJQok,1953
+kailash/mcp_server/ai_registry_server.py,sha256=BkkZcohRyGlSbiUkpRTaIA6M_yJUgtgtMda-dvbT86s,28118
+kailash/mcp_server/client.py,sha256=sTouSiuiu1nbahMXSWcP8-mr1k7cqdBCzSxm8G7le-s,16058
+kailash/mcp_server/client_new.py,sha256=YU671JvAM0uvuX0uhGZCIKI8co3fqz0cs6HqLZ59Xyo,10285
+kailash/mcp_server/server.py,sha256=z-72J5MKq98_j7zwo0HJMbMI9bp5yuNpyfIzD9N23yw,20600
+kailash/mcp_server/servers/ai_registry.py,sha256=IdF_keUuJlMsvjLjSAykxxbm46K4qA7eCj7T-lYSrzk,10007
+kailash/mcp_server/utils/__init__.py,sha256=VjGFpvugTYUxoNgtwVEKkmrkVS3iZ_kIbP4jR1PRM6E,750
+kailash/mcp_server/utils/cache.py,sha256=dLEseovPaXL4lRzMSw7tqd3tJHwnWfhdZ-HKGyScJXI,8414
+kailash/mcp_server/utils/config.py,sha256=DyZxgdy3vqI5pwhQ_E-42mhueVGNHiuOtTUOrM9HC_U,8124
+kailash/mcp_server/utils/formatters.py,sha256=D-2j1nvmprApiUI13HWY-L2_WPSAcJDtVdHcshAuOdo,9740
+kailash/mcp_server/utils/metrics.py,sha256=MNUjWGQyq1EGdeqzAKCCZJNgcWHOyaYAV8MlS2cb-4k,13754
 kailash/middleware/__init__.py,sha256=ZGo0qujL-qWn82nIrojY96N1rMPTWFKHumW6CGGpb4Y,10409
 kailash/middleware/auth/__init__.py,sha256=VkKM8H-zVFx2PLGL7kyxE2IfSiV1HiwveSysbmxMcg0,2077
-kailash/middleware/auth/access_control.py,sha256=6Z09mPyeTb1VBi2C4wRlyqNaJN0lR-me6GMwxDaoR5c,14954
+kailash/middleware/auth/access_control.py,sha256=2FwQjp_fZm2fg-V_CIgAN69GGL9YwyZwsbQis0hbW7I,14848
 kailash/middleware/auth/auth_manager.py,sha256=d1XFJ9jOCrOTwV26qO0b7wBOSbroTvTxaJADII-mCz0,14057
 kailash/middleware/auth/exceptions.py,sha256=tPDQgaX9nMQ9BJZR3Y5tv2LwLy8pZcUz-uFATQjALRA,2496
 kailash/middleware/auth/jwt_auth.py,sha256=r4dauFKcoARHj0yb4f4WwBmY9YQpLNBGyk4gdBZEI1k,21626
@@ -81,14 +80,14 @@ kailash/middleware/database/models.py,sha256=CJwwUEdgxqBteXqpFJr1tWskjypJxViZXjO
 kailash/middleware/database/repositories.py,sha256=7l2OcAem5sMQxbbnmBR4O22P-OcLv5zEo-4r9fIMl1o,23340
 kailash/middleware/database/session_manager.py,sha256=Pzj7c2TZnM3GRty2igSaxmLOf0-Fs67NVe2Q5lR_C-Q,379
 kailash/middleware/gateway/__init__.py,sha256=3R7HkvAadJKYCkDHEXFnOPHJ6yzdj8HpBOHtafStqSE,588
-kailash/middleware/gateway/checkpoint_manager.py,sha256=kSX90UeSI4gN4aaLa__BOMxIUhNpl0PJ3wGxa7KUCdQ,13606
+kailash/middleware/gateway/checkpoint_manager.py,sha256=zF2ZnHYxQwXmG3d5g7Qwx9pW8tDRTcqV9yh8dvhY95E,15056
 kailash/middleware/gateway/deduplicator.py,sha256=CblV3fwc7s4wg6KIvypwyNMYL5AuQi9EwtcxVOy64X8,13265
 kailash/middleware/gateway/durable_gateway.py,sha256=EsIgMNxS_no2W40AXDyE7FmVdnGNU26kBRC5rfnSzoQ,14626
 kailash/middleware/gateway/durable_request.py,sha256=SCnp-bF0tQX9oahr9reqcZjJ_YhyJkeYYl-un9rJ6lo,15437
 kailash/middleware/gateway/event_store.py,sha256=A3Kh2MhVVPbXWvjeo550SqEGPiJYyspAfu6Gv7UZzo4,16131
 kailash/middleware/mcp/__init__.py,sha256=EdZB8zOMSBEEmudRzs8ksz9QZJYWQMEx7Tm1MOwIWnI,922
-kailash/middleware/mcp/client_integration.py,sha256=opzhB5TUts_ND8gARXh93nKCc1u4kwo6SqNMMWqMcSU,18258
-kailash/middleware/mcp/enhanced_server.py,sha256=XUjQt0KyRX207FVYAFenkYHa_K8FWWKquROgXQWkoOQ,18453
+kailash/middleware/mcp/client_integration.py,sha256=dY1RmX-g5E6JzUFuWxk7viuOYIh8bMwoUSvHQMVEsYk,18265
+kailash/middleware/mcp/enhanced_server.py,sha256=vVd7WE95c-5861VtKfmTZ91iUWFRw-aKEKwHlu6edQo,18467
 kailash/nodes/__init__.py,sha256=E6CEp1ooq4GgFhKtwVAczOhPt5N3x-AVQ-R0n3_IFyA,936
 kailash/nodes/base.py,sha256=Fu9c2I5k_Ke192y4fj2NVhf-Y_I0nPr0sDE1zMSRCJY,55417
 kailash/nodes/base_async.py,sha256=mpntaeFMbUYLIyTvjsb221mXckx_H2dGX2LhxeKhhfA,6569
@@ -98,19 +97,22 @@ kailash/nodes/mixins.py,sha256=ncAdNQPe1sphLByeerP6G_s8mjFLt7dM4baiozzIBPA,12083
 kailash/nodes/validation.py,sha256=tuBZRZkDiEdvfeU7JaRB7v2-j1vxPYMJ1gVaJ-PKHRk,12117
 kailash/nodes/admin/__init__.py,sha256=C9_pK2w0rH6JEV_-roypeasAIyIhEFKKnH-npGBeew0,1508
 kailash/nodes/admin/audit_log.py,sha256=CSSeWwAK-_hGYXs8eess89wNVju8jvsdNKA1A1946to,41066
-kailash/nodes/admin/permission_check.py,sha256=0SY1PojT3kymhE8hjkXlY6t0OuESDGjJl3IkB8lnUlA,63109
+kailash/nodes/admin/permission_check.py,sha256=tLz6aGmNp1xV44Wfy_F1RgvGU4te9qzhNOAEpGo9fdU,66564
 kailash/nodes/admin/role_management.py,sha256=DVkFJU5yQRf_1TxlvmQBhHzCeE5oJQzIjCKjrEfZb1g,72894
+kailash/nodes/admin/schema.sql,sha256=50V-0NoQmYg66KYVE6rAJxx7LSacCVYs2UvvA1Y9zF0,14028
 kailash/nodes/admin/schema_manager.py,sha256=U8W8Xn4H5wamEtSq-I9jh54tWQnrLo4vZDKdX3K4bKs,15073
 kailash/nodes/admin/security_event.py,sha256=We_nGn6_J004b7LQVd0nKU6LmhJ6vyrl5zySC3xVdfo,56336
-kailash/nodes/admin/user_management.py,sha256=yoOee8auya9CMiXfDMPH_9A83OdaoJ0B_Sc72sto88w,53702
+kailash/nodes/admin/tenant_isolation.py,sha256=jztFmChohj4ZpwAUdM_z2tKYcFopFtCJ2ViXnlMuvzA,8270
+kailash/nodes/admin/transaction_utils.py,sha256=La3BLwT2XrYtKesiFRGEYL0dXop6JM5TqPUHLx5DYVs,7884
+kailash/nodes/admin/user_management.py,sha256=6u0iMyFKvb_zyaQdiKuGemQM65gEeax9wwEjoadb4Ro,54623
 kailash/nodes/ai/__init__.py,sha256=rslxIS8jlovshiNgWqVzQjD_kfT_3h3Ct03sk-iRe6U,2202
 kailash/nodes/ai/a2a.py,sha256=QpTPl-Cm6mqzM0unLfqPrgEu964QP47g5p2T4clVPMs,70004
 kailash/nodes/ai/agents.py,sha256=CRA3cdapQjpuvOniXUh6ZVWAlRxUIepVw1BROW6QzdY,20373
-kailash/nodes/ai/ai_providers.py,sha256=XK7AzC8uL8nIxukZs6bxX4g75vcxCbehefgvR4QJgXk,62824
+kailash/nodes/ai/ai_providers.py,sha256=XeDIaYH7PrX8frWCVV6CYFNQXFjHX_61T3eR1yullqs,65355
 kailash/nodes/ai/embedding_generator.py,sha256=rsos3B6oWrgGTMIbwSWIBzGH9kq3SFVD_-bEDrujBRs,31860
 kailash/nodes/ai/intelligent_agent_orchestrator.py,sha256=xw44C-CkcNH3SVmEJ49o4oNV3o4ZqjLE9aLpggwoIXs,83021
-kailash/nodes/ai/iterative_llm_agent.py,sha256=pv54W_YDfDPDl6DJf0ul9_rs2mL2kE_C59sSAJ4CRn8,52884
-kailash/nodes/ai/llm_agent.py,sha256=Beb07UNzCvPjHLTgYG9HeGsmuAXGLPjcExY30DghBfA,77566
+kailash/nodes/ai/iterative_llm_agent.py,sha256=14SxTRGGGYRqNC0SFMaKAFplUcsQXAVzXKcjri_0oWM,52891
+kailash/nodes/ai/llm_agent.py,sha256=0N5NU2HLwbh2_zFxrgwq2FlkVS6fifpEcMSI0BGX7ZQ,82198
 kailash/nodes/ai/models.py,sha256=wsEeUTuegy87mnLtKgSTg7ggCXvC1n3MsL-iZ4qujHs,16393
 kailash/nodes/ai/self_organizing.py,sha256=M7yCLkN4I1JCNU7PWuwrqwQSlaG9MJVxYIR44TV52MM,62877
 kailash/nodes/ai/vision_utils.py,sha256=OHD9cVH_mq0WpJyQkNTj_mpipIVWfSV_bF9eA6CdyeA,4166
@@ -151,7 +153,7 @@ kailash/nodes/data/readers.py,sha256=fX82yQSLGPUFbSf6krT_-9gybp6IBnBH-Lcr87aNlHQ
 kailash/nodes/data/retrieval.py,sha256=H8Qb2R64fQ_OeaQBh6jVn7-xLuygiF3fjzbh4ssNOrU,20591
 kailash/nodes/data/sharepoint_graph.py,sha256=J1_KZQ5s8yXkjGY1PjsMY7T8viKARi3os5P8CXpmn2U,37160
 kailash/nodes/data/sources.py,sha256=tvgDDDerWELD6QkAm73ciKmNakD_SYMd50idlOjpbF0,3632
-kailash/nodes/data/sql.py,sha256=3On6JatlXvGrp2D4MdObX_l_H3CIEoHxqoPUoSfXYso,34426
+kailash/nodes/data/sql.py,sha256=0VfOedQ6UwSiBJtwmMLhAvqQWeakrnB1QNrhjan5jIg,35288
 kailash/nodes/data/streaming.py,sha256=VE4e3CNuRAdjckV8UXFpkTj3mPLY2cicoZwnkq10ZnE,37163
 kailash/nodes/data/vector_db.py,sha256=pwCl-3tyk_Cv_VQ8GafgodJ1yM88W1BXCIcYC56XoqU,32056
 kailash/nodes/data/workflow_connection_pool.py,sha256=o-c-gu4HRtD0i7G6eCLZuOVB4CDJMOXFuL5CbXipZtk,40284
@@ -207,10 +209,10 @@ kailash/resources/__init__.py,sha256=JM90kDB7k9wTsd6-2jwm-312SeOaXvavyKnPwQwy4PY
 kailash/resources/factory.py,sha256=CYqBm3EFjTv_Aoqb3AUFT8emJlUrtwAZGCSVLt8K3KA,15824
 kailash/resources/health.py,sha256=i6XS0HdL6pUYq8SBdRjvchf-oj0sy3JoRLszNylfQJc,9702
 kailash/resources/reference.py,sha256=RKfXzJFIdid0cLOwsXSmlXq4NhSEci4GO-q3M3qcjA8,7526
-kailash/resources/registry.py,sha256=4qXk4rdsPeaUgwizLakq1TG9cLCNuKQGd313vCIstnw,13504
+kailash/resources/registry.py,sha256=5978e9VcUq0XBi9LRN89swptBSOAOAyNmwe2pxFHMxM,13789
 kailash/runtime/__init__.py,sha256=CvU-qBMESYYISqFOlYlLsYJrXJu0Gqr4x6yr4Ob_Rng,278
 kailash/runtime/access_controlled.py,sha256=Td8Fa8oPxEqlb74bDiH_HtT6J-ku0AudvN7mrUZOacc,17219
-kailash/runtime/async_local.py,sha256=Ebk3-pDij804YCQST7izXzqC0QWAS5--X_pnNqiwNxQ,29900
+kailash/runtime/async_local.py,sha256=XJyJYWmsezZBmAB8Y8xfAbiytPaY1j_zYU_moABP-3E,30313
 kailash/runtime/docker.py,sha256=sZknVl1PCGfAZeyc0-exTuKlllSyjYlFIgJoiB3CRNs,23500
 kailash/runtime/local.py,sha256=i2hnRrk2Ks0Q5jR9SkBhM6d6plJbpH0i7rNkJ8laLCw,39078
 kailash/runtime/parallel.py,sha256=mz_wPD13-YVc3Q_8HkOs4nPQPdTjnjCcnRL7ZRM70lo,21070
@@ -231,7 +233,7 @@ kailash/tracking/storage/base.py,sha256=wWkK1XdrMV0EGxlbFDyfuVnDoIG0tdSPPwz_8iwz
 kailash/tracking/storage/database.py,sha256=O3-qYmgwTccq9jYl25C0L6R398pXPsWkIAoWLL1aZvQ,20048
 kailash/tracking/storage/filesystem.py,sha256=VhWxNvqf_Ta3mIaGqKuOrcCqQiEvJj7S8NK5yRd1V68,19627
 kailash/utils/__init__.py,sha256=pFKhHJxU_kyFE9aGT5recw5E-3nbfVF5pMHepBJWB2E,253
-kailash/utils/export.py,sha256=HViwFBtg20VnGBC9gQjHcnu44wQ1u2vOZMXR8EtfGvw,31913
+kailash/utils/export.py,sha256=WBazN03LOCI03TsIElNv31wSZ_uTLPl8THnqdohgyTk,37361
 kailash/utils/resource_manager.py,sha256=xMgcSogLofUIRpRMXzmK4gxBi2lIqEJ7rn2rsBBm9xo,13209
 kailash/utils/secure_logging.py,sha256=3iEJt7sXnIgvdiRz8qsx0JbcygrsUf0_C8UHFXLNfcs,11325
 kailash/utils/templates.py,sha256=1O9mniUTJmZZFkC1gFaWy73pDIZxcawljDIQpPj6G9Q,21977
@@ -247,7 +249,7 @@ kailash/visualization/reports.py,sha256=D7kJ0flHr16d-qSEq8vnw20N8u_dgTrXtKVSXVm8
 kailash/workflow/__init__.py,sha256=DDQDE9K6RmbX6479guNLLgjiVVV-gQERRvCEJWSVlsM,1836
 kailash/workflow/async_builder.py,sha256=iv8bDJHdWAUZ77SyMo6sucd92dTdtXesdxycrSE7mM4,20613
 kailash/workflow/async_patterns.py,sha256=X0ZDXwr6UAu0WC1xnCB7-0V1-tRbKs9UI4JqaBCB6tE,22824
-kailash/workflow/builder.py,sha256=CPnfNHlL4HKsX58BI4kOv-CLNrFa_7cg_0lBqqVByYw,10779
+kailash/workflow/builder.py,sha256=FTzvso-FWVAcPYs2DOvL3CSEvbxaVbMP3L2T8_wtzbM,12356
 kailash/workflow/convergence.py,sha256=vfIDR-uNaQE-LVUEzrRtfgKPgX9gL0nLNH-nTg5ra-c,10031
 kailash/workflow/cycle_analyzer.py,sha256=BGBpgdB-g0-KRI65sVAvHV4lxfoCzMt4uKOHbw8GXT4,32596
 kailash/workflow/cycle_builder.py,sha256=uWAx8K4ZKMtFC3mWylK4gHT03xeu0xChJlhw50hVqEE,20883
@@ -257,7 +259,7 @@ kailash/workflow/cycle_exceptions.py,sha256=4_OLnbEXqIiXKzOc3uh8DzFik4wEHwl8bRZh
 kailash/workflow/cycle_profiler.py,sha256=aEWSCm0Xy15SjgLTpPooVJMzpFhtJWt4livR-3Me4N8,28547
 kailash/workflow/cycle_state.py,sha256=hzRUvciRreWfS56Cf7ZLQPit_mlPTQDoNTawh8yi-2s,10747
 kailash/workflow/cyclic_runner.py,sha256=NZqK_GTNbvf-pp08_brvzOlDCytrKscr1XRDy8PD8Is,40816
-kailash/workflow/graph.py,sha256=gknGNgv3Z6FyNno90jZLGHZ0tTdiUx0_HAy6bEWh6OQ,56156
+kailash/workflow/graph.py,sha256=zRpGLXeuwtuxFBvE7_16c_bB9yqZirM_uwtfD1_MY4g,59272
 kailash/workflow/mermaid_visualizer.py,sha256=UsQDvxgIAhy-Th7ZzFnbuziaTx1Cd5yUh6S_25DffoQ,22294
 kailash/workflow/migration.py,sha256=zsmXgbUtOZdjoOx9YoEY0-x7uOY1T-_yQo4MnZjokCQ,31487
 kailash/workflow/mock_registry.py,sha256=J4gyH8YdhRyhvORDVxGTya0FgDK8iAA8Nonur6p9s-o,1692
@@ -265,12 +267,12 @@ kailash/workflow/resilience.py,sha256=Ecef4gBg-QWP369a_xfzQnVWhHryvEcO2RSFVSriLJ
 kailash/workflow/runner.py,sha256=l6jb-H7DwbRlvQ3H3SuTs70rut-u7H3Gi8nybKCEjZU,10795
 kailash/workflow/safety.py,sha256=pS5GKu7UdkzFZcb16Dn-0jBxjULDU-59_M0CbUVMVyw,11298
 kailash/workflow/state.py,sha256=UTZxs5-Ona6uvBhx1__i6-RX8gB4qazkBIWE7uyRmWQ,7600
-kailash/workflow/templates.py,sha256=MTIMHGyQML6omLbqeKDbTVIVykfXG6pIFWTTsGBUHN0,48591
+kailash/workflow/templates.py,sha256=98EN5H4fO9b4xeczk20Hu5L8hNcAuRQNGayT6vnZYCw,48540
 kailash/workflow/validation.py,sha256=JIbIajWVIaWHSvWtgZ4WUVJaBaUOCz5B9cyTwM--dL4,33060
 kailash/workflow/visualization.py,sha256=ICMWCWqh5fOQ7eJygbvu2PMWHxe-H5_0epwdZuz8cMw,19737
-kailash-0.6.1.dist-info/licenses/LICENSE,sha256=Axe6g7bTrJkToK9h9j2SpRUKKNaDZDCo2lQ2zPxCE6s,1065
-kailash-0.6.1.dist-info/METADATA,sha256=uYAx8RZ-rKdgoXsNC0jOUmdDIJ8or0CTEm7eWW6z2BA,25896
-kailash-0.6.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-kailash-0.6.1.dist-info/entry_points.txt,sha256=M_q3b8PG5W4XbhSgESzIJjh3_4OBKtZFYFsOdkr2vO4,45
-kailash-0.6.1.dist-info/top_level.txt,sha256=z7GzH2mxl66498pVf5HKwo5wwfPtt9Aq95uZUpH6JV0,8
-kailash-0.6.1.dist-info/RECORD,,
+kailash-0.6.3.dist-info/licenses/LICENSE,sha256=Axe6g7bTrJkToK9h9j2SpRUKKNaDZDCo2lQ2zPxCE6s,1065
+kailash-0.6.3.dist-info/METADATA,sha256=PH0D-ORbBHT7eJjuIEMbrcr1CWNlQpY36Ncy4jGJ-wg,25932
+kailash-0.6.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+kailash-0.6.3.dist-info/entry_points.txt,sha256=M_q3b8PG5W4XbhSgESzIJjh3_4OBKtZFYFsOdkr2vO4,45
+kailash-0.6.3.dist-info/top_level.txt,sha256=z7GzH2mxl66498pVf5HKwo5wwfPtt9Aq95uZUpH6JV0,8
+kailash-0.6.3.dist-info/RECORD,,

kailash/mcp/server.py

@@ -1,292 +0,0 @@
-"""MCP Server Framework using official Anthropic SDK.
-
-This module provides a comprehensive framework for creating MCP servers using
-the official FastMCP framework from Anthropic. Servers run as long-lived
-services that expose tools, resources, and prompts to MCP clients, enabling
-dynamic capability extension for AI workflows.
-
-Note:
-    This module requires the FastMCP framework to be installed.
-    Install with: pip install 'mcp[server]'
-
-Examples:
-    Basic server with tools:
-
-    >>> from kailash.mcp.server import MCPServer
-    >>> class MyServer(MCPServer):
-    ...     def setup(self):
-    ...         @self.add_tool()
-    ...         def calculate(a: int, b: int) -> int:
-    ...             return a + b
-    >>> server = MyServer("calculator", port=8080)
-    >>> server.start()
-
-    Quick server creation:
-
-    >>> from kailash.mcp.server import SimpleMCPServer
-    >>> server = SimpleMCPServer("my-tools")
-    >>> @server.tool()
-    ... def search(query: str) -> list:
-    ...     return [f"Result for {query}"]
-    >>> server.start()
-"""
-
-import logging
-from abc import ABC, abstractmethod
-from collections.abc import Callable
-
-logger = logging.getLogger(__name__)
-
-
-class MCPServer(ABC):
-    """Base class for MCP servers using FastMCP.
-
-    This provides a framework for creating MCP servers that expose
-    tools, resources, and prompts via the Model Context Protocol.
-
-    Examples:
-        Creating a custom server:
-
-        >>> class MyServer(MCPServer):
-        ...     def setup(self):
-        ...         @self.add_tool()
-        ...         def search(query: str) -> str:
-        ...             return f"Results for: {query}"
-        ...         @self.add_resource("data://example")
-        ...         def get_example():
-        ...             return "Example data"
-        >>> server = MyServer("my-server", port=8080)
-        >>> server.start()  # Runs until stopped
-    """
-
-    def __init__(self, name: str, port: int = 8080, host: str = "localhost"):
-        """Initialize the MCP server.
-
-        Args:
-            name: Name of the server.
-            port: Port to listen on (default: 8080).
-            host: Host to bind to (default: "localhost").
-        """
-        self.name = name
-        self.port = port
-        self.host = host
-        self._mcp = None
-        self._running = False
-
-    @abstractmethod
-    def setup(self):
-        """Setup server tools, resources, and prompts.
-
-        This method should be implemented by subclasses to define
-        the server's capabilities using decorators.
-
-        Note:
-            Use @self.add_tool(), @self.add_resource(uri), and
-            @self.add_prompt(name) decorators to register capabilities.
-        """
-
-    def add_tool(self):
-        """Decorator to add a tool to the server.
-
-        Returns:
-            Function decorator for registering tools.
-
-        Examples:
-            >>> @server.add_tool()
-            ... def calculate(a: int, b: int) -> int:
-            ...     '''Add two numbers'''
-            ...     return a + b
-        """
-
-        def decorator(func: Callable):
-            if self._mcp is None:
-                self._init_mcp()
-
-            # Use FastMCP's tool decorator
-            return self._mcp.tool()(func)
-
-        return decorator
-
-    def add_resource(self, uri: str):
-        """Decorator to add a resource to the server.
-
-        Args:
-            uri: URI pattern for the resource (supports wildcards).
-
-        Returns:
-            Function decorator for registering resources.
-
-        Examples:
-            >>> @server.add_resource("file:///data/*")
-            ... def get_file(path: str) -> str:
-            ...     return f"Content of {path}"
-        """
-
-        def decorator(func: Callable):
-            if self._mcp is None:
-                self._init_mcp()
-
-            # Use FastMCP's resource decorator
-            return self._mcp.resource(uri)(func)
-
-        return decorator
-
-    def add_prompt(self, name: str):
-        """Decorator to add a prompt template to the server.
-
-        Args:
-            name: Name of the prompt.
-
-        Returns:
-            Function decorator for registering prompts.
-
-        Examples:
-            >>> @server.add_prompt("analyze")
-            ... def analyze_prompt(data: str) -> str:
-            ...     return f"Please analyze the following data: {data}"
-        """
-
-        def decorator(func: Callable):
-            if self._mcp is None:
-                self._init_mcp()
-
-            # Use FastMCP's prompt decorator
-            return self._mcp.prompt(name)(func)
-
-        return decorator
-
-    def _init_mcp(self):
-        """Initialize the FastMCP instance."""
-        try:
-            from mcp.server.fastmcp import FastMCP
-
-            self._mcp = FastMCP(self.name)
-        except ImportError:
-            logger.error(
-                "FastMCP not available. Install with: pip install 'mcp[server]'"
-            )
-            raise
-
-    def start(self):
-        """Start the MCP server.
-
-        This runs the server as a long-lived process until stopped.
-
-        Raises:
-            ImportError: If FastMCP is not available.
-            Exception: If server fails to start.
-        """
-        if self._mcp is None:
-            self._init_mcp()
-
-        # Run setup to register tools/resources
-        self.setup()
-
-        logger.info(f"Starting MCP server '{self.name}' on {self.host}:{self.port}")
-        self._running = True
-
-        try:
-            # Run the FastMCP server
-            logger.info("Running FastMCP server in stdio mode")
-            self._mcp.run()
-        except Exception as e:
-            logger.error(f"Failed to start server: {e}")
-            raise
-        finally:
-            self._running = False
-
-    def stop(self):
-        """Stop the MCP server."""
-        logger.info(f"Stopping MCP server '{self.name}'")
-        self._running = False
-        # In a real implementation, we'd need to handle graceful shutdown
-
-
-class SimpleMCPServer(MCPServer):
-    """Simple MCP server for basic use cases.
-
-    This provides an easy way to create MCP servers without subclassing.
-
-    Examples:
-        >>> server = SimpleMCPServer("my-server")
-        >>> @server.tool()
-        ... def add(a: int, b: int) -> int:
-        ...     return a + b
-        >>> server.start()
-    """
-
-    def __init__(self, name: str, port: int = 8080, host: str = "localhost"):
-        """Initialize the simple MCP server.
-
-        Args:
-            name: Name of the server.
-            port: Port to listen on (default: 8080).
-            host: Host to bind to (default: "localhost").
-        """
-        super().__init__(name, port, host)
-        self._tools = []
-        self._resources = []
-        self._prompts = []
-
-    def tool(self):
-        """Decorator to add a tool.
-
-        Returns:
-            Function decorator for registering tools.
-        """
-
-        def decorator(func):
-            self._tools.append(func)
-            return func
-
-        return decorator
-
-    def resource(self, uri: str):
-        """Decorator to add a resource.
-
-        Args:
-            uri: URI pattern for the resource.
-
-        Returns:
-            Function decorator for registering resources.
-        """
-
-        def decorator(func):
-            self._resources.append((uri, func))
-            return func
-
-        return decorator
-
-    def prompt(self, name: str):
-        """Decorator to add a prompt.
-
-        Args:
-            name: Name of the prompt.
-
-        Returns:
-            Function decorator for registering prompts.
-        """
-
-        def decorator(func):
-            self._prompts.append((name, func))
-            return func
-
-        return decorator
-
-    def setup(self):
-        """Setup the server with registered components.
-
-        Registers all tools, resources, and prompts that were decorated
-        before calling start().
-        """
-        # Register all tools
-        for tool_func in self._tools:
-            self.add_tool()(tool_func)
-
-        # Register all resources
-        for uri, resource_func in self._resources:
-            self.add_resource(uri)(resource_func)
-
-        # Register all prompts
-        for name, prompt_func in self._prompts:
-            self.add_prompt(name)(prompt_func)