contextsynapse 1.0.0__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 (776) hide show
  1. contextsynapse/__init__.py +107 -0
  2. contextsynapse/__main__.py +9 -0
  3. contextsynapse/a2a/__init__.py +28 -0
  4. contextsynapse/a2a/client.py +120 -0
  5. contextsynapse/a2a/discovery.py +128 -0
  6. contextsynapse/a2a/handlers.py +164 -0
  7. contextsynapse/a2a/models.py +246 -0
  8. contextsynapse/a2a/streaming.py +92 -0
  9. contextsynapse/a2a/task_manager.py +275 -0
  10. contextsynapse/adapters/__init__.py +98 -0
  11. contextsynapse/adapters/_base.py +298 -0
  12. contextsynapse/adapters/autogen/__init__.py +4 -0
  13. contextsynapse/adapters/autogen/tools.py +34 -0
  14. contextsynapse/adapters/crewai/__init__.py +4 -0
  15. contextsynapse/adapters/crewai/tools.py +63 -0
  16. contextsynapse/adapters/langchain/__init__.py +19 -0
  17. contextsynapse/adapters/langchain/message_history.py +17 -0
  18. contextsynapse/adapters/langchain/retriever.py +158 -0
  19. contextsynapse/adapters/langchain/tools.py +43 -0
  20. contextsynapse/adapters/langgraph/__init__.py +21 -0
  21. contextsynapse/adapters/langgraph/checkpoint.py +223 -0
  22. contextsynapse/adapters/langgraph/context_tools.py +341 -0
  23. contextsynapse/adapters/langgraph/message_history.py +138 -0
  24. contextsynapse/adapters/langgraph/tools.py +43 -0
  25. contextsynapse/adapters/llamaindex/__init__.py +4 -0
  26. contextsynapse/adapters/llamaindex/tools.py +64 -0
  27. contextsynapse/adapters/openai/__init__.py +24 -0
  28. contextsynapse/adapters/openai/tools.py +206 -0
  29. contextsynapse/adapters/pydantic_ai/__init__.py +4 -0
  30. contextsynapse/adapters/pydantic_ai/tools.py +36 -0
  31. contextsynapse/adapters/swarm/__init__.py +4 -0
  32. contextsynapse/adapters/swarm/tools.py +44 -0
  33. contextsynapse/agents/__init__.py +6 -0
  34. contextsynapse/agents/auto_capture.py +80 -0
  35. contextsynapse/agents/dispatcher.py +299 -0
  36. contextsynapse/agents/notifications.py +112 -0
  37. contextsynapse/aiql/__init__.py +71 -0
  38. contextsynapse/aiql/cache.py +341 -0
  39. contextsynapse/aiql/compiler/__init__.py +62 -0
  40. contextsynapse/aiql/compiler/optimizer.py +126 -0
  41. contextsynapse/aiql/compiler/planner.py +366 -0
  42. contextsynapse/aiql/compiler/statistics.py +235 -0
  43. contextsynapse/aiql/compiler/validator.py +425 -0
  44. contextsynapse/aiql/engine/__init__.py +11 -0
  45. contextsynapse/aiql/engine/chunk_storage.py +2584 -0
  46. contextsynapse/aiql/engine/executor.py +7737 -0
  47. contextsynapse/aiql/engine/extraction_integration.py +442 -0
  48. contextsynapse/aiql/engine/graph_builder.py +405 -0
  49. contextsynapse/aiql/engine/node_creation_from_normalized.py +691 -0
  50. contextsynapse/aiql/engine/performance_monitor.py +289 -0
  51. contextsynapse/aiql/engine/postprocessing.py +365 -0
  52. contextsynapse/aiql/engine/ray_stage_executors.py +918 -0
  53. contextsynapse/aiql/engine/redis_cache.py +106 -0
  54. contextsynapse/aiql/engine/streaming_json.py +366 -0
  55. contextsynapse/aiql/engine/streaming_parquet.py +568 -0
  56. contextsynapse/aiql/engine/streaming_ray_chunking.py +419 -0
  57. contextsynapse/aiql/grammar/__init__.py +14 -0
  58. contextsynapse/aiql/grammar/aiql_grammar.py +1822 -0
  59. contextsynapse/aiql/grammar/base.py +71 -0
  60. contextsynapse/aiql/parser/__init__.py +46 -0
  61. contextsynapse/aiql/parser/aiql_parser.py +11330 -0
  62. contextsynapse/aiql/parser/base_parser.py +47 -0
  63. contextsynapse/api/__init__.py +5 -0
  64. contextsynapse/api/a2a_router.py +233 -0
  65. contextsynapse/api/admin_router.py +140 -0
  66. contextsynapse/api/agent_worker_router.py +978 -0
  67. contextsynapse/api/algorithms_router.py +317 -0
  68. contextsynapse/api/api.py +4144 -0
  69. contextsynapse/api/audit.py +204 -0
  70. contextsynapse/api/auth.py +441 -0
  71. contextsynapse/api/auth_router.py +537 -0
  72. contextsynapse/api/billing.py +261 -0
  73. contextsynapse/api/billing_router.py +275 -0
  74. contextsynapse/api/boundary_router.py +452 -0
  75. contextsynapse/api/cognition_router.py +116 -0
  76. contextsynapse/api/connector_router.py +237 -0
  77. contextsynapse/api/context_router.py +1619 -0
  78. contextsynapse/api/dashboard_router.py +11367 -0
  79. contextsynapse/api/errors.py +183 -0
  80. contextsynapse/api/events.py +88 -0
  81. contextsynapse/api/experiment_bootstrap.py +321 -0
  82. contextsynapse/api/experiment_orchestrator.py +1078 -0
  83. contextsynapse/api/experiment_router.py +2089 -0
  84. contextsynapse/api/feature_gate.py +89 -0
  85. contextsynapse/api/federation_router.py +108 -0
  86. contextsynapse/api/integrations.py +567 -0
  87. contextsynapse/api/integrations_router.py +461 -0
  88. contextsynapse/api/intelligence_router.py +1028 -0
  89. contextsynapse/api/metering.py +155 -0
  90. contextsynapse/api/metrics.py +112 -0
  91. contextsynapse/api/models.py +38 -0
  92. contextsynapse/api/monitoring_router.py +174 -0
  93. contextsynapse/api/pipeline_router.py +1437 -0
  94. contextsynapse/api/playground_conn.py +116 -0
  95. contextsynapse/api/projects_router.py +131 -0
  96. contextsynapse/api/queries_router.py +282 -0
  97. contextsynapse/api/quota.py +95 -0
  98. contextsynapse/api/rate_limit.py +64 -0
  99. contextsynapse/api/search_router.py +261 -0
  100. contextsynapse/api/shield_router.py +447 -0
  101. contextsynapse/api/tenant_guard.py +103 -0
  102. contextsynapse/api/tenant_quotas.py +231 -0
  103. contextsynapse/api/tenants.py +305 -0
  104. contextsynapse/api/users.py +646 -0
  105. contextsynapse/api/v1.py +167 -0
  106. contextsynapse/api/vertical_builder_router.py +387 -0
  107. contextsynapse/benchmark/__init__.py +15 -0
  108. contextsynapse/benchmark/__main__.py +64 -0
  109. contextsynapse/benchmark/agents.py +338 -0
  110. contextsynapse/benchmark/live_ab.py +338 -0
  111. contextsynapse/benchmark/reporter.py +149 -0
  112. contextsynapse/benchmark/runner.py +264 -0
  113. contextsynapse/benchmark/savings.py +266 -0
  114. contextsynapse/benchmark/token_counter.py +212 -0
  115. contextsynapse/benchmark/workloads.py +289 -0
  116. contextsynapse/buffer/__init__.py +154 -0
  117. contextsynapse/buffer/base.py +257 -0
  118. contextsynapse/buffer/duckdb_buffer.py +246 -0
  119. contextsynapse/buffer/file_buffer.py +229 -0
  120. contextsynapse/buffer/lmdb_buffer.py +286 -0
  121. contextsynapse/buffer/local_buffer.py +154 -0
  122. contextsynapse/buffer/redis_buffer.py +211 -0
  123. contextsynapse/buffer/sqlite_buffer.py +266 -0
  124. contextsynapse/cache.py +2 -0
  125. contextsynapse/cli/__init__.py +40 -0
  126. contextsynapse/cli/cli.py +475 -0
  127. contextsynapse/cli/config_cli.py +177 -0
  128. contextsynapse/cli/scaffold.py +465 -0
  129. contextsynapse/cognition/__init__.py +31 -0
  130. contextsynapse/cognition/derivation.py +55 -0
  131. contextsynapse/cognition/events.py +135 -0
  132. contextsynapse/cognition/feedback.py +52 -0
  133. contextsynapse/cognition/hallucination.py +90 -0
  134. contextsynapse/cognition/invalidation.py +70 -0
  135. contextsynapse/cognition/read_tracker.py +50 -0
  136. contextsynapse/collaboration.py +2 -0
  137. contextsynapse/comparison.py +2 -0
  138. contextsynapse/config/__init__.py +10 -0
  139. contextsynapse/config/connectors/amfi_nav.yaml +18 -0
  140. contextsynapse/config/connectors/auto_sector.yaml +20 -0
  141. contextsynapse/config/connectors/banking_sector.yaml +20 -0
  142. contextsynapse/config/connectors/energy_sector.yaml +20 -0
  143. contextsynapse/config/connectors/fmcg_sector.yaml +20 -0
  144. contextsynapse/config/connectors/nsdl_fpi.yaml +9 -0
  145. contextsynapse/config/connectors/nse_price.yaml +61 -0
  146. contextsynapse/config/connectors/pharma_sector.yaml +20 -0
  147. contextsynapse/config/connectors/rbi_dbie.yaml +12 -0
  148. contextsynapse/config/context_presets/ai_research.json +25 -0
  149. contextsynapse/config/context_presets/chatgpt_conversations.json +19 -0
  150. contextsynapse/config/context_presets/claude_conversations.json +19 -0
  151. contextsynapse/config/context_presets/legal_compliance.json +21 -0
  152. contextsynapse/config/context_presets/meeting_notes.json +21 -0
  153. contextsynapse/config/context_presets/slack_engineering.json +21 -0
  154. contextsynapse/config/context_presets/tech_climate.json +24 -0
  155. contextsynapse/config/context_presets/toi_politics.json +25 -0
  156. contextsynapse/config/database_config.py +453 -0
  157. contextsynapse/config/deployment.py +221 -0
  158. contextsynapse/config/gpt_action_schema.json +164 -0
  159. contextsynapse/config/indicators/india_macro.yaml +213 -0
  160. contextsynapse/config/logging_config.py +88 -0
  161. contextsynapse/config/mode_config.py +223 -0
  162. contextsynapse/config/schemas/api_spec.yaml +48 -0
  163. contextsynapse/config/schemas/architecture_doc.yaml +51 -0
  164. contextsynapse/config/schemas/chat_history.yaml +90 -0
  165. contextsynapse/config/schemas/chatgpt_conversation.yaml +101 -0
  166. contextsynapse/config/schemas/claude_conversation.yaml +148 -0
  167. contextsynapse/config/schemas/contract.yaml +48 -0
  168. contextsynapse/config/schemas/conversation.yaml +167 -0
  169. contextsynapse/config/schemas/database.yaml +74 -0
  170. contextsynapse/config/schemas/decision.yaml +72 -0
  171. contextsynapse/config/schemas/ecommerce.yaml +56 -0
  172. contextsynapse/config/schemas/education.yaml +42 -0
  173. contextsynapse/config/schemas/healthcare.yaml +133 -0
  174. contextsynapse/config/schemas/hr_document.yaml +48 -0
  175. contextsynapse/config/schemas/invoice.yaml +40 -0
  176. contextsynapse/config/schemas/knowledge_base.yaml +75 -0
  177. contextsynapse/config/schemas/legal.yaml +42 -0
  178. contextsynapse/config/schemas/marketing_content.yaml +48 -0
  179. contextsynapse/config/schemas/meeting_notes.yaml +51 -0
  180. contextsynapse/config/schemas/news_article.yaml +61 -0
  181. contextsynapse/config/schemas/real_estate.yaml +49 -0
  182. contextsynapse/config/schemas/requirements_doc.yaml +48 -0
  183. contextsynapse/config/schemas/research_paper.yaml +56 -0
  184. contextsynapse/config/schemas/resume.yaml +49 -0
  185. contextsynapse/config/schemas/retail.yaml +42 -0
  186. contextsynapse/config/schemas/rule_meta.yaml +173 -0
  187. contextsynapse/config/schemas/rules.yaml +72 -0
  188. contextsynapse/config/schemas/screener_meta.yaml +236 -0
  189. contextsynapse/config/schemas/sdlc.yaml +103 -0
  190. contextsynapse/config/schemas/session_graph.yaml +172 -0
  191. contextsynapse/config/schemas/support_ticket.yaml +48 -0
  192. contextsynapse/config/schemas/web.yaml +64 -0
  193. contextsynapse/config/sensors/earnings_calendar.yaml +52 -0
  194. contextsynapse/config/sensors/global_risk.yaml +53 -0
  195. contextsynapse/config/sensors/government_policy.yaml +54 -0
  196. contextsynapse/config/sensors/supply_chain.yaml +50 -0
  197. contextsynapse/config/sensors/templates/behavioral.yaml +56 -0
  198. contextsynapse/config/sensors/templates/equity_sentiment.yaml +48 -0
  199. contextsynapse/config/sensors/templates/global_macro.yaml +47 -0
  200. contextsynapse/config/sensors/templates/market_data.yaml +118 -0
  201. contextsynapse/config/sensors/templates/sector_technical.yaml +47 -0
  202. contextsynapse/config/sensors/weather.yaml +52 -0
  203. contextsynapse/config/storage_strategy.py +202 -0
  204. contextsynapse/config/templates/competitive_intel.yaml +39 -0
  205. contextsynapse/config/templates/geopolitical_risk.yaml +46 -0
  206. contextsynapse/config/templates/market_analysis.yaml +58 -0
  207. contextsynapse/connectors/__init__.py +15 -0
  208. contextsynapse/connectors/api_router.py +121 -0
  209. contextsynapse/connectors/base.py +357 -0
  210. contextsynapse/connectors/financial/audio.py +170 -0
  211. contextsynapse/connectors/financial/youtube.py +272 -0
  212. contextsynapse/connectors/github.py +164 -0
  213. contextsynapse/connectors/jira.py +120 -0
  214. contextsynapse/connectors/kafka_connector.py +124 -0
  215. contextsynapse/connectors/pipeline_connector.py +747 -0
  216. contextsynapse/connectors/registry.py +116 -0
  217. contextsynapse/connectors/scheduler.py +87 -0
  218. contextsynapse/connectors/universal.py +360 -0
  219. contextsynapse/context/__init__.py +96 -0
  220. contextsynapse/context/__main__.py +4 -0
  221. contextsynapse/context/acl.py +125 -0
  222. contextsynapse/context/agent_card.py +201 -0
  223. contextsynapse/context/agent_memory.py +814 -0
  224. contextsynapse/context/agents.py +472 -0
  225. contextsynapse/context/agents_redis.py +391 -0
  226. contextsynapse/context/assembled.py +204 -0
  227. contextsynapse/context/attribution.py +113 -0
  228. contextsynapse/context/blob.py +184 -0
  229. contextsynapse/context/boundaries.py +187 -0
  230. contextsynapse/context/boundary.py +507 -0
  231. contextsynapse/context/bundle.py +155 -0
  232. contextsynapse/context/cli.py +519 -0
  233. contextsynapse/context/collaboration.py +165 -0
  234. contextsynapse/context/compiler.py +604 -0
  235. contextsynapse/context/composite.py +536 -0
  236. contextsynapse/context/compression.py +147 -0
  237. contextsynapse/context/context_manager.py +888 -0
  238. contextsynapse/context/context_schema.py +90 -0
  239. contextsynapse/context/context_schema.yaml +105 -0
  240. contextsynapse/context/context_units.py +891 -0
  241. contextsynapse/context/conversation.py +488 -0
  242. contextsynapse/context/dedup.py +207 -0
  243. contextsynapse/context/document_processor.py +698 -0
  244. contextsynapse/context/embedding_hooks.py +338 -0
  245. contextsynapse/context/execution_schema.py +274 -0
  246. contextsynapse/context/fan_out.py +138 -0
  247. contextsynapse/context/frozen.py +80 -0
  248. contextsynapse/context/gravity.py +227 -0
  249. contextsynapse/context/hub.py +1401 -0
  250. contextsynapse/context/ingest.py +391 -0
  251. contextsynapse/context/injection.py +226 -0
  252. contextsynapse/context/interaction_graphifier.py +562 -0
  253. contextsynapse/context/layers.py +574 -0
  254. contextsynapse/context/projection.py +1195 -0
  255. contextsynapse/context/promotion.py +634 -0
  256. contextsynapse/context/proof.py +131 -0
  257. contextsynapse/context/propagation.py +577 -0
  258. contextsynapse/context/pubsub.py +135 -0
  259. contextsynapse/context/quality.py +1054 -0
  260. contextsynapse/context/quality_tracker.py +270 -0
  261. contextsynapse/context/scoping.py +370 -0
  262. contextsynapse/context/seed.py +992 -0
  263. contextsynapse/context/session.py +1142 -0
  264. contextsynapse/context/session_graph.py +593 -0
  265. contextsynapse/context/session_memory.py +309 -0
  266. contextsynapse/context/session_resolver.py +111 -0
  267. contextsynapse/context/skill_learning.py +237 -0
  268. contextsynapse/context/smart_budget.py +129 -0
  269. contextsynapse/context/source_policy.py +242 -0
  270. contextsynapse/context/store_factory.py +53 -0
  271. contextsynapse/context/sync.py +648 -0
  272. contextsynapse/context/token.py +236 -0
  273. contextsynapse/context/vector_integration.py +297 -0
  274. contextsynapse/context/webhooks.py +194 -0
  275. contextsynapse/context/working_memory.py +224 -0
  276. contextsynapse/core/__init__.py +18 -0
  277. contextsynapse/core/archiver.py +237 -0
  278. contextsynapse/core/change_stream.py +282 -0
  279. contextsynapse/core/checkpoint.py +310 -0
  280. contextsynapse/core/cloud_storage.py +387 -0
  281. contextsynapse/core/context_state.py +651 -0
  282. contextsynapse/core/cron.py +324 -0
  283. contextsynapse/core/datasource.py +643 -0
  284. contextsynapse/core/db.py +289 -0
  285. contextsynapse/core/federation.py +283 -0
  286. contextsynapse/core/graph_coordinator.py +201 -0
  287. contextsynapse/core/graph_intelligence.py +388 -0
  288. contextsynapse/core/graph_structures.py +106 -0
  289. contextsynapse/core/graph_sync.py +157 -0
  290. contextsynapse/core/hybrid_graph_storage.py +2998 -0
  291. contextsynapse/core/lifecycle.py +691 -0
  292. contextsynapse/core/multiworker.py +378 -0
  293. contextsynapse/core/pruning.py +312 -0
  294. contextsynapse/core/redis_registry.py +524 -0
  295. contextsynapse/core/registry.py +1043 -0
  296. contextsynapse/core/registry_factory.py +51 -0
  297. contextsynapse/core/registry_metadata.py +294 -0
  298. contextsynapse/core/replay.py +354 -0
  299. contextsynapse/core/replication.py +339 -0
  300. contextsynapse/core/time_travel.py +260 -0
  301. contextsynapse/core/write_behind.py +152 -0
  302. contextsynapse/core/write_context.py +39 -0
  303. contextsynapse/db/__init__.py +1 -0
  304. contextsynapse/db/postgres.py +267 -0
  305. contextsynapse/db/profile.py +236 -0
  306. contextsynapse/db/rules.py +328 -0
  307. contextsynapse/demo/__init__.py +0 -0
  308. contextsynapse/demo/report.py +147 -0
  309. contextsynapse/demo/runner.py +520 -0
  310. contextsynapse/demo/scenario.py +165 -0
  311. contextsynapse/demo/session_demo.py +280 -0
  312. contextsynapse/demo/session_report.py +143 -0
  313. contextsynapse/demo/universal_runner.py +257 -0
  314. contextsynapse/engine/__init__.py +15 -0
  315. contextsynapse/engine/core.py +555 -0
  316. contextsynapse/extraction/__init__.py +69 -0
  317. contextsynapse/extraction/cli.py +274 -0
  318. contextsynapse/extraction/config.py +107 -0
  319. contextsynapse/extraction/examples/example_usage.py +118 -0
  320. contextsynapse/extraction/examples/sample_manifest.json +53 -0
  321. contextsynapse/extraction/examples/sample_normalized.json +185 -0
  322. contextsynapse/extraction/extract_engine.py +440 -0
  323. contextsynapse/extraction/extractors/__init__.py +37 -0
  324. contextsynapse/extraction/extractors/chat_export_extractor.py +189 -0
  325. contextsynapse/extraction/extractors/csv_extractor.py +83 -0
  326. contextsynapse/extraction/extractors/docx_extractor.py +129 -0
  327. contextsynapse/extraction/extractors/excel_extractor.py +152 -0
  328. contextsynapse/extraction/extractors/html_extractor.py +146 -0
  329. contextsynapse/extraction/extractors/pdf_extractor.py +635 -0
  330. contextsynapse/extraction/extractors/pdf_extractor_layoutparser.py +408 -0
  331. contextsynapse/extraction/extractors/register_extractors.py +63 -0
  332. contextsynapse/extraction/extractors/text_extractor.py +65 -0
  333. contextsynapse/extraction/extractors/txt_extractor.py +120 -0
  334. contextsynapse/extraction/extractors/website_extractor.py +246 -0
  335. contextsynapse/extraction/fact_extractor.py +236 -0
  336. contextsynapse/extraction/file_manager.py +353 -0
  337. contextsynapse/extraction/hierarchy.py +239 -0
  338. contextsynapse/extraction/id_generator.py +450 -0
  339. contextsynapse/extraction/layout_detector.py +409 -0
  340. contextsynapse/extraction/llm_entity_extractor.py +301 -0
  341. contextsynapse/extraction/normalized_store.py +480 -0
  342. contextsynapse/extraction/normalizer.py +594 -0
  343. contextsynapse/extraction/parquet_writer.py +598 -0
  344. contextsynapse/extraction/post_processors/__init__.py +40 -0
  345. contextsynapse/extraction/post_processors/column_reorganizer.py +507 -0
  346. contextsynapse/extraction/ray_normalization.py +233 -0
  347. contextsynapse/extraction/ray_runner.py +207 -0
  348. contextsynapse/extraction/redis_schema_registry.py +222 -0
  349. contextsynapse/extraction/registry.py +204 -0
  350. contextsynapse/extraction/schema_loader.py +663 -0
  351. contextsynapse/extraction/schema_registry.py +230 -0
  352. contextsynapse/extraction/semantic_normalizer.py +245 -0
  353. contextsynapse/extraction/version_manager.py +314 -0
  354. contextsynapse/gateway/__init__.py +24 -0
  355. contextsynapse/gateway/action_emitter.py +212 -0
  356. contextsynapse/gateway/cost_tracker.py +508 -0
  357. contextsynapse/gateway/gateway.py +377 -0
  358. contextsynapse/gateway/policy.py +303 -0
  359. contextsynapse/governance/__init__.py +13 -0
  360. contextsynapse/governance/layer.py +513 -0
  361. contextsynapse/ingestion/__init__.py +15 -0
  362. contextsynapse/ingestion/amplifier.py +288 -0
  363. contextsynapse/ingestion/async_ingest.py +463 -0
  364. contextsynapse/ingestion/chunker.py +303 -0
  365. contextsynapse/ingestion/chunking.py +58 -0
  366. contextsynapse/ingestion/cleaner.py +190 -0
  367. contextsynapse/ingestion/cleanse.py +252 -0
  368. contextsynapse/ingestion/connectors/__init__.py +38 -0
  369. contextsynapse/ingestion/connectors/base.py +300 -0
  370. contextsynapse/ingestion/connectors/chatgpt_memory.py +247 -0
  371. contextsynapse/ingestion/connectors/claude_memory.py +274 -0
  372. contextsynapse/ingestion/connectors/sap_ingestor.py +208 -0
  373. contextsynapse/ingestion/content_detector.py +80 -0
  374. contextsynapse/ingestion/content_monitor.py +502 -0
  375. contextsynapse/ingestion/dedup.py +106 -0
  376. contextsynapse/ingestion/feed.py +640 -0
  377. contextsynapse/ingestion/filters.py +461 -0
  378. contextsynapse/ingestion/graph_builder.py +705 -0
  379. contextsynapse/ingestion/input_classifier.py +307 -0
  380. contextsynapse/ingestion/job_manager.py +571 -0
  381. contextsynapse/ingestion/llm_extractor.py +610 -0
  382. contextsynapse/ingestion/orchestrator.py +116 -0
  383. contextsynapse/ingestion/parsers/__init__.py +19 -0
  384. contextsynapse/ingestion/parsers/graph_parser_registry.py +103 -0
  385. contextsynapse/ingestion/parsers/graphml_parser.py +141 -0
  386. contextsynapse/ingestion/parsers/jsonld_parser.py +171 -0
  387. contextsynapse/ingestion/parsers/rdf_parser.py +150 -0
  388. contextsynapse/ingestion/pipeline_context.py +228 -0
  389. contextsynapse/ingestion/pipeline_store.py +1195 -0
  390. contextsynapse/ingestion/queue.py +133 -0
  391. contextsynapse/ingestion/reprocess.py +179 -0
  392. contextsynapse/ingestion/scenario_router.py +224 -0
  393. contextsynapse/ingestion/schema_extractor.py +759 -0
  394. contextsynapse/ingestion/schema_validator.py +274 -0
  395. contextsynapse/ingestion/sdlc_ingest.py +87 -0
  396. contextsynapse/ingestion/smart_ingest.py +2451 -0
  397. contextsynapse/ingestion/source_connector.py +171 -0
  398. contextsynapse/ingestion/stage_executor.py +2095 -0
  399. contextsynapse/ingestion/stages/__init__.py +9 -0
  400. contextsynapse/ingestion/stages/tabular_to_graph.py +289 -0
  401. contextsynapse/ingestion/strategies.py +386 -0
  402. contextsynapse/ingestion/universal/__init__.py +4 -0
  403. contextsynapse/ingestion/universal/_operator_registry.py +59 -0
  404. contextsynapse/ingestion/universal/ingest_content.py +32 -0
  405. contextsynapse/ingestion/universal/operators/__init__.py +1 -0
  406. contextsynapse/ingestion/universal/operators/base.py +19 -0
  407. contextsynapse/ingestion/universal/operators/build_edges.py +75 -0
  408. contextsynapse/ingestion/universal/operators/cluster_topics.py +397 -0
  409. contextsynapse/ingestion/universal/operators/deduplicate.py +45 -0
  410. contextsynapse/ingestion/universal/operators/detect_signals.py +90 -0
  411. contextsynapse/ingestion/universal/operators/embed.py +80 -0
  412. contextsynapse/ingestion/universal/operators/extract_entities.py +253 -0
  413. contextsynapse/ingestion/universal/operators/extract_preferences.py +117 -0
  414. contextsynapse/ingestion/universal/operators/filter_content.py +69 -0
  415. contextsynapse/ingestion/universal/operators/index_bm25.py +52 -0
  416. contextsynapse/ingestion/universal/operators/infer_domains.py +69 -0
  417. contextsynapse/ingestion/universal/operators/link_cross_reference.py +210 -0
  418. contextsynapse/ingestion/universal/operators/parse_records.py +66 -0
  419. contextsynapse/ingestion/universal/operators/resolve_entities.py +261 -0
  420. contextsynapse/ingestion/universal/operators/scanners/__init__.py +16 -0
  421. contextsynapse/ingestion/universal/operators/scanners/edges.py +173 -0
  422. contextsynapse/ingestion/universal/operators/scanners/git.py +187 -0
  423. contextsynapse/ingestion/universal/operators/scanners/github.py +135 -0
  424. contextsynapse/ingestion/universal/operators/scanners/github_api.py +509 -0
  425. contextsynapse/ingestion/universal/operators/scanners/helpers.py +67 -0
  426. contextsynapse/ingestion/universal/operators/scanners/jira.py +65 -0
  427. contextsynapse/ingestion/universal/operators/scanners/llm_enrichment.py +82 -0
  428. contextsynapse/ingestion/universal/operators/scanners/repo.py +149 -0
  429. contextsynapse/ingestion/universal/operators/sdlc_scan.py +213 -0
  430. contextsynapse/ingestion/universal/operators/spec_scanner.py +149 -0
  431. contextsynapse/ingestion/universal/operators/store_documents.py +72 -0
  432. contextsynapse/ingestion/universal/operators/synthesize_cu.py +180 -0
  433. contextsynapse/ingestion/universal/operators/validate_gate.py +65 -0
  434. contextsynapse/ingestion/universal/parsers/__init__.py +5 -0
  435. contextsynapse/ingestion/universal/parsers/record_parser.py +203 -0
  436. contextsynapse/ingestion/universal/parsers/turn_parser.py +329 -0
  437. contextsynapse/ingestion/universal/shared_signals.py +161 -0
  438. contextsynapse/ingestion/universal/stage_executor.py +219 -0
  439. contextsynapse/ingestion/web_crawler.py +265 -0
  440. contextsynapse/intelligence/__init__.py +53 -0
  441. contextsynapse/intelligence/alerts.py +198 -0
  442. contextsynapse/intelligence/collector.py +655 -0
  443. contextsynapse/intelligence/comparison.py +198 -0
  444. contextsynapse/intelligence/config.py +56 -0
  445. contextsynapse/intelligence/conflict_detector.py +182 -0
  446. contextsynapse/intelligence/context_gaps.py +192 -0
  447. contextsynapse/intelligence/context_radar.py +135 -0
  448. contextsynapse/intelligence/correlation_engine.py +416 -0
  449. contextsynapse/intelligence/correlation_template.py +120 -0
  450. contextsynapse/intelligence/dedup.py +296 -0
  451. contextsynapse/intelligence/event_bus.py +91 -0
  452. contextsynapse/intelligence/feedback_loop.py +101 -0
  453. contextsynapse/intelligence/freshness.py +307 -0
  454. contextsynapse/intelligence/fundamental_feed.py +551 -0
  455. contextsynapse/intelligence/geo_data.py +170 -0
  456. contextsynapse/intelligence/geo_reference.py +112 -0
  457. contextsynapse/intelligence/graph_fusion.py +465 -0
  458. contextsynapse/intelligence/heuristics.py +176 -0
  459. contextsynapse/intelligence/impact_tracker.py +336 -0
  460. contextsynapse/intelligence/macro_indicators.py +319 -0
  461. contextsynapse/intelligence/ocr.py +104 -0
  462. contextsynapse/intelligence/persistence.py +267 -0
  463. contextsynapse/intelligence/pipeline_health.py +130 -0
  464. contextsynapse/intelligence/price_feed.py +221 -0
  465. contextsynapse/intelligence/rate_limiter.py +130 -0
  466. contextsynapse/intelligence/reactive.py +411 -0
  467. contextsynapse/intelligence/runtime_context.py +390 -0
  468. contextsynapse/intelligence/sentiment_decay.py +426 -0
  469. contextsynapse/intelligence/session.py +277 -0
  470. contextsynapse/intelligence/signal_hierarchy.py +438 -0
  471. contextsynapse/intelligence/signals.py +185 -0
  472. contextsynapse/intelligence/source_watcher.py +132 -0
  473. contextsynapse/intelligence/table_enricher.py +256 -0
  474. contextsynapse/intelligence/tagger.py +319 -0
  475. contextsynapse/intelligence/timeseries.py +209 -0
  476. contextsynapse/intelligence/tracker.py +318 -0
  477. contextsynapse/intelligence/watchdog.py +392 -0
  478. contextsynapse/intelligence/watchdog_manager.py +263 -0
  479. contextsynapse/intelligence/webhook_receiver.py +165 -0
  480. contextsynapse/intelligence/weight_learner.py +234 -0
  481. contextsynapse/llm/__init__.py +8 -0
  482. contextsynapse/llm/client.py +348 -0
  483. contextsynapse/llm/rate_limiter.py +166 -0
  484. contextsynapse/logging_config.py +2 -0
  485. contextsynapse/marketplace/__init__.py +7 -0
  486. contextsynapse/marketplace/registry.py +173 -0
  487. contextsynapse/mcp/__init__.py +11 -0
  488. contextsynapse/mcp/__main__.py +4 -0
  489. contextsynapse/mcp/auth_middleware.py +212 -0
  490. contextsynapse/mcp/connection_pool.py +128 -0
  491. contextsynapse/mcp/server.py +986 -0
  492. contextsynapse/mcp/session_context.py +67 -0
  493. contextsynapse/memory/__init__.py +34 -0
  494. contextsynapse/memory/engine.py +475 -0
  495. contextsynapse/memory/temporal.py +327 -0
  496. contextsynapse/metadata/__init__.py +12 -0
  497. contextsynapse/metadata/metadata_db.py +272 -0
  498. contextsynapse/metadata/metadata_query_tool.py +300 -0
  499. contextsynapse/metadata/metadata_tracker.py +425 -0
  500. contextsynapse/models/__init__.py +27 -0
  501. contextsynapse/models/embedding_service.py +360 -0
  502. contextsynapse/models/model_registry.py +228 -0
  503. contextsynapse/multiworker.py +2 -0
  504. contextsynapse/pipelines/__init__.py +1 -0
  505. contextsynapse/pipelines/executor.py +851 -0
  506. contextsynapse/pipelines/filters.py +189 -0
  507. contextsynapse/pipelines/models.py +176 -0
  508. contextsynapse/pipelines/scheduler.py +268 -0
  509. contextsynapse/playground/__init__.py +1 -0
  510. contextsynapse/playground/templates.py +85 -0
  511. contextsynapse/plugins/__init__.py +43 -0
  512. contextsynapse/plugins/api.py +66 -0
  513. contextsynapse/plugins/base.py +350 -0
  514. contextsynapse/plugins/domain_interface.py +508 -0
  515. contextsynapse/plugins/domain_loader.py +241 -0
  516. contextsynapse/plugins/loader.py +148 -0
  517. contextsynapse/plugins/registry.py +265 -0
  518. contextsynapse/project/__init__.py +11 -0
  519. contextsynapse/project/code_context.py +807 -0
  520. contextsynapse/project/connectors/__init__.py +4 -0
  521. contextsynapse/project/connectors/github_connector.py +113 -0
  522. contextsynapse/project/connectors/jira_connector.py +160 -0
  523. contextsynapse/project/graph.py +1023 -0
  524. contextsynapse/project/project_context.py +545 -0
  525. contextsynapse/project/scanner_registry.py +61 -0
  526. contextsynapse/project/schema_manager.py +322 -0
  527. contextsynapse/project/schemas/knowledge.yaml +59 -0
  528. contextsynapse/project/schemas/sdlc.yaml +369 -0
  529. contextsynapse/project/sdlc_schema.py +172 -0
  530. contextsynapse/project/spec_parser.py +85 -0
  531. contextsynapse/project/task_lock.py +181 -0
  532. contextsynapse/project/task_stream.py +203 -0
  533. contextsynapse/project/team_tools.py +537 -0
  534. contextsynapse/py.typed +0 -0
  535. contextsynapse/realtime/__init__.py +6 -0
  536. contextsynapse/realtime/collaboration.py +147 -0
  537. contextsynapse/rules/__init__.py +10 -0
  538. contextsynapse/rules/engine.py +116 -0
  539. contextsynapse/rules/evaluators/__init__.py +6 -0
  540. contextsynapse/rules/evaluators/aggregate.py +205 -0
  541. contextsynapse/rules/evaluators/custom.py +170 -0
  542. contextsynapse/rules/evaluators/pre_action.py +125 -0
  543. contextsynapse/rules/evaluators/temporal.py +129 -0
  544. contextsynapse/rules/loader.py +52 -0
  545. contextsynapse/rules/models.py +124 -0
  546. contextsynapse/rules/universal.py +393 -0
  547. contextsynapse/scheduler.py +229 -0
  548. contextsynapse/schema/__init__.py +42 -0
  549. contextsynapse/schema/compiler.py +252 -0
  550. contextsynapse/schema/composer.py +158 -0
  551. contextsynapse/schema/dedup.py +73 -0
  552. contextsynapse/schema/derivation.py +172 -0
  553. contextsynapse/schema/sdl.py +326 -0
  554. contextsynapse/schema/validation_gate.py +124 -0
  555. contextsynapse/sdk/__init__.py +63 -0
  556. contextsynapse/sdk/agent.py +214 -0
  557. contextsynapse/sdk/client.py +269 -0
  558. contextsynapse/sdk/exceptions.py +23 -0
  559. contextsynapse/sdk/models.py +78 -0
  560. contextsynapse/sdk/session.py +408 -0
  561. contextsynapse/sdk/tracked.py +218 -0
  562. contextsynapse/sdk/worker.py +266 -0
  563. contextsynapse/sdk/workspace_client.py +98 -0
  564. contextsynapse/sdk/ws.py +41 -0
  565. contextsynapse/search/__init__.py +20 -0
  566. contextsynapse/search/embedding_cache.py +140 -0
  567. contextsynapse/search/enhanced_search.py +152 -0
  568. contextsynapse/search/fulltext.py +130 -0
  569. contextsynapse/search/graph_search.py +634 -0
  570. contextsynapse/search/lmdb_index.py +1218 -0
  571. contextsynapse/search/nl_to_aiql.py +199 -0
  572. contextsynapse/search/rag.py +1346 -0
  573. contextsynapse/search/rag_cache.py +167 -0
  574. contextsynapse/search/reasoning_chain.py +231 -0
  575. contextsynapse/search/redis_search.py +207 -0
  576. contextsynapse/search/retrieval_quality.py +314 -0
  577. contextsynapse/search/semantic_query.py +455 -0
  578. contextsynapse/search/text_resolver.py +159 -0
  579. contextsynapse/search/whoosh_search.py +373 -0
  580. contextsynapse/security/__init__.py +17 -0
  581. contextsynapse/security/audit.py +297 -0
  582. contextsynapse/security/audit_logger.py +97 -0
  583. contextsynapse/security/audit_trail.py +270 -0
  584. contextsynapse/security/auth_unified.py +284 -0
  585. contextsynapse/security/auto_tagger.py +230 -0
  586. contextsynapse/security/data_security.py +218 -0
  587. contextsynapse/security/encryption.py +207 -0
  588. contextsynapse/security/identity.py +447 -0
  589. contextsynapse/security/jwt_identity.py +59 -0
  590. contextsynapse/security/middleware.py +336 -0
  591. contextsynapse/security/pii.py +233 -0
  592. contextsynapse/security/rbac.py +96 -0
  593. contextsynapse/security/rls.py +205 -0
  594. contextsynapse/security/sanitize.py +233 -0
  595. contextsynapse/security/scoped_encryption.py +131 -0
  596. contextsynapse/security/tenant.py +401 -0
  597. contextsynapse/shield/__init__.py +21 -0
  598. contextsynapse/shield/anomaly.py +56 -0
  599. contextsynapse/shield/permissions.py +99 -0
  600. contextsynapse/shield/profile.py +103 -0
  601. contextsynapse/shield/shield.py +108 -0
  602. contextsynapse/shield/trust_engine.py +69 -0
  603. contextsynapse/skills/__init__.py +14 -0
  604. contextsynapse/skills/engine.py +233 -0
  605. contextsynapse/skills/loader.py +59 -0
  606. contextsynapse/skills/models.py +139 -0
  607. contextsynapse/storage/__init__.py +7 -0
  608. contextsynapse/storage/cache.py +208 -0
  609. contextsynapse/storage/columnar_store_v2.py +178 -0
  610. contextsynapse/storage/csr_graph_storage.py +587 -0
  611. contextsynapse/storage/csr_store.py +388 -0
  612. contextsynapse/storage/document_store.py +491 -0
  613. contextsynapse/storage/hnsw_index.py +294 -0
  614. contextsynapse/storage/hybrid_store_v2.py +278 -0
  615. contextsynapse/storage/lmdb_graph_storage.py +366 -0
  616. contextsynapse/storage/migration.py +279 -0
  617. contextsynapse/storage/namespace_store.py +346 -0
  618. contextsynapse/storage/pure_graph_storage.py +185 -0
  619. contextsynapse/storage/redis_graph_adapter.py +186 -0
  620. contextsynapse/storage/redis_graph_storage.py +375 -0
  621. contextsynapse/storage/router/__init__.py +29 -0
  622. contextsynapse/storage/router/duckdb_store.py +398 -0
  623. contextsynapse/storage/router/factory.py +83 -0
  624. contextsynapse/storage/router/interface.py +125 -0
  625. contextsynapse/storage/router/postgres_store.py +339 -0
  626. contextsynapse/storage/router/resolver.py +192 -0
  627. contextsynapse/storage/router/router.py +300 -0
  628. contextsynapse/storage/single_file_storage.py +391 -0
  629. contextsynapse/storage/storage_manager.py +214 -0
  630. contextsynapse/storage/storage_query_optimizer.py +314 -0
  631. contextsynapse/storage/wal.py +359 -0
  632. contextsynapse/temporal/__init__.py +0 -0
  633. contextsynapse/temporal/storage.py +106 -0
  634. contextsynapse/tests/__init__.py +11 -0
  635. contextsynapse/tests/regression/__init__.py +1 -0
  636. contextsynapse/tests/regression/add_metadata.py +395 -0
  637. contextsynapse/tests/regression/add_skip_flags.py +85 -0
  638. contextsynapse/tests/regression/cases/test_basic_create_node.yaml +20 -0
  639. contextsynapse/tests/regression/cases/test_basic_select.yaml +21 -0
  640. contextsynapse/tests/regression/cases/test_blockchain_audit_trail.yaml +37 -0
  641. contextsynapse/tests/regression/cases/test_blockchain_audit_trail_filtered.yaml +39 -0
  642. contextsynapse/tests/regression/cases/test_blockchain_blocks_range.yaml +39 -0
  643. contextsynapse/tests/regression/cases/test_blockchain_get_block.yaml +35 -0
  644. contextsynapse/tests/regression/cases/test_blockchain_latest_block.yaml +35 -0
  645. contextsynapse/tests/regression/cases/test_blockchain_length.yaml +37 -0
  646. contextsynapse/tests/regression/cases/test_blockchain_merkle_root.yaml +37 -0
  647. contextsynapse/tests/regression/cases/test_blockchain_verify.yaml +35 -0
  648. contextsynapse/tests/regression/cases/test_blockchain_verify_block.yaml +35 -0
  649. contextsynapse/tests/regression/cases/test_create_edge.yaml +28 -0
  650. contextsynapse/tests/regression/cases/test_delete_edge_basic.yaml +29 -0
  651. contextsynapse/tests/regression/cases/test_delete_node_basic.yaml +29 -0
  652. contextsynapse/tests/regression/cases/test_evaluation_mrr.yaml +40 -0
  653. contextsynapse/tests/regression/cases/test_evaluation_ndcg.yaml +40 -0
  654. contextsynapse/tests/regression/cases/test_evaluation_precision.yaml +40 -0
  655. contextsynapse/tests/regression/cases/test_evaluation_recall.yaml +40 -0
  656. contextsynapse/tests/regression/cases/test_group_by_basic.yaml +56 -0
  657. contextsynapse/tests/regression/cases/test_match_node_basic.yaml +29 -0
  658. contextsynapse/tests/regression/cases/test_match_node_with_order_limit.yaml +31 -0
  659. contextsynapse/tests/regression/cases/test_namespace_switch.yaml +18 -0
  660. contextsynapse/tests/regression/cases/test_pipeline_cascaded.yaml +61 -0
  661. contextsynapse/tests/regression/cases/test_pipeline_fixed_size_chunking.yaml +48 -0
  662. contextsynapse/tests/regression/cases/test_pipeline_gpt4_chunking.yaml +54 -0
  663. contextsynapse/tests/regression/cases/test_pipeline_hybrid_indexing.yaml +57 -0
  664. contextsynapse/tests/regression/cases/test_pipeline_large_embedding.yaml +52 -0
  665. contextsynapse/tests/regression/cases/test_pipeline_paragraph_chunking.yaml +49 -0
  666. contextsynapse/tests/regression/cases/test_pipeline_recursive_chunking.yaml +47 -0
  667. contextsynapse/tests/regression/cases/test_pipeline_run_pipeline.yaml +56 -0
  668. contextsynapse/tests/regression/cases/test_pipeline_section_based_chunking.yaml +53 -0
  669. contextsynapse/tests/regression/cases/test_pipeline_semantic_chunking.yaml +51 -0
  670. contextsynapse/tests/regression/cases/test_pipeline_with_entity_extraction.yaml +57 -0
  671. contextsynapse/tests/regression/cases/test_pipeline_with_relationships.yaml +62 -0
  672. contextsynapse/tests/regression/cases/test_rag_evaluation.yaml +38 -0
  673. contextsynapse/tests/regression/cases/test_rag_graph_bfs_with_eval.yaml +48 -0
  674. contextsynapse/tests/regression/cases/test_rag_graph_centrality.yaml +40 -0
  675. contextsynapse/tests/regression/cases/test_rag_graph_centrality_eval.yaml +35 -0
  676. contextsynapse/tests/regression/cases/test_rag_graph_community.yaml +40 -0
  677. contextsynapse/tests/regression/cases/test_rag_graph_community_eval.yaml +35 -0
  678. contextsynapse/tests/regression/cases/test_rag_graph_dfs_with_eval.yaml +35 -0
  679. contextsynapse/tests/regression/cases/test_rag_graph_multi_hop_eval.yaml +35 -0
  680. contextsynapse/tests/regression/cases/test_rag_graph_path.yaml +40 -0
  681. contextsynapse/tests/regression/cases/test_rag_graph_search_basic.yaml +38 -0
  682. contextsynapse/tests/regression/cases/test_rag_graph_search_bfs.yaml +52 -0
  683. contextsynapse/tests/regression/cases/test_rag_graph_search_dfs.yaml +41 -0
  684. contextsynapse/tests/regression/cases/test_rag_graph_search_multi_hop.yaml +43 -0
  685. contextsynapse/tests/regression/cases/test_rag_graph_search_shortest_path.yaml +41 -0
  686. contextsynapse/tests/regression/cases/test_rag_graph_search_weighted.yaml +43 -0
  687. contextsynapse/tests/regression/cases/test_rag_graph_shortest_path_eval.yaml +35 -0
  688. contextsynapse/tests/regression/cases/test_rag_graph_traversal.yaml +30 -0
  689. contextsynapse/tests/regression/cases/test_rag_graph_weighted_eval.yaml +35 -0
  690. contextsynapse/tests/regression/cases/test_rag_hybrid_search.yaml +33 -0
  691. contextsynapse/tests/regression/cases/test_rag_hybrid_with_prompt_embedding.yaml +44 -0
  692. contextsynapse/tests/regression/cases/test_rag_keyword_search.yaml +32 -0
  693. contextsynapse/tests/regression/cases/test_rag_multi_hop_graph.yaml +32 -0
  694. contextsynapse/tests/regression/cases/test_rag_path_based.yaml +34 -0
  695. contextsynapse/tests/regression/cases/test_rag_profile_based.yaml +32 -0
  696. contextsynapse/tests/regression/cases/test_rag_vector_search.yaml +32 -0
  697. contextsynapse/tests/regression/cases/test_rag_weighted_graph.yaml +34 -0
  698. contextsynapse/tests/regression/cases/test_rag_weighted_hybrid.yaml +31 -0
  699. contextsynapse/tests/regression/cases/test_semantic_boundary_chunking.yaml +57 -0
  700. contextsynapse/tests/regression/cases/test_semantic_boundary_detection.yaml +33 -0
  701. contextsynapse/tests/regression/cases/test_semantic_hash_function.yaml +30 -0
  702. contextsynapse/tests/regression/cases/test_semantic_hash_with_model.yaml +31 -0
  703. contextsynapse/tests/regression/cases/test_tcs_ingestion_fixed_size.yaml +44 -0
  704. contextsynapse/tests/regression/cases/test_tcs_ingestion_recursive.yaml +46 -0
  705. contextsynapse/tests/regression/cases/test_tcs_ingestion_section.yaml +46 -0
  706. contextsynapse/tests/regression/cases/test_tcs_ingestion_semantic.yaml +44 -0
  707. contextsynapse/tests/regression/cases/test_time_travel_as_of.yaml +35 -0
  708. contextsynapse/tests/regression/cases/test_time_travel_at_commit.yaml +33 -0
  709. contextsynapse/tests/regression/cases/test_time_travel_at_timestamp.yaml +33 -0
  710. contextsynapse/tests/regression/cases/test_time_travel_for_system_time_all.yaml +37 -0
  711. contextsynapse/tests/regression/cases/test_time_travel_versions_between.yaml +39 -0
  712. contextsynapse/tests/regression/cases/test_traverse_basic.yaml +31 -0
  713. contextsynapse/tests/regression/cases/test_traverse_max_depth.yaml +34 -0
  714. contextsynapse/tests/regression/cases/test_update_edge_basic.yaml +29 -0
  715. contextsynapse/tests/regression/evaluation_metrics.py +227 -0
  716. contextsynapse/tests/regression/regression_runner.py +1081 -0
  717. contextsynapse/tests/regression/setup_regression_data.py +212 -0
  718. contextsynapse/tests/regression/test_basic_only.py +84 -0
  719. contextsynapse/tests/regression/test_namespace_creation.py +29 -0
  720. contextsynapse/tests/regression/test_single_fix.py +61 -0
  721. contextsynapse/tests/regression/test_single_query.py +20 -0
  722. contextsynapse/tests/regression/validate_test_syntax.py +179 -0
  723. contextsynapse/tests/test_advanced_features.py +347 -0
  724. contextsynapse/tests/test_aiql_summary.py +283 -0
  725. contextsynapse/tests/test_all_aiql_queries.py +948 -0
  726. contextsynapse/tests/test_all_create_queries.py +213 -0
  727. contextsynapse/tests/test_api_vs_native.py +280 -0
  728. contextsynapse/tests/test_chunking_dedup.py +199 -0
  729. contextsynapse/tests/test_connectors.py +171 -0
  730. contextsynapse/tests/test_edges_and_match.py +202 -0
  731. contextsynapse/tests/test_extraction_pipeline_integration.py +581 -0
  732. contextsynapse/tests/test_name_property.py +224 -0
  733. contextsynapse/tests/test_node_edge_metadata.py +250 -0
  734. contextsynapse/tests/test_playground_conn.py +114 -0
  735. contextsynapse/tests/test_query_cache.py +213 -0
  736. contextsynapse/tests/test_source_policy_auth.py +348 -0
  737. contextsynapse/tests/test_temporal_and_domain.py +224 -0
  738. contextsynapse/tests/test_time_travel_e2e.py +571 -0
  739. contextsynapse/tests/test_traversal_and_aggregation.py +275 -0
  740. contextsynapse/tools/__init__.py +36 -0
  741. contextsynapse/tools/a2a_tools.py +106 -0
  742. contextsynapse/tools/cognition_tools.py +118 -0
  743. contextsynapse/tools/context.py +905 -0
  744. contextsynapse/tools/formatting.py +92 -0
  745. contextsynapse/tools/gateway_tools.py +186 -0
  746. contextsynapse/tools/graph.py +1700 -0
  747. contextsynapse/tools/intelligence.py +235 -0
  748. contextsynapse/tools/memory.py +118 -0
  749. contextsynapse/tools/registry.py +373 -0
  750. contextsynapse/tools/shield_tools.py +159 -0
  751. contextsynapse/tools/tasks.py +341 -0
  752. contextsynapse/tools/workspace_tools.py +124 -0
  753. contextsynapse/utils/__init__.py +15 -0
  754. contextsynapse/utils/config_loader.py +198 -0
  755. contextsynapse/utils/env_loader.py +156 -0
  756. contextsynapse/utils/file_utils.py +288 -0
  757. contextsynapse/utils/logging_optimizer.py +119 -0
  758. contextsynapse/vector/__init__.py +30 -0
  759. contextsynapse/vector/chroma_vector_store.py +152 -0
  760. contextsynapse/vector/custom_vector_store.py +537 -0
  761. contextsynapse/vector/faiss_vector_store.py +296 -0
  762. contextsynapse/vector/qdrant_vector_store.py +163 -0
  763. contextsynapse/vector/vector_db.py +953 -0
  764. contextsynapse/vector/vector_db_manager.py +197 -0
  765. contextsynapse/workspace/__init__.py +27 -0
  766. contextsynapse/workspace/base.py +94 -0
  767. contextsynapse/workspace/git.py +228 -0
  768. contextsynapse/workspace/github.py +78 -0
  769. contextsynapse/workspace/local.py +73 -0
  770. contextsynapse/workspace/tools.py +222 -0
  771. contextsynapse-1.0.0.dist-info/METADATA +847 -0
  772. contextsynapse-1.0.0.dist-info/RECORD +776 -0
  773. contextsynapse-1.0.0.dist-info/WHEEL +5 -0
  774. contextsynapse-1.0.0.dist-info/entry_points.txt +4 -0
  775. contextsynapse-1.0.0.dist-info/licenses/LICENSE +200 -0
  776. contextsynapse-1.0.0.dist-info/top_level.txt +1 -0
