cap-sdk-python 0.1.0__tar.gz

cap_sdk_python-0.1.0/PKG-INFO

@@ -0,0 +1,59 @@
+Metadata-Version: 2.4
+Name: cap-sdk-python
+Version: 0.1.0
+Summary: CAP (Cortex Agent Protocol) Python SDK
+Project-URL: Homepage, https://github.com/coretexos/cap
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: protobuf>=4.25.0
+Requires-Dist: grpcio>=1.59.0
+Requires-Dist: nats-py>=2.6.0
+
+# CAP Python SDK
+
+Asyncio-first SDK with NATS helpers for CAP workers and clients.
+
+## Quick Start
+1. Generate protobuf stubs into this SDK (one-time per proto change):
+   ```bash
+   python -m grpc_tools.protoc \
+     -I../../proto \
+     --python_out=./cap/pb \
+     --grpc_python_out=./cap/pb \
+     ../../proto/cortex/agent/v1/*.proto
+   ```
+   (Or run `./tools/make_protos.sh` from repo root and copy `generated/python` into `sdk/python/cap/pb`.)
+
+2. Install:
+   ```bash
+   pip install -e .
+   ```
+
+3. Run a worker:
+   ```python
+   import asyncio
+   from cap import worker
+   from cap.pb.cortex.agent.v1 import job_pb2
+
+   async def handle(req: job_pb2.JobRequest):
+       return job_pb2.JobResult(
+           job_id=req.job_id,
+           status=job_pb2.JOB_STATUS_SUCCEEDED,
+           result_ptr=f"redis://res/{req.job_id}",
+           worker_id="worker-echo-1",
+       )
+
+   asyncio.run(worker.run_worker("nats://127.0.0.1:4222", "job.echo", handle))
+   ```
+
+## Files
+- `cap/bus.py` — NATS connector.
+- `cap/worker.py` — worker skeleton with handler hook.
+- `cap/client.py` — publish JobRequest to `sys.job.submit`.
+- `cap/pb/` — protobuf stubs (generated).
+
+## Defaults
+- Subjects: `sys.job.submit`, `sys.job.result`, `sys.heartbeat`.
+- Protocol version: `1`.
+
+Swap out `cap.bus` if you need a different transport.

cap_sdk_python-0.1.0/README.md

@@ -0,0 +1,48 @@
+# CAP Python SDK
+
+Asyncio-first SDK with NATS helpers for CAP workers and clients.
+
+## Quick Start
+1. Generate protobuf stubs into this SDK (one-time per proto change):
+   ```bash
+   python -m grpc_tools.protoc \
+     -I../../proto \
+     --python_out=./cap/pb \
+     --grpc_python_out=./cap/pb \
+     ../../proto/cortex/agent/v1/*.proto
+   ```
+   (Or run `./tools/make_protos.sh` from repo root and copy `generated/python` into `sdk/python/cap/pb`.)
+
+2. Install:
+   ```bash
+   pip install -e .
+   ```
+
+3. Run a worker:
+   ```python
+   import asyncio
+   from cap import worker
+   from cap.pb.cortex.agent.v1 import job_pb2
+
+   async def handle(req: job_pb2.JobRequest):
+       return job_pb2.JobResult(
+           job_id=req.job_id,
+           status=job_pb2.JOB_STATUS_SUCCEEDED,
+           result_ptr=f"redis://res/{req.job_id}",
+           worker_id="worker-echo-1",
+       )
+
+   asyncio.run(worker.run_worker("nats://127.0.0.1:4222", "job.echo", handle))
+   ```
+
+## Files
+- `cap/bus.py` — NATS connector.
+- `cap/worker.py` — worker skeleton with handler hook.
+- `cap/client.py` — publish JobRequest to `sys.job.submit`.
+- `cap/pb/` — protobuf stubs (generated).
+
+## Defaults
+- Subjects: `sys.job.submit`, `sys.job.result`, `sys.heartbeat`.
+- Protocol version: `1`.
+
+Swap out `cap.bus` if you need a different transport.

