gatewaysdk 0.3.2__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (376) hide show
  1. gatewaysdk/__init__.py +540 -0
  2. gatewaysdk/adapter/__init__.py +15 -0
  3. gatewaysdk/adapter/base.py +94 -0
  4. gatewaysdk/adapter/messages.py +270 -0
  5. gatewaysdk/adapter/triplet.py +1026 -0
  6. gatewaysdk/algorithm/__init__.py +39 -0
  7. gatewaysdk/algorithm/apo/__init__.py +5 -0
  8. gatewaysdk/algorithm/apo/apo.py +898 -0
  9. gatewaysdk/algorithm/apo/prompts/apply_edit_variant01.poml +22 -0
  10. gatewaysdk/algorithm/apo/prompts/apply_edit_variant02.poml +18 -0
  11. gatewaysdk/algorithm/apo/prompts/text_gradient_variant01.poml +18 -0
  12. gatewaysdk/algorithm/apo/prompts/text_gradient_variant02.poml +16 -0
  13. gatewaysdk/algorithm/apo/prompts/text_gradient_variant03.poml +107 -0
  14. gatewaysdk/algorithm/base.py +262 -0
  15. gatewaysdk/algorithm/decorator.py +264 -0
  16. gatewaysdk/algorithm/evals/__init__.py +7 -0
  17. gatewaysdk/algorithm/evals/evals.py +217 -0
  18. gatewaysdk/algorithm/fast.py +250 -0
  19. gatewaysdk/algorithm/gepa/__init__.py +61 -0
  20. gatewaysdk/algorithm/gepa/adapter.py +495 -0
  21. gatewaysdk/algorithm/gepa/gepa.py +570 -0
  22. gatewaysdk/algorithm/gepa/lib/__init__.py +18 -0
  23. gatewaysdk/algorithm/gepa/lib/adapters/README.md +12 -0
  24. gatewaysdk/algorithm/gepa/lib/adapters/__init__.py +0 -0
  25. gatewaysdk/algorithm/gepa/lib/adapters/anymaths_adapter/README.md +341 -0
  26. gatewaysdk/algorithm/gepa/lib/adapters/anymaths_adapter/__init__.py +1 -0
  27. gatewaysdk/algorithm/gepa/lib/adapters/anymaths_adapter/anymaths_adapter.py +174 -0
  28. gatewaysdk/algorithm/gepa/lib/adapters/anymaths_adapter/requirements.txt +1 -0
  29. gatewaysdk/algorithm/gepa/lib/adapters/default_adapter/README.md +0 -0
  30. gatewaysdk/algorithm/gepa/lib/adapters/default_adapter/__init__.py +0 -0
  31. gatewaysdk/algorithm/gepa/lib/adapters/default_adapter/default_adapter.py +209 -0
  32. gatewaysdk/algorithm/gepa/lib/adapters/dspy_adapter/README.md +7 -0
  33. gatewaysdk/algorithm/gepa/lib/adapters/dspy_adapter/__init__.py +0 -0
  34. gatewaysdk/algorithm/gepa/lib/adapters/dspy_adapter/dspy_adapter.py +307 -0
  35. gatewaysdk/algorithm/gepa/lib/adapters/dspy_full_program_adapter/README.md +99 -0
  36. gatewaysdk/algorithm/gepa/lib/adapters/dspy_full_program_adapter/dspy_program_proposal_signature.py +137 -0
  37. gatewaysdk/algorithm/gepa/lib/adapters/dspy_full_program_adapter/full_program_adapter.py +268 -0
  38. gatewaysdk/algorithm/gepa/lib/adapters/generic_rag_adapter/GEPA_RAG.md +621 -0
  39. gatewaysdk/algorithm/gepa/lib/adapters/generic_rag_adapter/__init__.py +56 -0
  40. gatewaysdk/algorithm/gepa/lib/adapters/generic_rag_adapter/evaluation_metrics.py +226 -0
  41. gatewaysdk/algorithm/gepa/lib/adapters/generic_rag_adapter/generic_rag_adapter.py +496 -0
  42. gatewaysdk/algorithm/gepa/lib/adapters/generic_rag_adapter/rag_pipeline.py +238 -0
  43. gatewaysdk/algorithm/gepa/lib/adapters/generic_rag_adapter/vector_store_interface.py +212 -0
  44. gatewaysdk/algorithm/gepa/lib/adapters/generic_rag_adapter/vector_stores/__init__.py +2 -0
  45. gatewaysdk/algorithm/gepa/lib/adapters/generic_rag_adapter/vector_stores/chroma_store.py +196 -0
  46. gatewaysdk/algorithm/gepa/lib/adapters/generic_rag_adapter/vector_stores/lancedb_store.py +422 -0
  47. gatewaysdk/algorithm/gepa/lib/adapters/generic_rag_adapter/vector_stores/milvus_store.py +409 -0
  48. gatewaysdk/algorithm/gepa/lib/adapters/generic_rag_adapter/vector_stores/qdrant_store.py +368 -0
  49. gatewaysdk/algorithm/gepa/lib/adapters/generic_rag_adapter/vector_stores/weaviate_store.py +418 -0
  50. gatewaysdk/algorithm/gepa/lib/adapters/mcp_adapter/README.md +552 -0
  51. gatewaysdk/algorithm/gepa/lib/adapters/mcp_adapter/__init__.py +37 -0
  52. gatewaysdk/algorithm/gepa/lib/adapters/mcp_adapter/mcp_adapter.py +699 -0
  53. gatewaysdk/algorithm/gepa/lib/adapters/mcp_adapter/mcp_client.py +364 -0
  54. gatewaysdk/algorithm/gepa/lib/adapters/terminal_bench_adapter/README.md +9 -0
  55. gatewaysdk/algorithm/gepa/lib/adapters/terminal_bench_adapter/__init__.py +0 -0
  56. gatewaysdk/algorithm/gepa/lib/adapters/terminal_bench_adapter/terminal_bench_adapter.py +217 -0
  57. gatewaysdk/algorithm/gepa/lib/api.py +382 -0
  58. gatewaysdk/algorithm/gepa/lib/core/__init__.py +0 -0
  59. gatewaysdk/algorithm/gepa/lib/core/adapter.py +180 -0
  60. gatewaysdk/algorithm/gepa/lib/core/data_loader.py +74 -0
  61. gatewaysdk/algorithm/gepa/lib/core/engine.py +379 -0
  62. gatewaysdk/algorithm/gepa/lib/core/result.py +233 -0
  63. gatewaysdk/algorithm/gepa/lib/core/state.py +636 -0
  64. gatewaysdk/algorithm/gepa/lib/examples/__init__.py +0 -0
  65. gatewaysdk/algorithm/gepa/lib/examples/aime.py +24 -0
  66. gatewaysdk/algorithm/gepa/lib/examples/anymaths-bench/eval_default.py +111 -0
  67. gatewaysdk/algorithm/gepa/lib/examples/anymaths-bench/prompt-templates/instruction_prompt.txt +9 -0
  68. gatewaysdk/algorithm/gepa/lib/examples/anymaths-bench/prompt-templates/optimal_prompt.txt +24 -0
  69. gatewaysdk/algorithm/gepa/lib/examples/anymaths-bench/train_anymaths.py +177 -0
  70. gatewaysdk/algorithm/gepa/lib/examples/dspy_full_program_evolution/arc_agi.ipynb +25705 -0
  71. gatewaysdk/algorithm/gepa/lib/examples/dspy_full_program_evolution/example.ipynb +348 -0
  72. gatewaysdk/algorithm/gepa/lib/examples/mcp_adapter/__init__.py +4 -0
  73. gatewaysdk/algorithm/gepa/lib/examples/mcp_adapter/mcp_optimization_example.py +456 -0
  74. gatewaysdk/algorithm/gepa/lib/examples/rag_adapter/RAG_GUIDE.md +613 -0
  75. gatewaysdk/algorithm/gepa/lib/examples/rag_adapter/__init__.py +9 -0
  76. gatewaysdk/algorithm/gepa/lib/examples/rag_adapter/rag_optimization.py +820 -0
  77. gatewaysdk/algorithm/gepa/lib/examples/rag_adapter/requirements-rag.txt +29 -0
  78. gatewaysdk/algorithm/gepa/lib/examples/terminal-bench/prompt-templates/instruction_prompt.txt +16 -0
  79. gatewaysdk/algorithm/gepa/lib/examples/terminal-bench/prompt-templates/terminus.txt +9 -0
  80. gatewaysdk/algorithm/gepa/lib/examples/terminal-bench/train_terminus.py +161 -0
  81. gatewaysdk/algorithm/gepa/lib/gepa_utils.py +117 -0
  82. gatewaysdk/algorithm/gepa/lib/logging/__init__.py +0 -0
  83. gatewaysdk/algorithm/gepa/lib/logging/experiment_tracker.py +187 -0
  84. gatewaysdk/algorithm/gepa/lib/logging/logger.py +75 -0
  85. gatewaysdk/algorithm/gepa/lib/logging/utils.py +103 -0
  86. gatewaysdk/algorithm/gepa/lib/proposer/__init__.py +0 -0
  87. gatewaysdk/algorithm/gepa/lib/proposer/base.py +31 -0
  88. gatewaysdk/algorithm/gepa/lib/proposer/merge.py +357 -0
  89. gatewaysdk/algorithm/gepa/lib/proposer/reflective_mutation/__init__.py +0 -0
  90. gatewaysdk/algorithm/gepa/lib/proposer/reflective_mutation/base.py +49 -0
  91. gatewaysdk/algorithm/gepa/lib/proposer/reflective_mutation/reflective_mutation.py +176 -0
  92. gatewaysdk/algorithm/gepa/lib/py.typed +0 -0
  93. gatewaysdk/algorithm/gepa/lib/strategies/__init__.py +0 -0
  94. gatewaysdk/algorithm/gepa/lib/strategies/batch_sampler.py +77 -0
  95. gatewaysdk/algorithm/gepa/lib/strategies/candidate_selector.py +50 -0
  96. gatewaysdk/algorithm/gepa/lib/strategies/component_selector.py +36 -0
  97. gatewaysdk/algorithm/gepa/lib/strategies/eval_policy.py +64 -0
  98. gatewaysdk/algorithm/gepa/lib/strategies/instruction_proposal.py +126 -0
  99. gatewaysdk/algorithm/gepa/lib/utils/__init__.py +10 -0
  100. gatewaysdk/algorithm/gepa/lib/utils/stop_condition.py +196 -0
  101. gatewaysdk/algorithm/gepa/tracing.py +105 -0
  102. gatewaysdk/algorithm/utils.py +177 -0
  103. gatewaysdk/algorithm/verl/__init__.py +5 -0
  104. gatewaysdk/algorithm/verl/interface.py +202 -0
  105. gatewaysdk/automations.py +111 -0
  106. gatewaysdk/benchmark_hub/README.md +88 -0
  107. gatewaysdk/benchmark_hub/__init__.py +119 -0
  108. gatewaysdk/benchmark_hub/_archive.py +189 -0
  109. gatewaysdk/benchmark_hub/_tar.py +6 -0
  110. gatewaysdk/benchmark_hub/client.py +992 -0
  111. gatewaysdk/benchmark_hub/dispatch_shard.py +34 -0
  112. gatewaysdk/benchmark_hub/eval_config.py +20 -0
  113. gatewaysdk/benchmark_hub/evals.py +351 -0
  114. gatewaysdk/benchmark_hub/harbor_adapter.py +673 -0
  115. gatewaysdk/benchmark_hub/path_utils.py +81 -0
  116. gatewaysdk/benchmark_hub/save_utils.py +101 -0
  117. gatewaysdk/benchmark_hub/verifiers_adapter.py +327 -0
  118. gatewaysdk/benchmark_hub/versioning.py +72 -0
  119. gatewaysdk/build.py +515 -0
  120. gatewaysdk/cli/__init__.py +58 -0
  121. gatewaysdk/cli/agent_runner.py +132 -0
  122. gatewaysdk/cli/http_client.py +115 -0
  123. gatewaysdk/cli/platform.py +4086 -0
  124. gatewaysdk/cli/prometheus.py +115 -0
  125. gatewaysdk/cli/release_gate.py +215 -0
  126. gatewaysdk/cli/store.py +131 -0
  127. gatewaysdk/cli/vllm.py +29 -0
  128. gatewaysdk/client.py +406 -0
  129. gatewaysdk/config.py +348 -0
  130. gatewaysdk/connectors/__init__.py +25 -0
  131. gatewaysdk/connectors/client.py +203 -0
  132. gatewaysdk/connectors/skill.py +25 -0
  133. gatewaysdk/connectors/template.py +106 -0
  134. gatewaysdk/context.py +606 -0
  135. gatewaysdk/emitter/__init__.py +43 -0
  136. gatewaysdk/emitter/annotation.py +370 -0
  137. gatewaysdk/emitter/exception.py +54 -0
  138. gatewaysdk/emitter/message.py +61 -0
  139. gatewaysdk/emitter/object.py +117 -0
  140. gatewaysdk/emitter/reward.py +320 -0
  141. gatewaysdk/env_var.py +156 -0
  142. gatewaysdk/environment/__init__.py +108 -0
  143. gatewaysdk/environment/_bundle.py +281 -0
  144. gatewaysdk/environment/_harbor.py +276 -0
  145. gatewaysdk/environment/_materialize.py +97 -0
  146. gatewaysdk/environment/_tar.py +33 -0
  147. gatewaysdk/environment/_world.py +346 -0
  148. gatewaysdk/environment/core.py +527 -0
  149. gatewaysdk/environment/errors.py +58 -0
  150. gatewaysdk/environment/runtime.py +150 -0
  151. gatewaysdk/environment/schema/__init__.py +35 -0
  152. gatewaysdk/environment/schema/__main__.py +225 -0
  153. gatewaysdk/environment/schema/_slack_fidelity.py +116 -0
  154. gatewaysdk/environment/schema/_slack_scenario.py +183 -0
  155. gatewaysdk/environment/schema/api.py +1628 -0
  156. gatewaysdk/environment/schema/batch.py +755 -0
  157. gatewaysdk/environment/schema/compiler.py +615 -0
  158. gatewaysdk/environment/schema/conform.py +428 -0
  159. gatewaysdk/environment/schema/connector.py +777 -0
  160. gatewaysdk/environment/schema/connectors/apple-business-manager/handlers.py +17 -0
  161. gatewaysdk/environment/schema/connectors/apple-business-manager/parity.json +28 -0
  162. gatewaysdk/environment/schema/connectors/apple-business-manager/provenance.json +233 -0
  163. gatewaysdk/environment/schema/connectors/apple-business-manager/scope.toml +112 -0
  164. gatewaysdk/environment/schema/connectors/apple-business-manager/world.json +1380 -0
  165. gatewaysdk/environment/schema/connectors/base.json +28 -0
  166. gatewaysdk/environment/schema/connectors/custom/handlers.py +18 -0
  167. gatewaysdk/environment/schema/connectors/custom/world.json +123 -0
  168. gatewaysdk/environment/schema/connectors/github/handlers.py +81 -0
  169. gatewaysdk/environment/schema/connectors/github/parity.json +37 -0
  170. gatewaysdk/environment/schema/connectors/github/provenance.json +134 -0
  171. gatewaysdk/environment/schema/connectors/github/scope.toml +158 -0
  172. gatewaysdk/environment/schema/connectors/github/world.json +5869 -0
  173. gatewaysdk/environment/schema/connectors/google-calendar/handlers.py +23 -0
  174. gatewaysdk/environment/schema/connectors/google-calendar/world.json +209 -0
  175. gatewaysdk/environment/schema/connectors/google-drive/handlers.py +18 -0
  176. gatewaysdk/environment/schema/connectors/google-drive/world.json +286 -0
  177. gatewaysdk/environment/schema/connectors/jamf/handlers.py +34 -0
  178. gatewaysdk/environment/schema/connectors/jamf/parity.json +49 -0
  179. gatewaysdk/environment/schema/connectors/jamf/provenance.json +181 -0
  180. gatewaysdk/environment/schema/connectors/jamf/scope.toml +128 -0
  181. gatewaysdk/environment/schema/connectors/jamf/world.json +2218 -0
  182. gatewaysdk/environment/schema/connectors/jira/handlers.py +28 -0
  183. gatewaysdk/environment/schema/connectors/jira/world.json +389 -0
  184. gatewaysdk/environment/schema/connectors/kandji/handlers.py +38 -0
  185. gatewaysdk/environment/schema/connectors/kandji/parity.json +29 -0
  186. gatewaysdk/environment/schema/connectors/kandji/provenance.json +152 -0
  187. gatewaysdk/environment/schema/connectors/kandji/scope.toml +120 -0
  188. gatewaysdk/environment/schema/connectors/kandji/world.json +765 -0
  189. gatewaysdk/environment/schema/connectors/linear/handlers.py +20 -0
  190. gatewaysdk/environment/schema/connectors/linear/world.json +363 -0
  191. gatewaysdk/environment/schema/connectors/microsoft-teams/fidelity.json +45 -0
  192. gatewaysdk/environment/schema/connectors/microsoft-teams/handlers.py +1 -0
  193. gatewaysdk/environment/schema/connectors/microsoft-teams/provenance.json +851 -0
  194. gatewaysdk/environment/schema/connectors/microsoft-teams/scope.md +231 -0
  195. gatewaysdk/environment/schema/connectors/microsoft-teams/teams-types.json +1710 -0
  196. gatewaysdk/environment/schema/connectors/microsoft-teams/world.json +858 -0
  197. gatewaysdk/environment/schema/connectors/netsuite/handlers.py +32 -0
  198. gatewaysdk/environment/schema/connectors/netsuite/parity.json +50 -0
  199. gatewaysdk/environment/schema/connectors/netsuite/provenance.json +137 -0
  200. gatewaysdk/environment/schema/connectors/netsuite/scope.toml +78 -0
  201. gatewaysdk/environment/schema/connectors/netsuite/world.json +16243 -0
  202. gatewaysdk/environment/schema/connectors/salesforce/handlers.py +806 -0
  203. gatewaysdk/environment/schema/connectors/salesforce/parity.json +46 -0
  204. gatewaysdk/environment/schema/connectors/salesforce/provenance.json +143 -0
  205. gatewaysdk/environment/schema/connectors/salesforce/scope.toml +174 -0
  206. gatewaysdk/environment/schema/connectors/salesforce/world.json +4695 -0
  207. gatewaysdk/environment/schema/connectors/slack/MODEL.md +236 -0
  208. gatewaysdk/environment/schema/connectors/slack/capabilities.json +1039 -0
  209. gatewaysdk/environment/schema/connectors/slack/handlers.py +762 -0
  210. gatewaysdk/environment/schema/connectors/slack/parity.json +65 -0
  211. gatewaysdk/environment/schema/connectors/slack/provenance.json +294 -0
  212. gatewaysdk/environment/schema/connectors/slack/scope.toml +44 -0
  213. gatewaysdk/environment/schema/connectors/slack/world.json +4993 -0
  214. gatewaysdk/environment/schema/connectors/workday/handlers.py +75 -0
  215. gatewaysdk/environment/schema/connectors/workday/parity.json +36 -0
  216. gatewaysdk/environment/schema/connectors/workday/provenance.json +179 -0
  217. gatewaysdk/environment/schema/connectors/workday/scope.toml +180 -0
  218. gatewaysdk/environment/schema/connectors/workday/world.json +806 -0
  219. gatewaysdk/environment/schema/data.py +593 -0
  220. gatewaysdk/environment/schema/extract.py +263 -0
  221. gatewaysdk/environment/schema/host.py +2437 -0
  222. gatewaysdk/environment/schema/host_surfaces.py +275 -0
  223. gatewaysdk/environment/schema/platform.py +1272 -0
  224. gatewaysdk/environment/schema/scaffold.py +250 -0
  225. gatewaysdk/environment/schema/skill.py +118 -0
  226. gatewaysdk/environment/schema/skills/agents/connector-template-author.md +202 -0
  227. gatewaysdk/environment/schema/skills/connector-schema-authoring/CONNECTORS.md +331 -0
  228. gatewaysdk/environment/schema/skills/connector-schema-authoring/CONVENTIONS.md +100 -0
  229. gatewaysdk/environment/schema/skills/connector-schema-authoring/SKILL.md +313 -0
  230. gatewaysdk/environment/schema/skills/world-data-ingestion/ROWS.md +223 -0
  231. gatewaysdk/environment/schema/skills/world-data-ingestion/SKILL.md +293 -0
  232. gatewaysdk/environment/schema/skills/worlds-getting-started/SKILL.md +390 -0
  233. gatewaysdk/environment/schema/slack.py +757 -0
  234. gatewaysdk/environment/schema/snapshot.py +479 -0
  235. gatewaysdk/environment/schema/store.py +904 -0
  236. gatewaysdk/environment/schema/tasks.py +525 -0
  237. gatewaysdk/environment/schema/tools_world.py +322 -0
  238. gatewaysdk/environment/schema/validation.py +577 -0
  239. gatewaysdk/environment/schema/worker.py +78 -0
  240. gatewaysdk/environment/schema/workers.py +500 -0
  241. gatewaysdk/environment/schema/world_tests.py +336 -0
  242. gatewaysdk/environments/__init__.py +24 -0
  243. gatewaysdk/environments/client.py +486 -0
  244. gatewaysdk/environments/types.py +318 -0
  245. gatewaysdk/execution/__init__.py +15 -0
  246. gatewaysdk/execution/base.py +64 -0
  247. gatewaysdk/execution/client_server.py +443 -0
  248. gatewaysdk/execution/events.py +69 -0
  249. gatewaysdk/execution/inter_process.py +16 -0
  250. gatewaysdk/execution/shared_memory.py +282 -0
  251. gatewaysdk/experiments/__init__.py +90 -0
  252. gatewaysdk/experiments/assignment.py +177 -0
  253. gatewaysdk/experiments/client.py +877 -0
  254. gatewaysdk/experiments/exposure.py +222 -0
  255. gatewaysdk/experiments/types.py +81 -0
  256. gatewaysdk/importers/__init__.py +49 -0
  257. gatewaysdk/importers/_normalize.py +91 -0
  258. gatewaysdk/importers/builder.py +173 -0
  259. gatewaysdk/importers/client.py +95 -0
  260. gatewaysdk/importers/langsmith.py +196 -0
  261. gatewaysdk/importers/recipes.py +249 -0
  262. gatewaysdk/instrumentation/__init__.py +300 -0
  263. gatewaysdk/instrumentation/agentops.py +314 -0
  264. gatewaysdk/instrumentation/agentops_langchain.py +45 -0
  265. gatewaysdk/instrumentation/base.py +119 -0
  266. gatewaysdk/instrumentation/litellm.py +83 -0
  267. gatewaysdk/instrumentation/registry.py +273 -0
  268. gatewaysdk/instrumentation/vllm.py +81 -0
  269. gatewaysdk/instrumentation/weave.py +500 -0
  270. gatewaysdk/integrations/__init__.py +15 -0
  271. gatewaysdk/integrations/gateway/__init__.py +11 -0
  272. gatewaysdk/integrations/gateway/client.py +171 -0
  273. gatewaysdk/integrations/tool_access.py +549 -0
  274. gatewaysdk/litagent/__init__.py +11 -0
  275. gatewaysdk/litagent/decorator.py +536 -0
  276. gatewaysdk/litagent/litagent.py +252 -0
  277. gatewaysdk/llm_proxy.py +1742 -0
  278. gatewaysdk/logging.py +370 -0
  279. gatewaysdk/memory.py +278 -0
  280. gatewaysdk/personas.py +186 -0
  281. gatewaysdk/platform/__init__.py +17 -0
  282. gatewaysdk/platform/builder.py +221 -0
  283. gatewaysdk/platform/compatibility.py +67 -0
  284. gatewaysdk/platform/manifest.py +185 -0
  285. gatewaysdk/platform/orchestrator.py +901 -0
  286. gatewaysdk/platform/registry.py +122 -0
  287. gatewaysdk/platform/worker.py +864 -0
  288. gatewaysdk/replay/__init__.py +1032 -0
  289. gatewaysdk/replay/pytest.py +56 -0
  290. gatewaysdk/reward.py +7 -0
  291. gatewaysdk/run.py +1781 -0
  292. gatewaysdk/runner/__init__.py +11 -0
  293. gatewaysdk/runner/agent.py +878 -0
  294. gatewaysdk/runner/base.py +182 -0
  295. gatewaysdk/runner/legacy.py +309 -0
  296. gatewaysdk/security.py +700 -0
  297. gatewaysdk/semconv.py +170 -0
  298. gatewaysdk/server.py +399 -0
  299. gatewaysdk/sessions.py +282 -0
  300. gatewaysdk/store/__init__.py +45 -0
  301. gatewaysdk/store/base.py +908 -0
  302. gatewaysdk/store/client_server.py +2093 -0
  303. gatewaysdk/store/collection/__init__.py +30 -0
  304. gatewaysdk/store/collection/base.py +587 -0
  305. gatewaysdk/store/collection/memory.py +970 -0
  306. gatewaysdk/store/collection/mongo.py +1412 -0
  307. gatewaysdk/store/collection_based.py +1823 -0
  308. gatewaysdk/store/gateway.py +983 -0
  309. gatewaysdk/store/gateway_listener.py +465 -0
  310. gatewaysdk/store/listener.py +58 -0
  311. gatewaysdk/store/memory.py +396 -0
  312. gatewaysdk/store/mongo.py +165 -0
  313. gatewaysdk/store/redis_stream.py +517 -0
  314. gatewaysdk/store/sqlite.py +3 -0
  315. gatewaysdk/store/threading.py +370 -0
  316. gatewaysdk/store/utils.py +142 -0
  317. gatewaysdk/tracer/__init__.py +14 -0
  318. gatewaysdk/tracer/base.py +286 -0
  319. gatewaysdk/tracer/dummy.py +106 -0
  320. gatewaysdk/tracer/otel.py +559 -0
  321. gatewaysdk/tracing/__init__.py +110 -0
  322. gatewaysdk/tracing/api.py +808 -0
  323. gatewaysdk/tracing/attributes.py +9 -0
  324. gatewaysdk/tracing/context.py +272 -0
  325. gatewaysdk/tracing/exporters/__init__.py +10 -0
  326. gatewaysdk/tracing/exporters/gateway.py +228 -0
  327. gatewaysdk/tracing/identity.py +288 -0
  328. gatewaysdk/tracing/init.py +620 -0
  329. gatewaysdk/tracing/instrumentors/__init__.py +15 -0
  330. gatewaysdk/tracing/instrumentors/claude_agent_sdk.py +766 -0
  331. gatewaysdk/tracing/instrumentors/instrumentation_principles.md +294 -0
  332. gatewaysdk/tracing/instrumentors/registry.py +352 -0
  333. gatewaysdk/tracing/mapping.py +729 -0
  334. gatewaysdk/tracing/processors.py +393 -0
  335. gatewaysdk/tracing/push.py +617 -0
  336. gatewaysdk/tracing/push_models.py +247 -0
  337. gatewaysdk/tracing/semconv.py +294 -0
  338. gatewaysdk/tracing/span_builder.py +356 -0
  339. gatewaysdk/trainer/__init__.py +6 -0
  340. gatewaysdk/trainer/init_utils.py +263 -0
  341. gatewaysdk/trainer/legacy.py +359 -0
  342. gatewaysdk/trainer/registry.py +12 -0
  343. gatewaysdk/trainer/trainer.py +638 -0
  344. gatewaysdk/types/__init__.py +63 -0
  345. gatewaysdk/types/core.py +556 -0
  346. gatewaysdk/types/resources.py +204 -0
  347. gatewaysdk/types/tracer.py +515 -0
  348. gatewaysdk/types/tracing.py +162 -0
  349. gatewaysdk/users.py +251 -0
  350. gatewaysdk/utils/__init__.py +1 -0
  351. gatewaysdk/utils/id.py +18 -0
  352. gatewaysdk/utils/metrics.py +1025 -0
  353. gatewaysdk/utils/otel.py +550 -0
  354. gatewaysdk/utils/otlp.py +556 -0
  355. gatewaysdk/utils/redact.py +22 -0
  356. gatewaysdk/utils/server_launcher.py +1045 -0
  357. gatewaysdk/utils/system_snapshot.py +90 -0
  358. gatewaysdk/verl/__init__.py +8 -0
  359. gatewaysdk/verl/__main__.py +6 -0
  360. gatewaysdk/verl/async_server.py +46 -0
  361. gatewaysdk/verl/config.yaml +27 -0
  362. gatewaysdk/verl/daemon.py +1154 -0
  363. gatewaysdk/verl/dataset.py +44 -0
  364. gatewaysdk/verl/entrypoint.py +248 -0
  365. gatewaysdk/verl/trainer.py +549 -0
  366. gatewaysdk/world_browser.py +748 -0
  367. gatewaysdk/world_data.py +453 -0
  368. gatewaysdk/world_sessions.py +978 -0
  369. gatewaysdk/world_tasks.py +285 -0
  370. gatewaysdk/world_tools.py +115 -0
  371. gatewaysdk/worlds.py +639 -0
  372. gatewaysdk-0.3.2.dist-info/METADATA +236 -0
  373. gatewaysdk-0.3.2.dist-info/RECORD +376 -0
  374. gatewaysdk-0.3.2.dist-info/WHEEL +4 -0
  375. gatewaysdk-0.3.2.dist-info/entry_points.txt +5 -0
  376. gatewaysdk-0.3.2.dist-info/licenses/LICENSE +19 -0
