ummaya 0.2.3 → 0.2.5

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 (534) hide show
  1. package/README.md +17 -3
  2. package/bin/ummaya +10 -1
  3. package/npm-shrinkwrap.json +253 -2
  4. package/package.json +5 -1
  5. package/prompts/manifest.yaml +2 -2
  6. package/prompts/session_guidance_v1.md +3 -1
  7. package/prompts/system_v1.md +9 -7
  8. package/pyproject.toml +26 -7
  9. package/specs/2803-document-production-hardening/contracts/document-tools.schema.json +1043 -0
  10. package/src/ummaya/_canonical/__init__.py +2 -0
  11. package/src/ummaya/context/builder.py +17 -11
  12. package/src/ummaya/engine/engine.py +30 -113
  13. package/src/ummaya/engine/query.py +20 -0
  14. package/src/ummaya/evidence/__init__.py +44 -0
  15. package/src/ummaya/evidence/__main__.py +7 -0
  16. package/src/ummaya/evidence/dataset_contract.py +193 -0
  17. package/src/ummaya/evidence/document_authoring_cases.py +33 -0
  18. package/src/ummaya/evidence/document_harness.py +313 -0
  19. package/src/ummaya/evidence/document_viewer_ux.py +391 -0
  20. package/src/ummaya/evidence/gates.py +70 -0
  21. package/src/ummaya/evidence/json_types.py +20 -0
  22. package/src/ummaya/evidence/models.py +145 -0
  23. package/src/ummaya/evidence/output_payload.py +89 -0
  24. package/src/ummaya/evidence/payload_documents.py +233 -0
  25. package/src/ummaya/evidence/route_contracts.py +224 -0
  26. package/src/ummaya/evidence/route_helpers.py +150 -0
  27. package/src/ummaya/evidence/runner.py +177 -0
  28. package/src/ummaya/evidence/source_provenance.py +246 -0
  29. package/src/ummaya/evidence/source_provenance_redaction.py +176 -0
  30. package/src/ummaya/evidence/task_registry.py +264 -0
  31. package/src/ummaya/evidence/tool_layer.py +39 -0
  32. package/src/ummaya/evidence/tool_layer_models.py +151 -0
  33. package/src/ummaya/ipc/adapter_manifest_emitter.py +26 -10
  34. package/src/ummaya/ipc/document_intent_normalization.py +185 -0
  35. package/src/ummaya/ipc/frame_schema.py +52 -5
  36. package/src/ummaya/ipc/route_diagnostics.py +73 -0
  37. package/src/ummaya/ipc/stdio.py +2282 -417
  38. package/src/ummaya/llm/client.py +234 -59
  39. package/src/ummaya/llm/config.py +8 -3
  40. package/src/ummaya/llm/reasoning.py +84 -0
  41. package/src/ummaya/primitives/__init__.py +6 -2
  42. package/src/ummaya/primitives/delegation.py +1 -1
  43. package/src/ummaya/primitives/document.py +28 -0
  44. package/src/ummaya/settings.py +0 -3
  45. package/src/ummaya/tools/discovery_bridge.py +34 -2
  46. package/src/ummaya/tools/documents/__init__.py +297 -0
  47. package/src/ummaya/tools/documents/adapter_registry.py +487 -0
  48. package/src/ummaya/tools/documents/archive_container_probe.py +167 -0
  49. package/src/ummaya/tools/documents/artifact_store.py +454 -0
  50. package/src/ummaya/tools/documents/authoring.py +283 -0
  51. package/src/ummaya/tools/documents/baselines.py +114 -0
  52. package/src/ummaya/tools/documents/capability.py +331 -0
  53. package/src/ummaya/tools/documents/contracts.py +112 -0
  54. package/src/ummaya/tools/documents/conversion.py +521 -0
  55. package/src/ummaya/tools/documents/diff.py +275 -0
  56. package/src/ummaya/tools/documents/engines.py +163 -0
  57. package/src/ummaya/tools/documents/evaluation.py +291 -0
  58. package/src/ummaya/tools/documents/explicit_values.py +108 -0
  59. package/src/ummaya/tools/documents/fixtures.py +174 -0
  60. package/src/ummaya/tools/documents/format_completion_audit.py +471 -0
  61. package/src/ummaya/tools/documents/formats/__init__.py +2 -0
  62. package/src/ummaya/tools/documents/formats/archive.py +528 -0
  63. package/src/ummaya/tools/documents/formats/base.py +41 -0
  64. package/src/ummaya/tools/documents/formats/code_file.py +211 -0
  65. package/src/ummaya/tools/documents/formats/data_file.py +272 -0
  66. package/src/ummaya/tools/documents/formats/hwp.py +284 -0
  67. package/src/ummaya/tools/documents/formats/hwpx.py +1837 -0
  68. package/src/ummaya/tools/documents/formats/odf.py +435 -0
  69. package/src/ummaya/tools/documents/formats/ooxml.py +1030 -0
  70. package/src/ummaya/tools/documents/formats/passive.py +766 -0
  71. package/src/ummaya/tools/documents/formats/pdf.py +702 -0
  72. package/src/ummaya/tools/documents/formats/text_web.py +268 -0
  73. package/src/ummaya/tools/documents/hwp_conversion_probe.py +178 -0
  74. package/src/ummaya/tools/documents/hwp_direct_candidate.py +141 -0
  75. package/src/ummaya/tools/documents/inspection.py +289 -0
  76. package/src/ummaya/tools/documents/intake.py +1079 -0
  77. package/src/ummaya/tools/documents/legacy_office_promotion_probe.py +366 -0
  78. package/src/ummaya/tools/documents/models.py +1598 -0
  79. package/src/ummaya/tools/documents/odf_promotion_probe.py +167 -0
  80. package/src/ummaya/tools/documents/orchestrator.py +96 -0
  81. package/src/ummaya/tools/documents/passive_capability_probe.py +251 -0
  82. package/src/ummaya/tools/documents/patch.py +170 -0
  83. package/src/ummaya/tools/documents/pdfa_conformance.py +284 -0
  84. package/src/ummaya/tools/documents/pdfa_promotion_probe.py +198 -0
  85. package/src/ummaya/tools/documents/permissions.py +110 -0
  86. package/src/ummaya/tools/documents/planner.py +616 -0
  87. package/src/ummaya/tools/documents/registry.py +2733 -0
  88. package/src/ummaya/tools/documents/render.py +978 -0
  89. package/src/ummaya/tools/documents/render_comparison.py +113 -0
  90. package/src/ummaya/tools/documents/render_comparison_models.py +74 -0
  91. package/src/ummaya/tools/documents/render_comparison_regions.py +73 -0
  92. package/src/ummaya/tools/documents/render_comparison_style.py +161 -0
  93. package/src/ummaya/tools/documents/reread.py +157 -0
  94. package/src/ummaya/tools/documents/runtime_authoring.py +244 -0
  95. package/src/ummaya/tools/documents/runtime_authoring_bundle.py +76 -0
  96. package/src/ummaya/tools/documents/scorecard.py +184 -0
  97. package/src/ummaya/tools/documents/socratic_planner.py +193 -0
  98. package/src/ummaya/tools/documents/style.py +48 -0
  99. package/src/ummaya/tools/documents/tool_defs.py +523 -0
  100. package/src/ummaya/tools/documents/validate.py +347 -0
  101. package/src/ummaya/tools/executor.py +61 -12
  102. package/src/ummaya/tools/geocoding/kakao_client.py +1 -2
  103. package/src/ummaya/tools/kma/apihub_catalog.py +984 -1
  104. package/src/ummaya/tools/kma/apihub_structured_adapter.py +86 -6
  105. package/src/ummaya/tools/kma/apihub_url_adapter.py +593 -0
  106. package/src/ummaya/tools/kma/apihub_url_catalog.py +296 -0
  107. package/src/ummaya/tools/live_proxy.py +0 -3
  108. package/src/ummaya/tools/location_adapters.py +8 -6
  109. package/src/ummaya/tools/manifest_metadata.py +16 -3
  110. package/src/ummaya/tools/models.py +5 -1
  111. package/src/ummaya/tools/mvp_surface.py +2 -2
  112. package/src/ummaya/tools/nmc/emergency_search.py +8 -6
  113. package/src/ummaya/tools/register_all.py +17 -0
  114. package/src/ummaya/tools/registry.py +10 -1
  115. package/src/ummaya/tools/resolve_location.py +4 -4
  116. package/src/ummaya/tools/routing/__init__.py +59 -0
  117. package/src/ummaya/tools/routing/builder.py +105 -0
  118. package/src/ummaya/tools/routing/cards.py +29 -0
  119. package/src/ummaya/tools/routing/decision_service.py +534 -0
  120. package/src/ummaya/tools/routing/decision_types.py +74 -0
  121. package/src/ummaya/tools/routing/feasibility.py +122 -0
  122. package/src/ummaya/tools/routing/intent.py +17 -0
  123. package/src/ummaya/tools/routing/intent_extractor.py +207 -0
  124. package/src/ummaya/tools/routing/intent_patterns.py +160 -0
  125. package/src/ummaya/tools/routing/intent_public_data.py +150 -0
  126. package/src/ummaya/tools/routing/intent_types.py +48 -0
  127. package/src/ummaya/tools/routing/lint.py +78 -0
  128. package/src/ummaya/tools/routing/metadata.py +174 -0
  129. package/src/ummaya/tools/routing/projection.py +340 -0
  130. package/src/ummaya/tools/routing/retrieval_policy.py +629 -0
  131. package/src/ummaya/tools/routing/schema.py +81 -0
  132. package/src/ummaya/tools/routing/types.py +96 -0
  133. package/src/ummaya/tools/routing_index.py +2 -2
  134. package/src/ummaya/tools/search.py +40 -106
  135. package/src/ummaya/tools/verified_data_go_kr/_manifest.py +115 -25
  136. package/src/ummaya/tools/verified_data_go_kr/airkorea_air_quality.py +109 -4
  137. package/src/ummaya/tools/verified_data_go_kr/nmc_aed_site.py +108 -2
  138. package/src/ummaya/tools/verified_data_go_kr/pps_bid_public_info.py +174 -9
  139. package/src/ummaya/tools/verified_data_go_kr/tago_bus_arrival.py +66 -3
  140. package/src/ummaya/tools/verified_data_go_kr/tago_bus_location.py +12 -2
  141. package/src/ummaya/tools/verified_data_go_kr/tago_bus_route.py +8 -2
  142. package/src/ummaya/tools/verified_data_go_kr/tago_bus_route_station.py +114 -0
  143. package/src/ummaya/tools/verified_data_go_kr/tago_bus_station.py +14 -3
  144. package/src/ummaya/tools/verify_canonical_map.py +21 -0
  145. package/tests/fixtures/documents/public_forms/baselines.yaml +113 -0
  146. package/tui/package.json +1 -2
  147. package/tui/src/.cc-byte-identical-whitelist.yaml +266 -0
  148. package/tui/src/QueryEngine.ts +12 -4
  149. package/tui/src/bridge/inboundAttachments.ts +3 -3
  150. package/tui/src/cli/handlers/auth.ts +4 -13
  151. package/tui/src/cli/handlers/mcp.tsx +3 -3
  152. package/tui/src/cli/print.ts +69 -18
  153. package/tui/src/cli/update.ts +13 -13
  154. package/tui/src/commands/copy/index.ts +1 -1
  155. package/tui/src/commands/cost/cost.ts +2 -2
  156. package/tui/src/commands/init-verifiers.ts +5 -5
  157. package/tui/src/commands/init.ts +30 -30
  158. package/tui/src/commands/insights.ts +44 -44
  159. package/tui/src/commands/install-github-app/install-github-app.tsx +2 -2
  160. package/tui/src/commands/install-github-app/setupGitHubActions.ts +3 -3
  161. package/tui/src/commands/install-github-app/types.ts +8 -30
  162. package/tui/src/commands/install.tsx +5 -5
  163. package/tui/src/commands/mcp/addCommand.ts +5 -5
  164. package/tui/src/commands/mcp/xaaIdpCommand.ts +2 -2
  165. package/tui/src/commands/plugin/ManageMarketplaces.tsx +2 -2
  166. package/tui/src/commands/plugin/types.ts +6 -28
  167. package/tui/src/commands/plugin/unifiedTypes.ts +4 -26
  168. package/tui/src/commands/reasoning/index.ts +13 -0
  169. package/tui/src/commands/reasoning/reasoning.tsx +177 -0
  170. package/tui/src/commands/rename/generateSessionName.ts +1 -1
  171. package/tui/src/commands/thinkback/thinkback.tsx +3 -3
  172. package/tui/src/commands.ts +2 -0
  173. package/tui/src/components/Feedback.tsx +1 -1
  174. package/tui/src/components/LogoV2/EmergencyTip.tsx +11 -2
  175. package/tui/src/components/LogoV2/WelcomeV2.tsx +1 -3
  176. package/tui/src/components/Messages.tsx +2 -1
  177. package/tui/src/components/ScrollKeybindingHandler.tsx +6 -6
  178. package/tui/src/components/Spinner/types.ts +6 -28
  179. package/tui/src/components/Spinner.tsx +2 -2
  180. package/tui/src/components/agents/generateAgent.ts +1 -1
  181. package/tui/src/components/agents/new-agent-creation/types.ts +4 -26
  182. package/tui/src/components/config/EnvSecretIsolatedEditor.tsx +1 -1
  183. package/tui/src/components/design-system/LoadingState.tsx +2 -2
  184. package/tui/src/components/mcp/types.ts +16 -38
  185. package/tui/src/components/messages/AssistantToolUseMessage.tsx +3 -2
  186. package/tui/src/components/messages/UserCrossSessionMessage.ts +16 -4
  187. package/tui/src/components/messages/UserForkBoilerplateMessage.ts +16 -4
  188. package/tui/src/components/messages/UserGitHubWebhookMessage.ts +16 -4
  189. package/tui/src/components/messages/UserToolResultMessage/utils.tsx +3 -2
  190. package/tui/src/components/permissions/MonitorPermissionRequest/MonitorPermissionRequest.ts +9 -4
  191. package/tui/src/components/permissions/ReviewArtifactPermissionRequest/ReviewArtifactPermissionRequest.ts +9 -4
  192. package/tui/src/components/primitive/DocumentSocraticReviewBlock.tsx +129 -0
  193. package/tui/src/components/primitive/DocumentToolResultCard.tsx +224 -0
  194. package/tui/src/components/primitive/documentSocraticReview.ts +215 -0
  195. package/tui/src/components/primitive/index.tsx +43 -1
  196. package/tui/src/components/primitive/types.ts +137 -0
  197. package/tui/src/components/ui/option.ts +4 -26
  198. package/tui/src/constants/common.ts +0 -2
  199. package/tui/src/constants/prompts.ts +4 -3
  200. package/tui/src/constants/querySource.ts +4 -26
  201. package/tui/src/entrypoints/sdk/controlTypes.ts +26 -48
  202. package/tui/src/entrypoints/sdk/coreTypes.generated.ts +3 -25
  203. package/tui/src/entrypoints/sdk/runtimeTypes.ts +38 -60
  204. package/tui/src/entrypoints/sdk/sdkUtilityTypes.ts +4 -26
  205. package/tui/src/entrypoints/sdk/settingsTypes.generated.ts +3 -25
  206. package/tui/src/entrypoints/sdk/toolTypes.ts +3 -25
  207. package/tui/src/hooks/toolPermission/handlers/interactiveHandler.ts +10 -0
  208. package/tui/src/hooks/useApiKeyVerification.ts +1 -1
  209. package/tui/src/hooks/useVirtualScroll.ts +1 -1
  210. package/tui/src/ink/ink.tsx +33 -14
  211. package/tui/src/ink/reconciler.ts +2 -3
  212. package/tui/src/ink/render-to-screen.ts +30 -10
  213. package/tui/src/ipc/bridge.ts +62 -15
  214. package/tui/src/ipc/bridgeSingleton.ts +5 -1
  215. package/tui/src/ipc/codec.ts +29 -3
  216. package/tui/src/ipc/frames.generated.ts +407 -312
  217. package/tui/src/ipc/llmClient.ts +279 -76
  218. package/tui/src/ipc/llmTypes.ts +16 -1
  219. package/tui/src/ipc/schema/frame.schema.json +1 -3475
  220. package/tui/src/keybindings/defaultBindings.ts +4 -0
  221. package/tui/src/main.tsx +32 -11
  222. package/tui/src/native-ts/file-index/index.ts +33 -3
  223. package/tui/src/observability/surface.ts +2 -2
  224. package/tui/src/probes/toolRegistryProbe.tsx +3 -1
  225. package/tui/src/projectOnboardingState.ts +7 -6
  226. package/tui/src/query/chatMessageTypes.ts +18 -0
  227. package/tui/src/query/chatMessagesBuilder.ts +1 -1
  228. package/tui/src/query/deps.ts +1 -1
  229. package/tui/src/query/messageGuards.ts +106 -0
  230. package/tui/src/query/publicDataTerminalRepair.ts +384 -0
  231. package/tui/src/query/run.ts +1075 -0
  232. package/tui/src/query/supportBoundary.ts +168 -0
  233. package/tui/src/query/toolResultErrors.ts +103 -0
  234. package/tui/src/query/toolRunner.ts +687 -0
  235. package/tui/src/query/unavailableToolRepair.ts +118 -0
  236. package/tui/src/query.ts +9 -1721
  237. package/tui/src/screens/REPL.tsx +42 -31
  238. package/tui/src/services/api/adapterManifest.ts +4 -0
  239. package/tui/src/services/api/backendChat/events.ts +117 -0
  240. package/tui/src/services/api/backendChat/finalMessage.ts +40 -0
  241. package/tui/src/services/api/backendChat/frame.ts +9 -0
  242. package/tui/src/services/api/backendChat/streaming.ts +430 -0
  243. package/tui/src/services/api/backendChat/types.ts +62 -0
  244. package/tui/src/services/api/backendChat.ts +1 -0
  245. package/tui/src/services/api/client.ts +98 -14
  246. package/tui/src/services/api/errorUtils.ts +5 -5
  247. package/tui/src/services/api/errors.ts +1 -1
  248. package/tui/src/services/api/logging.ts +1 -1
  249. package/tui/src/services/api/ummaya/evidence.ts +194 -0
  250. package/tui/src/services/api/ummaya/messages.ts +255 -0
  251. package/tui/src/services/api/ummaya/nonStreaming.ts +66 -0
  252. package/tui/src/services/api/ummaya/provider.ts +200 -0
  253. package/tui/src/services/api/ummaya/reasoning.ts +24 -0
  254. package/tui/src/services/api/ummaya/request.ts +200 -0
  255. package/tui/src/services/api/ummaya/selectionContext.ts +240 -0
  256. package/tui/src/services/api/ummaya/streaming.ts +365 -0
  257. package/tui/src/services/api/ummaya/streamingPayload.ts +129 -0
  258. package/tui/src/services/api/ummaya/streamingReader.ts +40 -0
  259. package/tui/src/services/api/ummaya/toolSelection.ts +217 -0
  260. package/tui/src/services/api/ummaya/types.ts +110 -0
  261. package/tui/src/services/api/ummaya/usage.ts +30 -0
  262. package/tui/src/services/api/ummaya.ts +26 -364
  263. package/tui/src/services/api/withRetry.ts +1 -1
  264. package/tui/src/services/awaySummary.ts +2 -2
  265. package/tui/src/services/claudeAiLimits.ts +1 -1
  266. package/tui/src/services/compact/autoCompact.ts +1 -1
  267. package/tui/src/services/compact/compact.ts +1 -1
  268. package/tui/src/services/lsp/types.ts +8 -30
  269. package/tui/src/services/tips/types.ts +6 -28
  270. package/tui/src/services/tokenEstimation.ts +1 -1
  271. package/tui/src/services/toolRegistry/bootGuard.ts +5 -5
  272. package/tui/src/services/toolUseSummary/toolUseSummaryGenerator.ts +1 -1
  273. package/tui/src/services/tools/toolExecution.ts +94 -1
  274. package/tui/src/skills/bundled/stuck.ts +12 -12
  275. package/tui/src/state/AppStateStore.ts +7 -0
  276. package/tui/src/store/pendingPermissionSlot.ts +1 -1
  277. package/tui/src/store/session-store.ts +10 -36
  278. package/tui/src/stubs/any-stub.ts +15 -10
  279. package/tui/src/stubs/color-diff-napi.ts +37 -23
  280. package/tui/src/stubs/globals.d.ts +3 -3
  281. package/tui/src/stubs/macro-preload.ts +23 -12
  282. package/tui/src/tools/AdapterTool/AdapterTool.ts +1239 -163
  283. package/tui/src/tools/AdapterTool/routeDiagnostics.ts +75 -0
  284. package/tui/src/tools/AgentTool/AgentTool.tsx +84 -1371
  285. package/tui/src/tools/AgentTool/agentToolHandoff.ts +114 -0
  286. package/tui/src/tools/AgentTool/agentToolPartialResult.ts +16 -0
  287. package/tui/src/tools/AgentTool/agentToolProgress.ts +32 -0
  288. package/tui/src/tools/AgentTool/agentToolResolver.ts +161 -0
  289. package/tui/src/tools/AgentTool/agentToolResult.ts +163 -0
  290. package/tui/src/tools/AgentTool/agentToolUtils.ts +14 -686
  291. package/tui/src/tools/AgentTool/asyncAgentLifecycle.ts +208 -0
  292. package/tui/src/tools/AgentTool/asyncLifecycle.ts +153 -0
  293. package/tui/src/tools/AgentTool/backgroundedCompletion.ts +126 -0
  294. package/tui/src/tools/AgentTool/backgroundedLifecycle.ts +174 -0
  295. package/tui/src/tools/AgentTool/foregroundBackground.ts +83 -0
  296. package/tui/src/tools/AgentTool/foregroundDrain.tsx +133 -0
  297. package/tui/src/tools/AgentTool/foregroundFinalize.ts +98 -0
  298. package/tui/src/tools/AgentTool/foregroundLifecycle.tsx +237 -0
  299. package/tui/src/tools/AgentTool/foregroundProgress.tsx +169 -0
  300. package/tui/src/tools/AgentTool/foregroundTask.ts +89 -0
  301. package/tui/src/tools/AgentTool/forkSubagent.ts +1 -12
  302. package/tui/src/tools/AgentTool/forkSubagentGate.ts +34 -0
  303. package/tui/src/tools/AgentTool/launchRouting.ts +203 -0
  304. package/tui/src/tools/AgentTool/lifecycle.ts +244 -0
  305. package/tui/src/tools/AgentTool/mcpRouting.ts +73 -0
  306. package/tui/src/tools/AgentTool/orchestrationSupport.ts +70 -0
  307. package/tui/src/tools/AgentTool/permissions.ts +39 -0
  308. package/tui/src/tools/AgentTool/promptSetup.ts +181 -0
  309. package/tui/src/tools/AgentTool/remoteRouting.ts +62 -0
  310. package/tui/src/tools/AgentTool/resultMapping.ts +116 -0
  311. package/tui/src/tools/AgentTool/resumeAgent.ts +39 -107
  312. package/tui/src/tools/AgentTool/resumeAgentHelpers.ts +140 -0
  313. package/tui/src/tools/AgentTool/runAgent.ts +1 -1
  314. package/tui/src/tools/AgentTool/runtimeConfig.ts +57 -0
  315. package/tui/src/tools/AgentTool/schemas.ts +196 -0
  316. package/tui/src/tools/AgentTool/sourceVerificationPropagation.ts +263 -0
  317. package/tui/src/tools/AgentTool/worktreeLifecycle.ts +105 -0
  318. package/tui/src/tools/AskUserQuestionTool/AskUserQuestionTool.tsx +174 -202
  319. package/tui/src/tools/BashTool/BashTool.tsx +71 -1072
  320. package/tui/src/tools/BashTool/bashCommandHelpers.ts +12 -12
  321. package/tui/src/tools/BashTool/bashPermissions/astPreflight.ts +173 -0
  322. package/tui/src/tools/BashTool/bashPermissions/classifierChecks.ts +199 -0
  323. package/tui/src/tools/BashTool/bashPermissions/compoundGuards.ts +53 -0
  324. package/tui/src/tools/BashTool/bashPermissions/constants.ts +99 -0
  325. package/tui/src/tools/BashTool/bashPermissions/index.ts +38 -0
  326. package/tui/src/tools/BashTool/bashPermissions/legacyMisparsing.ts +62 -0
  327. package/tui/src/tools/BashTool/bashPermissions/main.ts +135 -0
  328. package/tui/src/tools/BashTool/bashPermissions/normalizedCommands.ts +33 -0
  329. package/tui/src/tools/BashTool/bashPermissions/operatorFlow.ts +98 -0
  330. package/tui/src/tools/BashTool/bashPermissions/permissionChecks.ts +200 -0
  331. package/tui/src/tools/BashTool/bashPermissions/prefixSuggestions.ts +88 -0
  332. package/tui/src/tools/BashTool/bashPermissions/promptClassifierRules.ts +125 -0
  333. package/tui/src/tools/BashTool/bashPermissions/ruleDelegates.ts +19 -0
  334. package/tui/src/tools/BashTool/bashPermissions/ruleMatching.ts +145 -0
  335. package/tui/src/tools/BashTool/bashPermissions/sandboxAutoAllow.ts +75 -0
  336. package/tui/src/tools/BashTool/bashPermissions/subcommandFlow.ts +205 -0
  337. package/tui/src/tools/BashTool/bashPermissions/subcommandGuards.ts +73 -0
  338. package/tui/src/tools/BashTool/bashPermissions/subcommandResultHelpers.ts +116 -0
  339. package/tui/src/tools/BashTool/bashPermissions/types.ts +26 -0
  340. package/tui/src/tools/BashTool/bashPermissions/wrapperStripping.ts +139 -0
  341. package/tui/src/tools/BashTool/bashPermissions.ts +26 -2621
  342. package/tui/src/tools/BashTool/call.ts +202 -0
  343. package/tui/src/tools/BashTool/callLoader.ts +35 -0
  344. package/tui/src/tools/BashTool/commandClassification.ts +151 -0
  345. package/tui/src/tools/BashTool/commandClassificationLoader.ts +40 -0
  346. package/tui/src/tools/BashTool/cwdReset.ts +33 -0
  347. package/tui/src/tools/BashTool/lineTruncation.ts +11 -0
  348. package/tui/src/tools/BashTool/modeValidation.ts +13 -1
  349. package/tui/src/tools/BashTool/outputPersistence.ts +42 -0
  350. package/tui/src/tools/BashTool/permissionClassification.ts +66 -0
  351. package/tui/src/tools/BashTool/permissionLoader.ts +44 -0
  352. package/tui/src/tools/BashTool/resultLoader.ts +29 -0
  353. package/tui/src/tools/BashTool/resultMapping.ts +83 -0
  354. package/tui/src/tools/BashTool/sandboxPolicy.ts +79 -0
  355. package/tui/src/tools/BashTool/schemas.ts +65 -0
  356. package/tui/src/tools/BashTool/sedEditExecution.ts +59 -0
  357. package/tui/src/tools/BashTool/shellExecution.tsx +245 -0
  358. package/tui/src/tools/BashTool/shellOutputUtils.ts +85 -0
  359. package/tui/src/tools/BashTool/shellPermissionGauntlet.ts +97 -0
  360. package/tui/src/tools/BashTool/uiLoader.ts +37 -0
  361. package/tui/src/tools/BriefTool/upload.ts +1 -1
  362. package/tui/src/tools/CalculatorTool/parser.ts +2 -2
  363. package/tui/src/tools/DocumentPrimitive/DocumentPrimitive.ts +262 -0
  364. package/tui/src/tools/DocumentPrimitive/dispatchNormalization.ts +270 -0
  365. package/tui/src/tools/DocumentPrimitive/documentDestinationPath.ts +18 -0
  366. package/tui/src/tools/DocumentPrimitive/documentMutationGuard.ts +22 -0
  367. package/tui/src/tools/DocumentPrimitive/documentPatchNormalization.ts +248 -0
  368. package/tui/src/tools/DocumentPrimitive/documentSourceVerification.ts +245 -0
  369. package/tui/src/tools/DocumentPrimitive/documentSourceVerificationFields.ts +103 -0
  370. package/tui/src/tools/DocumentPrimitive/modelVisibleOutput.ts +40 -0
  371. package/tui/src/tools/DocumentPrimitive/prompt.ts +35 -0
  372. package/tui/src/tools/FileEditTool/FileEditTool.ts +9 -507
  373. package/tui/src/tools/FileEditTool/call.ts +228 -0
  374. package/tui/src/tools/FileEditTool/validateInput.ts +196 -0
  375. package/tui/src/tools/FileReadTool/imageProcessor.ts +13 -0
  376. package/tui/src/tools/FileWriteTool/FileWriteTool.ts +7 -300
  377. package/tui/src/tools/FileWriteTool/call.ts +223 -0
  378. package/tui/src/tools/FileWriteTool/validateInput.ts +80 -0
  379. package/tui/src/tools/ListMcpResourcesTool/ListMcpResourcesTool.ts +19 -3
  380. package/tui/src/tools/LookupPrimitive/LookupPrimitive.ts +48 -29
  381. package/tui/src/tools/LookupPrimitive/prompt.ts +6 -7
  382. package/tui/src/tools/MCPTool/trustPolicy.ts +118 -0
  383. package/tui/src/tools/McpAuthTool/McpAuthTool.ts +21 -3
  384. package/tui/src/tools/NotebookEditTool/NotebookEditTool.ts +7 -326
  385. package/tui/src/tools/NotebookEditTool/call.ts +254 -0
  386. package/tui/src/tools/NotebookEditTool/notebookModel.ts +51 -0
  387. package/tui/src/tools/NotebookEditTool/validateInput.ts +142 -0
  388. package/tui/src/tools/PowerShellTool/PowerShellTool.tsx +46 -937
  389. package/tui/src/tools/PowerShellTool/acceptEditsCommandValidation.ts +162 -0
  390. package/tui/src/tools/PowerShellTool/call.ts +179 -0
  391. package/tui/src/tools/PowerShellTool/callLoader.ts +37 -0
  392. package/tui/src/tools/PowerShellTool/commandClassification.ts +86 -0
  393. package/tui/src/tools/PowerShellTool/modeValidation.ts +25 -332
  394. package/tui/src/tools/PowerShellTool/outputPersistence.ts +42 -0
  395. package/tui/src/tools/PowerShellTool/permissionClassification.ts +28 -0
  396. package/tui/src/tools/PowerShellTool/resultLoader.ts +31 -0
  397. package/tui/src/tools/PowerShellTool/resultMapping.ts +75 -0
  398. package/tui/src/tools/PowerShellTool/schemas.ts +40 -0
  399. package/tui/src/tools/PowerShellTool/shellExecution.tsx +258 -0
  400. package/tui/src/tools/PowerShellTool/symlinkModeValidation.ts +44 -0
  401. package/tui/src/tools/PowerShellTool/uiLoader.ts +37 -0
  402. package/tui/src/tools/PowerShellTool/validation.ts +39 -0
  403. package/tui/src/tools/ReadMcpResourceTool/ReadMcpResourceTool.ts +19 -3
  404. package/tui/src/tools/ResolveLocationPrimitive/ResolveLocationPrimitive.ts +30 -19
  405. package/tui/src/tools/ResolveLocationPrimitive/prompt.ts +2 -6
  406. package/tui/src/tools/SkillTool/SkillTool.ts +2 -2
  407. package/tui/src/tools/SubmitPrimitive/SubmitPrimitive.ts +51 -18
  408. package/tui/src/tools/TaskCreateTool/TaskCreateTool.ts +16 -2
  409. package/tui/src/tools/TaskGetTool/TaskGetTool.ts +23 -3
  410. package/tui/src/tools/TaskListTool/TaskListTool.ts +22 -4
  411. package/tui/src/tools/TaskOutputTool/TaskOutputTool.tsx +46 -547
  412. package/tui/src/tools/TaskOutputTool/lookup.ts +216 -0
  413. package/tui/src/tools/TaskOutputTool/render.tsx +257 -0
  414. package/tui/src/tools/TaskOutputTool/schemas.ts +55 -0
  415. package/tui/src/tools/TaskOutputTool/serialization.ts +36 -0
  416. package/tui/src/tools/TaskStopTool/TaskStopTool.ts +10 -0
  417. package/tui/src/tools/TaskUpdateTool/TaskUpdateTool.ts +14 -364
  418. package/tui/src/tools/TaskUpdateTool/completion.ts +62 -0
  419. package/tui/src/tools/TaskUpdateTool/schemas.ts +62 -0
  420. package/tui/src/tools/TaskUpdateTool/serialization.ts +46 -0
  421. package/tui/src/tools/TaskUpdateTool/statusUpdate.ts +247 -0
  422. package/tui/src/tools/TodoWriteTool/TodoWriteTool.ts +21 -2
  423. package/tui/src/tools/ToolSearchTool/ToolSearchTool.ts +21 -302
  424. package/tui/src/tools/ToolSearchTool/ccSupportTools.ts +223 -0
  425. package/tui/src/tools/ToolSearchTool/descriptionCache.ts +50 -0
  426. package/tui/src/tools/ToolSearchTool/keywordSearch.ts +216 -0
  427. package/tui/src/tools/ToolSearchTool/prompt.ts +10 -4
  428. package/tui/src/tools/ToolSearchTool/resultMapping.ts +30 -0
  429. package/tui/src/tools/ToolSearchTool/schemas.ts +30 -0
  430. package/tui/src/tools/ToolSearchTool/searchPool.ts +47 -0
  431. package/tui/src/tools/ToolSearchTool/supportIntentHints.ts +140 -0
  432. package/tui/src/tools/TranslateTool/TranslateTool.ts +1 -1
  433. package/tui/src/tools/VerifyPrimitive/VerifyPrimitive.ts +27 -10
  434. package/tui/src/tools/WebFetchTool/WebFetchTool.ts +43 -138
  435. package/tui/src/tools/WebFetchTool/call.ts +227 -0
  436. package/tui/src/tools/WebFetchTool/resolvedAddressSafety.ts +78 -0
  437. package/tui/src/tools/WebFetchTool/sourceVerification.ts +204 -0
  438. package/tui/src/tools/WebFetchTool/types.ts +23 -0
  439. package/tui/src/tools/WebFetchTool/urlSafety.ts +181 -0
  440. package/tui/src/tools/WebFetchTool/utils.ts +1 -1
  441. package/tui/src/tools/WebSearchTool/UI.tsx +0 -1
  442. package/tui/src/tools/WebSearchTool/WebSearchTool.ts +9 -313
  443. package/tui/src/tools/WebSearchTool/call.ts +33 -0
  444. package/tui/src/tools/WebSearchTool/responseMapping.ts +190 -0
  445. package/tui/src/tools/WebSearchTool/resultBlock.ts +47 -0
  446. package/tui/src/tools/WebSearchTool/schemas.ts +47 -0
  447. package/tui/src/tools/WebSearchTool/toolSchema.ts +12 -0
  448. package/tui/src/tools/WorkspaceToolAdapter/WorkspaceToolAdapter.ts +79 -0
  449. package/tui/src/tools/WorkspaceToolAdapter/allowedRootPolicy.ts +85 -0
  450. package/tui/src/tools/WorkspaceToolAdapter/documentFormatGuards.ts +73 -0
  451. package/tui/src/tools/WorkspaceToolAdapter/inputNormalization.ts +105 -0
  452. package/tui/src/tools/WorkspaceToolAdapter/mcpExposurePolicy.ts +64 -0
  453. package/tui/src/tools/WorkspaceToolAdapter/toolDefFactory.ts +215 -0
  454. package/tui/src/tools/WorkspaceToolAdapter/toolNames.ts +6 -0
  455. package/tui/src/tools/WorkspaceToolAdapter/workspacePolicy.ts +15 -0
  456. package/tui/src/tools/_shared/citizenUserText.ts +49 -0
  457. package/tui/src/tools/_shared/dispatchPrimitive.ts +6 -6
  458. package/tui/src/tools/_shared/documentChangeToPatch.ts +125 -0
  459. package/tui/src/tools/_shared/documentDispatchArguments.ts +87 -0
  460. package/tui/src/tools/_shared/documentPrimitiveTimeout.ts +13 -0
  461. package/tui/src/tools/_shared/documentToolResultRender.ts +98 -0
  462. package/tui/src/tools/_shared/locationInputRepair.ts +112 -0
  463. package/tui/src/tools/_shared/pendingCallRegistry.ts +1 -6
  464. package/tui/src/tools/_shared/rootPrimitiveInput.ts +68 -0
  465. package/tui/src/tools/_shared/toolChoiceRepair/documentCompletionPatterns.ts +58 -0
  466. package/tui/src/tools/_shared/toolChoiceRepair/documentCompletionPrompt.ts +271 -0
  467. package/tui/src/tools/_shared/toolChoiceRepair/documentRepair.ts +452 -0
  468. package/tui/src/tools/_shared/toolChoiceRepair/messageAccess.ts +80 -0
  469. package/tui/src/tools/_shared/toolChoiceRepair/publicDataRepair.ts +92 -0
  470. package/tui/src/tools/_shared/toolChoiceRepair/supportRepair.ts +135 -0
  471. package/tui/src/tools/_shared/toolChoiceRepair.ts +61 -0
  472. package/tui/src/tools/shared/mockDisclaimer.ts +1 -1
  473. package/tui/src/tools.ts +39 -190
  474. package/tui/src/types/fileSuggestion.ts +4 -26
  475. package/tui/src/types/generated/events_mono/claude_code/v1/claude_code_internal_event.ts +186 -148
  476. package/tui/src/types/generated/events_mono/common/v1/auth.ts +25 -11
  477. package/tui/src/types/generated/events_mono/growthbook/v1/growthbook_experiment_event.ts +47 -30
  478. package/tui/src/types/generated/google/protobuf/timestamp.ts +21 -7
  479. package/tui/src/types/message.ts +80 -102
  480. package/tui/src/types/messageQueueTypes.ts +6 -28
  481. package/tui/src/types/notebook.ts +16 -38
  482. package/tui/src/types/statusLine.ts +4 -26
  483. package/tui/src/types/tools.ts +24 -46
  484. package/tui/src/types/utils.ts +6 -28
  485. package/tui/src/upstreamproxy/relay.ts +7 -3
  486. package/tui/src/upstreamproxy/upstreamproxy.ts +1 -1
  487. package/tui/src/utils/assistantMessageFactories.ts +9 -3
  488. package/tui/src/utils/attachments.ts +1 -1
  489. package/tui/src/utils/auth.ts +129 -139
  490. package/tui/src/utils/bash/ast.ts +23 -23
  491. package/tui/src/utils/bash/bashParser.ts +5 -5
  492. package/tui/src/utils/billing.ts +1 -1
  493. package/tui/src/utils/collapseReadSearch.ts +3 -3
  494. package/tui/src/utils/cronTasks.ts +1 -1
  495. package/tui/src/utils/execFileNoThrow.ts +1 -1
  496. package/tui/src/utils/filePersistence/types.ts +16 -38
  497. package/tui/src/utils/forkedAgent.ts +1 -1
  498. package/tui/src/utils/gracefulShutdown.ts +4 -4
  499. package/tui/src/utils/heapDumpService.ts +12 -8
  500. package/tui/src/utils/hooks/apiQueryHookHelper.ts +1 -1
  501. package/tui/src/utils/hooks/execPromptHook.ts +1 -1
  502. package/tui/src/utils/hooks/skillImprovement.ts +1 -1
  503. package/tui/src/utils/kExaoneReasoning.ts +138 -0
  504. package/tui/src/utils/mcp/dateTimeParser.ts +1 -1
  505. package/tui/src/utils/messages.ts +19 -0
  506. package/tui/src/utils/migrateSessions.ts +3 -3
  507. package/tui/src/utils/model/model.ts +6 -6
  508. package/tui/src/utils/multiToolLayout.ts +13 -0
  509. package/tui/src/utils/permissions/yoloClassifier.ts +1 -1
  510. package/tui/src/utils/plugins/headlessPluginInstall.ts +1 -1
  511. package/tui/src/utils/plugins/mcpPluginIntegration.ts +1 -1
  512. package/tui/src/utils/plugins/mcpbHandler.ts +1 -1
  513. package/tui/src/utils/plugins/pluginLoader.ts +8 -8
  514. package/tui/src/utils/processUserInput/processSlashCommand.tsx +2 -2
  515. package/tui/src/utils/processUserInput/processUserInput.ts +26 -0
  516. package/tui/src/utils/protectedNamespace.ts +5 -3
  517. package/tui/src/utils/rawJsonToolCall.ts +242 -0
  518. package/tui/src/utils/ripgrep.ts +16 -7
  519. package/tui/src/utils/sessionTitle.ts +1 -1
  520. package/tui/src/utils/settings/applySettingsChange.ts +4 -0
  521. package/tui/src/utils/settings/permissionValidation.ts +14 -2
  522. package/tui/src/utils/settings/types.ts +9 -3
  523. package/tui/src/utils/shell/prefix.ts +1 -1
  524. package/tui/src/utils/sideQuery.ts +1 -1
  525. package/tui/src/utils/stats.ts +1 -1
  526. package/tui/src/utils/systemThemeWatcher.ts +13 -3
  527. package/tui/src/utils/teleport.tsx +1 -1
  528. package/uv.lock +394 -22
  529. package/assets/copilot-gate-logo.svg +0 -58
  530. package/assets/govon-logo.svg +0 -40
  531. package/src/ummaya/eval/__init__.py +0 -5
  532. package/src/ummaya/eval/retrieval.py +0 -713
  533. package/tui/src/services/api/claude.ts +0 -3510
  534. package/tui/src/utils/messageStream.ts +0 -186