cap_sdk_python-0.1.0/cap/__init__.py

@@ -0,0 +1,5 @@
+from .client import submit_job
+from .worker import run_worker
+from .bus import connect_nats
+
+__all__ = ["submit_job", "run_worker", "connect_nats"]

cap_sdk_python-0.1.0/cap/bus.py

@@ -0,0 +1,31 @@
+import asyncio
+from typing import Optional
+
+import nats
+
+
+class NATSConfig:
+    def __init__(
+        self,
+        url: str,
+        token: Optional[str] = None,
+        username: Optional[str] = None,
+        password: Optional[str] = None,
+        name: str = "cap-sdk-python",
+    ):
+        self.url = url
+        self.token = token
+        self.username = username
+        self.password = password
+        self.name = name
+
+
+async def connect_nats(cfg: NATSConfig):
+    opts = {"servers": cfg.url, "name": cfg.name}
+    if cfg.token:
+        opts["token"] = cfg.token
+    if cfg.username:
+        opts["user"] = cfg.username
+    if cfg.password:
+        opts["password"] = cfg.password
+    return await nats.connect(**opts)

cap_sdk_python-0.1.0/cap/client.py

@@ -0,0 +1,18 @@
+from google.protobuf import timestamp_pb2
+from cap.pb.cortex.agent.v1 import buspacket_pb2
+
+
+DEFAULT_PROTOCOL_VERSION = 1
+SUBJECT_SUBMIT = "sys.job.submit"
+
+
+async def submit_job(nc, job_request, trace_id: str, sender_id: str):
+    ts = timestamp_pb2.Timestamp()
+    ts.GetCurrentTime()
+    packet = buspacket_pb2.BusPacket()
+    packet.trace_id = trace_id
+    packet.sender_id = sender_id
+    packet.created_at.CopyFrom(ts)
+    packet.protocol_version = DEFAULT_PROTOCOL_VERSION
+    packet.job_request.CopyFrom(job_request)
+    await nc.publish(SUBJECT_SUBMIT, packet.SerializeToString())

cap_sdk_python-0.1.0/cap/pb/cortex/agent/v1/alert_pb2.py

@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: cortex/agent/v1/alert.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1b\x63ortex/agent/v1/alert.proto\x12\x0f\x63ortex.agent.v1\"N\n\x0bSystemAlert\x12\r\n\x05level\x18\x01 \x01(\t\x12\x0f\n\x07message\x18\x02 \x01(\t\x12\x11\n\tcomponent\x18\x03 \x01(\t\x12\x0c\n\x04\x63ode\x18\x04 \x01(\tB|\n\x16\x61i.cortex.cap.agent.v1P\x01Z(github.com/coretexos/cap/cortex/agent/v1\xaa\x02\x0f\x43ortex.Agent.V1\xca\x02\x0f\x43ortex\\Agent\\V1\xea\x02\x11\x43ortex::Agent::V1b\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'cortex.agent.v1.alert_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\026ai.cortex.cap.agent.v1P\001Z(github.com/coretexos/cap/cortex/agent/v1\252\002\017Cortex.Agent.V1\312\002\017Cortex\\Agent\\V1\352\002\021Cortex::Agent::V1'
+  _globals['_SYSTEMALERT']._serialized_start=48
+  _globals['_SYSTEMALERT']._serialized_end=126
+# @@protoc_insertion_point(module_scope)

cap_sdk_python-0.1.0/cap/pb/cortex/agent/v1/alert_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

cap_sdk_python-0.1.0/cap/pb/cortex/agent/v1/buspacket_pb2.py

