fleet-memory/hindsight-api/tests
Nicolò Boschi 0560f6260d
fix: cancel in-flight async ops when bank is deleted (#545)
* fix: cancel async ops on bank delete via CASCADE FK + heartbeat checkpoints

- Add migration e5f6g7h8i9j0: FK ON DELETE CASCADE from async_operations
  and webhooks to banks, so deleting a bank auto-removes all its ops/webhooks
- Add _check_op_alive() helper: returns False if op row was deleted (cascade)
- Add consolidation checkpoint: after each LLM batch commit, abort early if
  op was deleted mid-run (returns status='cancelled')
- Add retain checkpoint: between sub-batches, abort early if op was deleted
- _mark_operation_completed/failed/completed_and_fire_webhook: gracefully
  handle missing row (UPDATE 0) with log instead of silent error
- Thread operation_id into run_consolidation_job() for checkpoint access
- Fix y0t1u2v3w4x5 and a1b2c3d4e5f6 migrations: add IF NOT EXISTS to prevent
  failure on idempotent re-runs
- Add 10 tests covering cascade delete, _check_op_alive, graceful mark methods,
  consolidation checkpoint, and retain checkpoint

* refactor: use RETURNING + fetchrow instead of execute + string comparison

* fix: add bank upsert before async_operations FK inserts and update tests

- memory_engine.py: upsert bank in submit_async_retain before async_operations INSERT
- http.py: upsert bank in api_create_webhook before webhooks INSERT
- test_worker.py, test_async_batch_retain.py, test_webhooks.py: add _ensure_bank
  helper calls before direct async_operations/webhooks inserts to satisfy FK constraint

* fix: mock bank_utils.get_bank_profile in unit test with mocked pool
2026-03-12 09:39:31 +01:00
..
fixtures rename to hindsight (#2) 2025-11-25 19:28:26 +01:00
__init__.py rename to hindsight (#2) 2025-11-25 19:28:26 +01:00
conftest.py fix: improve claude code and codex for /reflect (#285) 2026-02-04 13:34:45 +01:00
test_admin_backup_restore.py Fix run-db-migration for all-tenant upgrades (#530) 2026-03-10 10:10:03 +01:00
test_agents_api.py feat: increase customization for reflect, retain and consolidation (#419) 2026-02-23 20:35:32 +01:00
test_async_batch_retain.py fix: cancel in-flight async ops when bank is deleted (#545) 2026-03-12 09:39:31 +01:00
test_async_retain_tags.py fix: cancel in-flight async ops when bank is deleted (#545) 2026-03-12 09:39:31 +01:00
test_base_path.py feat: add reverse proxy support (#346) 2026-02-12 10:09:53 +01:00
test_batch_api.py feat: webhook system with retain.completed event, UI, and docs (#487) 2026-03-04 14:17:01 +01:00
test_batch_api_integration.py feat: support Batch API for retain (openai/groq) (#365) 2026-02-16 13:31:50 +01:00
test_batch_api_validation.py feat: support Batch API for retain (openai/groq) (#365) 2026-02-16 13:31:50 +01:00
test_batch_chunking.py feat: extensions (#54) 2025-12-22 11:05:23 +01:00
test_causal_relations.py feat: implement hierarchical configuration (system, tenant, bank) (#329) 2026-02-12 13:14:57 +01:00
test_causal_relationships.py feat: implement hierarchical configuration (system, tenant, bank) (#329) 2026-02-12 13:14:57 +01:00
test_chunking.py improve retain performances, caching and tests 2025-12-08 18:21:56 +01:00
test_combined_scoring.py fix: replace additive combined scoring with multiplicative CE boosts (#494) 2026-03-05 11:12:28 +01:00
test_config_validation.py fix: hindsight-embed profiles are not loaded correctly (#316) 2026-02-06 17:13:54 +01:00
test_consolidation.py feat: add source facts token limits to consolidation and recall (#509) 2026-03-06 13:01:33 +01:00
test_custom_embedding_dimension.py fix: migrate mental_models.embedding dimension alongside memory_units (#526) 2026-03-09 12:23:50 +01:00
test_document_tracking.py fix: document not tracked if has 0 extracted facts (#399) 2026-02-18 17:01:57 +01:00
test_entity_labels.py feat: entity labels — optional, free_values, multi_value, UI polish (#450) 2026-03-02 13:05:25 +01:00
test_extensions.py misc: fix vertex/gemini errors and use it for ci tests (#414) 2026-02-20 22:35:38 +01:00
test_fact_extraction_analysis.py feat: implement hierarchical configuration (system, tenant, bank) (#329) 2026-02-12 13:14:57 +01:00
test_fact_extraction_metadata.py feat: include doc metadata in fact extraction (#424) 2026-02-23 11:31:02 +01:00
test_fact_extraction_output_ratio.py feat: implement hierarchical configuration (system, tenant, bank) (#329) 2026-02-12 13:14:57 +01:00
test_fact_extraction_quality.py feat: implement hierarchical configuration (system, tenant, bank) (#329) 2026-02-12 13:14:57 +01:00
test_fact_extraction_retry.py fix: resolve TypeError when LLM returns invalid JSON across all retries (#488) (#490) 2026-03-04 14:17:55 +01:00
test_fact_ordering.py misc: fix vertex/gemini errors and use it for ci tests (#414) 2026-02-20 22:35:38 +01:00
test_file_retain.py feat: per-request file parser selection with fallback chains (#514) 2026-03-06 16:15:43 +01:00
test_file_storage_s3.py misc: fix vertex/gemini errors and use it for ci tests (#414) 2026-02-20 22:35:38 +01:00
test_gemini_safety_settings.py refactor: replace set_gemini_safety_settings() with LLMProvider.with_config() (#474) 2026-03-03 15:00:32 +01:00
test_graph_filtering.py feat: filter graph memories with tags (#431) 2026-02-25 10:31:40 +01:00
test_hierarchical_config.py feat: add source facts token limits to consolidation and recall (#509) 2026-03-06 13:01:33 +01:00
test_hnsw_indexes.py perf: replace window-function retrieval with UNION ALL + per-bank HNSW indexes (#541) 2026-03-11 12:09:50 +01:00
test_http_api_integration.py fix: improve async batch retain with large payloads (#366) 2026-02-16 12:51:42 +01:00
test_iris_parser.py feat: add iris as file parser (#395) 2026-02-18 14:09:56 +01:00
test_link_expansion_retrieval.py feat: implement hierarchical configuration (system, tenant, bank) (#329) 2026-02-12 13:14:57 +01:00
test_link_utils.py prepare for release 2025-12-03 11:52:25 +01:00
test_list_documents.py feat: add tags filtering and q description fix for list documents API (#468) 2026-03-02 17:03:16 +01:00
test_litellm_sdk_cross_encoder.py feat: support litellm-sdk as reranker and embeddings (#357) 2026-02-12 23:53:12 +01:00
test_litellm_sdk_embeddings.py fix: pass encoding_format="float" in LiteLLM embedding calls (#434) 2026-02-25 10:32:05 +01:00
test_llm_provider.py fix: resolve JSON serialization and logging exception propagation in claude_code_llm (#458, #459) (#461) 2026-03-02 10:14:16 +01:00
test_llm_token_metrics.py feat: support for codex and claude-code as llm (#276) 2026-02-02 12:54:44 +01:00
test_llm_tools.py chore: internal renames (#204) 2026-01-27 09:53:28 +01:00
test_llm_wrapper.py fix: strip null bytes from parsed file content before retain (#535) 2026-03-10 16:24:11 +01:00
test_lmstudio_tool_choice.py fix: normalize named tool_choice to required + filtered tools for OpenAI-compatible providers (#528) 2026-03-09 15:47:51 +01:00
test_load_large_batch.py feat: support for other text and vector search pg extensions (#355) 2026-02-12 14:13:04 +01:00
test_main_module.py fix(python-client): async method parity and server keepalive timeout (#387) 2026-02-17 16:00:14 +01:00
test_mcp_endpoint_routing.py Harden MCP server: fix routing, validation, and usage metering (#341) 2026-02-11 10:41:20 +01:00
test_mcp_extension.py feat: expand MCP tool surface area with 18 new tools and enhanced parameters (#435) 2026-02-25 10:26:46 +01:00
test_mcp_routing.py fix: zeroentropy rerank URL missing /v1 prefix and MCP retain async_processing param (#460) 2026-03-02 10:32:01 +01:00
test_mcp_tools.py feat: configure exposed mcp tools per bank (#439) 2026-02-25 11:57:46 +01:00
test_mental_model_hooks.py Add extension hooks for mental model operations (#260) 2026-01-31 09:30:53 -05:00
test_mental_models.py feat: mental model history tracking and UI diff view (#516) 2026-03-06 17:50:48 +01:00
test_metrics.py chore: remove dead code (#245) 2026-01-30 09:16:32 +01:00
test_migrations_thread_safety.py fix: serialize alembic upgrades in-process (#521) 2026-03-09 11:22:24 +01:00
test_mpfp_retrieval.py fix: improve mpfp retrieval (#146) 2026-01-12 18:58:05 +01:00
test_multilingual.py feat: observation_scopes field to drive observations granularity (#447) 2026-02-28 10:46:21 +01:00
test_observation_invalidation.py feat: change tags for a document (#517) 2026-03-07 09:00:13 +01:00
test_observations.py feat: implement hierarchical configuration (system, tenant, bank) (#329) 2026-02-12 13:14:57 +01:00
test_op_cancellation.py fix: cancel in-flight async ops when bank is deleted (#545) 2026-03-12 09:39:31 +01:00
test_per_operation_llm_config.py feat: add more config options for llm retries (#234) 2026-01-29 17:50:43 +01:00
test_provider_default_models.py feat: switch default model to gpt-4o-mini (#410) 2026-02-19 18:43:52 +01:00
test_query_analyzer.py fix: ci and ui build (#9) 2025-12-03 15:47:53 +01:00
test_recall_chunks_independence.py fix: resolve chunks for observation results via source_memory_ids (#496) 2026-03-05 14:12:21 +01:00
test_reflect_agent.py fix(reflect): prevent context_length_exceeded on large memory banks (#462) 2026-03-02 12:03:12 +01:00
test_reflect_empty_based_on.py fix: resolve based_on schema/serialization issues in reflect API (#348) 2026-02-12 11:26:12 +01:00
test_reflect_tracing.py feat: add otel traceability (#330) 2026-02-10 12:20:48 +01:00
test_reflections.py feat: improve mental models ux on control plane (#297) 2026-02-04 15:49:03 +01:00
test_reranker_error_handling.py fix: reranker crashes on provider error (#403) 2026-02-19 11:38:37 +01:00
test_retain.py feat: support timestamp="unset" to retain content without a date (#465) 2026-03-02 12:03:23 +01:00
test_schema_isolation.py fix: improve mental model consolidation (#197) 2026-01-26 09:54:25 +01:00
test_search_trace.py feat: extensions (#54) 2025-12-22 11:05:23 +01:00
test_server_module.py fix: improve mental model consolidation (#197) 2026-01-26 09:54:25 +01:00
test_source_facts_tokens.py feat: add source facts token limits to consolidation and recall (#509) 2026-03-06 13:01:33 +01:00
test_sql_schema_safety.py chore: internal renames (#204) 2026-01-27 09:53:28 +01:00
test_supabase_tenant.py Add Supabase tenant extension as built-in (#267) 2026-02-09 10:16:47 +01:00
test_tags_visibility.py feat: observation_scopes field to drive observations granularity (#447) 2026-02-28 10:46:21 +01:00
test_tei_cross_encoder.py fix: add defensive error handling to PyTorch device detection (#221) 2026-01-28 18:14:54 +01:00
test_temporal_ranges.py feat(mcp): add timestamp to retain (#190) 2026-01-23 16:00:43 +01:00
test_think.py chore: remove dead code (#245) 2026-01-30 09:16:32 +01:00
test_tracing.py misc: fix vertex/gemini errors and use it for ci tests (#414) 2026-02-20 22:35:38 +01:00
test_tracing_integration.py feat: add otel traceability (#330) 2026-02-10 12:20:48 +01:00
test_tracing_spans_verification.py feat: add otel traceability (#330) 2026-02-10 12:20:48 +01:00
test_vertexai_provider.py misc: fix vertex/gemini errors and use it for ci tests (#414) 2026-02-20 22:35:38 +01:00
test_webhooks.py fix: cancel in-flight async ops when bank is deleted (#545) 2026-03-12 09:39:31 +01:00
test_worker.py fix: cancel in-flight async ops when bank is deleted (#545) 2026-03-12 09:39:31 +01:00