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.
- contextsynapse/__init__.py +107 -0
- contextsynapse/__main__.py +9 -0
- contextsynapse/a2a/__init__.py +28 -0
- contextsynapse/a2a/client.py +120 -0
- contextsynapse/a2a/discovery.py +128 -0
- contextsynapse/a2a/handlers.py +164 -0
- contextsynapse/a2a/models.py +246 -0
- contextsynapse/a2a/streaming.py +92 -0
- contextsynapse/a2a/task_manager.py +275 -0
- contextsynapse/adapters/__init__.py +98 -0
- contextsynapse/adapters/_base.py +298 -0
- contextsynapse/adapters/autogen/__init__.py +4 -0
- contextsynapse/adapters/autogen/tools.py +34 -0
- contextsynapse/adapters/crewai/__init__.py +4 -0
- contextsynapse/adapters/crewai/tools.py +63 -0
- contextsynapse/adapters/langchain/__init__.py +19 -0
- contextsynapse/adapters/langchain/message_history.py +17 -0
- contextsynapse/adapters/langchain/retriever.py +158 -0
- contextsynapse/adapters/langchain/tools.py +43 -0
- contextsynapse/adapters/langgraph/__init__.py +21 -0
- contextsynapse/adapters/langgraph/checkpoint.py +223 -0
- contextsynapse/adapters/langgraph/context_tools.py +341 -0
- contextsynapse/adapters/langgraph/message_history.py +138 -0
- contextsynapse/adapters/langgraph/tools.py +43 -0
- contextsynapse/adapters/llamaindex/__init__.py +4 -0
- contextsynapse/adapters/llamaindex/tools.py +64 -0
- contextsynapse/adapters/openai/__init__.py +24 -0
- contextsynapse/adapters/openai/tools.py +206 -0
- contextsynapse/adapters/pydantic_ai/__init__.py +4 -0
- contextsynapse/adapters/pydantic_ai/tools.py +36 -0
- contextsynapse/adapters/swarm/__init__.py +4 -0
- contextsynapse/adapters/swarm/tools.py +44 -0
- contextsynapse/agents/__init__.py +6 -0
- contextsynapse/agents/auto_capture.py +80 -0
- contextsynapse/agents/dispatcher.py +299 -0
- contextsynapse/agents/notifications.py +112 -0
- contextsynapse/aiql/__init__.py +71 -0
- contextsynapse/aiql/cache.py +341 -0
- contextsynapse/aiql/compiler/__init__.py +62 -0
- contextsynapse/aiql/compiler/optimizer.py +126 -0
- contextsynapse/aiql/compiler/planner.py +366 -0
- contextsynapse/aiql/compiler/statistics.py +235 -0
- contextsynapse/aiql/compiler/validator.py +425 -0
- contextsynapse/aiql/engine/__init__.py +11 -0
- contextsynapse/aiql/engine/chunk_storage.py +2584 -0
- contextsynapse/aiql/engine/executor.py +7737 -0
- contextsynapse/aiql/engine/extraction_integration.py +442 -0
- contextsynapse/aiql/engine/graph_builder.py +405 -0
- contextsynapse/aiql/engine/node_creation_from_normalized.py +691 -0
- contextsynapse/aiql/engine/performance_monitor.py +289 -0
- contextsynapse/aiql/engine/postprocessing.py +365 -0
- contextsynapse/aiql/engine/ray_stage_executors.py +918 -0
- contextsynapse/aiql/engine/redis_cache.py +106 -0
- contextsynapse/aiql/engine/streaming_json.py +366 -0
- contextsynapse/aiql/engine/streaming_parquet.py +568 -0
- contextsynapse/aiql/engine/streaming_ray_chunking.py +419 -0
- contextsynapse/aiql/grammar/__init__.py +14 -0
- contextsynapse/aiql/grammar/aiql_grammar.py +1822 -0
- contextsynapse/aiql/grammar/base.py +71 -0
- contextsynapse/aiql/parser/__init__.py +46 -0
- contextsynapse/aiql/parser/aiql_parser.py +11330 -0
- contextsynapse/aiql/parser/base_parser.py +47 -0
- contextsynapse/api/__init__.py +5 -0
- contextsynapse/api/a2a_router.py +233 -0
- contextsynapse/api/admin_router.py +140 -0
- contextsynapse/api/agent_worker_router.py +978 -0
- contextsynapse/api/algorithms_router.py +317 -0
- contextsynapse/api/api.py +4144 -0
- contextsynapse/api/audit.py +204 -0
- contextsynapse/api/auth.py +441 -0
- contextsynapse/api/auth_router.py +537 -0
- contextsynapse/api/billing.py +261 -0
- contextsynapse/api/billing_router.py +275 -0
- contextsynapse/api/boundary_router.py +452 -0
- contextsynapse/api/cognition_router.py +116 -0
- contextsynapse/api/connector_router.py +237 -0
- contextsynapse/api/context_router.py +1619 -0
- contextsynapse/api/dashboard_router.py +11367 -0
- contextsynapse/api/errors.py +183 -0
- contextsynapse/api/events.py +88 -0
- contextsynapse/api/experiment_bootstrap.py +321 -0
- contextsynapse/api/experiment_orchestrator.py +1078 -0
- contextsynapse/api/experiment_router.py +2089 -0
- contextsynapse/api/feature_gate.py +89 -0
- contextsynapse/api/federation_router.py +108 -0
- contextsynapse/api/integrations.py +567 -0
- contextsynapse/api/integrations_router.py +461 -0
- contextsynapse/api/intelligence_router.py +1028 -0
- contextsynapse/api/metering.py +155 -0
- contextsynapse/api/metrics.py +112 -0
- contextsynapse/api/models.py +38 -0
- contextsynapse/api/monitoring_router.py +174 -0
- contextsynapse/api/pipeline_router.py +1437 -0
- contextsynapse/api/playground_conn.py +116 -0
- contextsynapse/api/projects_router.py +131 -0
- contextsynapse/api/queries_router.py +282 -0
- contextsynapse/api/quota.py +95 -0
- contextsynapse/api/rate_limit.py +64 -0
- contextsynapse/api/search_router.py +261 -0
- contextsynapse/api/shield_router.py +447 -0
- contextsynapse/api/tenant_guard.py +103 -0
- contextsynapse/api/tenant_quotas.py +231 -0
- contextsynapse/api/tenants.py +305 -0
- contextsynapse/api/users.py +646 -0
- contextsynapse/api/v1.py +167 -0
- contextsynapse/api/vertical_builder_router.py +387 -0
- contextsynapse/benchmark/__init__.py +15 -0
- contextsynapse/benchmark/__main__.py +64 -0
- contextsynapse/benchmark/agents.py +338 -0
- contextsynapse/benchmark/live_ab.py +338 -0
- contextsynapse/benchmark/reporter.py +149 -0
- contextsynapse/benchmark/runner.py +264 -0
- contextsynapse/benchmark/savings.py +266 -0
- contextsynapse/benchmark/token_counter.py +212 -0
- contextsynapse/benchmark/workloads.py +289 -0
- contextsynapse/buffer/__init__.py +154 -0
- contextsynapse/buffer/base.py +257 -0
- contextsynapse/buffer/duckdb_buffer.py +246 -0
- contextsynapse/buffer/file_buffer.py +229 -0
- contextsynapse/buffer/lmdb_buffer.py +286 -0
- contextsynapse/buffer/local_buffer.py +154 -0
- contextsynapse/buffer/redis_buffer.py +211 -0
- contextsynapse/buffer/sqlite_buffer.py +266 -0
- contextsynapse/cache.py +2 -0
- contextsynapse/cli/__init__.py +40 -0
- contextsynapse/cli/cli.py +475 -0
- contextsynapse/cli/config_cli.py +177 -0
- contextsynapse/cli/scaffold.py +465 -0
- contextsynapse/cognition/__init__.py +31 -0
- contextsynapse/cognition/derivation.py +55 -0
- contextsynapse/cognition/events.py +135 -0
- contextsynapse/cognition/feedback.py +52 -0
- contextsynapse/cognition/hallucination.py +90 -0
- contextsynapse/cognition/invalidation.py +70 -0
- contextsynapse/cognition/read_tracker.py +50 -0
- contextsynapse/collaboration.py +2 -0
- contextsynapse/comparison.py +2 -0
- contextsynapse/config/__init__.py +10 -0
- contextsynapse/config/connectors/amfi_nav.yaml +18 -0
- contextsynapse/config/connectors/auto_sector.yaml +20 -0
- contextsynapse/config/connectors/banking_sector.yaml +20 -0
- contextsynapse/config/connectors/energy_sector.yaml +20 -0
- contextsynapse/config/connectors/fmcg_sector.yaml +20 -0
- contextsynapse/config/connectors/nsdl_fpi.yaml +9 -0
- contextsynapse/config/connectors/nse_price.yaml +61 -0
- contextsynapse/config/connectors/pharma_sector.yaml +20 -0
- contextsynapse/config/connectors/rbi_dbie.yaml +12 -0
- contextsynapse/config/context_presets/ai_research.json +25 -0
- contextsynapse/config/context_presets/chatgpt_conversations.json +19 -0
- contextsynapse/config/context_presets/claude_conversations.json +19 -0
- contextsynapse/config/context_presets/legal_compliance.json +21 -0
- contextsynapse/config/context_presets/meeting_notes.json +21 -0
- contextsynapse/config/context_presets/slack_engineering.json +21 -0
- contextsynapse/config/context_presets/tech_climate.json +24 -0
- contextsynapse/config/context_presets/toi_politics.json +25 -0
- contextsynapse/config/database_config.py +453 -0
- contextsynapse/config/deployment.py +221 -0
- contextsynapse/config/gpt_action_schema.json +164 -0
- contextsynapse/config/indicators/india_macro.yaml +213 -0
- contextsynapse/config/logging_config.py +88 -0
- contextsynapse/config/mode_config.py +223 -0
- contextsynapse/config/schemas/api_spec.yaml +48 -0
- contextsynapse/config/schemas/architecture_doc.yaml +51 -0
- contextsynapse/config/schemas/chat_history.yaml +90 -0
- contextsynapse/config/schemas/chatgpt_conversation.yaml +101 -0
- contextsynapse/config/schemas/claude_conversation.yaml +148 -0
- contextsynapse/config/schemas/contract.yaml +48 -0
- contextsynapse/config/schemas/conversation.yaml +167 -0
- contextsynapse/config/schemas/database.yaml +74 -0
- contextsynapse/config/schemas/decision.yaml +72 -0
- contextsynapse/config/schemas/ecommerce.yaml +56 -0
- contextsynapse/config/schemas/education.yaml +42 -0
- contextsynapse/config/schemas/healthcare.yaml +133 -0
- contextsynapse/config/schemas/hr_document.yaml +48 -0
- contextsynapse/config/schemas/invoice.yaml +40 -0
- contextsynapse/config/schemas/knowledge_base.yaml +75 -0
- contextsynapse/config/schemas/legal.yaml +42 -0
- contextsynapse/config/schemas/marketing_content.yaml +48 -0
- contextsynapse/config/schemas/meeting_notes.yaml +51 -0
- contextsynapse/config/schemas/news_article.yaml +61 -0
- contextsynapse/config/schemas/real_estate.yaml +49 -0
- contextsynapse/config/schemas/requirements_doc.yaml +48 -0
- contextsynapse/config/schemas/research_paper.yaml +56 -0
- contextsynapse/config/schemas/resume.yaml +49 -0
- contextsynapse/config/schemas/retail.yaml +42 -0
- contextsynapse/config/schemas/rule_meta.yaml +173 -0
- contextsynapse/config/schemas/rules.yaml +72 -0
- contextsynapse/config/schemas/screener_meta.yaml +236 -0
- contextsynapse/config/schemas/sdlc.yaml +103 -0
- contextsynapse/config/schemas/session_graph.yaml +172 -0
- contextsynapse/config/schemas/support_ticket.yaml +48 -0
- contextsynapse/config/schemas/web.yaml +64 -0
- contextsynapse/config/sensors/earnings_calendar.yaml +52 -0
- contextsynapse/config/sensors/global_risk.yaml +53 -0
- contextsynapse/config/sensors/government_policy.yaml +54 -0
- contextsynapse/config/sensors/supply_chain.yaml +50 -0
- contextsynapse/config/sensors/templates/behavioral.yaml +56 -0
- contextsynapse/config/sensors/templates/equity_sentiment.yaml +48 -0
- contextsynapse/config/sensors/templates/global_macro.yaml +47 -0
- contextsynapse/config/sensors/templates/market_data.yaml +118 -0
- contextsynapse/config/sensors/templates/sector_technical.yaml +47 -0
- contextsynapse/config/sensors/weather.yaml +52 -0
- contextsynapse/config/storage_strategy.py +202 -0
- contextsynapse/config/templates/competitive_intel.yaml +39 -0
- contextsynapse/config/templates/geopolitical_risk.yaml +46 -0
- contextsynapse/config/templates/market_analysis.yaml +58 -0
- contextsynapse/connectors/__init__.py +15 -0
- contextsynapse/connectors/api_router.py +121 -0
- contextsynapse/connectors/base.py +357 -0
- contextsynapse/connectors/financial/audio.py +170 -0
- contextsynapse/connectors/financial/youtube.py +272 -0
- contextsynapse/connectors/github.py +164 -0
- contextsynapse/connectors/jira.py +120 -0
- contextsynapse/connectors/kafka_connector.py +124 -0
- contextsynapse/connectors/pipeline_connector.py +747 -0
- contextsynapse/connectors/registry.py +116 -0
- contextsynapse/connectors/scheduler.py +87 -0
- contextsynapse/connectors/universal.py +360 -0
- contextsynapse/context/__init__.py +96 -0
- contextsynapse/context/__main__.py +4 -0
- contextsynapse/context/acl.py +125 -0
- contextsynapse/context/agent_card.py +201 -0
- contextsynapse/context/agent_memory.py +814 -0
- contextsynapse/context/agents.py +472 -0
- contextsynapse/context/agents_redis.py +391 -0
- contextsynapse/context/assembled.py +204 -0
- contextsynapse/context/attribution.py +113 -0
- contextsynapse/context/blob.py +184 -0
- contextsynapse/context/boundaries.py +187 -0
- contextsynapse/context/boundary.py +507 -0
- contextsynapse/context/bundle.py +155 -0
- contextsynapse/context/cli.py +519 -0
- contextsynapse/context/collaboration.py +165 -0
- contextsynapse/context/compiler.py +604 -0
- contextsynapse/context/composite.py +536 -0
- contextsynapse/context/compression.py +147 -0
- contextsynapse/context/context_manager.py +888 -0
- contextsynapse/context/context_schema.py +90 -0
- contextsynapse/context/context_schema.yaml +105 -0
- contextsynapse/context/context_units.py +891 -0
- contextsynapse/context/conversation.py +488 -0
- contextsynapse/context/dedup.py +207 -0
- contextsynapse/context/document_processor.py +698 -0
- contextsynapse/context/embedding_hooks.py +338 -0
- contextsynapse/context/execution_schema.py +274 -0
- contextsynapse/context/fan_out.py +138 -0
- contextsynapse/context/frozen.py +80 -0
- contextsynapse/context/gravity.py +227 -0
- contextsynapse/context/hub.py +1401 -0
- contextsynapse/context/ingest.py +391 -0
- contextsynapse/context/injection.py +226 -0
- contextsynapse/context/interaction_graphifier.py +562 -0
- contextsynapse/context/layers.py +574 -0
- contextsynapse/context/projection.py +1195 -0
- contextsynapse/context/promotion.py +634 -0
- contextsynapse/context/proof.py +131 -0
- contextsynapse/context/propagation.py +577 -0
- contextsynapse/context/pubsub.py +135 -0
- contextsynapse/context/quality.py +1054 -0
- contextsynapse/context/quality_tracker.py +270 -0
- contextsynapse/context/scoping.py +370 -0
- contextsynapse/context/seed.py +992 -0
- contextsynapse/context/session.py +1142 -0
- contextsynapse/context/session_graph.py +593 -0
- contextsynapse/context/session_memory.py +309 -0
- contextsynapse/context/session_resolver.py +111 -0
- contextsynapse/context/skill_learning.py +237 -0
- contextsynapse/context/smart_budget.py +129 -0
- contextsynapse/context/source_policy.py +242 -0
- contextsynapse/context/store_factory.py +53 -0
- contextsynapse/context/sync.py +648 -0
- contextsynapse/context/token.py +236 -0
- contextsynapse/context/vector_integration.py +297 -0
- contextsynapse/context/webhooks.py +194 -0
- contextsynapse/context/working_memory.py +224 -0
- contextsynapse/core/__init__.py +18 -0
- contextsynapse/core/archiver.py +237 -0
- contextsynapse/core/change_stream.py +282 -0
- contextsynapse/core/checkpoint.py +310 -0
- contextsynapse/core/cloud_storage.py +387 -0
- contextsynapse/core/context_state.py +651 -0
- contextsynapse/core/cron.py +324 -0
- contextsynapse/core/datasource.py +643 -0
- contextsynapse/core/db.py +289 -0
- contextsynapse/core/federation.py +283 -0
- contextsynapse/core/graph_coordinator.py +201 -0
- contextsynapse/core/graph_intelligence.py +388 -0
- contextsynapse/core/graph_structures.py +106 -0
- contextsynapse/core/graph_sync.py +157 -0
- contextsynapse/core/hybrid_graph_storage.py +2998 -0
- contextsynapse/core/lifecycle.py +691 -0
- contextsynapse/core/multiworker.py +378 -0
- contextsynapse/core/pruning.py +312 -0
- contextsynapse/core/redis_registry.py +524 -0
- contextsynapse/core/registry.py +1043 -0
- contextsynapse/core/registry_factory.py +51 -0
- contextsynapse/core/registry_metadata.py +294 -0
- contextsynapse/core/replay.py +354 -0
- contextsynapse/core/replication.py +339 -0
- contextsynapse/core/time_travel.py +260 -0
- contextsynapse/core/write_behind.py +152 -0
- contextsynapse/core/write_context.py +39 -0
- contextsynapse/db/__init__.py +1 -0
- contextsynapse/db/postgres.py +267 -0
- contextsynapse/db/profile.py +236 -0
- contextsynapse/db/rules.py +328 -0
- contextsynapse/demo/__init__.py +0 -0
- contextsynapse/demo/report.py +147 -0
- contextsynapse/demo/runner.py +520 -0
- contextsynapse/demo/scenario.py +165 -0
- contextsynapse/demo/session_demo.py +280 -0
- contextsynapse/demo/session_report.py +143 -0
- contextsynapse/demo/universal_runner.py +257 -0
- contextsynapse/engine/__init__.py +15 -0
- contextsynapse/engine/core.py +555 -0
- contextsynapse/extraction/__init__.py +69 -0
- contextsynapse/extraction/cli.py +274 -0
- contextsynapse/extraction/config.py +107 -0
- contextsynapse/extraction/examples/example_usage.py +118 -0
- contextsynapse/extraction/examples/sample_manifest.json +53 -0
- contextsynapse/extraction/examples/sample_normalized.json +185 -0
- contextsynapse/extraction/extract_engine.py +440 -0
- contextsynapse/extraction/extractors/__init__.py +37 -0
- contextsynapse/extraction/extractors/chat_export_extractor.py +189 -0
- contextsynapse/extraction/extractors/csv_extractor.py +83 -0
- contextsynapse/extraction/extractors/docx_extractor.py +129 -0
- contextsynapse/extraction/extractors/excel_extractor.py +152 -0
- contextsynapse/extraction/extractors/html_extractor.py +146 -0
- contextsynapse/extraction/extractors/pdf_extractor.py +635 -0
- contextsynapse/extraction/extractors/pdf_extractor_layoutparser.py +408 -0
- contextsynapse/extraction/extractors/register_extractors.py +63 -0
- contextsynapse/extraction/extractors/text_extractor.py +65 -0
- contextsynapse/extraction/extractors/txt_extractor.py +120 -0
- contextsynapse/extraction/extractors/website_extractor.py +246 -0
- contextsynapse/extraction/fact_extractor.py +236 -0
- contextsynapse/extraction/file_manager.py +353 -0
- contextsynapse/extraction/hierarchy.py +239 -0
- contextsynapse/extraction/id_generator.py +450 -0
- contextsynapse/extraction/layout_detector.py +409 -0
- contextsynapse/extraction/llm_entity_extractor.py +301 -0
- contextsynapse/extraction/normalized_store.py +480 -0
- contextsynapse/extraction/normalizer.py +594 -0
- contextsynapse/extraction/parquet_writer.py +598 -0
- contextsynapse/extraction/post_processors/__init__.py +40 -0
- contextsynapse/extraction/post_processors/column_reorganizer.py +507 -0
- contextsynapse/extraction/ray_normalization.py +233 -0
- contextsynapse/extraction/ray_runner.py +207 -0
- contextsynapse/extraction/redis_schema_registry.py +222 -0
- contextsynapse/extraction/registry.py +204 -0
- contextsynapse/extraction/schema_loader.py +663 -0
- contextsynapse/extraction/schema_registry.py +230 -0
- contextsynapse/extraction/semantic_normalizer.py +245 -0
- contextsynapse/extraction/version_manager.py +314 -0
- contextsynapse/gateway/__init__.py +24 -0
- contextsynapse/gateway/action_emitter.py +212 -0
- contextsynapse/gateway/cost_tracker.py +508 -0
- contextsynapse/gateway/gateway.py +377 -0
- contextsynapse/gateway/policy.py +303 -0
- contextsynapse/governance/__init__.py +13 -0
- contextsynapse/governance/layer.py +513 -0
- contextsynapse/ingestion/__init__.py +15 -0
- contextsynapse/ingestion/amplifier.py +288 -0
- contextsynapse/ingestion/async_ingest.py +463 -0
- contextsynapse/ingestion/chunker.py +303 -0
- contextsynapse/ingestion/chunking.py +58 -0
- contextsynapse/ingestion/cleaner.py +190 -0
- contextsynapse/ingestion/cleanse.py +252 -0
- contextsynapse/ingestion/connectors/__init__.py +38 -0
- contextsynapse/ingestion/connectors/base.py +300 -0
- contextsynapse/ingestion/connectors/chatgpt_memory.py +247 -0
- contextsynapse/ingestion/connectors/claude_memory.py +274 -0
- contextsynapse/ingestion/connectors/sap_ingestor.py +208 -0
- contextsynapse/ingestion/content_detector.py +80 -0
- contextsynapse/ingestion/content_monitor.py +502 -0
- contextsynapse/ingestion/dedup.py +106 -0
- contextsynapse/ingestion/feed.py +640 -0
- contextsynapse/ingestion/filters.py +461 -0
- contextsynapse/ingestion/graph_builder.py +705 -0
- contextsynapse/ingestion/input_classifier.py +307 -0
- contextsynapse/ingestion/job_manager.py +571 -0
- contextsynapse/ingestion/llm_extractor.py +610 -0
- contextsynapse/ingestion/orchestrator.py +116 -0
- contextsynapse/ingestion/parsers/__init__.py +19 -0
- contextsynapse/ingestion/parsers/graph_parser_registry.py +103 -0
- contextsynapse/ingestion/parsers/graphml_parser.py +141 -0
- contextsynapse/ingestion/parsers/jsonld_parser.py +171 -0
- contextsynapse/ingestion/parsers/rdf_parser.py +150 -0
- contextsynapse/ingestion/pipeline_context.py +228 -0
- contextsynapse/ingestion/pipeline_store.py +1195 -0
- contextsynapse/ingestion/queue.py +133 -0
- contextsynapse/ingestion/reprocess.py +179 -0
- contextsynapse/ingestion/scenario_router.py +224 -0
- contextsynapse/ingestion/schema_extractor.py +759 -0
- contextsynapse/ingestion/schema_validator.py +274 -0
- contextsynapse/ingestion/sdlc_ingest.py +87 -0
- contextsynapse/ingestion/smart_ingest.py +2451 -0
- contextsynapse/ingestion/source_connector.py +171 -0
- contextsynapse/ingestion/stage_executor.py +2095 -0
- contextsynapse/ingestion/stages/__init__.py +9 -0
- contextsynapse/ingestion/stages/tabular_to_graph.py +289 -0
- contextsynapse/ingestion/strategies.py +386 -0
- contextsynapse/ingestion/universal/__init__.py +4 -0
- contextsynapse/ingestion/universal/_operator_registry.py +59 -0
- contextsynapse/ingestion/universal/ingest_content.py +32 -0
- contextsynapse/ingestion/universal/operators/__init__.py +1 -0
- contextsynapse/ingestion/universal/operators/base.py +19 -0
- contextsynapse/ingestion/universal/operators/build_edges.py +75 -0
- contextsynapse/ingestion/universal/operators/cluster_topics.py +397 -0
- contextsynapse/ingestion/universal/operators/deduplicate.py +45 -0
- contextsynapse/ingestion/universal/operators/detect_signals.py +90 -0
- contextsynapse/ingestion/universal/operators/embed.py +80 -0
- contextsynapse/ingestion/universal/operators/extract_entities.py +253 -0
- contextsynapse/ingestion/universal/operators/extract_preferences.py +117 -0
- contextsynapse/ingestion/universal/operators/filter_content.py +69 -0
- contextsynapse/ingestion/universal/operators/index_bm25.py +52 -0
- contextsynapse/ingestion/universal/operators/infer_domains.py +69 -0
- contextsynapse/ingestion/universal/operators/link_cross_reference.py +210 -0
- contextsynapse/ingestion/universal/operators/parse_records.py +66 -0
- contextsynapse/ingestion/universal/operators/resolve_entities.py +261 -0
- contextsynapse/ingestion/universal/operators/scanners/__init__.py +16 -0
- contextsynapse/ingestion/universal/operators/scanners/edges.py +173 -0
- contextsynapse/ingestion/universal/operators/scanners/git.py +187 -0
- contextsynapse/ingestion/universal/operators/scanners/github.py +135 -0
- contextsynapse/ingestion/universal/operators/scanners/github_api.py +509 -0
- contextsynapse/ingestion/universal/operators/scanners/helpers.py +67 -0
- contextsynapse/ingestion/universal/operators/scanners/jira.py +65 -0
- contextsynapse/ingestion/universal/operators/scanners/llm_enrichment.py +82 -0
- contextsynapse/ingestion/universal/operators/scanners/repo.py +149 -0
- contextsynapse/ingestion/universal/operators/sdlc_scan.py +213 -0
- contextsynapse/ingestion/universal/operators/spec_scanner.py +149 -0
- contextsynapse/ingestion/universal/operators/store_documents.py +72 -0
- contextsynapse/ingestion/universal/operators/synthesize_cu.py +180 -0
- contextsynapse/ingestion/universal/operators/validate_gate.py +65 -0
- contextsynapse/ingestion/universal/parsers/__init__.py +5 -0
- contextsynapse/ingestion/universal/parsers/record_parser.py +203 -0
- contextsynapse/ingestion/universal/parsers/turn_parser.py +329 -0
- contextsynapse/ingestion/universal/shared_signals.py +161 -0
- contextsynapse/ingestion/universal/stage_executor.py +219 -0
- contextsynapse/ingestion/web_crawler.py +265 -0
- contextsynapse/intelligence/__init__.py +53 -0
- contextsynapse/intelligence/alerts.py +198 -0
- contextsynapse/intelligence/collector.py +655 -0
- contextsynapse/intelligence/comparison.py +198 -0
- contextsynapse/intelligence/config.py +56 -0
- contextsynapse/intelligence/conflict_detector.py +182 -0
- contextsynapse/intelligence/context_gaps.py +192 -0
- contextsynapse/intelligence/context_radar.py +135 -0
- contextsynapse/intelligence/correlation_engine.py +416 -0
- contextsynapse/intelligence/correlation_template.py +120 -0
- contextsynapse/intelligence/dedup.py +296 -0
- contextsynapse/intelligence/event_bus.py +91 -0
- contextsynapse/intelligence/feedback_loop.py +101 -0
- contextsynapse/intelligence/freshness.py +307 -0
- contextsynapse/intelligence/fundamental_feed.py +551 -0
- contextsynapse/intelligence/geo_data.py +170 -0
- contextsynapse/intelligence/geo_reference.py +112 -0
- contextsynapse/intelligence/graph_fusion.py +465 -0
- contextsynapse/intelligence/heuristics.py +176 -0
- contextsynapse/intelligence/impact_tracker.py +336 -0
- contextsynapse/intelligence/macro_indicators.py +319 -0
- contextsynapse/intelligence/ocr.py +104 -0
- contextsynapse/intelligence/persistence.py +267 -0
- contextsynapse/intelligence/pipeline_health.py +130 -0
- contextsynapse/intelligence/price_feed.py +221 -0
- contextsynapse/intelligence/rate_limiter.py +130 -0
- contextsynapse/intelligence/reactive.py +411 -0
- contextsynapse/intelligence/runtime_context.py +390 -0
- contextsynapse/intelligence/sentiment_decay.py +426 -0
- contextsynapse/intelligence/session.py +277 -0
- contextsynapse/intelligence/signal_hierarchy.py +438 -0
- contextsynapse/intelligence/signals.py +185 -0
- contextsynapse/intelligence/source_watcher.py +132 -0
- contextsynapse/intelligence/table_enricher.py +256 -0
- contextsynapse/intelligence/tagger.py +319 -0
- contextsynapse/intelligence/timeseries.py +209 -0
- contextsynapse/intelligence/tracker.py +318 -0
- contextsynapse/intelligence/watchdog.py +392 -0
- contextsynapse/intelligence/watchdog_manager.py +263 -0
- contextsynapse/intelligence/webhook_receiver.py +165 -0
- contextsynapse/intelligence/weight_learner.py +234 -0
- contextsynapse/llm/__init__.py +8 -0
- contextsynapse/llm/client.py +348 -0
- contextsynapse/llm/rate_limiter.py +166 -0
- contextsynapse/logging_config.py +2 -0
- contextsynapse/marketplace/__init__.py +7 -0
- contextsynapse/marketplace/registry.py +173 -0
- contextsynapse/mcp/__init__.py +11 -0
- contextsynapse/mcp/__main__.py +4 -0
- contextsynapse/mcp/auth_middleware.py +212 -0
- contextsynapse/mcp/connection_pool.py +128 -0
- contextsynapse/mcp/server.py +986 -0
- contextsynapse/mcp/session_context.py +67 -0
- contextsynapse/memory/__init__.py +34 -0
- contextsynapse/memory/engine.py +475 -0
- contextsynapse/memory/temporal.py +327 -0
- contextsynapse/metadata/__init__.py +12 -0
- contextsynapse/metadata/metadata_db.py +272 -0
- contextsynapse/metadata/metadata_query_tool.py +300 -0
- contextsynapse/metadata/metadata_tracker.py +425 -0
- contextsynapse/models/__init__.py +27 -0
- contextsynapse/models/embedding_service.py +360 -0
- contextsynapse/models/model_registry.py +228 -0
- contextsynapse/multiworker.py +2 -0
- contextsynapse/pipelines/__init__.py +1 -0
- contextsynapse/pipelines/executor.py +851 -0
- contextsynapse/pipelines/filters.py +189 -0
- contextsynapse/pipelines/models.py +176 -0
- contextsynapse/pipelines/scheduler.py +268 -0
- contextsynapse/playground/__init__.py +1 -0
- contextsynapse/playground/templates.py +85 -0
- contextsynapse/plugins/__init__.py +43 -0
- contextsynapse/plugins/api.py +66 -0
- contextsynapse/plugins/base.py +350 -0
- contextsynapse/plugins/domain_interface.py +508 -0
- contextsynapse/plugins/domain_loader.py +241 -0
- contextsynapse/plugins/loader.py +148 -0
- contextsynapse/plugins/registry.py +265 -0
- contextsynapse/project/__init__.py +11 -0
- contextsynapse/project/code_context.py +807 -0
- contextsynapse/project/connectors/__init__.py +4 -0
- contextsynapse/project/connectors/github_connector.py +113 -0
- contextsynapse/project/connectors/jira_connector.py +160 -0
- contextsynapse/project/graph.py +1023 -0
- contextsynapse/project/project_context.py +545 -0
- contextsynapse/project/scanner_registry.py +61 -0
- contextsynapse/project/schema_manager.py +322 -0
- contextsynapse/project/schemas/knowledge.yaml +59 -0
- contextsynapse/project/schemas/sdlc.yaml +369 -0
- contextsynapse/project/sdlc_schema.py +172 -0
- contextsynapse/project/spec_parser.py +85 -0
- contextsynapse/project/task_lock.py +181 -0
- contextsynapse/project/task_stream.py +203 -0
- contextsynapse/project/team_tools.py +537 -0
- contextsynapse/py.typed +0 -0
- contextsynapse/realtime/__init__.py +6 -0
- contextsynapse/realtime/collaboration.py +147 -0
- contextsynapse/rules/__init__.py +10 -0
- contextsynapse/rules/engine.py +116 -0
- contextsynapse/rules/evaluators/__init__.py +6 -0
- contextsynapse/rules/evaluators/aggregate.py +205 -0
- contextsynapse/rules/evaluators/custom.py +170 -0
- contextsynapse/rules/evaluators/pre_action.py +125 -0
- contextsynapse/rules/evaluators/temporal.py +129 -0
- contextsynapse/rules/loader.py +52 -0
- contextsynapse/rules/models.py +124 -0
- contextsynapse/rules/universal.py +393 -0
- contextsynapse/scheduler.py +229 -0
- contextsynapse/schema/__init__.py +42 -0
- contextsynapse/schema/compiler.py +252 -0
- contextsynapse/schema/composer.py +158 -0
- contextsynapse/schema/dedup.py +73 -0
- contextsynapse/schema/derivation.py +172 -0
- contextsynapse/schema/sdl.py +326 -0
- contextsynapse/schema/validation_gate.py +124 -0
- contextsynapse/sdk/__init__.py +63 -0
- contextsynapse/sdk/agent.py +214 -0
- contextsynapse/sdk/client.py +269 -0
- contextsynapse/sdk/exceptions.py +23 -0
- contextsynapse/sdk/models.py +78 -0
- contextsynapse/sdk/session.py +408 -0
- contextsynapse/sdk/tracked.py +218 -0
- contextsynapse/sdk/worker.py +266 -0
- contextsynapse/sdk/workspace_client.py +98 -0
- contextsynapse/sdk/ws.py +41 -0
- contextsynapse/search/__init__.py +20 -0
- contextsynapse/search/embedding_cache.py +140 -0
- contextsynapse/search/enhanced_search.py +152 -0
- contextsynapse/search/fulltext.py +130 -0
- contextsynapse/search/graph_search.py +634 -0
- contextsynapse/search/lmdb_index.py +1218 -0
- contextsynapse/search/nl_to_aiql.py +199 -0
- contextsynapse/search/rag.py +1346 -0
- contextsynapse/search/rag_cache.py +167 -0
- contextsynapse/search/reasoning_chain.py +231 -0
- contextsynapse/search/redis_search.py +207 -0
- contextsynapse/search/retrieval_quality.py +314 -0
- contextsynapse/search/semantic_query.py +455 -0
- contextsynapse/search/text_resolver.py +159 -0
- contextsynapse/search/whoosh_search.py +373 -0
- contextsynapse/security/__init__.py +17 -0
- contextsynapse/security/audit.py +297 -0
- contextsynapse/security/audit_logger.py +97 -0
- contextsynapse/security/audit_trail.py +270 -0
- contextsynapse/security/auth_unified.py +284 -0
- contextsynapse/security/auto_tagger.py +230 -0
- contextsynapse/security/data_security.py +218 -0
- contextsynapse/security/encryption.py +207 -0
- contextsynapse/security/identity.py +447 -0
- contextsynapse/security/jwt_identity.py +59 -0
- contextsynapse/security/middleware.py +336 -0
- contextsynapse/security/pii.py +233 -0
- contextsynapse/security/rbac.py +96 -0
- contextsynapse/security/rls.py +205 -0
- contextsynapse/security/sanitize.py +233 -0
- contextsynapse/security/scoped_encryption.py +131 -0
- contextsynapse/security/tenant.py +401 -0
- contextsynapse/shield/__init__.py +21 -0
- contextsynapse/shield/anomaly.py +56 -0
- contextsynapse/shield/permissions.py +99 -0
- contextsynapse/shield/profile.py +103 -0
- contextsynapse/shield/shield.py +108 -0
- contextsynapse/shield/trust_engine.py +69 -0
- contextsynapse/skills/__init__.py +14 -0
- contextsynapse/skills/engine.py +233 -0
- contextsynapse/skills/loader.py +59 -0
- contextsynapse/skills/models.py +139 -0
- contextsynapse/storage/__init__.py +7 -0
- contextsynapse/storage/cache.py +208 -0
- contextsynapse/storage/columnar_store_v2.py +178 -0
- contextsynapse/storage/csr_graph_storage.py +587 -0
- contextsynapse/storage/csr_store.py +388 -0
- contextsynapse/storage/document_store.py +491 -0
- contextsynapse/storage/hnsw_index.py +294 -0
- contextsynapse/storage/hybrid_store_v2.py +278 -0
- contextsynapse/storage/lmdb_graph_storage.py +366 -0
- contextsynapse/storage/migration.py +279 -0
- contextsynapse/storage/namespace_store.py +346 -0
- contextsynapse/storage/pure_graph_storage.py +185 -0
- contextsynapse/storage/redis_graph_adapter.py +186 -0
- contextsynapse/storage/redis_graph_storage.py +375 -0
- contextsynapse/storage/router/__init__.py +29 -0
- contextsynapse/storage/router/duckdb_store.py +398 -0
- contextsynapse/storage/router/factory.py +83 -0
- contextsynapse/storage/router/interface.py +125 -0
- contextsynapse/storage/router/postgres_store.py +339 -0
- contextsynapse/storage/router/resolver.py +192 -0
- contextsynapse/storage/router/router.py +300 -0
- contextsynapse/storage/single_file_storage.py +391 -0
- contextsynapse/storage/storage_manager.py +214 -0
- contextsynapse/storage/storage_query_optimizer.py +314 -0
- contextsynapse/storage/wal.py +359 -0
- contextsynapse/temporal/__init__.py +0 -0
- contextsynapse/temporal/storage.py +106 -0
- contextsynapse/tests/__init__.py +11 -0
- contextsynapse/tests/regression/__init__.py +1 -0
- contextsynapse/tests/regression/add_metadata.py +395 -0
- contextsynapse/tests/regression/add_skip_flags.py +85 -0
- contextsynapse/tests/regression/cases/test_basic_create_node.yaml +20 -0
- contextsynapse/tests/regression/cases/test_basic_select.yaml +21 -0
- contextsynapse/tests/regression/cases/test_blockchain_audit_trail.yaml +37 -0
- contextsynapse/tests/regression/cases/test_blockchain_audit_trail_filtered.yaml +39 -0
- contextsynapse/tests/regression/cases/test_blockchain_blocks_range.yaml +39 -0
- contextsynapse/tests/regression/cases/test_blockchain_get_block.yaml +35 -0
- contextsynapse/tests/regression/cases/test_blockchain_latest_block.yaml +35 -0
- contextsynapse/tests/regression/cases/test_blockchain_length.yaml +37 -0
- contextsynapse/tests/regression/cases/test_blockchain_merkle_root.yaml +37 -0
- contextsynapse/tests/regression/cases/test_blockchain_verify.yaml +35 -0
- contextsynapse/tests/regression/cases/test_blockchain_verify_block.yaml +35 -0
- contextsynapse/tests/regression/cases/test_create_edge.yaml +28 -0
- contextsynapse/tests/regression/cases/test_delete_edge_basic.yaml +29 -0
- contextsynapse/tests/regression/cases/test_delete_node_basic.yaml +29 -0
- contextsynapse/tests/regression/cases/test_evaluation_mrr.yaml +40 -0
- contextsynapse/tests/regression/cases/test_evaluation_ndcg.yaml +40 -0
- contextsynapse/tests/regression/cases/test_evaluation_precision.yaml +40 -0
- contextsynapse/tests/regression/cases/test_evaluation_recall.yaml +40 -0
- contextsynapse/tests/regression/cases/test_group_by_basic.yaml +56 -0
- contextsynapse/tests/regression/cases/test_match_node_basic.yaml +29 -0
- contextsynapse/tests/regression/cases/test_match_node_with_order_limit.yaml +31 -0
- contextsynapse/tests/regression/cases/test_namespace_switch.yaml +18 -0
- contextsynapse/tests/regression/cases/test_pipeline_cascaded.yaml +61 -0
- contextsynapse/tests/regression/cases/test_pipeline_fixed_size_chunking.yaml +48 -0
- contextsynapse/tests/regression/cases/test_pipeline_gpt4_chunking.yaml +54 -0
- contextsynapse/tests/regression/cases/test_pipeline_hybrid_indexing.yaml +57 -0
- contextsynapse/tests/regression/cases/test_pipeline_large_embedding.yaml +52 -0
- contextsynapse/tests/regression/cases/test_pipeline_paragraph_chunking.yaml +49 -0
- contextsynapse/tests/regression/cases/test_pipeline_recursive_chunking.yaml +47 -0
- contextsynapse/tests/regression/cases/test_pipeline_run_pipeline.yaml +56 -0
- contextsynapse/tests/regression/cases/test_pipeline_section_based_chunking.yaml +53 -0
- contextsynapse/tests/regression/cases/test_pipeline_semantic_chunking.yaml +51 -0
- contextsynapse/tests/regression/cases/test_pipeline_with_entity_extraction.yaml +57 -0
- contextsynapse/tests/regression/cases/test_pipeline_with_relationships.yaml +62 -0
- contextsynapse/tests/regression/cases/test_rag_evaluation.yaml +38 -0
- contextsynapse/tests/regression/cases/test_rag_graph_bfs_with_eval.yaml +48 -0
- contextsynapse/tests/regression/cases/test_rag_graph_centrality.yaml +40 -0
- contextsynapse/tests/regression/cases/test_rag_graph_centrality_eval.yaml +35 -0
- contextsynapse/tests/regression/cases/test_rag_graph_community.yaml +40 -0
- contextsynapse/tests/regression/cases/test_rag_graph_community_eval.yaml +35 -0
- contextsynapse/tests/regression/cases/test_rag_graph_dfs_with_eval.yaml +35 -0
- contextsynapse/tests/regression/cases/test_rag_graph_multi_hop_eval.yaml +35 -0
- contextsynapse/tests/regression/cases/test_rag_graph_path.yaml +40 -0
- contextsynapse/tests/regression/cases/test_rag_graph_search_basic.yaml +38 -0
- contextsynapse/tests/regression/cases/test_rag_graph_search_bfs.yaml +52 -0
- contextsynapse/tests/regression/cases/test_rag_graph_search_dfs.yaml +41 -0
- contextsynapse/tests/regression/cases/test_rag_graph_search_multi_hop.yaml +43 -0
- contextsynapse/tests/regression/cases/test_rag_graph_search_shortest_path.yaml +41 -0
- contextsynapse/tests/regression/cases/test_rag_graph_search_weighted.yaml +43 -0
- contextsynapse/tests/regression/cases/test_rag_graph_shortest_path_eval.yaml +35 -0
- contextsynapse/tests/regression/cases/test_rag_graph_traversal.yaml +30 -0
- contextsynapse/tests/regression/cases/test_rag_graph_weighted_eval.yaml +35 -0
- contextsynapse/tests/regression/cases/test_rag_hybrid_search.yaml +33 -0
- contextsynapse/tests/regression/cases/test_rag_hybrid_with_prompt_embedding.yaml +44 -0
- contextsynapse/tests/regression/cases/test_rag_keyword_search.yaml +32 -0
- contextsynapse/tests/regression/cases/test_rag_multi_hop_graph.yaml +32 -0
- contextsynapse/tests/regression/cases/test_rag_path_based.yaml +34 -0
- contextsynapse/tests/regression/cases/test_rag_profile_based.yaml +32 -0
- contextsynapse/tests/regression/cases/test_rag_vector_search.yaml +32 -0
- contextsynapse/tests/regression/cases/test_rag_weighted_graph.yaml +34 -0
- contextsynapse/tests/regression/cases/test_rag_weighted_hybrid.yaml +31 -0
- contextsynapse/tests/regression/cases/test_semantic_boundary_chunking.yaml +57 -0
- contextsynapse/tests/regression/cases/test_semantic_boundary_detection.yaml +33 -0
- contextsynapse/tests/regression/cases/test_semantic_hash_function.yaml +30 -0
- contextsynapse/tests/regression/cases/test_semantic_hash_with_model.yaml +31 -0
- contextsynapse/tests/regression/cases/test_tcs_ingestion_fixed_size.yaml +44 -0
- contextsynapse/tests/regression/cases/test_tcs_ingestion_recursive.yaml +46 -0
- contextsynapse/tests/regression/cases/test_tcs_ingestion_section.yaml +46 -0
- contextsynapse/tests/regression/cases/test_tcs_ingestion_semantic.yaml +44 -0
- contextsynapse/tests/regression/cases/test_time_travel_as_of.yaml +35 -0
- contextsynapse/tests/regression/cases/test_time_travel_at_commit.yaml +33 -0
- contextsynapse/tests/regression/cases/test_time_travel_at_timestamp.yaml +33 -0
- contextsynapse/tests/regression/cases/test_time_travel_for_system_time_all.yaml +37 -0
- contextsynapse/tests/regression/cases/test_time_travel_versions_between.yaml +39 -0
- contextsynapse/tests/regression/cases/test_traverse_basic.yaml +31 -0
- contextsynapse/tests/regression/cases/test_traverse_max_depth.yaml +34 -0
- contextsynapse/tests/regression/cases/test_update_edge_basic.yaml +29 -0
- contextsynapse/tests/regression/evaluation_metrics.py +227 -0
- contextsynapse/tests/regression/regression_runner.py +1081 -0
- contextsynapse/tests/regression/setup_regression_data.py +212 -0
- contextsynapse/tests/regression/test_basic_only.py +84 -0
- contextsynapse/tests/regression/test_namespace_creation.py +29 -0
- contextsynapse/tests/regression/test_single_fix.py +61 -0
- contextsynapse/tests/regression/test_single_query.py +20 -0
- contextsynapse/tests/regression/validate_test_syntax.py +179 -0
- contextsynapse/tests/test_advanced_features.py +347 -0
- contextsynapse/tests/test_aiql_summary.py +283 -0
- contextsynapse/tests/test_all_aiql_queries.py +948 -0
- contextsynapse/tests/test_all_create_queries.py +213 -0
- contextsynapse/tests/test_api_vs_native.py +280 -0
- contextsynapse/tests/test_chunking_dedup.py +199 -0
- contextsynapse/tests/test_connectors.py +171 -0
- contextsynapse/tests/test_edges_and_match.py +202 -0
- contextsynapse/tests/test_extraction_pipeline_integration.py +581 -0
- contextsynapse/tests/test_name_property.py +224 -0
- contextsynapse/tests/test_node_edge_metadata.py +250 -0
- contextsynapse/tests/test_playground_conn.py +114 -0
- contextsynapse/tests/test_query_cache.py +213 -0
- contextsynapse/tests/test_source_policy_auth.py +348 -0
- contextsynapse/tests/test_temporal_and_domain.py +224 -0
- contextsynapse/tests/test_time_travel_e2e.py +571 -0
- contextsynapse/tests/test_traversal_and_aggregation.py +275 -0
- contextsynapse/tools/__init__.py +36 -0
- contextsynapse/tools/a2a_tools.py +106 -0
- contextsynapse/tools/cognition_tools.py +118 -0
- contextsynapse/tools/context.py +905 -0
- contextsynapse/tools/formatting.py +92 -0
- contextsynapse/tools/gateway_tools.py +186 -0
- contextsynapse/tools/graph.py +1700 -0
- contextsynapse/tools/intelligence.py +235 -0
- contextsynapse/tools/memory.py +118 -0
- contextsynapse/tools/registry.py +373 -0
- contextsynapse/tools/shield_tools.py +159 -0
- contextsynapse/tools/tasks.py +341 -0
- contextsynapse/tools/workspace_tools.py +124 -0
- contextsynapse/utils/__init__.py +15 -0
- contextsynapse/utils/config_loader.py +198 -0
- contextsynapse/utils/env_loader.py +156 -0
- contextsynapse/utils/file_utils.py +288 -0
- contextsynapse/utils/logging_optimizer.py +119 -0
- contextsynapse/vector/__init__.py +30 -0
- contextsynapse/vector/chroma_vector_store.py +152 -0
- contextsynapse/vector/custom_vector_store.py +537 -0
- contextsynapse/vector/faiss_vector_store.py +296 -0
- contextsynapse/vector/qdrant_vector_store.py +163 -0
- contextsynapse/vector/vector_db.py +953 -0
- contextsynapse/vector/vector_db_manager.py +197 -0
- contextsynapse/workspace/__init__.py +27 -0
- contextsynapse/workspace/base.py +94 -0
- contextsynapse/workspace/git.py +228 -0
- contextsynapse/workspace/github.py +78 -0
- contextsynapse/workspace/local.py +73 -0
- contextsynapse/workspace/tools.py +222 -0
- contextsynapse-1.0.0.dist-info/METADATA +847 -0
- contextsynapse-1.0.0.dist-info/RECORD +776 -0
- contextsynapse-1.0.0.dist-info/WHEEL +5 -0
- contextsynapse-1.0.0.dist-info/entry_points.txt +4 -0
- contextsynapse-1.0.0.dist-info/licenses/LICENSE +200 -0
- 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
|
+
}
|