@@ -0,0 +1,31 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: cortex/agent/v1/buspacket.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+from cortex.agent.v1 import job_pb2 as cortex_dot_agent_dot_v1_dot_job__pb2
+from cortex.agent.v1 import heartbeat_pb2 as cortex_dot_agent_dot_v1_dot_heartbeat__pb2
+from cortex.agent.v1 import alert_pb2 as cortex_dot_agent_dot_v1_dot_alert__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1f\x63ortex/agent/v1/buspacket.proto\x12\x0f\x63ortex.agent.v1\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x19\x63ortex/agent/v1/job.proto\x1a\x1f\x63ortex/agent/v1/heartbeat.proto\x1a\x1b\x63ortex/agent/v1/alert.proto\"\xcb\x02\n\tBusPacket\x12\x10\n\x08trace_id\x18\x01 \x01(\t\x12\x11\n\tsender_id\x18\x02 \x01(\t\x12.\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10protocol_version\x18\x04 \x01(\x05\x12\x32\n\x0bjob_request\x18\n \x01(\x0b\x32\x1b.cortex.agent.v1.JobRequestH\x00\x12\x30\n\njob_result\x18\x0b \x01(\x0b\x32\x1a.cortex.agent.v1.JobResultH\x00\x12/\n\theartbeat\x18\x0c \x01(\x0b\x32\x1a.cortex.agent.v1.HeartbeatH\x00\x12-\n\x05\x61lert\x18\r \x01(\x0b\x32\x1c.cortex.agent.v1.SystemAlertH\x00\x42\t\n\x07payloadB|\n\x16\x61i.cortex.cap.agent.v1P\x01Z(github.com/coretexos/cap/cortex/agent/v1\xaa\x02\x0f\x43ortex.Agent.V1\xca\x02\x0f\x43ortex\\Agent\\V1\xea\x02\x11\x43ortex::Agent::V1b\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'cortex.agent.v1.buspacket_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\026ai.cortex.cap.agent.v1P\001Z(github.com/coretexos/cap/cortex/agent/v1\252\002\017Cortex.Agent.V1\312\002\017Cortex\\Agent\\V1\352\002\021Cortex::Agent::V1'
+  _globals['_BUSPACKET']._serialized_start=175
+  _globals['_BUSPACKET']._serialized_end=506
+# @@protoc_insertion_point(module_scope)

cap_sdk_python-0.1.0/cap/pb/cortex/agent/v1/buspacket_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

cap_sdk_python-0.1.0/cap/pb/cortex/agent/v1/heartbeat_pb2.py

@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: cortex/agent/v1/heartbeat.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1f\x63ortex/agent/v1/heartbeat.proto\x12\x0f\x63ortex.agent.v1\"\xbb\x01\n\tHeartbeat\x12\x11\n\tworker_id\x18\x01 \x01(\t\x12\x0e\n\x06region\x18\x02 \x01(\t\x12\x0c\n\x04type\x18\x03 \x01(\t\x12\x10\n\x08\x63pu_load\x18\x04 \x01(\x02\x12\x17\n\x0fgpu_utilization\x18\x05 \x01(\x02\x12\x13\n\x0b\x61\x63tive_jobs\x18\x06 \x01(\x05\x12\x14\n\x0c\x63\x61pabilities\x18\x07 \x03(\t\x12\x0c\n\x04pool\x18\x0b \x01(\t\x12\x19\n\x11max_parallel_jobs\x18\x0c \x01(\x05\x42|\n\x16\x61i.cortex.cap.agent.v1P\x01Z(github.com/coretexos/cap/cortex/agent/v1\xaa\x02\x0f\x43ortex.Agent.V1\xca\x02\x0f\x43ortex\\Agent\\V1\xea\x02\x11\x43ortex::Agent::V1b\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'cortex.agent.v1.heartbeat_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\026ai.cortex.cap.agent.v1P\001Z(github.com/coretexos/cap/cortex/agent/v1\252\002\017Cortex.Agent.V1\312\002\017Cortex\\Agent\\V1\352\002\021Cortex::Agent::V1'
+  _globals['_HEARTBEAT']._serialized_start=53
+  _globals['_HEARTBEAT']._serialized_end=240
+# @@protoc_insertion_point(module_scope)

cap_sdk_python-0.1.0/cap/pb/cortex/agent/v1/heartbeat_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

