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,1054 @@
1
+ """
2
+ Context Quality Scorer
3
+ ======================
4
+ Staleness detection, completeness metrics, and quality indicators
5
+ for session context.
6
+
7
+ Also exposes score_node() — the Universal Context Layer ingest gate
8
+ for per-node quality scoring (0-100, no I/O, <1ms).
9
+ """
10
+
11
+ from __future__ import annotations
12
+
13
+ import logging
14
+ import re
15
+ import time
16
+ from dataclasses import dataclass, field
17
+ from typing import Any, Dict, List, Optional
18
+
19
+ logger = logging.getLogger(__name__)
20
+
21
+ # ── Gate 1: Ingest ─────────────────────────────────────────────
22
+
23
+ # Expected properties per label (for completeness scoring)
24
+ _EXPECTED_PROPS = {
25
+ "Person": {"name", "confidence", "source_url", "created_by", "mention_count"},
26
+ "Organization": {"name", "confidence", "source_url", "created_by"},
27
+ "Location": {"name", "confidence", "source_url"},
28
+ "Event": {"name", "confidence", "source_url", "created_by"},
29
+ "Fact": {"statement", "subject", "source_url", "created_by", "confidence"},
30
+ "Document": {"title", "source_url", "content_type"},
31
+ "Passage": {"content", "source_url"},
32
+ }
33
+
34
+ # Pattern: 2+ capitalized words (real names like "Benjamin Netanyahu")
35
+ _REAL_NAME_RE = re.compile(r"^[A-Z][a-z]+(?:\s+[A-Z][a-z]+)+$")
36
+
37
+ # Pattern: generic/garbage text — short, no caps, or known noise patterns
38
+ _NOISE_PATTERNS = [
39
+ re.compile(r"^Edition\b", re.IGNORECASE),
40
+ re.compile(r"^(IN|US|GCC|English|Hindi)\b"),
41
+ re.compile(r"^(Match|Results|Click|Share|Subscribe|Download)\b", re.IGNORECASE),
42
+ re.compile(r"^(Latest|Breaking|Trending|Top)\s+(News|Videos|Stories)", re.IGNORECASE),
43
+ ]
44
+
45
+ # Pattern: content with specific claims (numbers, dates, named entities)
46
+ _SPECIFIC_RE = re.compile(
47
+ r"\d{2,}|%|\$|Rs\.?|USD|January|February|March|April|May|June|July|August|September|October|November|December|20\d\d"
48
+ )
49
+
50
+
51
+ def score_node(node: Dict[str, Any]) -> int:
52
+ """Score a node 0-100 for quality. <1ms, no I/O.
53
+
54
+ Parameters
55
+ ----------
56
+ node : dict
57
+ Must have 'label', 'properties', and optionally 'edge_count' and '_is_duplicate'.
58
+
59
+ Returns
60
+ -------
61
+ int
62
+ Quality score clamped to 0-100.
63
+ """
64
+ label = node.get("label", "")
65
+ props = node.get("properties") or {}
66
+ edge_count = node.get("edge_count", 0)
67
+ is_duplicate = node.get("_is_duplicate", False)
68
+
69
+ score = 50 # base
70
+
71
+ # ── Connectivity ──
72
+ if edge_count == 0:
73
+ score -= 30
74
+ elif edge_count <= 2:
75
+ pass # neutral
76
+ elif edge_count <= 5:
77
+ score += 10
78
+ else:
79
+ score += 20
80
+
81
+ # ── Name quality ──
82
+ name = props.get("name", "")
83
+ statement = props.get("statement", name)
84
+ text = statement or name
85
+
86
+ if name:
87
+ # Noise check takes priority — catches "Match Results" etc.
88
+ _noise_hit = False
89
+ for pat in _NOISE_PATTERNS:
90
+ if pat.search(name):
91
+ score -= 20
92
+ _noise_hit = True
93
+ break
94
+
95
+ if not _noise_hit:
96
+ if _REAL_NAME_RE.match(name):
97
+ score += 15
98
+ else:
99
+ # Short generic name (< 3 chars or single lowercase word)
100
+ if len(name) < 3 or (len(name.split()) == 1 and name[0].islower()):
101
+ score -= 10
102
+
103
+ # ── Content specificity ──
104
+ if text and len(text) > 30:
105
+ if _SPECIFIC_RE.search(text):
106
+ score += 15
107
+ elif len(text) < 20:
108
+ score -= 20
109
+ elif text:
110
+ if len(text) < 10:
111
+ score -= 20
112
+
113
+ # ── Source reliability ──
114
+ if props.get("source_url"):
115
+ score += 10
116
+ if props.get("created_by", "").startswith("pipeline:"):
117
+ score += 5
118
+
119
+ # ── Duplication ──
120
+ if is_duplicate:
121
+ score -= 40
122
+
123
+ # ── Property completeness ──
124
+ expected = _EXPECTED_PROPS.get(label, set())
125
+ if expected:
126
+ filled = sum(1 for k in expected if props.get(k))
127
+ completeness = filled / len(expected)
128
+ score += int(completeness * 20)
129
+
130
+ return max(0, min(100, score))
131
+
132
+
133
+ @dataclass
134
+ class QualityItem:
135
+ item_id: str
136
+ label: str
137
+ age_seconds: float
138
+ staleness: float # 0.0 (fresh) to 1.0 (stale)
139
+ confidence: float # 0.0 to 1.0
140
+ has_content: bool = True
141
+
142
+ def to_dict(self) -> dict:
143
+ return {
144
+ "item_id": self.item_id,
145
+ "label": self.label,
146
+ "age_seconds": self.age_seconds,
147
+ "staleness": round(self.staleness, 3),
148
+ "confidence": round(self.confidence, 3),
149
+ "has_content": self.has_content,
150
+ }
151
+
152
+
153
+ @dataclass
154
+ class QualityReport:
155
+ session_id: str
156
+ overall_score: float # 0.0 to 1.0
157
+ grade: str # A, B, C, D, F
158
+ total_items: int
159
+ stale_items: int
160
+ fresh_items: int
161
+ avg_staleness: float
162
+ avg_confidence: float
163
+ items: List[QualityItem] = field(default_factory=list)
164
+ warnings: List[str] = field(default_factory=list)
165
+
166
+ def to_dict(self) -> dict:
167
+ return {
168
+ "session_id": self.session_id,
169
+ "overall_score": round(self.overall_score, 3),
170
+ "grade": self.grade,
171
+ "total_items": self.total_items,
172
+ "stale_items": self.stale_items,
173
+ "fresh_items": self.fresh_items,
174
+ "avg_staleness": round(self.avg_staleness, 3),
175
+ "avg_confidence": round(self.avg_confidence, 3),
176
+ "items": [i.to_dict() for i in self.items],
177
+ "warnings": self.warnings,
178
+ }
179
+
180
+
181
+ class ContextQualityScorer:
182
+ """Score context items for staleness, completeness, and quality."""
183
+
184
+ def __init__(self, staleness_threshold_hours: float = 24.0):
185
+ self.staleness_threshold = staleness_threshold_hours * 3600
186
+
187
+ def score_items(self, items: List[Dict[str, Any]], session_id: str = "") -> QualityReport:
188
+ """Score a list of context items."""
189
+ now = time.time()
190
+ quality_items = []
191
+ warnings = []
192
+
193
+ for item in items:
194
+ created_at = item.get("created_at", item.get("timestamp", now))
195
+ if isinstance(created_at, str):
196
+ try:
197
+ from datetime import datetime
198
+ created_at = datetime.fromisoformat(created_at.replace("Z", "+00:00")).timestamp()
199
+ except Exception:
200
+ created_at = now
201
+
202
+ age = now - created_at
203
+ staleness = min(age / self.staleness_threshold, 1.0) if self.staleness_threshold > 0 else 0.0
204
+ confidence = item.get("confidence", 1.0 - (staleness * 0.5))
205
+ has_content = bool(item.get("content") or item.get("properties") or item.get("data"))
206
+
207
+ qi = QualityItem(
208
+ item_id=item.get("id", item.get("node_id", str(len(quality_items)))),
209
+ label=item.get("label", item.get("type", item.get("content_type", "unknown"))),
210
+ age_seconds=age,
211
+ staleness=staleness,
212
+ confidence=max(0.0, min(1.0, confidence)),
213
+ has_content=has_content,
214
+ )
215
+ quality_items.append(qi)
216
+
217
+ if not quality_items:
218
+ return QualityReport(
219
+ session_id=session_id,
220
+ overall_score=0.0,
221
+ grade="F",
222
+ total_items=0,
223
+ stale_items=0,
224
+ fresh_items=0,
225
+ avg_staleness=0.0,
226
+ avg_confidence=0.0,
227
+ warnings=["No context items found"],
228
+ )
229
+
230
+ stale_count = sum(1 for q in quality_items if q.staleness > 0.7)
231
+ fresh_count = sum(1 for q in quality_items if q.staleness < 0.3)
232
+ avg_staleness = sum(q.staleness for q in quality_items) / len(quality_items)
233
+ avg_confidence = sum(q.confidence for q in quality_items) / len(quality_items)
234
+
235
+ # Overall score: weighted combination
236
+ freshness_score = 1.0 - avg_staleness
237
+ completeness_score = sum(1 for q in quality_items if q.has_content) / len(quality_items)
238
+ overall = (freshness_score * 0.5) + (avg_confidence * 0.3) + (completeness_score * 0.2)
239
+
240
+ if stale_count > len(quality_items) * 0.5:
241
+ warnings.append(f"{stale_count} of {len(quality_items)} items are stale")
242
+ if not any(q.has_content for q in quality_items):
243
+ warnings.append("No items have content")
244
+
245
+ grade = "A" if overall >= 0.9 else "B" if overall >= 0.75 else "C" if overall >= 0.6 else "D" if overall >= 0.4 else "F"
246
+
247
+ return QualityReport(
248
+ session_id=session_id,
249
+ overall_score=overall,
250
+ grade=grade,
251
+ total_items=len(quality_items),
252
+ stale_items=stale_count,
253
+ fresh_items=fresh_count,
254
+ avg_staleness=avg_staleness,
255
+ avg_confidence=avg_confidence,
256
+ items=quality_items[:20], # Limit to 20 items in report
257
+ warnings=warnings,
258
+ )
259
+
260
+ def score_session(self, session, graph=None) -> QualityReport:
261
+ """Score a session's context quality from its graph nodes."""
262
+ items = []
263
+ if graph:
264
+ try:
265
+ nodes = graph.get_all_nodes()
266
+ for n in (nodes or []):
267
+ items.append({
268
+ "id": n.get("node_id", ""),
269
+ "type": n.get("node_type", ""),
270
+ "content": str(n.get("properties", {})),
271
+ "properties": n.get("properties", {}),
272
+ "created_at": n.get("created_at", time.time()),
273
+ })
274
+ except Exception:
275
+ pass
276
+
277
+ session_id = session.session_id if hasattr(session, 'session_id') else str(session)
278
+ return self.score_items(items, session_id)
279
+
280
+
281
+ import json # noqa: E402 — appended after class definitions
282
+
283
+
284
+ class EntityIndex:
285
+ """O(1) entity lookup via Redis hash, with in-memory fallback."""
286
+
287
+ _TITLES = re.compile(
288
+ r"^(Mr\.?|Mrs\.?|Ms\.?|Dr\.?|Prof\.?|PM|President|Minister|Gen\.?|Col\.?|Sgt\.?)\s+",
289
+ re.IGNORECASE,
290
+ )
291
+
292
+ def __init__(self, redis_client=None):
293
+ self._redis = redis_client
294
+ self._mem: Dict[str, Dict[str, Dict]] = {}
295
+
296
+ @staticmethod
297
+ def normalize_name(name: str) -> str:
298
+ n = name.strip()
299
+ n = EntityIndex._TITLES.sub("", n)
300
+ n = re.sub(r"\s+", " ", n).strip().lower()
301
+ return n
302
+
303
+ def _key(self, graph: str) -> str:
304
+ return f"entity_idx:{graph}"
305
+
306
+ def _field(self, label: str, name: str) -> str:
307
+ return f"{label}:{self.normalize_name(name)}"
308
+
309
+ def put(self, graph: str, label: str, name: str, node_id: str, quality: int = 50) -> Dict:
310
+ field = self._field(label, name)
311
+ existing = self.get(graph, label, name)
312
+ if existing:
313
+ existing["mention_count"] += 1
314
+ if quality > existing.get("quality", 0):
315
+ existing["quality"] = quality
316
+ data = existing
317
+ else:
318
+ data = {"node_id": node_id, "mention_count": 1, "quality": quality,
319
+ "name": name.strip(), "label": label}
320
+ if self._redis:
321
+ self._redis.hset(self._key(graph), field, json.dumps(data))
322
+ else:
323
+ self._mem.setdefault(graph, {})[field] = data
324
+ return data
325
+
326
+ def get(self, graph: str, label: str, name: str) -> Optional[Dict]:
327
+ field = self._field(label, name)
328
+ if self._redis:
329
+ raw = self._redis.hget(self._key(graph), field)
330
+ return json.loads(raw) if raw else None
331
+ return self._mem.get(graph, {}).get(field)
332
+
333
+ def top(self, graph: str, label: str, limit: int = 5) -> List[Dict]:
334
+ if self._redis:
335
+ raw = self._redis.hgetall(self._key(graph))
336
+ entries = []
337
+ prefix = f"{label}:"
338
+ for field, val in raw.items():
339
+ f = field.decode() if isinstance(field, bytes) else field
340
+ if f.startswith(prefix):
341
+ entries.append(json.loads(val))
342
+ else:
343
+ entries = [v for f, v in self._mem.get(graph, {}).items()
344
+ if f.startswith(f"{label}:")]
345
+ entries.sort(key=lambda e: e.get("mention_count", 0) * e.get("quality", 50), reverse=True)
346
+ return entries[:limit]
347
+
348
+ def all_labels(self, graph: str) -> Dict[str, int]:
349
+ counts: Dict[str, int] = {}
350
+ if self._redis:
351
+ raw = self._redis.hgetall(self._key(graph))
352
+ for field in raw:
353
+ f = field.decode() if isinstance(field, bytes) else field
354
+ label = f.split(":")[0]
355
+ counts[label] = counts.get(label, 0) + 1
356
+ else:
357
+ for f in self._mem.get(graph, {}):
358
+ label = f.split(":")[0]
359
+ counts[label] = counts.get(label, 0) + 1
360
+ return counts
361
+
362
+
363
+ # ── Gate 2: Search ─────────────────────────────────────────────
364
+
365
+ _RANKING_WEIGHTS = {
366
+ "news": {"relevance": 0.50, "quality": 0.20, "freshness": 0.30},
367
+ "knowledge": {"relevance": 0.40, "quality": 0.50, "freshness": 0.10},
368
+ "code": {"relevance": 0.50, "quality": 0.40, "freshness": 0.10},
369
+ "mixed": {"relevance": 0.60, "quality": 0.25, "freshness": 0.15},
370
+ }
371
+
372
+
373
+ def _jaccard(a: str, b: str) -> float:
374
+ wa = set(a.lower().split())
375
+ wb = set(b.lower().split())
376
+ if not wa or not wb:
377
+ return 0.0
378
+ return len(wa & wb) / len(wa | wb)
379
+
380
+
381
+ class QualityFilter:
382
+ """Search gate — filter, dedup, and rank results by quality."""
383
+
384
+ @staticmethod
385
+ def threshold(context_quality: int) -> int:
386
+ return max(20, round(context_quality * 0.4))
387
+
388
+ @staticmethod
389
+ def filter_results(results: list, context_quality: int = 50) -> list:
390
+ t = QualityFilter.threshold(context_quality)
391
+ return [r for r in results if r.get("_quality", 50) >= t]
392
+
393
+ @staticmethod
394
+ def dedup_results(results: list, similarity_threshold: float = 0.8) -> list:
395
+ if not results:
396
+ return results
397
+ kept = []
398
+ for r in results:
399
+ text = r.get("name", "") or r.get("statement", "")
400
+ is_dup = False
401
+ for k in kept:
402
+ k_text = k.get("name", "") or k.get("statement", "")
403
+ if text and k_text and _jaccard(text, k_text) >= similarity_threshold:
404
+ if r.get("_quality", 0) > k.get("_quality", 0):
405
+ kept.remove(k)
406
+ kept.append(r)
407
+ is_dup = True
408
+ break
409
+ if not is_dup:
410
+ kept.append(r)
411
+ return kept
412
+
413
+ @staticmethod
414
+ def cap_per_label(results: list, max_per_label: int = 3) -> list:
415
+ counts: Dict[str, int] = {}
416
+ capped = []
417
+ for r in results:
418
+ label = r.get("label", "")
419
+ counts[label] = counts.get(label, 0) + 1
420
+ if counts[label] <= max_per_label:
421
+ capped.append(r)
422
+ return capped
423
+
424
+ @staticmethod
425
+ def rank(results: list, content_type: str = "mixed", freshness_scores=None) -> list:
426
+ weights = _RANKING_WEIGHTS.get(content_type, _RANKING_WEIGHTS["mixed"])
427
+ for r in results:
428
+ relevance = r.get("score", 0.5)
429
+ quality = r.get("_quality", 50) / 100.0
430
+ freshness = (freshness_scores or {}).get(r.get("node_id", ""), 0.5)
431
+ r["_final_score"] = (
432
+ relevance * weights["relevance"]
433
+ + quality * weights["quality"]
434
+ + freshness * weights["freshness"]
435
+ )
436
+ results.sort(key=lambda r: r.get("_final_score", 0), reverse=True)
437
+ return results
438
+
439
+ @staticmethod
440
+ def apply(results: list, context_quality: int = 50,
441
+ content_type: str = "mixed", max_per_label: int = 3) -> list:
442
+ filtered = QualityFilter.filter_results(results, context_quality)
443
+ deduped = QualityFilter.dedup_results(filtered)
444
+ ranked = QualityFilter.rank(deduped, content_type)
445
+ return QualityFilter.cap_per_label(ranked, max_per_label)
446
+
447
+
448
+ from datetime import datetime, timezone # noqa: E402 — appended after class definitions
449
+
450
+ # ── Gate 3: Delivery ───────────────────────────────────────────
451
+
452
+ # Core entity types + extensible via graph discovery
453
+ _ENTITY_LABELS_CORE = ("Person", "Organization", "Location", "Event")
454
+ # Extended types that may appear in domain-specific graphs
455
+ _ENTITY_LABELS_EXTENDED = (
456
+ "Technology", "Product", "Concept", "Topic", "Disease",
457
+ "Policy", "Law", "Currency", "Metric", "Skill",
458
+ )
459
+ _ENTITY_LABELS = _ENTITY_LABELS_CORE + _ENTITY_LABELS_EXTENDED
460
+
461
+
462
+ class ManifestBuilder:
463
+ """Build and cache the context manifest."""
464
+
465
+ def __init__(self, entity_index: EntityIndex, redis_client=None):
466
+ self._idx = entity_index
467
+ self._redis = redis_client
468
+
469
+ @staticmethod
470
+ def detect_content_type(label_counts: Dict[str, int]) -> str:
471
+ # Exclude system/structural labels from content type detection
472
+ _SYSTEM_LABELS = {
473
+ "CodeBase", "VectorIndex", "BM25Index", "MemoryStore", "ToolStore",
474
+ "UserStore", "WebStore", "GeneratedStore", "KnowledgeBase",
475
+ "ArtifactStore", "SystemStore", "Session", "Context", "Project",
476
+ "ContextIntelligence", "ExperimentRun", "ExperimentScore",
477
+ "AgentMessage", "Action", "Link",
478
+ }
479
+ content_counts = {k: v for k, v in label_counts.items() if k not in _SYSTEM_LABELS}
480
+ total = sum(content_counts.values()) or 1
481
+ facts = content_counts.get("Fact", 0) + content_counts.get("Document", 0) + content_counts.get("Passage", 0)
482
+ entities = sum(content_counts.get(l, 0) for l in _ENTITY_LABELS)
483
+ if content_counts.get("Feature", 0) > 0 and content_counts.get("Requirement", 0) > 0:
484
+ return "code"
485
+ if facts / total > 0.5:
486
+ return "news"
487
+ if entities / total > 0.6:
488
+ return "knowledge"
489
+ return "mixed"
490
+
491
+ def build(self, graph_name: str, context_id: str,
492
+ node_qualities=None, label_counts=None, themes=None) -> Dict:
493
+ all_labels = self._idx.all_labels(graph_name)
494
+ if label_counts:
495
+ all_labels.update({k: v for k, v in label_counts.items() if k not in all_labels})
496
+
497
+ # Discover entity types: hardcoded + any label found in the entity index
498
+ entity_labels_to_check = set(_ENTITY_LABELS)
499
+ idx_labels = self._idx.all_labels(graph_name)
500
+ entity_labels_to_check.update(idx_labels.keys())
501
+ # Exclude non-entity labels
502
+ _NON_ENTITY = {"Fact", "Document", "Passage", "Finding", "Insight", "Task",
503
+ "Action", "Link", "AgentMessage", "ExperimentRun", "ExperimentScore",
504
+ "ContextIntelligence", "Project", "Session", "Context",
505
+ "CodeBase", "VectorIndex", "BM25Index", "MemoryStore", "ToolStore",
506
+ "UserStore", "WebStore", "GeneratedStore", "KnowledgeBase",
507
+ "ArtifactStore", "SystemStore"}
508
+ entity_labels_to_check -= _NON_ENTITY
509
+
510
+ top_entities = {}
511
+ for label in sorted(entity_labels_to_check):
512
+ top = self._idx.top(graph_name, label, limit=5)
513
+ if top:
514
+ top_entities[label] = [
515
+ {"name": e["name"], "mentions": e["mention_count"], "quality": e["quality"]}
516
+ for e in top
517
+ ]
518
+
519
+ quals = node_qualities or []
520
+ total = len(quals) or 1
521
+ stats = {
522
+ "total_nodes": total if quals else sum(all_labels.values()),
523
+ "high_quality": sum(1 for q in quals if q > 70),
524
+ "medium_quality": sum(1 for q in quals if 30 <= q <= 70),
525
+ "low_quality": sum(1 for q in quals if q < 30),
526
+ }
527
+ overall_quality = int(sum(quals) / total) if quals else 50
528
+ content_type = self.detect_content_type(label_counts or all_labels)
529
+
530
+ tool_hints = []
531
+ for label, count in sorted(all_labels.items(), key=lambda x: x[1], reverse=True):
532
+ if label in _ENTITY_LABELS and count > 0:
533
+ tool_hints.append({"tool": "search_nodes", "example": f'label:"{label}"', "result_count": count})
534
+ tool_hints.append({"tool": "search", "example": '"keyword"', "description": "text search across all nodes"})
535
+ tool_hints.append({"tool": "rag_query", "example": '"question"', "description": "semantic search with sources"})
536
+
537
+ manifest = {
538
+ "context_id": context_id,
539
+ "context_quality": overall_quality,
540
+ "content_type": content_type,
541
+ "updated_at": datetime.now(timezone.utc).isoformat(),
542
+ "stats": stats,
543
+ "top_entities": top_entities,
544
+ "themes": themes or [],
545
+ "tool_hints": tool_hints,
546
+ }
547
+ if self._redis:
548
+ self._redis.set(f"manifest:{context_id}", json.dumps(manifest), ex=86400) # 24h TTL
549
+ # Also update GraphMeta for fast dashboard reads
550
+ try:
551
+ GraphMeta(self._redis).build_from_manifest(graph_name, manifest)
552
+ except Exception:
553
+ pass
554
+ return manifest
555
+
556
+ @staticmethod
557
+ def to_text(manifest: Dict) -> str:
558
+ lines = []
559
+ q = manifest.get("context_quality", 0)
560
+ ct = manifest.get("content_type", "mixed")
561
+ stats = manifest.get("stats", {})
562
+ lines.append(f"CONTEXT QUALITY: {q}/100 ({ct})")
563
+ lines.append(f"Nodes: {stats.get('total_nodes', 0)} "
564
+ f"(high: {stats.get('high_quality', 0)}, "
565
+ f"medium: {stats.get('medium_quality', 0)}, "
566
+ f"low: {stats.get('low_quality', 0)})")
567
+ cu_count = manifest.get("cu_count", 0)
568
+ if cu_count:
569
+ lines.append(f"Context Units: {cu_count} pre-assembled topics")
570
+ lines.append("")
571
+ top = manifest.get("top_entities", {})
572
+ if top:
573
+ lines.append("TOP ENTITIES:")
574
+ for label, entities in top.items():
575
+ names = ", ".join(f"{e['name']} ({e['mentions']})" for e in entities[:5])
576
+ lines.append(f" {label}: {names}")
577
+ lines.append("")
578
+ themes = manifest.get("themes", [])
579
+ if themes:
580
+ lines.append("THEMES:")
581
+ for i, t in enumerate(themes[:5], 1):
582
+ lines.append(f" {i}. {t['name']} ({t.get('fact_count', 0)} facts)")
583
+ actors = ", ".join(t.get("key_actors", [])[:3])
584
+ if actors:
585
+ lines.append(f" Key actors: {actors}")
586
+ sample = t.get("sample_fact", "")
587
+ if sample:
588
+ lines.append(f' e.g. "{sample[:80]}"')
589
+ lines.append("")
590
+ # Suggested research tasks (for leader agents)
591
+ if top or themes:
592
+ lines.append("SUGGESTED RESEARCH TASKS:")
593
+ task_num = 0
594
+ # Generate tasks from themes
595
+ for t in (themes or [])[:3]:
596
+ task_num += 1
597
+ actors = ", ".join(t.get("key_actors", [])[:3])
598
+ actor_hint = f" (key actors: {actors})" if actors else ""
599
+ lines.append(f" {task_num}. Research: {t['name']}{actor_hint} — {t.get('fact_count', 0)} facts available")
600
+ lines.append(f" Query: search_nodes(label=\"Fact\", query=\"{t['name'][:30]}\")")
601
+ # Generate tasks from top entity types if no themes
602
+ if not themes:
603
+ for label, entities in list(top.items())[:3]:
604
+ task_num += 1
605
+ names = ", ".join(e["name"] for e in entities[:3])
606
+ lines.append(f" {task_num}. Research {label} entities: {names}")
607
+ lines.append(f" Query: search_nodes(label=\"{label}\")")
608
+ lines.append("")
609
+
610
+ # Show actionable queries with example names from the data
611
+ lines.append("READY-TO-USE QUERIES:")
612
+ if top:
613
+ for label, entities in list(top.items())[:4]:
614
+ top_names = [e["name"] for e in entities[:3]]
615
+ example_name = top_names[0] if top_names else ""
616
+ lines.append(f" search_nodes(label=\"{label}\", query=\"{example_name}\") — {len(entities)} top {label}s: {', '.join(top_names)}")
617
+ lines.append(f" search_nodes(label=\"Fact\", query=\"your topic\") — find facts about any topic")
618
+ lines.append(f" search(query=\"specific keywords\") — text search across all {stats.get('total_nodes', 0)} nodes")
619
+ lines.append(f" rag_query(question=\"your question\") — semantic search (finds similar meaning)")
620
+ return "\n".join(lines)
621
+
622
+
623
+ def get_manifest(context_id: str, redis_client=None):
624
+ """Read cached manifest. Returns None if not cached."""
625
+ if redis_client:
626
+ raw = redis_client.get(f"manifest:{context_id}")
627
+ if raw:
628
+ return json.loads(raw)
629
+ return None
630
+
631
+
632
+ # ── Graph Metadata Store ──────────────────────────────────────
633
+
634
+ class GraphMeta:
635
+ """Single Redis hash per graph for all metadata — microsecond reads.
636
+
637
+ Key: ``graph_meta:{name}``
638
+ Fields: node_count, edge_count, quality, content_type, updated_at, label_counts
639
+
640
+ Updated by: ingest gate, manifest builder, node create/delete.
641
+ Read by: all dashboard pages.
642
+ """
643
+
644
+ def __init__(self, redis_client):
645
+ self._redis = redis_client
646
+
647
+ def _key(self, graph_name: str) -> str:
648
+ return f"graph_meta:{graph_name}"
649
+
650
+ def update(self, graph_name: str, **fields):
651
+ """Update specific metadata fields. Only writes provided fields."""
652
+ if not self._redis:
653
+ return
654
+ key = self._key(graph_name)
655
+ # Serialize complex values
656
+ to_store = {}
657
+ for k, v in fields.items():
658
+ if isinstance(v, dict):
659
+ to_store[k] = json.dumps(v)
660
+ elif isinstance(v, (list, tuple)):
661
+ to_store[k] = json.dumps(v)
662
+ else:
663
+ to_store[k] = str(v)
664
+ if to_store:
665
+ self._redis.hset(key, mapping=to_store)
666
+ self._redis.expire(key, 86400 * 7) # 7 day TTL
667
+
668
+ def get(self, graph_name: str) -> Optional[Dict]:
669
+ """Read all metadata for a graph. Returns dict or None."""
670
+ if not self._redis:
671
+ return None
672
+ raw = self._redis.hgetall(self._key(graph_name))
673
+ if not raw:
674
+ return None
675
+ result = {}
676
+ for k, v in raw.items():
677
+ k = k.decode() if isinstance(k, bytes) else k
678
+ v = v.decode() if isinstance(v, bytes) else v
679
+ # Try to parse JSON values
680
+ try:
681
+ result[k] = json.loads(v)
682
+ except (json.JSONDecodeError, TypeError):
683
+ # Try numeric
684
+ try:
685
+ result[k] = int(v)
686
+ except ValueError:
687
+ result[k] = v
688
+ return result
689
+
690
+ def get_field(self, graph_name: str, field: str):
691
+ """Read a single field. Returns value or None."""
692
+ if not self._redis:
693
+ return None
694
+ raw = self._redis.hget(self._key(graph_name), field)
695
+ if raw is None:
696
+ return None
697
+ v = raw.decode() if isinstance(raw, bytes) else raw
698
+ try:
699
+ return json.loads(v)
700
+ except (json.JSONDecodeError, TypeError):
701
+ try:
702
+ return int(v)
703
+ except ValueError:
704
+ return v
705
+
706
+ def build_from_manifest(self, graph_name: str, manifest: Dict):
707
+ """Populate metadata from a manifest."""
708
+ stats = manifest.get("stats", {})
709
+ self.update(
710
+ graph_name,
711
+ node_count=stats.get("total_nodes", 0),
712
+ high_quality=stats.get("high_quality", 0),
713
+ medium_quality=stats.get("medium_quality", 0),
714
+ low_quality=stats.get("low_quality", 0),
715
+ quality=manifest.get("context_quality", 0),
716
+ content_type=manifest.get("content_type", "mixed"),
717
+ updated_at=manifest.get("updated_at", ""),
718
+ top_entities=manifest.get("top_entities", {}),
719
+ )
720
+
721
+ def list_all(self) -> List[Dict]:
722
+ """List metadata for all graphs. Fast — one SCAN + pipeline HGETALL."""
723
+ if not self._redis:
724
+ return []
725
+ cursor = 0
726
+ keys = []
727
+ while True:
728
+ cursor, batch = self._redis.scan(cursor, match="graph_meta:*", count=100)
729
+ keys.extend(batch)
730
+ if cursor == 0:
731
+ break
732
+ if not keys:
733
+ return []
734
+ pipe = self._redis.pipeline()
735
+ for k in keys:
736
+ pipe.hgetall(k)
737
+ results = pipe.execute()
738
+ out = []
739
+ for k, raw in zip(keys, results):
740
+ if not raw:
741
+ continue
742
+ name = (k.decode() if isinstance(k, bytes) else k).replace("graph_meta:", "")
743
+ entry = {"name": name}
744
+ for field, val in raw.items():
745
+ f = field.decode() if isinstance(field, bytes) else field
746
+ v = val.decode() if isinstance(val, bytes) else val
747
+ try:
748
+ entry[f] = json.loads(v)
749
+ except (json.JSONDecodeError, TypeError):
750
+ try:
751
+ entry[f] = int(v)
752
+ except ValueError:
753
+ entry[f] = v
754
+ out.append(entry)
755
+ return out
756
+
757
+
758
+ # ── Pre-computed Result Cache ──────────────────────────────────
759
+
760
+ class ResultCache:
761
+ """Pre-computed search results cached in Redis for <1ms lookups.
762
+
763
+ Built after ingest alongside the manifest. Stores top results
764
+ per entity type and per theme keyword so search_nodes() becomes
765
+ a Redis hash read instead of a graph scan.
766
+
767
+ Keys:
768
+ results:{graph}:label:{Label} → JSON list of top nodes
769
+ results:{graph}:query:{keyword} → JSON list of matching nodes
770
+ """
771
+
772
+ def __init__(self, redis_client):
773
+ self._redis = redis_client
774
+
775
+ def build(self, graph_name: str, db, entity_index: EntityIndex,
776
+ max_per_label: int = 20, max_per_query: int = 10):
777
+ """Pre-compute and cache results for all entity types + top keywords."""
778
+ if not self._redis:
779
+ return
780
+
781
+ pipe = self._redis.pipeline()
782
+ adapter = getattr(db, 'csr_adapter', None) or db
783
+
784
+ # 1. Cache top nodes per entity label
785
+ all_labels = entity_index.all_labels(graph_name)
786
+ for label in all_labels:
787
+ top = entity_index.top(graph_name, label, limit=max_per_label)
788
+ if top:
789
+ # Enrich with node content
790
+ results = []
791
+ for entry in top:
792
+ try:
793
+ node = adapter.get_node(entry["node_id"])
794
+ if node:
795
+ props = node.properties or {}
796
+ results.append({
797
+ "node_id": entry["node_id"],
798
+ "label": label,
799
+ "name": entry["name"],
800
+ "content": props.get("statement", props.get("content", props.get("description", "")))[:200],
801
+ "mentions": entry["mention_count"],
802
+ "_quality": props.get("_quality", entry.get("quality", 50)),
803
+ })
804
+ except Exception:
805
+ results.append({
806
+ "node_id": entry["node_id"], "label": label,
807
+ "name": entry["name"], "mentions": entry["mention_count"],
808
+ "_quality": entry.get("quality", 50),
809
+ })
810
+ key = f"results:{graph_name}:label:{label}"
811
+ pipe.set(key, json.dumps(results), ex=86400)
812
+
813
+ # 2. Cache top facts (most common search target)
814
+ try:
815
+ all_nodes = adapter.get_all_nodes()
816
+ facts = []
817
+ for n in all_nodes:
818
+ lbl = getattr(n, 'label', getattr(n, 'node_type', ''))
819
+ if lbl == "Fact":
820
+ props = n.properties or {}
821
+ q = props.get("_quality", 50)
822
+ if q >= 40:
823
+ facts.append({
824
+ "node_id": n.id,
825
+ "label": "Fact",
826
+ "name": props.get("name", "")[:120],
827
+ "content": props.get("statement", "")[:200],
828
+ "_quality": q,
829
+ })
830
+ # Sort by quality, cache top N
831
+ facts.sort(key=lambda f: f.get("_quality", 0), reverse=True)
832
+ pipe.set(f"results:{graph_name}:label:Fact", json.dumps(facts[:max_per_label]), ex=86400)
833
+ except Exception:
834
+ pass
835
+
836
+ pipe.execute()
837
+ logger.info("[RESULT_CACHE] Pre-computed results for %s (%d labels)", graph_name, len(all_labels))
838
+
839
+ def get_by_label(self, graph_name: str, label: str) -> Optional[list]:
840
+ """Get cached results for a label. Returns list or None."""
841
+ if not self._redis:
842
+ return None
843
+ raw = self._redis.get(f"results:{graph_name}:label:{label}")
844
+ if raw:
845
+ return json.loads(raw)
846
+ return None
847
+
848
+ def format_results(self, results: list, query: str = "") -> str:
849
+ """Format cached results as tool output string."""
850
+ if not results:
851
+ return ""
852
+ lines = [f"Found {len(results)} node(s):"]
853
+ for r in results:
854
+ name = r.get("name", "")
855
+ content = r.get("content", "")
856
+ text = content if content and content != name else name
857
+ lines.append(f" [{r.get('label', '?')}] {text[:150]}")
858
+ return "\n".join(lines)
859
+
860
+
861
+ # ── Graph Quality Scorer ──────────────────────────────────────────
862
+
863
+ def score_graph(db, graph_name: str = "") -> Dict[str, Any]:
864
+ """Score the overall quality of an ingested graph.
865
+
866
+ Returns a report card with scores (0-100) for:
867
+ - completeness: Do nodes have proper names, descriptions, edges?
868
+ - connectivity: How well-connected is the graph? Orphan ratio?
869
+ - richness: Entity diversity, relationship variety
870
+ - structure: Session→Turn→Topic→Entity chains intact?
871
+ - overall: Weighted average
872
+
873
+ No LLM calls — pure graph analysis, fast (<500ms).
874
+ """
875
+ all_nodes = list(db.get_all_nodes())
876
+ all_edges = list(db.get_all_edges())
877
+
878
+ if not all_nodes:
879
+ return {
880
+ "overall": 0, "grade": "F",
881
+ "completeness": 0, "connectivity": 0, "richness": 0, "structure": 0,
882
+ "node_count": 0, "edge_count": 0, "issues": ["Graph is empty"],
883
+ "label_distribution": {},
884
+ }
885
+
886
+ n_nodes = len(all_nodes)
887
+ n_edges = len(all_edges)
888
+ issues = []
889
+ suggestions = []
890
+
891
+ # ── Label distribution ──
892
+ label_counts: Dict[str, int] = {}
893
+ for n in all_nodes:
894
+ label = getattr(n, "label", "") or "Unknown"
895
+ label_counts[label] = label_counts.get(label, 0) + 1
896
+
897
+ # Infrastructure vs knowledge nodes
898
+ infra_labels = {"Session", "Turn", "ContextUnit", "ContextMeta", "BM25Index", "VectorIndex", "ContextRef"}
899
+ knowledge_labels = {"Topic", "Person", "Technology", "Organization", "Concept",
900
+ "Location", "Event", "Fact", "Document", "Finding", "Insight",
901
+ "Decision", "Action", "Question", "Problem", "Solution"}
902
+ n_knowledge = sum(label_counts.get(l, 0) for l in knowledge_labels)
903
+ n_infra = sum(label_counts.get(l, 0) for l in infra_labels)
904
+ n_turns = label_counts.get("Turn", 0)
905
+
906
+ # ── 1. Completeness (0-100) ──
907
+ # Do nodes have names? Do they have meaningful properties?
908
+ named_count = 0
909
+ quality_scores = []
910
+ for n in all_nodes:
911
+ props = dict(getattr(n, "properties", {}) or {})
912
+ label = getattr(n, "label", "")
913
+ if label in infra_labels:
914
+ continue
915
+ name = props.get("name", "")
916
+ if name and len(name) > 2:
917
+ named_count += 1
918
+ quality_scores.append(score_node({
919
+ "label": label, "properties": props,
920
+ "edge_count": 0, # Approximate
921
+ }))
922
+
923
+ n_scoreable = max(len(quality_scores), 1)
924
+ avg_quality = sum(quality_scores) / n_scoreable if quality_scores else 0
925
+ named_ratio = named_count / n_scoreable if n_scoreable > 0 else 0
926
+
927
+ completeness = int(named_ratio * 50 + min(avg_quality, 100) / 2)
928
+ if named_ratio < 0.5:
929
+ issues.append(f"{int((1 - named_ratio) * 100)}% of knowledge nodes have no name")
930
+ suggestions.append("Ensure entity extraction assigns proper names")
931
+
932
+ # ── 2. Connectivity (0-100) ──
933
+ # Edge-to-node ratio, orphan detection
934
+ node_ids = {getattr(n, "id", "") for n in all_nodes}
935
+ connected_ids = set()
936
+ for e in all_edges:
937
+ connected_ids.add(getattr(e, "source", ""))
938
+ connected_ids.add(getattr(e, "target", ""))
939
+ connected_ids &= node_ids
940
+
941
+ orphan_count = len(node_ids - connected_ids)
942
+ # Exclude infrastructure from orphan calculation
943
+ knowledge_ids = {getattr(n, "id", "") for n in all_nodes
944
+ if getattr(n, "label", "") not in infra_labels}
945
+ orphan_knowledge = len(knowledge_ids - connected_ids)
946
+ n_knowledge_ids = max(len(knowledge_ids), 1)
947
+
948
+ edge_ratio = n_edges / max(n_nodes, 1)
949
+ orphan_ratio = orphan_knowledge / n_knowledge_ids
950
+
951
+ connectivity = 100
952
+ if edge_ratio < 0.5:
953
+ connectivity -= 40
954
+ issues.append(f"Low edge density ({edge_ratio:.1f} edges per node)")
955
+ suggestions.append("Build more edges between entities (co-occurrence, schema-driven)")
956
+ elif edge_ratio < 1.0:
957
+ connectivity -= 15
958
+ if orphan_ratio > 0.3:
959
+ connectivity -= 30
960
+ issues.append(f"{int(orphan_ratio * 100)}% knowledge nodes are orphans (no connections)")
961
+ suggestions.append("Run entity resolution and edge building stages")
962
+ elif orphan_ratio > 0.1:
963
+ connectivity -= 10
964
+ connectivity = max(0, connectivity)
965
+
966
+ # ── 3. Richness (0-100) ──
967
+ # How many different entity types? How diverse?
968
+ knowledge_label_set = {l for l in label_counts if l in knowledge_labels}
969
+ n_entity_types = len(knowledge_label_set)
970
+
971
+ edge_labels = set()
972
+ for e in all_edges:
973
+ edge_labels.add(getattr(e, "label", "") or "")
974
+ edge_labels.discard("")
975
+ n_edge_types = len(edge_labels)
976
+
977
+ # Score: more types = richer
978
+ richness = min(100, n_entity_types * 15 + n_edge_types * 10)
979
+ if n_knowledge == 0:
980
+ richness = 0
981
+ issues.append("No knowledge entities extracted")
982
+ suggestions.append("Check LLM model configuration for entity extraction")
983
+ elif n_entity_types <= 1:
984
+ richness = max(richness, 20)
985
+ issues.append(f"Only {n_entity_types} entity type(s): {', '.join(knowledge_label_set)}")
986
+ suggestions.append("Enable LLM extraction for Person, Concept, Location entities")
987
+
988
+ # Knowledge-to-turn ratio
989
+ if n_turns > 0:
990
+ k_per_turn = n_knowledge / n_turns
991
+ if k_per_turn < 0.3:
992
+ issues.append(f"Low extraction rate: {k_per_turn:.1f} entities per turn")
993
+ suggestions.append("Review extraction quality — most turns should produce entities")
994
+
995
+ # ── 4. Structure (0-100) ──
996
+ # Are the expected chains present? Session→Turn, Turn→Topic, Topic→Entity
997
+ has_sessions = label_counts.get("Session", 0) > 0
998
+ has_turns = n_turns > 0
999
+ has_topics = label_counts.get("Topic", 0) > 0
1000
+ has_entities = n_knowledge > 0
1001
+
1002
+ structure_checks = [has_sessions, has_turns, has_topics, has_entities]
1003
+ structure = int(sum(structure_checks) / len(structure_checks) * 100)
1004
+
1005
+ # Check for expected edge types
1006
+ expected_edges = {"CONTAINS_TURN", "NEXT", "RAISES", "MENTIONS", "INVOLVES"}
1007
+ found_expected = expected_edges & edge_labels
1008
+ if expected_edges:
1009
+ structure = int(structure * 0.5 + len(found_expected) / len(expected_edges) * 50)
1010
+
1011
+ if not has_topics:
1012
+ issues.append("No Topic nodes — topic clustering may have failed")
1013
+ if not has_entities:
1014
+ issues.append("No entity nodes extracted")
1015
+
1016
+ # ── Overall (weighted average) ──
1017
+ overall = int(
1018
+ completeness * 0.25 +
1019
+ connectivity * 0.25 +
1020
+ richness * 0.30 +
1021
+ structure * 0.20
1022
+ )
1023
+
1024
+ # Grade
1025
+ if overall >= 85:
1026
+ grade = "A"
1027
+ elif overall >= 70:
1028
+ grade = "B"
1029
+ elif overall >= 50:
1030
+ grade = "C"
1031
+ elif overall >= 30:
1032
+ grade = "D"
1033
+ else:
1034
+ grade = "F"
1035
+
1036
+ return {
1037
+ "overall": overall,
1038
+ "grade": grade,
1039
+ "completeness": completeness,
1040
+ "connectivity": connectivity,
1041
+ "richness": richness,
1042
+ "structure": structure,
1043
+ "node_count": n_nodes,
1044
+ "edge_count": n_edges,
1045
+ "knowledge_nodes": n_knowledge,
1046
+ "orphan_nodes": orphan_knowledge,
1047
+ "edge_ratio": round(edge_ratio, 2),
1048
+ "entity_types": sorted(knowledge_label_set),
1049
+ "edge_types": sorted(edge_labels),
1050
+ "label_distribution": label_counts,
1051
+ "avg_node_quality": round(avg_quality, 1),
1052
+ "issues": issues,
1053
+ "suggestions": suggestions,
1054
+ }