@@ -0,0 +1,807 @@
1
+ """
2
+ Code Context
3
+ ==============
4
+ A spec-driven code context that agents use to build applications.
5
+
6
+ Give it a spec (stack, requirements, rules) and connected agents
7
+ (Claude, Codex) collaboratively design, implement, and test the app.
8
+
9
+ Usage:
10
+ from contextsynapse.project.code_context import CodeContext
11
+
12
+ cc = CodeContext.create(
13
+ name="my-saas-app",
14
+ stack={"backend": "Python/FastAPI", "frontend": "React", "database": "PostgreSQL"},
15
+ spec="Build a task management app with auth, CRUD tasks, team sharing",
16
+ rules=["TypeScript strict", "pytest for tests", "no ORM"],
17
+ )
18
+
19
+ # Auto-generate tasks from spec
20
+ cc.generate_tasks()
21
+
22
+ # Agents work on tasks
23
+ cc.claim_task(task_id, agent_id)
24
+ cc.complete_task(task_id, agent_id, summary="Implemented auth", files=["auth.py"])
25
+
26
+ # Check status
27
+ cc.status()
28
+ """
29
+
30
+ from __future__ import annotations
31
+
32
+ import json
33
+ import logging
34
+ import uuid
35
+ from datetime import datetime, timezone
36
+ from typing import Any, Dict, List, Optional
37
+
38
+ from ..adapters._base import AIContextDBConnection
39
+ from ..core.graph_structures import GraphNode, GraphEdge
40
+ from ..storage.namespace_store import NamespaceStore
41
+
42
+ logger = logging.getLogger(__name__)
43
+
44
+
45
+ def _now() -> str:
46
+ return datetime.now(timezone.utc).isoformat()
47
+
48
+
49
+ def _esc(s: str) -> str:
50
+ return s.replace('"', '\\"').replace("\n", " ")
51
+
52
+
53
+ class CodeContext:
54
+ """A spec-driven code context for agentic application building.
55
+
56
+ The context IS the graph. The graph namespace = context name.
57
+ Everything lives in the graph: spec, architecture, tasks, code, tests.
58
+ """
59
+
60
+ def __init__(
61
+ self,
62
+ name: str,
63
+ conn: Optional[AIContextDBConnection] = None,
64
+ namespace_store: Optional[NamespaceStore] = None,
65
+ thread_id: Optional[str] = None,
66
+ ):
67
+ self.name = name
68
+ self.conn = conn or AIContextDBConnection(namespace=name)
69
+ self.ns = namespace_store or NamespaceStore(name)
70
+ self._spec_node_id: Optional[str] = None
71
+ self._thread_id = thread_id
72
+
73
+ @classmethod
74
+ def create(
75
+ cls,
76
+ name: str,
77
+ stack: Dict[str, str],
78
+ spec: str,
79
+ rules: Optional[List[str]] = None,
80
+ repo: Optional[str] = None,
81
+ ) -> "CodeContext":
82
+ """Create a new code context with spec.
83
+
84
+ Args:
85
+ name: Project name (becomes the graph namespace).
86
+ stack: Tech stack dict (backend, frontend, database, testing).
87
+ spec: Natural language specification of what to build.
88
+ rules: Coding rules and constraints.
89
+ repo: Optional GitHub repo URL.
90
+ """
91
+ cc = cls(name)
92
+
93
+ # Create ProjectSpec root node
94
+ spec_id = str(uuid.uuid4())
95
+ spec_node = GraphNode(
96
+ id=spec_id,
97
+ label="ProjectSpec",
98
+ properties={
99
+ "name": name,
100
+ "spec": spec,
101
+ "stack": json.dumps(stack),
102
+ "rules": json.dumps(rules or []),
103
+ "repo": repo or "",
104
+ "status": "active",
105
+ "created_at": _now(),
106
+ },
107
+ )
108
+ cc.conn.contextsynapse.add_node(spec_node, write_through=True)
109
+ cc._spec_node_id = spec_id
110
+
111
+ # Create Technology nodes from stack
112
+ for role, tech in stack.items():
113
+ tech_id = str(uuid.uuid4())
114
+ tech_node = GraphNode(
115
+ id=tech_id,
116
+ label="Technology",
117
+ properties={"name": tech, "role": role},
118
+ )
119
+ cc.conn.contextsynapse.add_node(tech_node, write_through=True)
120
+ cc.conn.contextsynapse.add_edge(GraphEdge(
121
+ id=str(uuid.uuid4()), source=spec_id, target=tech_id,
122
+ label="USES", properties={"role": role},
123
+ ))
124
+
125
+ # Store config in namespace SQL
126
+ cc.ns.set_metadata("stack", stack)
127
+ cc.ns.set_metadata("spec", spec)
128
+ cc.ns.set_metadata("rules", rules or [])
129
+ if repo:
130
+ cc.ns.set_metadata("repo", repo)
131
+
132
+ logger.info("Created code context: %s", name)
133
+ return cc
134
+
135
+ @classmethod
136
+ def load(cls, name: str) -> "CodeContext":
137
+ """Load an existing code context."""
138
+ cc = cls(name)
139
+ # Find the ProjectSpec node
140
+ nodes = cc.conn.get_nodes(label="ProjectSpec")
141
+ if nodes:
142
+ cc._spec_node_id = nodes[0].id if hasattr(nodes[0], 'id') else None
143
+ return cc
144
+
145
+ # ==================================================================
146
+ # Task Generation (LLM-powered)
147
+ # ==================================================================
148
+
149
+ def generate_tasks(
150
+ self,
151
+ provider: Optional[str] = None,
152
+ model: Optional[str] = None,
153
+ ) -> List[str]:
154
+ """Use LLM to generate tasks from the project spec.
155
+
156
+ Returns list of created task node IDs.
157
+ """
158
+ from ..llm import get_llm_client
159
+
160
+ spec = self.ns.get_metadata("spec") or ""
161
+ stack = self.ns.get_metadata("stack") or {}
162
+ rules = self.ns.get_metadata("rules") or []
163
+
164
+ # If spec is thin, pull in requirement docs from the graph
165
+ requirements_text = self._gather_requirements_text()
166
+ if requirements_text:
167
+ spec = f"{spec}\n\n--- Requirements Document ---\n{requirements_text}"
168
+
169
+ prompt = f"""Given this project specification, generate a list of implementation tasks.
170
+
171
+ Project: {self.name}
172
+ Stack: {json.dumps(stack)}
173
+ Spec: {spec}
174
+ Rules: {json.dumps(rules)}
175
+
176
+ Return JSON with this structure:
177
+ {{
178
+ "tasks": [
179
+ {{
180
+ "title": "Task title",
181
+ "description": "Detailed description",
182
+ "priority": "high|medium|low",
183
+ "tags": ["auth", "backend"],
184
+ "depends_on": []
185
+ }}
186
+ ]
187
+ }}
188
+
189
+ Generate 5-15 concrete, actionable tasks ordered by dependency (foundational first).
190
+ """
191
+ llm = get_llm_client(provider=provider, model=model)
192
+ result = llm.generate_json(
193
+ prompt=prompt,
194
+ system="You are a senior software architect. Generate precise, actionable development tasks.",
195
+ )
196
+
197
+ task_ids = []
198
+ title_to_id: Dict[str, str] = {}
199
+
200
+ for task in result.get("tasks", []):
201
+ task_id = str(uuid.uuid4())
202
+ title = task.get("title", "")
203
+ tags = task.get("tags", [])
204
+
205
+ node = GraphNode(
206
+ id=task_id,
207
+ label="Task",
208
+ properties={
209
+ "title": title,
210
+ "description": task.get("description", ""),
211
+ "priority": task.get("priority", "medium"),
212
+ "status": "open",
213
+ "tags": ",".join(tags) if tags else "",
214
+ "created_at": _now(),
215
+ },
216
+ )
217
+ self.conn.contextsynapse.add_node(node, write_through=True)
218
+
219
+ # Link to spec
220
+ if self._spec_node_id:
221
+ self.conn.contextsynapse.add_edge(GraphEdge(
222
+ id=str(uuid.uuid4()), source=self._spec_node_id,
223
+ target=task_id, label="HAS_TASK", properties={},
224
+ ))
225
+
226
+ title_to_id[title] = task_id
227
+ task_ids.append(task_id)
228
+
229
+ # Create dependency edges
230
+ for task in result.get("tasks", []):
231
+ title = task.get("title", "")
232
+ task_id = title_to_id.get(title)
233
+ if not task_id:
234
+ continue
235
+ for dep_title in task.get("depends_on", []):
236
+ dep_id = title_to_id.get(dep_title)
237
+ if dep_id:
238
+ self.conn.contextsynapse.add_edge(GraphEdge(
239
+ id=str(uuid.uuid4()), source=task_id,
240
+ target=dep_id, label="DEPENDS_ON", properties={},
241
+ ))
242
+
243
+ logger.info("Generated %d tasks for %s", len(task_ids), self.name)
244
+ return task_ids
245
+
246
+ # ==================================================================
247
+ # Thread logging
248
+ # ==================================================================
249
+
250
+ def _thread_log(self, content: str, agent_id: str = "", action: str = ""):
251
+ """Log an agent action to the session thread (if thread_id is set)."""
252
+ if not self._thread_id:
253
+ return
254
+ try:
255
+ from ..context.conversation import ConversationStore
256
+ store = ConversationStore()
257
+ store.append_message(
258
+ conversation_id=self._thread_id,
259
+ role="assistant",
260
+ content=content,
261
+ agent_id=agent_id,
262
+ metadata={"type": "agent_action", "action": action},
263
+ )
264
+ except Exception as e:
265
+ logger.debug("Thread log failed (non-critical): %s", e)
266
+
267
+ def get_thread_messages(self, limit: int = 50) -> List[Dict]:
268
+ """Get recent thread messages for context building."""
269
+ if not self._thread_id:
270
+ return []
271
+ try:
272
+ from ..context.conversation import ConversationStore
273
+ store = ConversationStore()
274
+ conv = store.get(self._thread_id, include_messages=True)
275
+ if not conv or not conv.messages:
276
+ return []
277
+ return [m.to_dict() for m in conv.messages[-limit:]]
278
+ except Exception:
279
+ return []
280
+
281
+ # ==================================================================
282
+ # Requirements → Tasks
283
+ # ==================================================================
284
+
285
+ def _gather_requirements_text(self, max_chars: int = 15000) -> str:
286
+ """Collect text from Document and TextChunk nodes in the graph."""
287
+ text_parts = []
288
+ total = 0
289
+
290
+ # Collect from Document nodes
291
+ for node in self.conn.get_nodes(label="Document"):
292
+ p = node.properties if hasattr(node, "properties") else {}
293
+ content = p.get("content") or p.get("text") or p.get("title") or ""
294
+ if content and total + len(content) < max_chars:
295
+ text_parts.append(content)
296
+ total += len(content)
297
+
298
+ # Collect from TextChunk / Chunk nodes
299
+ for label in ("TextChunk", "Chunk", "Passage"):
300
+ for node in self.conn.get_nodes(label=label):
301
+ p = node.properties if hasattr(node, "properties") else {}
302
+ content = p.get("text") or p.get("content") or ""
303
+ if content and total + len(content) < max_chars:
304
+ text_parts.append(content)
305
+ total += len(content)
306
+ if total >= max_chars:
307
+ break
308
+
309
+ # Collect from Requirement nodes (manually added)
310
+ for node in self.conn.get_nodes(label="Requirement"):
311
+ p = node.properties if hasattr(node, "properties") else {}
312
+ title = p.get("title", "")
313
+ if title:
314
+ text_parts.append(f"Requirement: {title}")
315
+
316
+ return "\n\n".join(text_parts)
317
+
318
+ def generate_tasks_from_documents(
319
+ self,
320
+ provider: Optional[str] = None,
321
+ model: Optional[str] = None,
322
+ ) -> List[str]:
323
+ """Generate tasks specifically from uploaded requirement documents."""
324
+ requirements_text = self._gather_requirements_text()
325
+ if not requirements_text:
326
+ logger.warning("No document content found in graph for %s", self.name)
327
+ return []
328
+
329
+ original_spec = self.ns.get_metadata("spec") or ""
330
+ combined_spec = f"{original_spec}\n\n--- Uploaded Requirements ---\n{requirements_text}"
331
+ self.ns.set_metadata("spec", combined_spec)
332
+
333
+ try:
334
+ task_ids = self.generate_tasks(provider=provider, model=model)
335
+ finally:
336
+ self.ns.set_metadata("spec", original_spec)
337
+
338
+ return task_ids
339
+
340
+ # ==================================================================
341
+ # Task Management
342
+ # ==================================================================
343
+
344
+ def claim_task(self, task_id: str, agent_id: str) -> str:
345
+ node = self.conn.get_node(task_id)
346
+ if not node:
347
+ return f"Task {task_id[:8]} not found"
348
+ props = node.properties if hasattr(node, 'properties') else {}
349
+ title = props.get("title", "?")
350
+
351
+ self.conn.query(
352
+ f'UPDATE NODE Task SET {{status: "in_progress", assigned_to: "{agent_id}", claimed_at: "{_now()}"}} '
353
+ f'WHERE uuid = "{task_id}"'
354
+ )
355
+ self._thread_log(f"Claimed task: {title}", agent_id=agent_id, action="claimed_task")
356
+ return f"Claimed: {title}"
357
+
358
+ def complete_task(
359
+ self, task_id: str, agent_id: str,
360
+ summary: str = "", files: Optional[List[str]] = None,
361
+ ) -> str:
362
+ node = self.conn.get_node(task_id)
363
+ if not node:
364
+ return f"Task {task_id[:8]} not found"
365
+ props = node.properties if hasattr(node, 'properties') else {}
366
+ title = props.get("title", "?")
367
+
368
+ self.conn.query(
369
+ f'UPDATE NODE Task SET {{status: "completed", completed_by: "{agent_id}", '
370
+ f'completed_at: "{_now()}", summary: "{_esc(summary)}"}} WHERE uuid = "{task_id}"'
371
+ )
372
+
373
+ if files:
374
+ for fp in files:
375
+ file_id = str(uuid.uuid4())
376
+ file_node = GraphNode(
377
+ id=file_id, label="CodeFile",
378
+ properties={"path": fp, "created_by": agent_id, "created_at": _now()},
379
+ )
380
+ self.conn.contextsynapse.add_node(file_node, write_through=True)
381
+ self.conn.contextsynapse.add_edge(GraphEdge(
382
+ id=str(uuid.uuid4()), source=file_id, target=task_id,
383
+ label="IMPLEMENTS", properties={},
384
+ ))
385
+
386
+ files_str = ", ".join(files) if files else ""
387
+ log_msg = f"Completed task: {title}"
388
+ if files_str:
389
+ log_msg += f" (files: {files_str})"
390
+ if summary:
391
+ log_msg += f" — {summary[:100]}"
392
+ self._thread_log(log_msg, agent_id=agent_id, action="completed_task")
393
+ return f"Completed: {title}"
394
+
395
+ def add_decision(self, title: str, rationale: str, agent_id: str) -> str:
396
+ dec_id = str(uuid.uuid4())
397
+ node = GraphNode(
398
+ id=dec_id, label="Decision",
399
+ properties={
400
+ "title": title, "rationale": rationale,
401
+ "decided_by": agent_id, "decided_at": _now(),
402
+ },
403
+ )
404
+ self.conn.contextsynapse.add_node(node, write_through=True)
405
+ if self._spec_node_id:
406
+ self.conn.contextsynapse.add_edge(GraphEdge(
407
+ id=str(uuid.uuid4()), source=self._spec_node_id,
408
+ target=dec_id, label="HAS_DECISION", properties={},
409
+ ))
410
+ self._thread_log(f"Decision: {title} — {rationale[:100]}", agent_id=agent_id, action="decision")
411
+ return dec_id
412
+
413
+ def add_requirement(self, title: str, priority: str = "medium") -> str:
414
+ req_id = str(uuid.uuid4())
415
+ node = GraphNode(
416
+ id=req_id, label="Requirement",
417
+ properties={"title": title, "priority": priority, "status": "open"},
418
+ )
419
+ self.conn.contextsynapse.add_node(node, write_through=True)
420
+ if self._spec_node_id:
421
+ self.conn.contextsynapse.add_edge(GraphEdge(
422
+ id=str(uuid.uuid4()), source=self._spec_node_id,
423
+ target=req_id, label="REQUIRES", properties={},
424
+ ))
425
+ return req_id
426
+
427
+ # ==================================================================
428
+ # Status & Context
429
+ # ==================================================================
430
+
431
+ def status(self) -> Dict[str, Any]:
432
+ """Get project status."""
433
+ tasks = self.conn.get_nodes(label="Task")
434
+ decisions = self.conn.get_nodes(label="Decision")
435
+ code_files = self.conn.get_nodes(label="CodeFile")
436
+ requirements = self.conn.get_nodes(label="Requirement")
437
+
438
+ status_counts: Dict[str, int] = {}
439
+ for t in tasks:
440
+ p = t.properties if hasattr(t, 'properties') else {}
441
+ s = p.get("status", "unknown")
442
+ status_counts[s] = status_counts.get(s, 0) + 1
443
+
444
+ return {
445
+ "project": self.name,
446
+ "spec": self.ns.get_metadata("spec") or "",
447
+ "stack": self.ns.get_metadata("stack") or {},
448
+ "tasks": {"total": len(tasks), "by_status": status_counts},
449
+ "decisions": len(decisions),
450
+ "code_files": len(code_files),
451
+ "requirements": len(requirements),
452
+ }
453
+
454
+ def get_open_tasks(self, unblocked_only: bool = False) -> List[Dict]:
455
+ """Get open tasks. If unblocked_only, only return tasks whose dependencies are all completed."""
456
+ tasks = self.conn.get_nodes(label="Task")
457
+ edges = self.conn.get_edges(label="DEPENDS_ON")
458
+
459
+ deps: Dict[str, List[str]] = {}
460
+ for e in edges:
461
+ src = e.source if hasattr(e, 'source') else e.get('source', '')
462
+ tgt = e.target if hasattr(e, 'target') else e.get('target', '')
463
+ if src not in deps:
464
+ deps[src] = []
465
+ deps[src].append(tgt)
466
+
467
+ task_status: Dict[str, str] = {}
468
+ for t in tasks:
469
+ p = t.properties if hasattr(t, 'properties') else {}
470
+ tid = t.id if hasattr(t, 'id') else '?'
471
+ task_status[tid] = p.get("status", "unknown")
472
+
473
+ results = []
474
+ for t in tasks:
475
+ p = t.properties if hasattr(t, 'properties') else {}
476
+ tid = t.id if hasattr(t, 'id') else '?'
477
+ if p.get("status") not in ("open", "in_progress"):
478
+ continue
479
+
480
+ task_deps = deps.get(tid, [])
481
+ blocked = any(task_status.get(d) != "completed" for d in task_deps)
482
+
483
+ if unblocked_only and blocked:
484
+ continue
485
+
486
+ results.append({
487
+ "id": tid,
488
+ "title": p.get("title", "?"),
489
+ "status": p.get("status", "?"),
490
+ "priority": p.get("priority", "medium"),
491
+ "assigned_to": p.get("assigned_to", "unassigned"),
492
+ "blocked": blocked,
493
+ "depends_on": [d[:8] for d in task_deps],
494
+ })
495
+ prio = {"critical": 0, "high": 1, "medium": 2, "low": 3}
496
+ results.sort(key=lambda x: (x["blocked"], prio.get(x["priority"], 9)))
497
+ return results
498
+
499
+ # ==================================================================
500
+ # Reviews (Human in the Loop)
501
+ # ==================================================================
502
+
503
+ def request_review(self, task_id: str, notes: str = "", reviewer: str = "human") -> str:
504
+ """Request human review for a task or decision."""
505
+ review_id = str(uuid.uuid4())
506
+ node = GraphNode(
507
+ id=review_id, label="Review",
508
+ properties={
509
+ "status": "pending",
510
+ "reviewer": reviewer,
511
+ "notes": notes,
512
+ "requested_at": _now(),
513
+ },
514
+ )
515
+ self.conn.contextsynapse.add_node(node, write_through=True)
516
+ self.conn.contextsynapse.add_edge(GraphEdge(
517
+ id=str(uuid.uuid4()), source=task_id, target=review_id,
518
+ label="NEEDS_REVIEW", properties={},
519
+ ))
520
+ return review_id
521
+
522
+ def approve_review(self, review_id: str, approved_by: str = "human", feedback: str = "") -> str:
523
+ """Approve a pending review."""
524
+ self.conn.query(
525
+ f'UPDATE NODE Review SET {{status: "approved", approved_by: "{approved_by}", '
526
+ f'feedback: "{_esc(feedback)}", approved_at: "{_now()}"}} WHERE uuid = "{review_id}"'
527
+ )
528
+ return f"Review {review_id[:8]} approved"
529
+
530
+ def reject_review(self, review_id: str, rejected_by: str = "human", feedback: str = "") -> str:
531
+ """Reject a review with feedback."""
532
+ self.conn.query(
533
+ f'UPDATE NODE Review SET {{status: "rejected", rejected_by: "{rejected_by}", '
534
+ f'feedback: "{_esc(feedback)}", rejected_at: "{_now()}"}} WHERE uuid = "{review_id}"'
535
+ )
536
+ return f"Review {review_id[:8]} rejected: {feedback}"
537
+
538
+ def get_pending_reviews(self) -> List[Dict]:
539
+ """Get all pending reviews."""
540
+ reviews = self.conn.get_nodes(label="Review")
541
+ return [
542
+ {
543
+ "id": r.id if hasattr(r, 'id') else '?',
544
+ "status": (r.properties if hasattr(r, 'properties') else {}).get("status", "?"),
545
+ "notes": (r.properties if hasattr(r, 'properties') else {}).get("notes", ""),
546
+ "reviewer": (r.properties if hasattr(r, 'properties') else {}).get("reviewer", "?"),
547
+ }
548
+ for r in reviews
549
+ if (r.properties if hasattr(r, 'properties') else {}).get("status") == "pending"
550
+ ]
551
+
552
+ # ==================================================================
553
+ # Connectors
554
+ # ==================================================================
555
+
556
+ def add_connector(self, connector_type: str, config: Dict[str, Any]) -> str:
557
+ """Attach a connector (GitHub, Jira, etc.) to this project."""
558
+ conn_id = str(uuid.uuid4())
559
+ node = GraphNode(
560
+ id=conn_id, label="Connector",
561
+ properties={
562
+ "type": connector_type,
563
+ "config": json.dumps(config),
564
+ "status": "active",
565
+ "created_at": _now(),
566
+ },
567
+ )
568
+ self.conn.contextsynapse.add_node(node, write_through=True)
569
+ if self._spec_node_id:
570
+ self.conn.contextsynapse.add_edge(GraphEdge(
571
+ id=str(uuid.uuid4()), source=self._spec_node_id,
572
+ target=conn_id, label="HAS_CONNECTOR", properties={},
573
+ ))
574
+ self.ns.set_metadata(f"connector_{connector_type}", config)
575
+ return conn_id
576
+
577
+ def get_connectors(self) -> List[Dict]:
578
+ """List attached connectors."""
579
+ connectors = self.conn.get_nodes(label="Connector")
580
+ return [
581
+ {
582
+ "id": c.id if hasattr(c, 'id') else '?',
583
+ "type": (c.properties if hasattr(c, 'properties') else {}).get("type", "?"),
584
+ "config": json.loads((c.properties if hasattr(c, 'properties') else {}).get("config", "{}")),
585
+ "status": (c.properties if hasattr(c, 'properties') else {}).get("status", "?"),
586
+ }
587
+ for c in connectors
588
+ ]
589
+
590
+ def log_connector_action(self, connector_id: str, action_type: str, result: str) -> str:
591
+ """Log a connector action (push, create_ticket, etc.)."""
592
+ action_id = str(uuid.uuid4())
593
+ node = GraphNode(
594
+ id=action_id, label="ConnectorAction",
595
+ properties={
596
+ "type": action_type,
597
+ "result": result,
598
+ "timestamp": _now(),
599
+ },
600
+ )
601
+ self.conn.contextsynapse.add_node(node, write_through=True)
602
+ self.conn.contextsynapse.add_edge(GraphEdge(
603
+ id=str(uuid.uuid4()), source=connector_id, target=action_id,
604
+ label="ACTION", properties={},
605
+ ))
606
+ return action_id
607
+
608
+ def build_context(self, agent_id: str = "", max_tokens: int = 6000) -> str:
609
+ """Build LLM-ready context for an agent working on this project."""
610
+ from ..context.hub import ContextHub, ContextRole
611
+
612
+ spec = self.ns.get_metadata("spec") or ""
613
+ stack = self.ns.get_metadata("stack") or {}
614
+ rules = self.ns.get_metadata("rules") or []
615
+
616
+ hub = ContextHub(
617
+ system_prompt=(
618
+ f"You are working on project '{self.name}'. "
619
+ f"Stack: {json.dumps(stack)}. "
620
+ f"Follow these rules: {json.dumps(rules)}."
621
+ ),
622
+ max_tokens=max_tokens,
623
+ )
624
+
625
+ hub.add_text(f"Project Spec:\n{spec}", role=ContextRole.INSTRUCTION, label="Specification")
626
+
627
+ unblocked = self.get_open_tasks(unblocked_only=True)
628
+ if unblocked:
629
+ task_text = "\n".join(
630
+ f"- [{t['priority'].upper()}] {t['title']} (id: {t['id'][:8]}, assigned: {t['assigned_to']})"
631
+ for t in unblocked
632
+ )
633
+ hub.add_text(task_text, role=ContextRole.INSTRUCTION, label="Unblocked Tasks (ready to claim)")
634
+
635
+ blocked = [t for t in self.get_open_tasks() if t.get("blocked")]
636
+ if blocked:
637
+ blocked_text = "\n".join(
638
+ f"- {t['title']} (blocked by: {', '.join(t['depends_on'])})"
639
+ for t in blocked
640
+ )
641
+ hub.add_text(blocked_text, role=ContextRole.BACKGROUND, label="Blocked Tasks")
642
+
643
+ decisions = self.conn.get_nodes(label="Decision")
644
+ if decisions:
645
+ dec_text = "\n".join(
646
+ f"- {p.get('title', '?')}: {p.get('rationale', '')}"
647
+ for d in decisions
648
+ for p in [d.properties if hasattr(d, 'properties') else {}]
649
+ )
650
+ hub.add_text(dec_text, role=ContextRole.DECISION, label="Decisions Made")
651
+
652
+ reviews = self.get_pending_reviews()
653
+ if reviews:
654
+ review_text = "\n".join(
655
+ f"- Review pending: {r['notes']} (id: {r['id'][:8]})"
656
+ for r in reviews
657
+ )
658
+ hub.add_text(review_text, role=ContextRole.BACKGROUND, label="Pending Reviews (waiting for human)")
659
+
660
+ connectors = self.get_connectors()
661
+ if connectors:
662
+ conn_text = "\n".join(
663
+ f"- {c['type']}: {json.dumps(c['config'])}"
664
+ for c in connectors
665
+ )
666
+ hub.add_text(conn_text, role=ContextRole.BACKGROUND, label="Connected Services")
667
+
668
+ thread_msgs = self.get_thread_messages(limit=20)
669
+ if thread_msgs:
670
+ thread_lines = []
671
+ for m in thread_msgs:
672
+ role = m.get("role", "?")
673
+ content = m.get("content", "")[:200]
674
+ meta = m.get("metadata", {})
675
+ msg_type = meta.get("type", "")
676
+ prefix = f"[{role}]"
677
+ if msg_type == "approval":
678
+ prefix = f"[{role} — {meta.get('action', 'approval')}]"
679
+ elif msg_type == "agent_action":
680
+ prefix = f"[agent:{m.get('agent_id', '?')}]"
681
+ thread_lines.append(f"{prefix} {content}")
682
+ hub.add_text(
683
+ "## Session Thread\n" + "\n".join(thread_lines),
684
+ role=ContextRole.INTERACTION,
685
+ label="Session Thread (recent messages)",
686
+ )
687
+
688
+ return hub.to_prompt()
689
+
690
+ # ── Spec-driven traceability ──────────────────────────────────────
691
+
692
+ def ingest_spec(self, markdown_text: str) -> List[str]:
693
+ """Parse markdown spec and write Requirement/Feature/Constraint nodes to the graph.
694
+
695
+ Each node is tagged with the project name for scoped queries.
696
+
697
+ Returns list of node IDs created.
698
+ """
699
+ from .spec_parser import parse_spec
700
+
701
+ nodes = parse_spec(markdown_text)
702
+ if not nodes:
703
+ return []
704
+
705
+ ids = []
706
+ for n in nodes:
707
+ props = {
708
+ "name": n["name"],
709
+ "description": n["description"],
710
+ "priority": n["priority"],
711
+ "tags": json.dumps(n["tags"]),
712
+ "project": self.name,
713
+ "created_at": _now(),
714
+ }
715
+ graph_node = GraphNode(
716
+ id=n["node_id"],
717
+ label=n["label"],
718
+ properties=props,
719
+ )
720
+ self.conn.contextsynapse.add_node(graph_node)
721
+ ids.append(n["node_id"])
722
+
723
+ return ids
724
+
725
+ def link_artifact(self, req_id: str, artifact_id: str, edge_label: str) -> None:
726
+ """Create a traceability edge between a requirement/feature node and an artifact.
727
+
728
+ Common edge labels:
729
+ IMPLEMENTED_BY — Requirement → Task or CodeFile
730
+ VERIFIED_BY — Requirement → TestCase
731
+ COVERS — TestCase → Requirement
732
+ """
733
+ self.conn.contextsynapse.add_edge(GraphEdge(
734
+ id=str(uuid.uuid4()),
735
+ source=req_id,
736
+ target=artifact_id,
737
+ label=edge_label,
738
+ properties={},
739
+ ))
740
+
741
+ def get_requirements(self) -> List[Dict[str, Any]]:
742
+ """Return all Requirement nodes for this project.
743
+
744
+ Returns dicts with keys: node_id, label, properties.
745
+ """
746
+ nodes = self.conn.get_nodes(label="Requirement")
747
+ results = []
748
+ for node in nodes:
749
+ props = node.properties if hasattr(node, "properties") else {}
750
+ # Filter to this project
751
+ if props.get("project") == self.name:
752
+ results.append({
753
+ "node_id": node.id if hasattr(node, "id") else "",
754
+ "label": node.label if hasattr(node, "label") else "Requirement",
755
+ "properties": props,
756
+ })
757
+ return results
758
+
759
+ def check_coverage(self) -> Dict[str, Any]:
760
+ """Analyse which requirements are implemented and verified.
761
+
762
+ Returns:
763
+ {
764
+ "covered": [req_id, ...], # both IMPLEMENTED_BY and VERIFIED_BY
765
+ "partial": [req_id, ...], # one but not the other
766
+ "uncovered": [req_id, ...], # neither
767
+ "coverage_pct": float, # weighted 0-100
768
+ "total": int,
769
+ }
770
+ """
771
+ reqs = self.get_requirements()
772
+ total = len(reqs)
773
+ if total == 0:
774
+ return {"covered": [], "partial": [], "uncovered": [],
775
+ "coverage_pct": 100.0, "total": 0}
776
+
777
+ covered, partial, uncovered = [], [], []
778
+ for req in reqs:
779
+ req_id = req.get("node_id", "")
780
+
781
+ # Check IMPLEMENTED_BY edges (outgoing from req_id)
782
+ impl_neighbors = self.conn.contextsynapse.get_neighbors(
783
+ req_id, edge_label="IMPLEMENTED_BY", direction="OUTGOING"
784
+ )
785
+ # Check VERIFIED_BY edges (outgoing from req_id)
786
+ test_neighbors = self.conn.contextsynapse.get_neighbors(
787
+ req_id, edge_label="VERIFIED_BY", direction="OUTGOING"
788
+ )
789
+
790
+ has_impl = bool(impl_neighbors)
791
+ has_test = bool(test_neighbors)
792
+
793
+ if has_impl and has_test:
794
+ covered.append(req_id)
795
+ elif has_impl or has_test:
796
+ partial.append(req_id)
797
+ else:
798
+ uncovered.append(req_id)
799
+
800
+ coverage_pct = (len(covered) + 0.5 * len(partial)) / total * 100
801
+ return {
802
+ "covered": covered,
803
+ "partial": partial,
804
+ "uncovered": uncovered,
805
+ "coverage_pct": round(coverage_pct, 2),
806
+ "total": total,
807
+ }