cap_sdk_python-0.1.0/cap/pb/cortex/agent/v1/job_pb2.py

@@ -0,0 +1,37 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: cortex/agent/v1/job.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x19\x63ortex/agent/v1/job.proto\x12\x0f\x63ortex.agent.v1\"\xa3\x02\n\nJobRequest\x12\x0e\n\x06job_id\x18\x01 \x01(\t\x12\r\n\x05topic\x18\x02 \x01(\t\x12.\n\x08priority\x18\x03 \x01(\x0e\x32\x1c.cortex.agent.v1.JobPriority\x12\x13\n\x0b\x63ontext_ptr\x18\x04 \x01(\t\x12\x12\n\nadapter_id\x18\x05 \x01(\t\x12\x31\n\x03\x65nv\x18\x06 \x03(\x0b\x32$.cortex.agent.v1.JobRequest.EnvEntry\x12\x15\n\rparent_job_id\x18\x07 \x01(\t\x12\x13\n\x0bworkflow_id\x18\x08 \x01(\t\x12\x12\n\nstep_index\x18\t \x01(\x05\x1a*\n\x08\x45nvEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\xaf\x01\n\tJobResult\x12\x0e\n\x06job_id\x18\x01 \x01(\t\x12*\n\x06status\x18\x02 \x01(\x0e\x32\x1a.cortex.agent.v1.JobStatus\x12\x12\n\nresult_ptr\x18\x03 \x01(\t\x12\x11\n\tworker_id\x18\x04 \x01(\t\x12\x14\n\x0c\x65xecution_ms\x18\x05 \x01(\x03\x12\x12\n\nerror_code\x18\x06 \x01(\t\x12\x15\n\rerror_message\x18\x07 \x01(\t*|\n\x0bJobPriority\x12\x1c\n\x18JOB_PRIORITY_UNSPECIFIED\x10\x00\x12\x1c\n\x18JOB_PRIORITY_INTERACTIVE\x10\x01\x12\x16\n\x12JOB_PRIORITY_BATCH\x10\x02\x12\x19\n\x15JOB_PRIORITY_CRITICAL\x10\x03*\x86\x02\n\tJobStatus\x12\x1a\n\x16JOB_STATUS_UNSPECIFIED\x10\x00\x12\x16\n\x12JOB_STATUS_PENDING\x10\x01\x12\x18\n\x14JOB_STATUS_SCHEDULED\x10\x02\x12\x19\n\x15JOB_STATUS_DISPATCHED\x10\x03\x12\x16\n\x12JOB_STATUS_RUNNING\x10\x04\x12\x18\n\x14JOB_STATUS_SUCCEEDED\x10\x05\x12\x15\n\x11JOB_STATUS_FAILED\x10\x06\x12\x18\n\x14JOB_STATUS_CANCELLED\x10\x07\x12\x15\n\x11JOB_STATUS_DENIED\x10\x08\x12\x16\n\x12JOB_STATUS_TIMEOUT\x10\tB|\n\x16\x61i.cortex.cap.agent.v1P\x01Z(github.com/coretexos/cap/cortex/agent/v1\xaa\x02\x0f\x43ortex.Agent.V1\xca\x02\x0f\x43ortex\\Agent\\V1\xea\x02\x11\x43ortex::Agent::V1b\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'cortex.agent.v1.job_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\026ai.cortex.cap.agent.v1P\001Z(github.com/coretexos/cap/cortex/agent/v1\252\002\017Cortex.Agent.V1\312\002\017Cortex\\Agent\\V1\352\002\021Cortex::Agent::V1'
+  _globals['_JOBREQUEST_ENVENTRY']._options = None
+  _globals['_JOBREQUEST_ENVENTRY']._serialized_options = b'8\001'
+  _globals['_JOBPRIORITY']._serialized_start=518
+  _globals['_JOBPRIORITY']._serialized_end=642
+  _globals['_JOBSTATUS']._serialized_start=645
+  _globals['_JOBSTATUS']._serialized_end=907
+  _globals['_JOBREQUEST']._serialized_start=47
+  _globals['_JOBREQUEST']._serialized_end=338
+  _globals['_JOBREQUEST_ENVENTRY']._serialized_start=296
+  _globals['_JOBREQUEST_ENVENTRY']._serialized_end=338
+  _globals['_JOBRESULT']._serialized_start=341
+  _globals['_JOBRESULT']._serialized_end=516
+# @@protoc_insertion_point(module_scope)