@@ -0,0 +1,765 @@
1
+ {
2
+ "$schema": "https://json-schema.org/draft/2020-12/schema",
3
+ "$id": "urn:gateway:connector:kandji",
4
+ "$defs": {
5
+ "DeviceUser": {
6
+ "type": "object",
7
+ "properties": {
8
+ "email": {
9
+ "$ref": "base.json#/$defs/Email"
10
+ },
11
+ "name": {
12
+ "$ref": "base.json#/$defs/Label"
13
+ },
14
+ "id": {
15
+ "$ref": "base.json#/$defs/Id"
16
+ },
17
+ "is_archived": {
18
+ "type": "boolean"
19
+ }
20
+ },
21
+ "required": [
22
+ "email",
23
+ "name",
24
+ "id",
25
+ "is_archived"
26
+ ],
27
+ "additionalProperties": false
28
+ },
29
+ "Device": {
30
+ "type": "object",
31
+ "properties": {
32
+ "device_id": {
33
+ "$ref": "base.json#/$defs/Id"
34
+ },
35
+ "device_name": {
36
+ "$ref": "base.json#/$defs/Label"
37
+ },
38
+ "serial_number": {
39
+ "$ref": "base.json#/$defs/Label"
40
+ },
41
+ "model": {
42
+ "$ref": "base.json#/$defs/Label"
43
+ },
44
+ "platform": {
45
+ "type": "string",
46
+ "enum": [
47
+ "Mac",
48
+ "iPhone",
49
+ "iPad",
50
+ "AppleTV",
51
+ "Windows",
52
+ "Android"
53
+ ]
54
+ },
55
+ "os_version": {
56
+ "$ref": "base.json#/$defs/Label"
57
+ },
58
+ "supplemental_build_version": {
59
+ "type": "string",
60
+ "maxLength": 256
61
+ },
62
+ "supplemental_os_version_extra": {
63
+ "type": "string",
64
+ "maxLength": 256
65
+ },
66
+ "blueprint_id": {
67
+ "$ref": "base.json#/$defs/Id"
68
+ },
69
+ "blueprint_name": {
70
+ "$ref": "base.json#/$defs/Label"
71
+ },
72
+ "user": {
73
+ "anyOf": [
74
+ {
75
+ "$ref": "#/$defs/DeviceUser"
76
+ },
77
+ {
78
+ "type": "string",
79
+ "enum": [
80
+ ""
81
+ ]
82
+ }
83
+ ]
84
+ },
85
+ "last_check_in": {
86
+ "anyOf": [
87
+ {
88
+ "$ref": "base.json#/$defs/Timestamp"
89
+ },
90
+ {
91
+ "type": "string",
92
+ "enum": [
93
+ ""
94
+ ]
95
+ }
96
+ ]
97
+ },
98
+ "agent_installed": {
99
+ "type": "boolean"
100
+ },
101
+ "agent_version": {
102
+ "type": "string",
103
+ "maxLength": 256
104
+ },
105
+ "mdm_enabled": {
106
+ "type": "boolean"
107
+ },
108
+ "is_missing": {
109
+ "type": "boolean"
110
+ },
111
+ "is_removed": {
112
+ "type": "boolean"
113
+ },
114
+ "asset_tag": {
115
+ "type": "string",
116
+ "maxLength": 256
117
+ },
118
+ "first_enrollment": {
119
+ "type": "string",
120
+ "maxLength": 256
121
+ },
122
+ "last_enrollment": {
123
+ "type": "string",
124
+ "maxLength": 256
125
+ },
126
+ "lost_mode_status": {
127
+ "type": "string",
128
+ "maxLength": 256
129
+ },
130
+ "tags": {
131
+ "type": "array",
132
+ "items": {
133
+ "$ref": "base.json#/$defs/Label"
134
+ }
135
+ }
136
+ },
137
+ "required": [
138
+ "device_id",
139
+ "device_name",
140
+ "serial_number",
141
+ "model",
142
+ "platform",
143
+ "os_version",
144
+ "supplemental_build_version",
145
+ "supplemental_os_version_extra",
146
+ "blueprint_id",
147
+ "blueprint_name",
148
+ "user",
149
+ "last_check_in",
150
+ "agent_installed",
151
+ "agent_version",
152
+ "mdm_enabled",
153
+ "is_missing",
154
+ "is_removed",
155
+ "asset_tag",
156
+ "first_enrollment",
157
+ "last_enrollment",
158
+ "tags"
159
+ ],
160
+ "additionalProperties": false
161
+ },
162
+ "Blueprint": {
163
+ "type": "object",
164
+ "properties": {
165
+ "id": {
166
+ "$ref": "base.json#/$defs/Id"
167
+ },
168
+ "name": {
169
+ "$ref": "base.json#/$defs/Label"
170
+ },
171
+ "icon": {
172
+ "$ref": "base.json#/$defs/Label"
173
+ },
174
+ "color": {
175
+ "$ref": "base.json#/$defs/Label"
176
+ },
177
+ "type": {
178
+ "type": "string",
179
+ "enum": [
180
+ "classic",
181
+ "map"
182
+ ]
183
+ },
184
+ "description": {
185
+ "$ref": "base.json#/$defs/Text"
186
+ },
187
+ "params": {
188
+ "type": "object",
189
+ "properties": {},
190
+ "additionalProperties": true
191
+ },
192
+ "computers_count": {
193
+ "type": "integer",
194
+ "minimum": 0
195
+ },
196
+ "enrollment_code": {
197
+ "type": "object",
198
+ "properties": {
199
+ "code": {
200
+ "type": "string",
201
+ "maxLength": 256
202
+ },
203
+ "is_active": {
204
+ "type": "boolean"
205
+ }
206
+ },
207
+ "required": [
208
+ "code",
209
+ "is_active"
210
+ ],
211
+ "additionalProperties": false
212
+ }
213
+ },
214
+ "required": [
215
+ "id",
216
+ "name",
217
+ "icon",
218
+ "color",
219
+ "type",
220
+ "description",
221
+ "params",
222
+ "computers_count",
223
+ "enrollment_code"
224
+ ],
225
+ "additionalProperties": false
226
+ },
227
+ "LibraryItem": {
228
+ "type": "object",
229
+ "properties": {
230
+ "id": {
231
+ "$ref": "base.json#/$defs/Id"
232
+ },
233
+ "name": {
234
+ "$ref": "base.json#/$defs/Label"
235
+ },
236
+ "type": {
237
+ "$ref": "base.json#/$defs/Label"
238
+ },
239
+ "status": {
240
+ "type": "string",
241
+ "enum": [
242
+ "AVAILABLE",
243
+ "CACHED",
244
+ "CHANGE_PENDING",
245
+ "DOWNLOADING",
246
+ "ERROR",
247
+ "EXCLUDED",
248
+ "INCOMPATIBLE",
249
+ "INSTALLING",
250
+ "PASS",
251
+ "PENDING",
252
+ "failed",
253
+ "pending",
254
+ "success"
255
+ ]
256
+ },
257
+ "reported_at": {
258
+ "anyOf": [
259
+ {
260
+ "$ref": "base.json#/$defs/Timestamp"
261
+ },
262
+ {
263
+ "type": "null"
264
+ }
265
+ ]
266
+ },
267
+ "log": {
268
+ "type": [
269
+ "string",
270
+ "null"
271
+ ],
272
+ "maxLength": 40000
273
+ },
274
+ "last_audit_run": {
275
+ "anyOf": [
276
+ {
277
+ "$ref": "base.json#/$defs/Timestamp"
278
+ },
279
+ {
280
+ "type": "null"
281
+ }
282
+ ]
283
+ },
284
+ "last_audit_log": {
285
+ "type": [
286
+ "string",
287
+ "null"
288
+ ],
289
+ "maxLength": 40000
290
+ },
291
+ "control_log": {
292
+ "type": [
293
+ "string",
294
+ "null"
295
+ ],
296
+ "maxLength": 40000
297
+ },
298
+ "control_reported_at": {
299
+ "anyOf": [
300
+ {
301
+ "$ref": "base.json#/$defs/Timestamp"
302
+ },
303
+ {
304
+ "type": "null"
305
+ }
306
+ ]
307
+ },
308
+ "rules_present": {
309
+ "type": "boolean"
310
+ }
311
+ },
312
+ "required": [
313
+ "id",
314
+ "name"
315
+ ],
316
+ "additionalProperties": false
317
+ },
318
+ "BlueprintItem": {
319
+ "type": "object",
320
+ "properties": {
321
+ "blueprint_id": {
322
+ "$ref": "base.json#/$defs/Id"
323
+ },
324
+ "library_item_id": {
325
+ "$ref": "base.json#/$defs/Id"
326
+ },
327
+ "assignment_node_id": {
328
+ "$ref": "base.json#/$defs/Id"
329
+ }
330
+ },
331
+ "required": [
332
+ "blueprint_id",
333
+ "library_item_id"
334
+ ],
335
+ "additionalProperties": false
336
+ },
337
+ "User": {
338
+ "type": "object",
339
+ "properties": {
340
+ "id": {
341
+ "$ref": "base.json#/$defs/Id"
342
+ },
343
+ "email": {
344
+ "$ref": "base.json#/$defs/Email"
345
+ },
346
+ "name": {
347
+ "$ref": "base.json#/$defs/Label"
348
+ },
349
+ "active": {
350
+ "type": "boolean"
351
+ },
352
+ "archived": {
353
+ "type": "boolean"
354
+ },
355
+ "created_at": {
356
+ "$ref": "base.json#/$defs/Timestamp"
357
+ },
358
+ "department": {
359
+ "type": [
360
+ "string",
361
+ "null"
362
+ ],
363
+ "maxLength": 256
364
+ },
365
+ "deprecated_user_id": {
366
+ "$ref": "base.json#/$defs/Id"
367
+ },
368
+ "integration": {
369
+ "type": "object",
370
+ "properties": {
371
+ "id": {
372
+ "type": "integer"
373
+ },
374
+ "name": {
375
+ "$ref": "base.json#/$defs/Label"
376
+ },
377
+ "uuid": {
378
+ "$ref": "base.json#/$defs/Id"
379
+ },
380
+ "type": {
381
+ "$ref": "base.json#/$defs/Label"
382
+ }
383
+ },
384
+ "required": [
385
+ "id",
386
+ "name",
387
+ "uuid",
388
+ "type"
389
+ ],
390
+ "additionalProperties": false
391
+ },
392
+ "job_title": {
393
+ "type": [
394
+ "string",
395
+ "null"
396
+ ],
397
+ "maxLength": 256
398
+ },
399
+ "device_count": {
400
+ "type": "integer",
401
+ "minimum": 0
402
+ },
403
+ "updated_at": {
404
+ "$ref": "base.json#/$defs/Timestamp"
405
+ }
406
+ },
407
+ "required": [
408
+ "id",
409
+ "email",
410
+ "name",
411
+ "active",
412
+ "archived",
413
+ "created_at",
414
+ "department",
415
+ "deprecated_user_id",
416
+ "integration",
417
+ "job_title",
418
+ "device_count",
419
+ "updated_at"
420
+ ],
421
+ "additionalProperties": false
422
+ },
423
+ "Tag": {
424
+ "type": "object",
425
+ "properties": {
426
+ "id": {
427
+ "$ref": "base.json#/$defs/Id"
428
+ },
429
+ "name": {
430
+ "$ref": "base.json#/$defs/Label"
431
+ }
432
+ },
433
+ "required": [
434
+ "id",
435
+ "name"
436
+ ],
437
+ "additionalProperties": false
438
+ },
439
+ "DeviceTag": {
440
+ "type": "object",
441
+ "properties": {
442
+ "device_id": {
443
+ "$ref": "base.json#/$defs/Id"
444
+ },
445
+ "tag_id": {
446
+ "$ref": "base.json#/$defs/Id"
447
+ }
448
+ },
449
+ "required": [
450
+ "device_id",
451
+ "tag_id"
452
+ ],
453
+ "additionalProperties": false
454
+ },
455
+ "DeviceApp": {
456
+ "type": "object",
457
+ "properties": {
458
+ "device_id": {
459
+ "$ref": "base.json#/$defs/Id"
460
+ },
461
+ "bundle_id": {
462
+ "$ref": "base.json#/$defs/Label"
463
+ },
464
+ "app_id": {
465
+ "$ref": "base.json#/$defs/Id"
466
+ },
467
+ "app_name": {
468
+ "$ref": "base.json#/$defs/Label"
469
+ },
470
+ "version": {
471
+ "type": "string",
472
+ "maxLength": 256
473
+ },
474
+ "bundle_size": {
475
+ "type": "string",
476
+ "maxLength": 256
477
+ },
478
+ "app_store_vendable": {
479
+ "type": "string",
480
+ "maxLength": 256
481
+ },
482
+ "device_based_vpp": {
483
+ "type": "string",
484
+ "maxLength": 256
485
+ },
486
+ "source": {
487
+ "type": "string",
488
+ "maxLength": 256
489
+ },
490
+ "process": {
491
+ "type": "string",
492
+ "maxLength": 256
493
+ },
494
+ "signature": {
495
+ "$ref": "base.json#/$defs/Text"
496
+ },
497
+ "creation_date": {
498
+ "anyOf": [
499
+ {
500
+ "$ref": "base.json#/$defs/Timestamp"
501
+ },
502
+ {
503
+ "type": "string",
504
+ "enum": [
505
+ ""
506
+ ]
507
+ }
508
+ ]
509
+ },
510
+ "modification_date": {
511
+ "anyOf": [
512
+ {
513
+ "$ref": "base.json#/$defs/Timestamp"
514
+ },
515
+ {
516
+ "type": "string",
517
+ "enum": [
518
+ ""
519
+ ]
520
+ }
521
+ ]
522
+ },
523
+ "path": {
524
+ "type": "string",
525
+ "maxLength": 40000
526
+ }
527
+ },
528
+ "required": [
529
+ "device_id",
530
+ "bundle_id",
531
+ "app_id",
532
+ "app_name",
533
+ "version",
534
+ "bundle_size",
535
+ "app_store_vendable",
536
+ "device_based_vpp",
537
+ "source",
538
+ "process",
539
+ "signature",
540
+ "creation_date",
541
+ "modification_date",
542
+ "path"
543
+ ],
544
+ "additionalProperties": false
545
+ }
546
+ },
547
+ "x-gateway": {
548
+ "version": 1,
549
+ "domain_version": "1.1.0",
550
+ "package": {
551
+ "id": "gateway-connector-kandji",
552
+ "version": "1.0.1"
553
+ },
554
+ "entities": {
555
+ "kandji_devices": {
556
+ "schema": {
557
+ "$ref": "#/$defs/Device"
558
+ },
559
+ "primary_key": [
560
+ "device_id"
561
+ ],
562
+ "unique": [],
563
+ "relationships": [
564
+ {
565
+ "fields": [
566
+ "blueprint_id"
567
+ ],
568
+ "target": "kandji_blueprints",
569
+ "on_delete": "restrict"
570
+ },
571
+ {
572
+ "fields": [
573
+ "/user/id"
574
+ ],
575
+ "target": "kandji_users",
576
+ "on_delete": "restrict"
577
+ }
578
+ ]
579
+ },
580
+ "kandji_blueprints": {
581
+ "schema": {
582
+ "$ref": "#/$defs/Blueprint"
583
+ },
584
+ "primary_key": [
585
+ "id"
586
+ ],
587
+ "unique": [
588
+ [
589
+ "name"
590
+ ]
591
+ ],
592
+ "relationships": []
593
+ },
594
+ "kandji_library_items": {
595
+ "schema": {
596
+ "$ref": "#/$defs/LibraryItem"
597
+ },
598
+ "primary_key": [
599
+ "id"
600
+ ],
601
+ "unique": [],
602
+ "relationships": []
603
+ },
604
+ "kandji_blueprint_items": {
605
+ "schema": {
606
+ "$ref": "#/$defs/BlueprintItem"
607
+ },
608
+ "primary_key": [
609
+ "blueprint_id",
610
+ "library_item_id"
611
+ ],
612
+ "unique": [],
613
+ "relationships": [
614
+ {
615
+ "fields": [
616
+ "blueprint_id"
617
+ ],
618
+ "target": "kandji_blueprints",
619
+ "on_delete": "restrict"
620
+ },
621
+ {
622
+ "fields": [
623
+ "library_item_id"
624
+ ],
625
+ "target": "kandji_library_items",
626
+ "on_delete": "restrict"
627
+ }
628
+ ]
629
+ },
630
+ "kandji_users": {
631
+ "schema": {
632
+ "$ref": "#/$defs/User"
633
+ },
634
+ "primary_key": [
635
+ "id"
636
+ ],
637
+ "unique": [],
638
+ "relationships": []
639
+ },
640
+ "kandji_tags": {
641
+ "schema": {
642
+ "$ref": "#/$defs/Tag"
643
+ },
644
+ "primary_key": [
645
+ "id"
646
+ ],
647
+ "unique": [],
648
+ "relationships": []
649
+ },
650
+ "kandji_device_tags": {
651
+ "schema": {
652
+ "$ref": "#/$defs/DeviceTag"
653
+ },
654
+ "primary_key": [
655
+ "device_id",
656
+ "tag_id"
657
+ ],
658
+ "unique": [],
659
+ "relationships": [
660
+ {
661
+ "fields": [
662
+ "device_id"
663
+ ],
664
+ "target": "kandji_devices",
665
+ "on_delete": "restrict"
666
+ },
667
+ {
668
+ "fields": [
669
+ "tag_id"
670
+ ],
671
+ "target": "kandji_tags",
672
+ "on_delete": "restrict"
673
+ }
674
+ ]
675
+ },
676
+ "kandji_device_apps": {
677
+ "schema": {
678
+ "$ref": "#/$defs/DeviceApp"
679
+ },
680
+ "primary_key": [
681
+ "device_id",
682
+ "bundle_id"
683
+ ],
684
+ "unique": [],
685
+ "relationships": [
686
+ {
687
+ "fields": [
688
+ "device_id"
689
+ ],
690
+ "target": "kandji_devices",
691
+ "on_delete": "restrict"
692
+ }
693
+ ]
694
+ }
695
+ },
696
+ "tools": {
697
+ "list_devices": {
698
+ "input": {
699
+ "type": "object",
700
+ "properties": {
701
+ "blueprint_id": {
702
+ "$ref": "base.json#/$defs/Id"
703
+ }
704
+ },
705
+ "required": [],
706
+ "additionalProperties": false
707
+ },
708
+ "output": {
709
+ "type": "array",
710
+ "items": {
711
+ "$ref": "#/$defs/Device"
712
+ }
713
+ },
714
+ "description": "Devices in the tenant, optionally filtered to one blueprint."
715
+ },
716
+ "get_device": {
717
+ "input": {
718
+ "type": "object",
719
+ "properties": {
720
+ "device_id": {
721
+ "$ref": "base.json#/$defs/Id"
722
+ }
723
+ },
724
+ "required": [
725
+ "device_id"
726
+ ],
727
+ "additionalProperties": false
728
+ },
729
+ "output": {
730
+ "anyOf": [
731
+ {
732
+ "$ref": "#/$defs/Device"
733
+ },
734
+ {
735
+ "type": "null"
736
+ }
737
+ ]
738
+ },
739
+ "description": "One device by device_id, or null when absent."
740
+ },
741
+ "assign_blueprint": {
742
+ "input": {
743
+ "type": "object",
744
+ "properties": {
745
+ "device_id": {
746
+ "$ref": "base.json#/$defs/Id"
747
+ },
748
+ "blueprint_id": {
749
+ "$ref": "base.json#/$defs/Id"
750
+ }
751
+ },
752
+ "required": [
753
+ "device_id",
754
+ "blueprint_id"
755
+ ],
756
+ "additionalProperties": false
757
+ },
758
+ "output": {
759
+ "$ref": "#/$defs/Device"
760
+ },
761
+ "description": "Assign an existing blueprint to a device and return the device."
762
+ }
763
+ }
764
+ }
765
+ }