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,851 @@
1
+ """Pipeline executor — runs a single Pipeline through its 9-step chain."""
2
+ from __future__ import annotations
3
+
4
+ import json
5
+ import logging
6
+ from datetime import datetime, timedelta, timezone
7
+ from typing import Any, Dict, Optional
8
+
9
+ logger = logging.getLogger(__name__)
10
+
11
+ # ---------------------------------------------------------------------------
12
+ # Lazy top-level imports (avoids circular dependency on startup)
13
+ # ---------------------------------------------------------------------------
14
+ from ..ingestion.smart_ingest import ingest_crawl # noqa: E402
15
+
16
+
17
+ def _get_graph(namespace: str):
18
+ """Get or create a named graph."""
19
+ from ..core.registry import graph_registry
20
+ graph = graph_registry.get_graph(namespace, load_if_missing=True)
21
+ if not graph:
22
+ graph = graph_registry.create_graph(namespace)
23
+ return graph
24
+
25
+
26
+ # ---------------------------------------------------------------------------
27
+ # Public API
28
+ # ---------------------------------------------------------------------------
29
+
30
+ def execute_pipeline(pipeline_id: str, store) -> Dict[str, Any]:
31
+ """
32
+ Execute a pipeline by id.
33
+
34
+ Steps:
35
+ 1. Load Pipeline from store
36
+ 2. Get/create target graph
37
+ 3. Build filter_config from keyword/semantic settings
38
+ 4. Call ingest_crawl() using pipeline.steps, pipeline.schema_id/schema_mode
39
+ 5. Update pipeline stats (total_runs, last_run_at, total_documents)
40
+ 6. one_time → set status="completed"
41
+ scheduled → calculate next_run_at, keep status="active"
42
+ 7. Save run history
43
+ 8. Propagate "pipeline_completed" event
44
+ 9. Handle errors gracefully (set last_error, return error dict)
45
+
46
+ Returns a result dict with keys: status, pages_ingested, nodes_created,
47
+ pages_filtered, pages_failed, pipeline_id.
48
+ """
49
+ # Step 1 — load pipeline
50
+ pipeline = store.get(pipeline_id)
51
+ if pipeline is None:
52
+ return {"status": "error", "error": f"Pipeline {pipeline_id} not found"}
53
+
54
+ now = datetime.now(timezone.utc)
55
+
56
+ try:
57
+ # Step 2 — get/create graph
58
+ graph = _get_graph(pipeline.target_context_id or pipeline_id)
59
+
60
+ # Step 3 — build filter_config
61
+ filter_config: Optional[Dict[str, Any]] = None
62
+ if pipeline.include_keywords or pipeline.exclude_keywords or pipeline.semantic_filter:
63
+ filter_config = {
64
+ "include_keywords": pipeline.include_keywords,
65
+ "exclude_keywords": pipeline.exclude_keywords,
66
+ "semantic_filter": pipeline.semantic_filter,
67
+ "semantic_threshold": pipeline.semantic_threshold,
68
+ }
69
+
70
+ # Step 4 — call ingest_crawl
71
+ crawl_result = ingest_crawl(
72
+ start_url=pipeline.source_url,
73
+ db=graph,
74
+ max_pages=pipeline.max_pages,
75
+ schema=pipeline.schema_id or None,
76
+ pipeline=pipeline.schema_mode or "smart_article",
77
+ mode="auto",
78
+ filter_config=filter_config,
79
+ debug=False,
80
+ owner=pipeline.target_context_id,
81
+ )
82
+
83
+ ingested = crawl_result.get("ingested", [])
84
+ filtered = crawl_result.get("filtered", [])
85
+ failed = crawl_result.get("failed", [])
86
+
87
+ pages_ingested = len(ingested)
88
+ nodes_created = sum(
89
+ item.get("nodes", 0) or
90
+ (item.get("passages", 0) + item.get("entities", 0) + item.get("facts", 0))
91
+ for item in ingested
92
+ )
93
+ pages_filtered = len(filtered)
94
+ pages_failed = len(failed)
95
+
96
+ # Step 5 — update pipeline stats
97
+ pipeline.total_runs += 1
98
+ pipeline.last_run_at = now.isoformat()
99
+ pipeline.total_documents += pages_ingested
100
+ pipeline.last_error = ""
101
+
102
+ # Step 6 — lifecycle state
103
+ if pipeline.trigger_type == "one_time":
104
+ pipeline.status = "completed"
105
+ elif pipeline.trigger_type == "scheduled":
106
+ interval = max(pipeline.interval_minutes, 1 / 60) # min 1s
107
+ pipeline.next_run_at = (now + timedelta(minutes=interval)).isoformat()
108
+ pipeline.status = "active"
109
+
110
+ # Step 7 — persist updated pipeline + run history
111
+ store.save(pipeline)
112
+ run_record = {
113
+ "run_at": now.isoformat(),
114
+ "pages_ingested": pages_ingested,
115
+ "nodes_created": nodes_created,
116
+ "pages_filtered": pages_filtered,
117
+ "pages_failed": pages_failed,
118
+ }
119
+ store.save_run(pipeline_id, run_record)
120
+
121
+ # Step 8 — propagate event (best-effort)
122
+ try:
123
+ from ..context.propagation import get_propagator
124
+ propagator = get_propagator()
125
+ propagator.propagate(
126
+ source_agent="pipeline_executor",
127
+ event_type="pipeline_completed",
128
+ content=(
129
+ f"Pipeline '{pipeline.name}' finished: "
130
+ f"{pages_ingested} pages ingested, {nodes_created} nodes created."
131
+ ),
132
+ namespace=pipeline.target_context_id or "default",
133
+ )
134
+ except Exception as prop_err:
135
+ logger.debug("Propagation skipped: %s", prop_err)
136
+
137
+ result: Dict[str, Any] = {
138
+ "status": "success",
139
+ "pipeline_id": pipeline_id,
140
+ "pages_ingested": pages_ingested,
141
+ "nodes_created": nodes_created,
142
+ "pages_filtered": pages_filtered,
143
+ "pages_failed": pages_failed,
144
+ }
145
+ logger.info("Pipeline %s completed: %s", pipeline_id, result)
146
+ return result
147
+
148
+ except Exception as exc:
149
+ # Step 9 — graceful error handling
150
+ err_msg = str(exc)
151
+ logger.exception("Pipeline %s failed: %s", pipeline_id, err_msg)
152
+
153
+ pipeline.last_error = err_msg
154
+ pipeline.last_run_at = now.isoformat()
155
+ pipeline.total_runs += 1
156
+ store.save(pipeline)
157
+
158
+ return {
159
+ "status": "error",
160
+ "pipeline_id": pipeline_id,
161
+ "error": err_msg,
162
+ }
163
+
164
+
165
+ # ---------------------------------------------------------------------------
166
+ # Rollback
167
+ # ---------------------------------------------------------------------------
168
+
169
+ def rollback_run(context_id: str, run_id: str, context_manager, graph_registry=None) -> Dict[str, Any]:
170
+ """Rollback a pipeline run — delete all nodes created by that run.
171
+
172
+ Nodes are identified by _owner == run_id.
173
+ """
174
+ ctx = context_manager.get_context(context_id)
175
+ if not ctx:
176
+ return {"status": "error", "error": "Context not found"}
177
+
178
+ namespace = ctx.graph_namespace or context_id
179
+ if graph_registry:
180
+ graph = graph_registry.get_graph(namespace, load_if_missing=True)
181
+ else:
182
+ graph = _get_graph(namespace)
183
+
184
+ if not graph:
185
+ return {"status": "error", "error": "Graph not found"}
186
+
187
+ # Find all nodes with _owner == run_id
188
+ deleted = 0
189
+ try:
190
+ all_nodes = list(graph.csr_adapter.get_all_nodes())
191
+ for node in all_nodes:
192
+ props = getattr(node, "properties", {}) or {}
193
+ if props.get("_owner") == run_id:
194
+ node_id = getattr(node, "id", None)
195
+ if node_id:
196
+ graph.remove_node(node_id)
197
+ deleted += 1
198
+ except Exception as e:
199
+ return {"status": "error", "error": f"Rollback failed: {e}", "deleted": deleted}
200
+
201
+ # Refresh stats
202
+ try:
203
+ context_manager.refresh_stats(context_id)
204
+ except Exception:
205
+ pass
206
+
207
+ logger.info("[PIPELINE] Rollback %s: deleted %d nodes from %s", run_id, deleted, namespace)
208
+ return {"status": "success", "run_id": run_id, "deleted": deleted}
209
+
210
+
211
+ # ---------------------------------------------------------------------------
212
+ # Context-based pipeline executor
213
+ # ---------------------------------------------------------------------------
214
+
215
+ def _execute_multi_pipeline(
216
+ context_id: str,
217
+ ctx,
218
+ pipelines_list: list,
219
+ context_manager,
220
+ graph_registry=None,
221
+ ) -> Dict[str, Any]:
222
+ """Execute multiple pipelines feeding a single context.
223
+
224
+ Each pipeline in config["pipelines"] is a dict with:
225
+ - name: str (identifier)
226
+ - source_url: str or sources: list[str]
227
+ - interval: str (schedule)
228
+ - strategy: str (content strategy)
229
+ - enabled: bool (default True)
230
+ - All other fields from the single-pipeline config
231
+
232
+ All pipelines feed the SAME graph namespace (the context's graph).
233
+ Dedup runs per-pipeline — same content from different pipelines is caught.
234
+ """
235
+ from ..ingestion.amplifier import build_context_purpose
236
+
237
+ namespace = ctx.graph_namespace or context_id
238
+ if graph_registry is not None:
239
+ graph = graph_registry.get_graph(namespace, load_if_missing=True)
240
+ if not graph:
241
+ graph = graph_registry.create_graph(namespace)
242
+ else:
243
+ graph = _get_graph(namespace)
244
+
245
+ context_purpose = build_context_purpose(
246
+ context_name=ctx.name,
247
+ context_description=ctx.description,
248
+ tags=ctx.tags or [],
249
+ )
250
+
251
+ now = datetime.now(timezone.utc)
252
+ combined_result = {
253
+ "status": "success",
254
+ "context_id": context_id,
255
+ "pipelines_run": 0,
256
+ "pipelines_skipped": 0,
257
+ "total_pages_ingested": 0,
258
+ "total_nodes_created": 0,
259
+ "total_deduped": 0,
260
+ "pipeline_results": [],
261
+ }
262
+
263
+ for pipe_config in pipelines_list:
264
+ pipe_name = pipe_config.get("name", "unnamed")
265
+ enabled = pipe_config.get("enabled", True)
266
+
267
+ if not enabled:
268
+ combined_result["pipelines_skipped"] += 1
269
+ combined_result["pipeline_results"].append({
270
+ "pipeline": pipe_name, "status": "skipped", "reason": "disabled",
271
+ })
272
+ continue
273
+
274
+ # Check schedule — only run if due
275
+ pipe_schedule = pipe_config.get("schedule", {})
276
+ next_run = pipe_schedule.get("next_run_at", "")
277
+ if next_run:
278
+ try:
279
+ next_dt = datetime.fromisoformat(next_run.replace("Z", "+00:00"))
280
+ if next_dt > now:
281
+ combined_result["pipelines_skipped"] += 1
282
+ combined_result["pipeline_results"].append({
283
+ "pipeline": pipe_name, "status": "skipped", "reason": "not due",
284
+ "next_run_at": next_run,
285
+ })
286
+ continue
287
+ except (ValueError, TypeError):
288
+ pass
289
+
290
+ # Get sources — support both "source_url" (single) and "sources" (list)
291
+ sources = pipe_config.get("sources", [])
292
+ source_url = pipe_config.get("source_url", "")
293
+ if source_url and not sources:
294
+ sources = [source_url]
295
+
296
+ if not sources:
297
+ combined_result["pipeline_results"].append({
298
+ "pipeline": pipe_name, "status": "skipped", "reason": "no sources",
299
+ })
300
+ combined_result["pipelines_skipped"] += 1
301
+ continue
302
+
303
+ # Build filter config for this pipeline
304
+ from ..ingestion.filters import FilterConfig, SemanticRule
305
+ fc = None
306
+ include_kw = pipe_config.get("include_keywords") or []
307
+ exclude_kw = pipe_config.get("exclude_keywords") or []
308
+ semantic_filter = pipe_config.get("semantic_filter", "")
309
+ if include_kw or exclude_kw or semantic_filter:
310
+ fc = FilterConfig(
311
+ keywords_include=include_kw,
312
+ keywords_exclude=exclude_kw,
313
+ semantic_query=semantic_filter,
314
+ semantic_auto_threshold=pipe_config.get("semantic_threshold", 0.6),
315
+ )
316
+
317
+ max_pages = pipe_config.get("max_pages", 5)
318
+ strategy = pipe_config.get("strategy", "")
319
+
320
+ pipe_result = {
321
+ "pipeline": pipe_name,
322
+ "source_type": pipe_config.get("source_type", "crawl"),
323
+ "pages_ingested": 0,
324
+ "nodes_created": 0,
325
+ "deduped": 0,
326
+ "errors": [],
327
+ }
328
+
329
+ for source in sources:
330
+ try:
331
+ crawl_kwargs = {}
332
+ if fc:
333
+ crawl_kwargs["filter_config"] = fc
334
+
335
+ src_type = pipe_config.get("source_type", "crawl")
336
+
337
+ if src_type == "stock_price":
338
+ # Stock price sources: each "source" is a Yahoo Finance ticker
339
+ # (e.g., "TCS.NS"). Fetches OHLCV via PriceFeed and stores
340
+ # Indicator nodes in the graph namespace named
341
+ # "{company}_price" relative to the context name.
342
+ try:
343
+ from ..intelligence.price_feed import PriceFeed
344
+ feed = PriceFeed()
345
+ # Build price namespace: e.g. ctx "TCS" → "tcs_price"
346
+ price_ns_base = source.split(".")[0].lower() # TCS.NS → tcs
347
+ price_ns = f"{price_ns_base}_price"
348
+ price_graph = graph_registry.get_graph(price_ns, load_if_missing=True) if graph_registry else None
349
+ if price_graph is None and graph_registry:
350
+ graph_registry.create_graph(price_ns)
351
+ price_graph = graph_registry.get_graph(price_ns, load_if_missing=True)
352
+ if price_graph is None:
353
+ price_graph = graph # fallback to context graph
354
+ period = pipe_config.get("config", {}).get("period", "5d")
355
+ interval = pipe_config.get("config", {}).get("interval", "1h")
356
+ created = feed.ingest_into_graph(price_graph, source, period=period, interval=interval)
357
+ pipe_result["nodes_created"] += created
358
+ pipe_result["pages_ingested"] += 1 if created > 0 else 0
359
+ except Exception as price_err:
360
+ pipe_result["errors"].append(f"Stock price {source}: {price_err}")
361
+ continue # sources loop — no further handling needed
362
+
363
+ elif src_type == "rss":
364
+ # RSS feed — parse with feedparser, ingest each article as a node
365
+ from ..ingestion.smart_ingest import ingest_text
366
+ try:
367
+ import feedparser
368
+ parsed = feedparser.parse(source)
369
+ entries = parsed.entries or []
370
+ filter_topics = []
371
+ if fc:
372
+ filter_topics = [t.lower() for t in (getattr(fc, 'topics', None) or [])]
373
+ ingested_count = 0
374
+ for entry in entries[:20]: # cap at 20 articles per feed per run
375
+ title = entry.get("title", "")
376
+ summary = entry.get("summary", entry.get("description", ""))
377
+ link = entry.get("link", source)
378
+ content = f"{title}\n\n{summary}".strip()
379
+ if not content:
380
+ continue
381
+ # Topic filter — skip if none of the filter topics appear
382
+ if filter_topics:
383
+ combined = (title + " " + summary).lower()
384
+ if not any(t in combined for t in filter_topics):
385
+ continue
386
+ try:
387
+ node_r = ingest_text(
388
+ content, graph,
389
+ title=title,
390
+ source_url=link,
391
+ strategy=strategy or "news_article",
392
+ context_purpose=context_purpose,
393
+ )
394
+ if node_r and (getattr(node_r, 'entity_ids', None) or getattr(node_r, 'fact_ids', None)):
395
+ ingested_count += 1
396
+ pipe_result["nodes_created"] += len(getattr(node_r, 'entity_ids', [])) + len(getattr(node_r, 'fact_ids', []))
397
+ elif isinstance(node_r, dict) and node_r.get('node_id'):
398
+ ingested_count += 1
399
+ pipe_result["nodes_created"] += 1
400
+ except Exception:
401
+ pass
402
+ pipe_result["pages_ingested"] += ingested_count
403
+ if not entries:
404
+ pipe_result["errors"].append(f"RSS {source}: no entries returned")
405
+ except ImportError:
406
+ pipe_result["errors"].append("feedparser not installed — run: pip install feedparser")
407
+ except Exception as rss_err:
408
+ pipe_result["errors"].append(f"RSS {source}: {rss_err}")
409
+ continue
410
+
411
+ elif src_type == "api":
412
+ # API sources — fetch and ingest as text
413
+ from ..ingestion.smart_ingest import ingest_text
414
+ try:
415
+ import urllib.request
416
+ with urllib.request.urlopen(source, timeout=30) as resp:
417
+ api_data = resp.read().decode("utf-8")
418
+ r = ingest_text(
419
+ api_data, graph,
420
+ title=f"{pipe_name}: {source}",
421
+ source_url=source,
422
+ strategy=strategy or None,
423
+ context_purpose=context_purpose,
424
+ )
425
+ except Exception as fetch_err:
426
+ pipe_result["errors"].append(f"API fetch: {fetch_err}")
427
+ continue
428
+ else:
429
+ # URL/crawl/rss sources
430
+ r = ingest_crawl(
431
+ source, graph, max_pages=max_pages,
432
+ pipeline=strategy or "smart_pipeline",
433
+ context_purpose=context_purpose,
434
+ **crawl_kwargs,
435
+ )
436
+
437
+ if hasattr(r, 'errors') and r.errors:
438
+ if any("Duplicate" in str(e) for e in r.errors):
439
+ pipe_result["deduped"] += 1
440
+ else:
441
+ pipe_result["errors"].extend(r.errors)
442
+ elif hasattr(r, 'ingested'):
443
+ pipe_result["pages_ingested"] += len(r.ingested)
444
+ pipe_result["nodes_created"] += sum(
445
+ item.get("nodes", 0) for item in (r.ingested or [])
446
+ )
447
+ elif hasattr(r, 'entity_ids'):
448
+ pipe_result["pages_ingested"] += 1
449
+ pipe_result["nodes_created"] += len(r.entity_ids) + len(r.fact_ids)
450
+ except Exception as exc:
451
+ pipe_result["errors"].append(f"{source}: {exc}")
452
+
453
+ # Update per-pipeline schedule
454
+ interval_str = pipe_config.get("interval", "daily")
455
+ try:
456
+ from contextsynapse.intelligence.collector import _INTERVAL_SECONDS
457
+ except Exception:
458
+ _INTERVAL_SECONDS = {}
459
+ _fallback = {
460
+ "1s": 1, "5s": 5, "10s": 10, "30s": 30,
461
+ "1m": 60, "2m": 120, "5m": 300, "10m": 600, "15m": 900,
462
+ "30m": 1800, "1h": 3600, "6h": 21600, "12h": 43200,
463
+ "daily": 86400, "weekly": 604800,
464
+ }
465
+ interval_secs = (_INTERVAL_SECONDS or _fallback).get(interval_str, 86400)
466
+ pipe_config.setdefault("schedule", {})
467
+ pipe_config["schedule"]["last_run_at"] = now.isoformat()
468
+ pipe_config["schedule"]["next_run_at"] = (now + timedelta(seconds=interval_secs)).isoformat()
469
+ pipe_config["schedule"]["last_nodes_created"] = pipe_result.get("nodes_created", 0)
470
+ pipe_config["schedule"]["last_pages_ingested"] = pipe_result.get("pages_ingested", 0)
471
+ pipe_config["schedule"]["last_deduped"] = pipe_result.get("deduped", 0)
472
+ pipe_config["schedule"]["last_error_count"] = len(pipe_result.get("errors", []))
473
+ pipe_config["schedule"]["last_status"] = "error" if pipe_result.get("errors") else "ok"
474
+
475
+ combined_result["pipelines_run"] += 1
476
+ combined_result["total_pages_ingested"] += pipe_result["pages_ingested"]
477
+ combined_result["total_nodes_created"] += pipe_result["nodes_created"]
478
+ combined_result["total_deduped"] += pipe_result["deduped"]
479
+ combined_result["pipeline_results"].append(pipe_result)
480
+
481
+ logger.info("[MULTI-PIPE] '%s' pipeline '%s': %d ingested, %d deduped",
482
+ ctx.name, pipe_name, pipe_result["pages_ingested"], pipe_result["deduped"])
483
+
484
+ # Persist updated pipeline schedules
485
+ try:
486
+ updated_config = dict(ctx.config or {})
487
+ updated_config["pipelines"] = pipelines_list
488
+ context_manager.update_context(context_id, config=updated_config)
489
+ except Exception:
490
+ pass
491
+
492
+ if any(p.get("errors") for p in combined_result["pipeline_results"]):
493
+ combined_result["status"] = "partial"
494
+
495
+ logger.info("[MULTI-PIPE] Context '%s': %d pipelines run, %d total ingested, %d deduped",
496
+ ctx.name, combined_result["pipelines_run"],
497
+ combined_result["total_pages_ingested"], combined_result["total_deduped"])
498
+
499
+ return combined_result
500
+
501
+
502
+ def execute_context_pipeline(
503
+ context_id: str,
504
+ context_manager,
505
+ graph_registry=None,
506
+ ) -> Dict[str, Any]:
507
+ """Execute a pipeline run for a context.
508
+
509
+ Reads pipeline settings from ``context.config["pipeline"]`` and schedule
510
+ state from ``context.config["schedule"]``. After the run it writes updated
511
+ schedule stats back to the context via ``context_manager.update_context``.
512
+
513
+ Returns a result dict with keys: status, context_id, pages_ingested,
514
+ nodes_created, pages_filtered, pages_failed (and optionally ``error``).
515
+ """
516
+ ctx = context_manager.get_context(context_id)
517
+ if not ctx:
518
+ return {"status": "error", "context_id": context_id, "error": "Context not found"}
519
+
520
+ # Multi-pipeline support: if config["pipelines"] is a list, run each pipeline
521
+ pipelines_list = (ctx.config or {}).get("pipelines", [])
522
+ if isinstance(pipelines_list, list) and len(pipelines_list) > 0:
523
+ return _execute_multi_pipeline(context_id, ctx, pipelines_list, context_manager, graph_registry)
524
+
525
+ pipeline_config = ctx.config.get("pipeline", {}) if ctx.config else {}
526
+ schedule = ctx.config.get("schedule", {}) if ctx.config else {}
527
+
528
+ source_url = pipeline_config.get("source_url", "")
529
+ if not source_url:
530
+ return {"status": "error", "context_id": context_id, "error": "No source URL configured"}
531
+
532
+ # Generate run_id for rollback tracking
533
+ import uuid as _uuid
534
+ run_id = f"run_{_uuid.uuid4().hex[:12]}"
535
+
536
+ # Register as a Job so it shows in the Jobs page
537
+ job = None
538
+ jm = None
539
+ try:
540
+ from ..ingestion.job_manager import get_job_manager, Job, JobStatus
541
+ jm = get_job_manager()
542
+ job = Job(
543
+ job_type="pipeline_crawl",
544
+ pipeline=pipeline_config.get("template_id", "smart-article"),
545
+ context_id=context_id,
546
+ input_summary=source_url,
547
+ )
548
+ job.job_id = f"pipe_{context_id[:8]}_{datetime.now(timezone.utc).strftime('%H%M%S')}"
549
+ job.status = JobStatus.RUNNING
550
+ job.created_at = datetime.now(timezone.utc).isoformat()
551
+ job.started_at = job.created_at
552
+ job.add_log("start", f"Pipeline crawl: {source_url} (max {pipeline_config.get('max_pages', 5)} pages)")
553
+ jm._jobs[job.job_id] = job
554
+ jm._save_to_redis(job)
555
+ except Exception as e:
556
+ logger.debug("Job registration skipped: %s", e)
557
+ job = None
558
+
559
+ now = datetime.now(timezone.utc)
560
+
561
+ try:
562
+ # Get target graph
563
+ namespace = ctx.graph_namespace or context_id
564
+ if graph_registry is not None:
565
+ graph = graph_registry.get_graph(namespace, load_if_missing=True)
566
+ if not graph:
567
+ graph = graph_registry.create_graph(namespace)
568
+ else:
569
+ graph = _get_graph(namespace)
570
+
571
+ # Build filter_config as a proper FilterConfig dataclass
572
+ from ..ingestion.filters import FilterConfig, SemanticRule
573
+
574
+ filters = ctx.config.get("filters", {}) if ctx.config else {}
575
+ crawl_kwargs: Dict[str, Any] = {}
576
+ fc = None
577
+
578
+ if filters:
579
+ semantic_rules = []
580
+ for rule in (filters.get("semantic_rules") or []):
581
+ semantic_rules.append(SemanticRule(
582
+ query=rule.get("description", ""),
583
+ direction="include",
584
+ threshold=rule.get("threshold", 0.6),
585
+ ))
586
+ fc = FilterConfig(
587
+ keywords_include=filters.get("keyword_include") or [],
588
+ keywords_exclude=filters.get("keyword_exclude") or [],
589
+ semantic_rules=semantic_rules,
590
+ semantic_query=semantic_rules[0].query if semantic_rules else "",
591
+ semantic_auto_threshold=semantic_rules[0].threshold if semantic_rules else 0.55,
592
+ )
593
+ else:
594
+ include_kw = pipeline_config.get("include_keywords") or []
595
+ exclude_kw = pipeline_config.get("exclude_keywords") or []
596
+ semantic_filter_val = pipeline_config.get("semantic_filter", "")
597
+ semantic_threshold = pipeline_config.get("semantic_threshold", 0.6)
598
+ if include_kw or exclude_kw or semantic_filter_val:
599
+ fc = FilterConfig(
600
+ keywords_include=include_kw,
601
+ keywords_exclude=exclude_kw,
602
+ semantic_query=semantic_filter_val,
603
+ semantic_auto_threshold=semantic_threshold,
604
+ )
605
+
606
+ if fc:
607
+ crawl_kwargs["filter_config"] = fc
608
+
609
+ max_pages = pipeline_config.get("max_pages", 5)
610
+ schema_id = pipeline_config.get("schema_id") or None
611
+ schema_mode = pipeline_config.get("schema_mode") or "schema_plus"
612
+ template_id = pipeline_config.get("template_id", "")
613
+ # Derive pipeline name from template_id (strip "builtin:" prefix, convert hyphens to underscores)
614
+ pipeline_name = template_id.replace("builtin:", "").replace("-", "_") if template_id else "smart_article"
615
+
616
+ # Load extraction schema as IngestionSchema (what ingest_crawl expects)
617
+ schema_obj = None
618
+ if schema_id:
619
+ try:
620
+ from ..ingestion.schema_extractor import IngestionSchema, _parse_yaml_schema
621
+ from ..extraction.schema_loader import get_default_schema_yaml
622
+ # Load the YAML string for this schema
623
+ yaml_str = get_default_schema_yaml(schema_id)
624
+ if yaml_str:
625
+ schema_obj = _parse_yaml_schema(yaml_str, name=schema_id)
626
+ if not schema_obj:
627
+ # Try loading from schema registry as YAML
628
+ from ..extraction.schema_loader import SchemaRegistry
629
+ sr = SchemaRegistry(graph)
630
+ ext_schema = sr.get(schema_id)
631
+ if ext_schema and hasattr(ext_schema, 'node_types'):
632
+ # Convert ExtractionSchema to IngestionSchema
633
+ node_types = {}
634
+ for nt_name, nt_def in ext_schema.node_types.items():
635
+ fields = nt_def.fields if hasattr(nt_def, 'fields') else []
636
+ node_types[nt_name] = {"fields": fields}
637
+ edge_types = {}
638
+ for et_name, et_def in ext_schema.edge_types.items():
639
+ src = et_def.source if hasattr(et_def, 'source') else '*'
640
+ tgt = et_def.target if hasattr(et_def, 'target') else '*'
641
+ edge_types[et_name] = {"from": src, "to": tgt}
642
+ schema_obj = IngestionSchema(name=schema_id, node_types=node_types, edge_types=edge_types)
643
+ if schema_obj:
644
+ logger.info("[PIPELINE] Loaded schema '%s': %d node types, %d edge types",
645
+ schema_id, len(schema_obj.node_types), len(schema_obj.edge_types))
646
+ else:
647
+ logger.warning("[PIPELINE] Schema '%s' not found — using default", schema_id)
648
+ except Exception as e:
649
+ logger.warning("[PIPELINE] Failed to load schema '%s': %s", schema_id, e)
650
+
651
+ # Log crawl stages to the Job
652
+ def _on_stage(stage_name, details=None):
653
+ if job:
654
+ msg = (details or {}).get("message", "")
655
+ job.add_log(stage_name, msg[:200] if msg else stage_name)
656
+ job.progress = stage_name
657
+ if jm:
658
+ jm._save_to_redis(job)
659
+
660
+ crawl_result = ingest_crawl(
661
+ start_url=source_url,
662
+ db=graph,
663
+ max_pages=max_pages,
664
+ schema=schema_obj,
665
+ pipeline=pipeline_name,
666
+ mode="auto",
667
+ debug=False,
668
+ owner=run_id,
669
+ on_stage=_on_stage,
670
+ **crawl_kwargs,
671
+ )
672
+
673
+ ingested = crawl_result.get("ingested", [])
674
+ filtered = crawl_result.get("filtered", [])
675
+ failed = crawl_result.get("failed", [])
676
+
677
+ pages_ingested = len(ingested)
678
+ nodes_created = sum(
679
+ item.get("nodes", 0) or
680
+ (item.get("passages", 0) + item.get("entities", 0) + item.get("facts", 0))
681
+ for item in ingested
682
+ )
683
+ edges_created = sum(item.get("edges", 0) for item in ingested)
684
+ total_passages = sum(item.get("passages", 0) for item in ingested)
685
+ total_entities = sum(item.get("entities", 0) for item in ingested)
686
+ total_facts = sum(item.get("facts", 0) for item in ingested)
687
+ pages_filtered = len(filtered)
688
+ pages_failed = len(failed)
689
+
690
+ # Update schedule state
691
+ total_runs = int(schedule.get("total_runs", 0)) + 1
692
+ total_documents = int(schedule.get("total_documents", 0)) + pages_ingested
693
+
694
+ trigger_type = schedule.get("trigger_type") or pipeline_config.get("trigger_type", "one_time")
695
+ if trigger_type == "one_time":
696
+ next_run_at = ""
697
+ new_status = "completed"
698
+ else:
699
+ interval = max(float(schedule.get("interval_minutes") or pipeline_config.get("interval_minutes", 60)), 1 / 60)
700
+ next_run_at = (now + timedelta(minutes=interval)).isoformat()
701
+ new_status = "active"
702
+
703
+ updated_schedule = dict(schedule)
704
+ updated_schedule["last_run_at"] = now.isoformat()
705
+ updated_schedule["next_run_at"] = next_run_at
706
+ updated_schedule["total_runs"] = total_runs
707
+ updated_schedule["total_documents"] = total_documents
708
+ updated_schedule["last_error"] = ""
709
+ if trigger_type == "one_time":
710
+ updated_schedule["status"] = new_status
711
+
712
+ updated_config = dict(ctx.config or {})
713
+ updated_config["schedule"] = updated_schedule
714
+ context_manager.update_context(context_id, config=updated_config)
715
+
716
+ # Save run record to Redis
717
+ run_record = {
718
+ "run_id": run_id,
719
+ "run_at": now.isoformat(),
720
+ "pages_ingested": pages_ingested,
721
+ "nodes_created": nodes_created,
722
+ "edges_created": edges_created,
723
+ "passages": total_passages,
724
+ "entities": total_entities,
725
+ "facts": total_facts,
726
+ "pages_filtered": pages_filtered,
727
+ "pages_failed": pages_failed,
728
+ "duration_ms": int((datetime.now(timezone.utc) - now).total_seconds() * 1000),
729
+ }
730
+ _save_context_run(context_id, run_record)
731
+
732
+ # Refresh context stats (item_count, estimated_tokens)
733
+ try:
734
+ context_manager.refresh_stats(context_id)
735
+ except Exception:
736
+ pass
737
+
738
+ # Rebuild ContextIntelligence node (auto-briefing data)
739
+ try:
740
+ from ..core.context_state import get_context_state
741
+ state = get_context_state(graph, namespace)
742
+ state._last_rebuilt = 0 # force rebuild
743
+ state.rebuild()
744
+ except Exception:
745
+ pass
746
+
747
+ # Propagate event (best-effort)
748
+ try:
749
+ from ..context.propagation import get_propagator
750
+ propagator = get_propagator()
751
+ propagator.propagate(
752
+ source_agent="pipeline_executor",
753
+ event_type="pipeline_completed",
754
+ content=(
755
+ f"Context pipeline '{ctx.name}' finished: "
756
+ f"{pages_ingested} pages ingested, {nodes_created} nodes created."
757
+ ),
758
+ namespace=namespace,
759
+ )
760
+ except Exception as prop_err:
761
+ logger.debug("Propagation skipped: %s", prop_err)
762
+
763
+ result: Dict[str, Any] = {
764
+ "status": "success",
765
+ "context_id": context_id,
766
+ "pages_ingested": pages_ingested,
767
+ "nodes_created": nodes_created,
768
+ "pages_filtered": pages_filtered,
769
+ "pages_failed": pages_failed,
770
+ }
771
+
772
+ # Update Job status
773
+ if job and jm:
774
+ try:
775
+ from ..ingestion.job_manager import JobStatus
776
+ job.status = JobStatus.COMPLETED
777
+ job.completed_at = datetime.now(timezone.utc).isoformat()
778
+ job.result = result
779
+ job.add_log("done", f"Completed: {pages_ingested} pages, {nodes_created} nodes")
780
+ jm._save_to_redis(job)
781
+ except Exception:
782
+ pass
783
+
784
+ logger.info("Context pipeline %s completed: %s", context_id, result)
785
+ return result
786
+
787
+ except Exception as exc:
788
+ err_msg = str(exc)
789
+ logger.exception("Context pipeline %s failed: %s", context_id, err_msg)
790
+
791
+ # Update schedule with error info
792
+ updated_schedule = dict(schedule)
793
+ updated_schedule["last_run_at"] = now.isoformat()
794
+ updated_schedule["last_error"] = err_msg
795
+ updated_schedule["total_runs"] = int(schedule.get("total_runs", 0)) + 1
796
+ updated_config = dict(ctx.config or {})
797
+ updated_config["schedule"] = updated_schedule
798
+ try:
799
+ context_manager.update_context(context_id, config=updated_config)
800
+ except Exception:
801
+ pass
802
+
803
+ # Update Job status on error
804
+ if job and jm:
805
+ try:
806
+ from ..ingestion.job_manager import JobStatus
807
+ job.status = JobStatus.FAILED
808
+ job.completed_at = datetime.now(timezone.utc).isoformat()
809
+ job.error = err_msg
810
+ job.add_log("error", f"Failed: {err_msg}")
811
+ jm._save_to_redis(job)
812
+ except Exception:
813
+ pass
814
+
815
+ return {
816
+ "status": "error",
817
+ "context_id": context_id,
818
+ "error": err_msg,
819
+ }
820
+
821
+
822
+ def _save_context_run(context_id: str, run_data: Dict) -> None:
823
+ """Persist a context pipeline run record to Redis (best-effort)."""
824
+ import os
825
+ redis_url = os.environ.get("CONTEXTSYNAPSE_REDIS_URL") or os.environ.get("AICONTEXTDB_REDIS_URL", "")
826
+ if not redis_url:
827
+ return
828
+ try:
829
+ import redis as _redis
830
+ r = _redis.from_url(redis_url)
831
+ key = f"context_pipeline:{context_id}:runs"
832
+ r.lpush(key, json.dumps(run_data))
833
+ r.ltrim(key, 0, 99)
834
+ except Exception as e:
835
+ logger.debug("Could not save context run to Redis: %s", e)
836
+
837
+
838
+ def _get_context_runs(context_id: str, limit: int = 20) -> list:
839
+ """Retrieve run history for a context pipeline from Redis."""
840
+ import os
841
+ redis_url = os.environ.get("CONTEXTSYNAPSE_REDIS_URL") or os.environ.get("AICONTEXTDB_REDIS_URL", "")
842
+ if not redis_url:
843
+ return []
844
+ try:
845
+ import redis as _redis
846
+ r = _redis.from_url(redis_url)
847
+ raws = r.lrange(f"context_pipeline:{context_id}:runs", 0, limit - 1)
848
+ return [json.loads(x) for x in raws]
849
+ except Exception as e:
850
+ logger.debug("Could not read context runs from Redis: %s", e)
851
+ return []