cap_sdk_python-0.1.0/cap/pb/cortex/agent/v1/job_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

cap_sdk_python-0.1.0/cap/pb/cortex/agent/v1/safety_pb2.py

@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: cortex/agent/v1/safety.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from cortex.agent.v1 import job_pb2 as cortex_dot_agent_dot_v1_dot_job__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1c\x63ortex/agent/v1/safety.proto\x12\x0f\x63ortex.agent.v1\x1a\x19\x63ortex/agent/v1/job.proto\"\x8b\x01\n\x12PolicyCheckRequest\x12\x0e\n\x06job_id\x18\x01 \x01(\t\x12\r\n\x05topic\x18\x02 \x01(\t\x12\x0e\n\x06tenant\x18\x03 \x01(\t\x12.\n\x08priority\x18\x04 \x01(\x0e\x32\x1c.cortex.agent.v1.JobPriority\x12\x16\n\x0e\x65stimated_cost\x18\x05 \x01(\x01\"t\n\x13PolicyCheckResponse\x12/\n\x08\x64\x65\x63ision\x18\x01 \x01(\x0e\x32\x1d.cortex.agent.v1.DecisionType\x12\x0e\n\x06reason\x18\x02 \x01(\t\x12\x1c\n\x14redacted_context_ptr\x18\x03 \x01(\t*\x9b\x01\n\x0c\x44\x65\x63isionType\x12\x1d\n\x19\x44\x45\x43ISION_TYPE_UNSPECIFIED\x10\x00\x12\x17\n\x13\x44\x45\x43ISION_TYPE_ALLOW\x10\x01\x12\x16\n\x12\x44\x45\x43ISION_TYPE_DENY\x10\x02\x12\x1f\n\x1b\x44\x45\x43ISION_TYPE_REQUIRE_HUMAN\x10\x03\x12\x1a\n\x16\x44\x45\x43ISION_TYPE_THROTTLE\x10\x04\x32\x62\n\x0cSafetyKernel\x12R\n\x05\x43heck\x12#.cortex.agent.v1.PolicyCheckRequest\x1a$.cortex.agent.v1.PolicyCheckResponseB|\n\x16\x61i.cortex.cap.agent.v1P\x01Z(github.com/coretexos/cap/cortex/agent/v1\xaa\x02\x0f\x43ortex.Agent.V1\xca\x02\x0f\x43ortex\\Agent\\V1\xea\x02\x11\x43ortex::Agent::V1b\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'cortex.agent.v1.safety_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\026ai.cortex.cap.agent.v1P\001Z(github.com/coretexos/cap/cortex/agent/v1\252\002\017Cortex.Agent.V1\312\002\017Cortex\\Agent\\V1\352\002\021Cortex::Agent::V1'
+  _globals['_DECISIONTYPE']._serialized_start=337
+  _globals['_DECISIONTYPE']._serialized_end=492
+  _globals['_POLICYCHECKREQUEST']._serialized_start=77
+  _globals['_POLICYCHECKREQUEST']._serialized_end=216
+  _globals['_POLICYCHECKRESPONSE']._serialized_start=218
+  _globals['_POLICYCHECKRESPONSE']._serialized_end=334
+  _globals['_SAFETYKERNEL']._serialized_start=494
+  _globals['_SAFETYKERNEL']._serialized_end=592
+# @@protoc_insertion_point(module_scope)

cap_sdk_python-0.1.0/cap/pb/cortex/agent/v1/safety_pb2_grpc.py

