feat: integrate chatbox research with datasets and backtests

backend/app/alphas.py

@@ -7,6 +7,7 @@ from datetime import datetime
 from sqlalchemy import or_, select, update
 
 from .models import Alpha, Research, ResearchTag, SelfCorrelation, now
+from .research.provenance import source_alpha_ids
 
 
 def submission_condition(submission):
@@ -119,6 +120,9 @@ def list_statement(filters):
     query = select(Alpha, Research).join(Research, Research.alpha_id == Alpha.id)
     if filters.submission:
         query = query.where(submission_condition(filters.submission))
+    source_filters = {k: getattr(filters, k) for k in ("source", "source_reference", "research_id", "backtest_run_id")}
+    if any(source_filters.values()):
+        query = query.where(Alpha.id.in_(source_alpha_ids(**source_filters)))
     q = filters.q
     if q:
         pattern = "%" + q.replace("\\", "\\\\").replace("%", "\\%").replace("_", "\\_") + "%"