@@ -0,0 +1,74 @@
1
+ # SPDX-License-Identifier: Apache-2.0
2
+
3
+ from __future__ import annotations
4
+
5
+ from typing import Literal
6
+
7
+ from pydantic import BaseModel, ConfigDict, Field
8
+
9
+ from ummaya.tools.routing.intent import ToolSelectionIntent
10
+ from ummaya.tools.routing.types import AdapterCard
11
+
12
+ SchemaProjectionLevel = Literal["none", "name_only", "summary", "full_schema"]
13
+ FeasibilityStatus = Literal["feasible", "infeasible"]
14
+ RouteStopReason = Literal[
15
+ "answerable",
16
+ "needs_input",
17
+ "permission_required",
18
+ "handoff_required",
19
+ "blocked_no_adapter",
20
+ "blocked_no_credential",
21
+ "max_turns",
22
+ "repeated_tool_mismatch",
23
+ "no_new_evidence",
24
+ "runtime_tool_failure_unrecovered",
25
+ ]
26
+ RouteClarificationReason = Literal[
27
+ "missing_slots",
28
+ "equal_candidates",
29
+ "side_effect_confirmation",
30
+ "execution_risk_input_fault",
31
+ ]
32
+
33
+
34
+ class RouteCandidate(BaseModel):
35
+ model_config = ConfigDict(frozen=True, extra="forbid")
36
+
37
+ tool_id: str = Field(min_length=1)
38
+ retrieval_score: float = Field(ge=0)
39
+ card: AdapterCard
40
+ status: FeasibilityStatus
41
+ feasible: bool
42
+ filter_reasons: tuple[str, ...]
43
+ score_breakdown: dict[str, float]
44
+
45
+
46
+ class RouteClarificationDecision(BaseModel):
47
+ model_config = ConfigDict(frozen=True, extra="forbid")
48
+
49
+ reason: RouteClarificationReason
50
+ question: str = Field(min_length=1, max_length=240)
51
+ missing_slots: tuple[str, ...] = Field(default_factory=tuple)
52
+ candidate_tool_ids: tuple[str, ...] = Field(default_factory=tuple)
53
+ evidence_events: tuple[str, ...] = Field(default_factory=tuple)
54
+
55
+
56
+ class RouteDecision(BaseModel):
57
+ model_config = ConfigDict(frozen=True, extra="forbid")
58
+
59
+ decision_id: str = Field(min_length=1)
60
+ query_hash: str = Field(min_length=64, max_length=64)
61
+ manifest_hash: str = Field(min_length=64, max_length=64)
62
+ intent: ToolSelectionIntent
63
+ candidate_set: tuple[RouteCandidate, ...]
64
+ selected_tools: tuple[str, ...]
65
+ schema_projection_level: SchemaProjectionLevel
66
+ backend_label: str = Field(min_length=1)
67
+ effective_top_k: int = Field(ge=0, le=20)
68
+ clarification: RouteClarificationDecision | None
69
+ clarification_question: str | None
70
+ permission_gate: bool
71
+ stop_reason: RouteStopReason
72
+ degradation_reason: str | None
73
+ score_breakdown: dict[str, float]
74
+ evidence_events: tuple[str, ...]
@@ -0,0 +1,122 @@
1
+ # SPDX-License-Identifier: Apache-2.0
2
+
3
+ from __future__ import annotations
4
+
5
+ from collections.abc import Sequence
6
+
7
+ from ummaya.tools.routing.intent import ACTIVE_PRIMITIVES, ToolSelectionIntent
8
+ from ummaya.tools.routing.lint import lint_adapter_card
9
+ from ummaya.tools.routing.types import AdapterCard, PrimitiveFamily, SourceMode
10
+
11
+ DEFAULT_SOURCE_MODES: tuple[SourceMode, ...] = ("live", "mock")
12
+ SIDE_EFFECT_LEVELS = frozenset({"login", "action", "sign", "send", "verify"})
13
+ PERMISSION_CONTEXT_LEVELS = frozenset({"action", "sign", "send", "verify"})
14
+
15
+
16
+ def hard_exclusion_reasons(
17
+ card: AdapterCard,
18
+ *,
19
+ active_tool_ids: frozenset[str],
20
+ allowed_source_modes: frozenset[SourceMode],
21
+ allowed_credentials: frozenset[str] | None,
22
+ ) -> tuple[str, ...]:
23
+ reasons: list[str] = []
24
+ if card.tool_id not in active_tool_ids:
25
+ reasons.append("inactive_adapter")
26
+ if card.source_mode not in allowed_source_modes:
27
+ reasons.append(f"disallowed_source_mode:{card.source_mode}")
28
+ if allowed_credentials is not None:
29
+ missing_credentials = [
30
+ requirement
31
+ for requirement in card.credential_requirements
32
+ if requirement not in allowed_credentials
33
+ ]
34
+ reasons.extend(f"disallowed_credential:{credential}" for credential in missing_credentials)
35
+ return tuple(dict.fromkeys(reasons))
36
+
37
+
38
+ def soft_filter_reasons(
39
+ card: AdapterCard,
40
+ *,
41
+ route_intent: ToolSelectionIntent,
42
+ session_identity: object | None,
43
+ active_tool_ids: frozenset[str],
44
+ active_primitives: frozenset[PrimitiveFamily],
45
+ ) -> tuple[str, ...]:
46
+ reasons: list[str] = []
47
+ reasons.extend(
48
+ f"adapter_card_quality:{violation.code}" for violation in lint_adapter_card(card)
49
+ )
50
+ reasons.extend(_missing_slot_reasons(card.required_slots, route_intent.missing_slots))
51
+ reasons.extend(_permission_reasons(card, route_intent, session_identity))
52
+ reasons.extend(_prerequisite_reasons(card, active_tool_ids, active_primitives))
53
+ return tuple(dict.fromkeys(reasons))
54
+
55
+
56
+ def score_breakdown(
57
+ card: AdapterCard,
58
+ retrieval_score: float,
59
+ filter_reasons: tuple[str, ...],
60
+ ) -> dict[str, float]:
61
+ return {
62
+ "retrieval": retrieval_score,
63
+ "feasibility": 1.0 if not filter_reasons else 0.0,
64
+ "source_mode": 1.0 if card.source_mode in {"live", "mock"} else 0.0,
65
+ "permission": 0.0 if any("permission" in reason for reason in filter_reasons) else 1.0,
66
+ "prerequisite": 0.0 if any("prerequisite" in reason for reason in filter_reasons) else 1.0,
67
+ "quality": 0.0
68
+ if any("adapter_card_quality" in reason for reason in filter_reasons)
69
+ else 1.0,
70
+ }
71
+
72
+
73
+ def requires_permission_gate(card: AdapterCard, *, feasible: bool) -> bool:
74
+ return feasible and card.side_effect_level in SIDE_EFFECT_LEVELS
75
+
76
+
77
+ def source_modes(value: Sequence[SourceMode]) -> frozenset[SourceMode]:
78
+ return frozenset(value)
79
+
80
+
81
+ def _missing_slot_reasons(
82
+ required_slots: tuple[str, ...],
83
+ intent_missing_slots: tuple[str, ...],
84
+ ) -> tuple[str, ...]:
85
+ missing = frozenset(intent_missing_slots)
86
+ return tuple(f"missing_slot:{slot}" for slot in required_slots if slot in missing)
87
+
88
+
89
+ def _permission_reasons(
90
+ card: AdapterCard,
91
+ intent: ToolSelectionIntent,
92
+ session_identity: object | None,
93
+ ) -> tuple[str, ...]:
94
+ if card.side_effect_level not in SIDE_EFFECT_LEVELS:
95
+ return ()
96
+ reasons: list[str] = []
97
+ if card.policy_authority_url is None:
98
+ reasons.append("missing_policy_citation")
99
+ if (
100
+ card.side_effect_level in PERMISSION_CONTEXT_LEVELS
101
+ and not intent.requires_permission
102
+ and session_identity is None
103
+ ):
104
+ reasons.append("permission_context_missing")
105
+ return tuple(reasons)
106
+
107
+
108
+ def _prerequisite_reasons(
109
+ card: AdapterCard,
110
+ active_tool_ids: frozenset[str],
111
+ active_primitives: frozenset[PrimitiveFamily],
112
+ ) -> tuple[str, ...]:
113
+ reasons: list[str] = []
114
+ primitive_names = frozenset((*ACTIVE_PRIMITIVES, "document"))
115
+ for prerequisite in card.prerequisite_tools:
116
+ if prerequisite in primitive_names:
117
+ if prerequisite not in active_primitives:
118
+ reasons.append(f"missing_prerequisite_primitive:{prerequisite}")
119
+ continue
120
+ if prerequisite not in active_tool_ids:
121
+ reasons.append(f"missing_prerequisite_tool:{prerequisite}")
122
+ return tuple(reasons)
@@ -0,0 +1,17 @@
1
+ # SPDX-License-Identifier: Apache-2.0
2
+
3
+ from ummaya.tools.routing.intent_extractor import extract_tool_selection_intent
4
+ from ummaya.tools.routing.intent_types import (
5
+ ACTIVE_PRIMITIVES,
6
+ LEGACY_PRIMITIVE_ALIASES,
7
+ ActivePrimitive,
8
+ ToolSelectionIntent,
9
+ )
10
+
11
+ __all__ = [
12
+ "ACTIVE_PRIMITIVES",
13
+ "LEGACY_PRIMITIVE_ALIASES",
14
+ "ActivePrimitive",
15
+ "ToolSelectionIntent",
16
+ "extract_tool_selection_intent",
17
+ ]
@@ -0,0 +1,207 @@
1
+ # SPDX-License-Identifier: Apache-2.0
2
+
3
+ from __future__ import annotations
4
+
5
+ from collections.abc import Collection, Iterable
6
+
7
+ from ummaya.tools.routing.intent_patterns import (
8
+ _ADMIN_LOCATION_RE,
9
+ _ARTIFACT_ID_RE,
10
+ _COORDINATE_PAIR_RE,
11
+ _CREDENTIAL_RE,
12
+ _DOCUMENT_FORMAT_RE,
13
+ _DOCUMENT_INTENT_RE,
14
+ _DOCUMENT_LOCAL_HINT_RE,
15
+ _DOCUMENT_PATH_RE,
16
+ _DOCUMENT_WRITE_INTENT_RE,
17
+ _EXPLICIT_TOOL_ID_RE,
18
+ _PAYMENT_SIDE_EFFECT_RE,
19
+ _POI_LOCATION_RE,
20
+ _SIDE_EFFECT_RE,
21
+ _SUBMIT_SIDE_EFFECT_RE,
22
+ _WHITESPACE_RE,
23
+ )
24
+ from ummaya.tools.routing.intent_public_data import extract_public_data_refs
25
+ from ummaya.tools.routing.intent_types import (
26
+ ACTIVE_PRIMITIVES,
27
+ ActivePrimitive,
28
+ ToolSelectionIntent,
29
+ )
30
+
31
+
32
+ def extract_tool_selection_intent(
33
+ query: str, *, known_tool_ids: Collection[str] = ()
34
+ ) -> ToolSelectionIntent:
35
+ normalized_query = _normalize_query(query)
36
+ document_refs = _extract_document_refs(query)
37
+ location_refs = _extract_location_refs(query)
38
+ public_data_refs = extract_public_data_refs(query)
39
+ credential_refs = _extract_credential_refs(query)
40
+ side_effect_markers = _extract_side_effect_markers(query, document_refs=document_refs)
41
+ explicit_tool_ids = _extract_explicit_tool_ids(query, known_tool_ids=known_tool_ids)
42
+ explicit_artifact_ids = _ordered_unique(_ARTIFACT_ID_RE.findall(query))
43
+ candidate_primitives = _candidate_primitives(
44
+ normalized_query=normalized_query,
45
+ document_refs=document_refs,
46
+ location_refs=location_refs,
47
+ public_data_refs=public_data_refs,
48
+ credential_refs=credential_refs,
49
+ side_effect_markers=side_effect_markers,
50
+ explicit_tool_ids=explicit_tool_ids,
51
+ )
52
+ return ToolSelectionIntent(
53
+ raw_query=query,
54
+ normalized_query=normalized_query,
55
+ intent_verbs=candidate_primitives,
56
+ entities=(),
57
+ document_refs=document_refs,
58
+ location_refs=location_refs,
59
+ time_refs=(),
60
+ public_data_refs=public_data_refs,
61
+ credential_refs=credential_refs,
62
+ side_effect_markers=side_effect_markers,
63
+ explicit_tool_ids=explicit_tool_ids,
64
+ explicit_artifact_ids=explicit_artifact_ids,
65
+ candidate_primitives=candidate_primitives,
66
+ missing_slots=_missing_slots(location_refs, public_data_refs),
67
+ unsafe_assumptions=_unsafe_assumptions(location_refs, public_data_refs),
68
+ requires_clarification=not normalized_query,
69
+ requires_permission=bool(credential_refs or side_effect_markers),
70
+ )
71
+
72
+
73
+ def _normalize_query(query: str) -> str:
74
+ return _WHITESPACE_RE.sub(" ", query.strip()).lower()
75
+
76
+
77
+ def _extract_explicit_tool_ids(query: str, *, known_tool_ids: Collection[str]) -> tuple[str, ...]:
78
+ matches: list[tuple[int, str]] = []
79
+ for match in _EXPLICIT_TOOL_ID_RE.finditer(query):
80
+ matches.append((match.start(1), match.group(1)))
81
+ query_lower = query.lower()
82
+ for tool_id in known_tool_ids:
83
+ index = query_lower.find(tool_id.lower())
84
+ if index >= 0:
85
+ matches.append((index, tool_id))
86
+ return _ordered_unique(value for _index, value in sorted(matches))
87
+
88
+
89
+ def _extract_document_refs(query: str) -> tuple[str, ...]:
90
+ refs: list[str] = []
91
+ refs.extend(match.group(1).strip() for match in _DOCUMENT_PATH_RE.finditer(query))
92
+ refs.extend(f"format:{match.group(1).lower()}" for match in _DOCUMENT_FORMAT_RE.finditer(query))
93
+ if _is_document_harness_query(query):
94
+ refs.append("document_harness")
95
+ return _ordered_unique(refs)
96
+
97
+
98
+ def _extract_location_refs(query: str) -> tuple[str, ...]:
99
+ refs: list[str] = []
100
+ for match in _COORDINATE_PAIR_RE.finditer(query):
101
+ lat = float(match.group("lat"))
102
+ lon = float(match.group("lon"))
103
+ if -90.0 <= lat <= 90.0 and -180.0 <= lon <= 180.0:
104
+ refs.append(f"coordinate:{_format_decimal(lat)},{_format_decimal(lon)}")
105
+ if _POI_LOCATION_RE.search(query):
106
+ refs.append("poi")
107
+ if _ADMIN_LOCATION_RE.search(query):
108
+ refs.append("admin")
109
+ return _ordered_unique(refs)
110
+
111
+
112
+ def _extract_credential_refs(query: str) -> tuple[str, ...]:
113
+ if not _CREDENTIAL_RE.search(query):
114
+ return ()
115
+ refs: list[str] = []
116
+ lowered = query.lower()
117
+ if "간편인증" in query:
118
+ refs.append("simple_auth")
119
+ if "모바일" in query and "신분증" in query:
120
+ refs.append("mobile_id")
121
+ if "인증서" in query:
122
+ refs.append("certificate")
123
+ if "동의" in query or "consent" in lowered or "delegation" in lowered:
124
+ refs.append("delegation")
125
+ if not refs:
126
+ refs.append("credential")
127
+ return _ordered_unique(refs)
128
+
129
+
130
+ def _extract_side_effect_markers(query: str, *, document_refs: tuple[str, ...]) -> tuple[str, ...]:
131
+ refs: list[str] = []
132
+ if _SIDE_EFFECT_RE.search(query):
133
+ refs.append("action")
134
+ if _SUBMIT_SIDE_EFFECT_RE.search(query):
135
+ refs.append("submit")
136
+ if _PAYMENT_SIDE_EFFECT_RE.search(query):
137
+ refs.append("payment")
138
+ if document_refs and _DOCUMENT_WRITE_INTENT_RE.search(query):
139
+ refs.append("document_write")
140
+ return _ordered_unique(refs)
141
+
142
+
143
+ def _candidate_primitives(
144
+ *,
145
+ normalized_query: str,
146
+ document_refs: tuple[str, ...],
147
+ location_refs: tuple[str, ...],
148
+ public_data_refs: tuple[str, ...],
149
+ credential_refs: tuple[str, ...],
150
+ side_effect_markers: tuple[str, ...],
151
+ explicit_tool_ids: tuple[str, ...],
152
+ ) -> tuple[ActivePrimitive, ...]:
153
+ primitives: set[ActivePrimitive] = set()
154
+ if normalized_query or public_data_refs or document_refs or explicit_tool_ids:
155
+ primitives.add("find")
156
+ if location_refs:
157
+ primitives.add("locate")
158
+ if side_effect_markers:
159
+ primitives.add("send")
160
+ if credential_refs:
161
+ primitives.add("check")
162
+ return tuple(primitive for primitive in ACTIVE_PRIMITIVES if primitive in primitives)
163
+
164
+
165
+ def _missing_slots(
166
+ location_refs: tuple[str, ...], public_data_refs: tuple[str, ...]
167
+ ) -> tuple[str, ...]:
168
+ has_coordinate = any(ref.startswith("coordinate:") for ref in location_refs)
169
+ coordinate_bound_refs = {"emergency_medical", "aed"}
170
+ if not has_coordinate and any(ref in coordinate_bound_refs for ref in public_data_refs):
171
+ return ("lat", "lon")
172
+ return ()
173
+
174
+
175
+ def _unsafe_assumptions(
176
+ location_refs: tuple[str, ...], public_data_refs: tuple[str, ...]
177
+ ) -> tuple[str, ...]:
178
+ if "poi" in location_refs and not any(ref.startswith("coordinate:") for ref in location_refs):
179
+ return ("poi_requires_location_resolution",)
180
+ return ()
181
+
182
+
183
+ def _is_document_harness_query(query: str) -> bool:
184
+ return bool(
185
+ _DOCUMENT_PATH_RE.search(query)
186
+ or (_DOCUMENT_FORMAT_RE.search(query) and _DOCUMENT_INTENT_RE.search(query))
187
+ or (
188
+ _DOCUMENT_INTENT_RE.search(query)
189
+ and _DOCUMENT_WRITE_INTENT_RE.search(query)
190
+ and _DOCUMENT_LOCAL_HINT_RE.search(query)
191
+ )
192
+ )
193
+
194
+
195
+ def _format_decimal(value: float) -> str:
196
+ return f"{value:.6f}".rstrip("0").rstrip(".")
197
+
198
+
199
+ def _ordered_unique(values: Iterable[str]) -> tuple[str, ...]:
200
+ seen: set[str] = set()
201
+ ordered: list[str] = []
202
+ for value in values:
203
+ if value in seen:
204
+ continue
205
+ seen.add(value)
206
+ ordered.append(value)
207
+ return tuple(ordered)
@@ -0,0 +1,160 @@
1
+ # SPDX-License-Identifier: Apache-2.0
2
+
3
+ from __future__ import annotations
4
+
5
+ import re
6
+
7
+ _WHITESPACE_RE = re.compile(r"\s+")
8
+ _EXPLICIT_TOOL_ID_RE = re.compile(
9
+ r"(?:tool_id|adapter|도구)\s*[:=]\s*`?([a-z][a-z0-9_]*(?:\.[a-z][a-z0-9_]*)?)`?",
10
+ re.IGNORECASE,
11
+ )
12
+ _ARTIFACT_ID_RE = re.compile(
13
+ r"\b(?:artifact|source|working|derivative|render|viewport)-[A-Za-z0-9_-]+\b"
14
+ )
15
+ _DOCUMENT_PATH_RE = re.compile(
16
+ r"(?:^|[\s:'\"(])((?:~|/|[A-Za-z]:\\|\.{1,2}/)?[^\s:'\"]*"
17
+ r"\.(?:hwpx|hwp|docx|pdf|xlsx|pptx))\b",
18
+ re.IGNORECASE,
19
+ )
20
+ _DOCUMENT_FORMAT_RE = re.compile(r"\b(hwpx|hwp|docx|pdf|xlsx|pptx)\b", re.IGNORECASE)
21
+ _DOCUMENT_INTENT_RE = re.compile(
22
+ r"(문서|공문서|양식|서식|파일|작성|저장|렌더|미리보기|변경사항|"
23
+ r"\bdiff\b|\bcompact\b|\bdocument\b|\bfile\b|\bform\b|\brender\b|\bsave\b|\bwrite\b)",
24
+ re.IGNORECASE,
25
+ )
26
+ _DOCUMENT_WRITE_INTENT_RE = re.compile(
27
+ r"(작성|수정|편집|채우|채워|입력|변경|저장|write|edit|fill|apply|save)",
28
+ re.IGNORECASE,
29
+ )
30
+ _DOCUMENT_LOCAL_HINT_RE = re.compile(
31
+ r"(다운로드|downloads?|폴더|파일|양식|서식|활동일지|신청서|등본|증명서)",
32
+ re.IGNORECASE,
33
+ )
34
+ _COORDINATE_PAIR_RE = re.compile(
35
+ r"(?P<lat>[+-]?\d{1,2}(?:\.\d+)?)\s*,\s*(?P<lon>[+-]?\d{2,3}(?:\.\d+)?)"
36
+ )
37
+ _POI_LOCATION_RE = re.compile(
38
+ r"(근처|주변|인근|가까운|역|터미널|공항|캠퍼스|대학교|대학|해수욕장|시장|공원|랜드마크)"
39
+ )
40
+ _ADMIN_LOCATION_RE = re.compile(
41
+ r"(?:[가-힣]{2,}(?:시|군|구|동|읍|면)\b|[가-힣0-9]{2,}(?<!으)(?:로|길)\b)"
42
+ )
43
+ _PUBLIC_DATA_OPERATION_RE = re.compile(
44
+ r"\b(getSurfaceChart|getAuxillaryChart|SEA\d{4}|FinancesService|"
45
+ r"StudentService|WthrChartInfoService)\b",
46
+ re.IGNORECASE,
47
+ )
48
+ _KMA_GIMHAE_AIRPORT_RE = re.compile(r"(김해(?:공항)?|Gimhae|RKPK)", re.IGNORECASE)
49
+ _KMA_GIMPO_AIRPORT_RE = re.compile(r"(김포(?:공항)?|Gimpo|RKSS)", re.IGNORECASE)
50
+ _KMA_AIRPORT_AVIATION_RE = re.compile(
51
+ r"(AMOS|METAR|SPECI|RVR|항공기상|공항기상|활주로|runway|aviation|"
52
+ r"비행기|항공편|비행편|이륙|착륙|결항|지연|운항|뜰\s*만|뜨나|뜰\s*수|"
53
+ r"flight|take\s*off|landing|delay|cancel)",
54
+ re.IGNORECASE,
55
+ )
56
+ _KMA_EXPLICIT_METAR_RE = re.compile(r"(\bMETAR\b|\bSPECI\b|해독자료)", re.IGNORECASE)
57
+ _KMA_RUNWAY_AREA_RE = re.compile(
58
+ r"(AMOS|활주로|RVR|runway|시정|visibility|공항기상관측|매분)",
59
+ re.IGNORECASE,
60
+ )
61
+ _KMA_ANALYSIS_CHART_RE = re.compile(
62
+ r"(분석일기도|지상일기도|보조일기도|WthrChartInfoService|getSurfaceChart|"
63
+ r"getAuxillaryChart|synoptic\s+chart)",
64
+ re.IGNORECASE,
65
+ )
66
+ _KMA_ANALYSIS_DATA_RE = re.compile(
67
+ r"(분석자료|이미\s*분석|고해상도\s*격자|객관분석|AWS\s*객관|지도\s*자료|"
68
+ r"일기도|분석일기도|비구름|바람\s*흐름|날씨\s*흐름|공식\s*기상자료|전국\s*날씨|"
69
+ r"synoptic|weather\s*chart|objective\s*analysis|high[-\s]?resolution|grid)",
70
+ re.IGNORECASE,
71
+ )
72
+ _KMA_ANALYSIS_MAP_RE = re.compile(
73
+ r"(일기도|분석일기도|지도\s*자료|비구름|바람\s*흐름|날씨\s*흐름|전국\s*날씨|"
74
+ r"synoptic|weather\s*chart)",
75
+ re.IGNORECASE,
76
+ )
77
+ _KMA_ANALYSIS_POINT_RE = re.compile(
78
+ r"(주변|근처|특정지점|좌표|위도|경도|\blat\b|\blon\b|공항\s*주변)",
79
+ re.IGNORECASE,
80
+ )
81
+ _KMA_LIFESTYLE_WEATHER_RE = re.compile(
82
+ r"(날씨|현재\s*기상|실황|관측|예보|기온|습도|풍속|지금\s*비|"
83
+ r"비\s*(?:와|오|올|내리)|우산|강수|소나기|산책|퇴근|"
84
+ r"current\s+weather|forecast|rain|umbrella|precipitation|temperature)",
85
+ re.IGNORECASE,
86
+ )
87
+ _EMERGENCY_RE = re.compile(r"(응급|응급실|응급의료|\bemergency\b|\ber\b)", re.IGNORECASE)
88
+ _IMPLICIT_EMERGENCY_RE = re.compile(
89
+ r"(사람이\s*(?:쓰러|쓰러졌|쓰러져)|의식(?:을)?\s*(?:잃|없)|"
90
+ r"갑자기\s*쓰러|쓰러진\s*사람|위급|심정지|호흡(?:이)?\s*없|"
91
+ r"collapsed|unconscious|cardiac\s*arrest)",
92
+ re.IGNORECASE,
93
+ )
94
+ _AED_RE = re.compile(r"(\bAED\b|자동심장충격기|자동제세동기|제세동기)", re.IGNORECASE)
95
+ _TRAFFIC_HAZARD_RE = re.compile(
96
+ r"(교통사고|사고\s*위험|사고다발|위험\s*(?:구간|도로|지점)|어린이보호구역|보호구역|"
97
+ r"도로\s*구간|accident|hazard|hotspot)",
98
+ re.IGNORECASE,
99
+ )
100
+ _TRAFFIC_HAZARD_SPECIFIC_RE = re.compile(
101
+ r"(사고\s*위험|위험\s*(?:구간|도로|지점)|어린이보호구역|보호구역|스쿨존|"
102
+ r"도로\s*구간|행정동코드|adm_cd|hazard|hotspot)",
103
+ re.IGNORECASE,
104
+ )
105
+ _MOF_OCEAN_WATER_QUALITY_RE = re.compile(
106
+ r"(해양\s*수질|해양수질|수질\s*자동\s*측정|용존산소|\bpH\b|"
107
+ r"water\s+quality|ocean\s+water)",
108
+ re.IGNORECASE,
109
+ )
110
+ _PPS_BID_RE = re.compile(r"(입찰|나라장터|조달청|\bbid\b|procurement|tender)", re.IGNORECASE)
111
+ _PPS_SHOPPING_RE = re.compile(
112
+ r"(종합\s*쇼핑몰|쇼핑몰|계약\s*물품|물품\s*조회|shopping\s*mall)",
113
+ re.IGNORECASE,
114
+ )
115
+ _AIRKOREA_AIR_QUALITY_RE = re.compile(
116
+ r"(미세먼지|초미세먼지|초미세|대기질|대기오염|공기질|마스크|"
117
+ r"pm\s*2\.?5|pm\s*10|air\s*korea|airkorea|air\s*quality|airquality)",
118
+ re.IGNORECASE,
119
+ )
120
+ _KCUE_REGIONAL_RE = re.compile(
121
+ r"(대학알리미|대학정보공시|학교구분코드|schl[_\s-]?div[_\s-]?cd|KCUE|"
122
+ r"지역별\s*(등록금|재정)|외국인\s*유학생|foreign\s+student|international\s+student)",
123
+ re.IGNORECASE,
124
+ )
125
+ _KCUE_REGIONAL_FINANCE_RE = re.compile(
126
+ r"(지역별\s*(등록금|재정)|등록금\s*(현황|지역별)?|tuition|finance)",
127
+ re.IGNORECASE,
128
+ )
129
+ _KCUE_REGIONAL_FOREIGN_STUDENT_RE = re.compile(
130
+ r"(외국인\s*유학생|유학생\s*현황|foreign\s+student|international\s+student)",
131
+ re.IGNORECASE,
132
+ )
133
+ _HIRA_MEDICAL_DETAIL_RE = re.compile(
134
+ r"((병원|의료기관|의원).*(상세|진료과|진료과목|진료시간|주차)|"
135
+ r"(상세|진료시간|주차|응급실).*(병원|의료기관|의원)|ykiho|detail)",
136
+ re.IGNORECASE,
137
+ )
138
+ _MOIS_EMERGENCY_CALL_BOX_RE = re.compile(
139
+ r"(안전\s*비상벨|비상벨|긴급\s*신고함|긴급신고함|방범벨|"
140
+ r"emergency\s+call\s+box)",
141
+ re.IGNORECASE,
142
+ )
143
+ _GYERYONG_ASSISTIVE_CHARGER_RE = re.compile(
144
+ r"((전동보장구|전동\s*휠체어|보장구|장애인).*(충전|충전소|충전장소)|"
145
+ r"(충전|충전소|충전장소).*(전동보장구|전동\s*휠체어|보장구|장애인)|"
146
+ r"계룡시?.*(충전소|충전\s*장소))",
147
+ re.IGNORECASE,
148
+ )
149
+ _CREDENTIAL_RE = re.compile(
150
+ r"(간편인증|본인인증|모바일\s*신분증|공동인증서|금융인증서|인증서|"
151
+ r"delegation|consent|verify|login|로그인|동의)",
152
+ re.IGNORECASE,
153
+ )
154
+ _SIDE_EFFECT_RE = re.compile(
155
+ r"(신청|제출|신고|송신|납부|결제|발급|저장|수정|편집|채우|"
156
+ r"submit|send|pay|issue|save|write|edit|fill|apply)",
157
+ re.IGNORECASE,
158
+ )
159
+ _SUBMIT_SIDE_EFFECT_RE = re.compile(r"(신청|제출|신고|submit|send|apply)", re.IGNORECASE)
160
+ _PAYMENT_SIDE_EFFECT_RE = re.compile(r"(납부|결제|pay)", re.IGNORECASE)