@@ -0,0 +1,69 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from cortex.agent.v1 import safety_pb2 as cortex_dot_agent_dot_v1_dot_safety__pb2
+
+
+class SafetyKernelStub(object):
+    """SafetyKernel defines the policy decision service.
+    """
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.Check = channel.unary_unary(
+                '/cortex.agent.v1.SafetyKernel/Check',
+                request_serializer=cortex_dot_agent_dot_v1_dot_safety__pb2.PolicyCheckRequest.SerializeToString,
+                response_deserializer=cortex_dot_agent_dot_v1_dot_safety__pb2.PolicyCheckResponse.FromString,
+                )
+
+
+class SafetyKernelServicer(object):
+    """SafetyKernel defines the policy decision service.
+    """
+
+    def Check(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_SafetyKernelServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'Check': grpc.unary_unary_rpc_method_handler(
+                    servicer.Check,
+                    request_deserializer=cortex_dot_agent_dot_v1_dot_safety__pb2.PolicyCheckRequest.FromString,
+                    response_serializer=cortex_dot_agent_dot_v1_dot_safety__pb2.PolicyCheckResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'cortex.agent.v1.SafetyKernel', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class SafetyKernel(object):
+    """SafetyKernel defines the policy decision service.
+    """
+
+    @staticmethod
+    def Check(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/cortex.agent.v1.SafetyKernel/Check',
+            cortex_dot_agent_dot_v1_dot_safety__pb2.PolicyCheckRequest.SerializeToString,
+            cortex_dot_agent_dot_v1_dot_safety__pb2.PolicyCheckResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

cap_sdk_python-0.1.0/cap/worker.py

@@ -0,0 +1,45 @@
+import asyncio
+from typing import Callable, Awaitable
+
+from google.protobuf import timestamp_pb2
+from cap.pb.cortex.agent.v1 import buspacket_pb2, job_pb2
+
+DEFAULT_PROTOCOL_VERSION = 1
+SUBJECT_RESULT = "sys.job.result"
+
+
+async def run_worker(nats_url: str, subject: str, handler: Callable[[job_pb2.JobRequest], Awaitable[job_pb2.JobResult]]):
+    import nats
+
+    nc = await nats.connect(servers=nats_url, name="cap-worker")
+
+    async def on_msg(msg):
+        packet = buspacket_pb2.BusPacket()
+        packet.ParseFromString(msg.data)
+        req = packet.job_request
+        if not req.job_id:
+            return
+        try:
+            res = await handler(req)
+        except Exception as exc:  # noqa: BLE001
+            res = job_pb2.JobResult(
+                job_id=req.job_id,
+                status=job_pb2.JOB_STATUS_FAILED,
+                error_message=str(exc),
+            )
+        ts = timestamp_pb2.Timestamp()
+        ts.GetCurrentTime()
+        out = buspacket_pb2.BusPacket()
+        out.trace_id = packet.trace_id
+        out.sender_id = "cap-worker"
+        out.protocol_version = DEFAULT_PROTOCOL_VERSION
+        out.created_at.CopyFrom(ts)
+        out.job_result.CopyFrom(res)
+        await nc.publish(SUBJECT_RESULT, out.SerializeToString())
+
+    await nc.subscribe(subject, queue=subject, cb=on_msg)
+    try:
+        while True:
+            await asyncio.sleep(1)
+    finally:
+        await nc.drain()

cap_sdk_python-0.1.0/cap_sdk_python.egg-info/PKG-INFO

@@ -0,0 +1,59 @@
+Metadata-Version: 2.4
+Name: cap-sdk-python
+Version: 0.1.0
+Summary: CAP (Cortex Agent Protocol) Python SDK
+Project-URL: Homepage, https://github.com/coretexos/cap
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: protobuf>=4.25.0
+Requires-Dist: grpcio>=1.59.0
+Requires-Dist: nats-py>=2.6.0
+
+# CAP Python SDK
+
+Asyncio-first SDK with NATS helpers for CAP workers and clients.
+
+## Quick Start
+1. Generate protobuf stubs into this SDK (one-time per proto change):
+   ```bash
+   python -m grpc_tools.protoc \
+     -I../../proto \
+     --python_out=./cap/pb \
+     --grpc_python_out=./cap/pb \
+     ../../proto/cortex/agent/v1/*.proto
+   ```
+   (Or run `./tools/make_protos.sh` from repo root and copy `generated/python` into `sdk/python/cap/pb`.)
+
+2. Install:
+   ```bash
+   pip install -e .
+   ```
+
+3. Run a worker:
+   ```python
+   import asyncio
+   from cap import worker
+   from cap.pb.cortex.agent.v1 import job_pb2
+
+   async def handle(req: job_pb2.JobRequest):
+       return job_pb2.JobResult(
+           job_id=req.job_id,
+           status=job_pb2.JOB_STATUS_SUCCEEDED,
+           result_ptr=f"redis://res/{req.job_id}",
+           worker_id="worker-echo-1",
+       )
+
+   asyncio.run(worker.run_worker("nats://127.0.0.1:4222", "job.echo", handle))
+   ```
+
+## Files
+- `cap/bus.py` — NATS connector.
+- `cap/worker.py` — worker skeleton with handler hook.
+- `cap/client.py` — publish JobRequest to `sys.job.submit`.
+- `cap/pb/` — protobuf stubs (generated).
+
+## Defaults
+- Subjects: `sys.job.submit`, `sys.job.result`, `sys.heartbeat`.
+- Protocol version: `1`.
+
+Swap out `cap.bus` if you need a different transport.

cap_sdk_python-0.1.0/cap_sdk_python.egg-info/SOURCES.txt

@@ -0,0 +1,25 @@
+README.md
+pyproject.toml
+cap/__init__.py
+cap/bus.py
+cap/client.py
+cap/worker.py
+cap/pb/__init__.py
+cap/pb/cortex/__init__.py
+cap/pb/cortex/agent/__init__.py
+cap/pb/cortex/agent/v1/__init__.py
+cap/pb/cortex/agent/v1/alert_pb2.py
+cap/pb/cortex/agent/v1/alert_pb2_grpc.py
+cap/pb/cortex/agent/v1/buspacket_pb2.py
+cap/pb/cortex/agent/v1/buspacket_pb2_grpc.py
+cap/pb/cortex/agent/v1/heartbeat_pb2.py
+cap/pb/cortex/agent/v1/heartbeat_pb2_grpc.py
+cap/pb/cortex/agent/v1/job_pb2.py
+cap/pb/cortex/agent/v1/job_pb2_grpc.py
+cap/pb/cortex/agent/v1/safety_pb2.py
+cap/pb/cortex/agent/v1/safety_pb2_grpc.py
+cap_sdk_python.egg-info/PKG-INFO
+cap_sdk_python.egg-info/SOURCES.txt
+cap_sdk_python.egg-info/dependency_links.txt
+cap_sdk_python.egg-info/requires.txt
+cap_sdk_python.egg-info/top_level.txt

cap_sdk_python-0.1.0/cap_sdk_python.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

cap_sdk_python-0.1.0/cap_sdk_python.egg-info/requires.txt

@@ -0,0 +1,3 @@
+protobuf>=4.25.0
+grpcio>=1.59.0
+nats-py>=2.6.0

cap_sdk_python-0.1.0/cap_sdk_python.egg-info/top_level.txt

@@ -0,0 +1,2 @@
+cap
+dist

cap_sdk_python-0.1.0/pyproject.toml

@@ -0,0 +1,21 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "cap-sdk-python"
+version = "0.1.0"
+description = "CAP (Cortex Agent Protocol) Python SDK"
+readme = "README.md"
+requires-python = ">=3.9"
+dependencies = [
+  "protobuf>=4.25.0",
+  "grpcio>=1.59.0",
+  "nats-py>=2.6.0",
+]
+
+[project.urls]
+Homepage = "https://github.com/coretexos/cap"
+
+[tool.setuptools.packages.find]
+where = ["."]

cap_sdk_python-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+