limit slack search to avoid overloading the search

Orbital-Web · Orbital-Web · commit a678e5d526e3 · 2025-07-02T10:19:00.000-07:00
diff --git a/backend/onyx/context/search/federated/slack_search.py b/backend/onyx/context/search/federated/slack_search.py
@@ -9,6 +9,7 @@
 from sqlalchemy.orm import Session
 
 from onyx.configs.app_configs import ENABLE_CONTEXTUAL_RAG
+from onyx.configs.app_configs import NUM_FEDERATED_SECTIONS
 from onyx.configs.app_configs import NUM_MAX_SLACK_QUERIES
 from onyx.configs.app_configs import NUM_SLACK_SEARCH_DOCS
 from onyx.configs.app_configs import SLACK_USER_TOKEN
@@ -284,4 +285,7 @@ def slack_retrieval(query: SearchQuery, db_session: Session) -> list[InferenceCh
                 updated_at=docid_to_message[document_id].timestamp,
             )
         )
+        if len(top_chunks) >= NUM_FEDERATED_SECTIONS:
+            break
+
     return top_chunks

Original file line number	Diff line number	Diff line change
`@@ -9,6 +9,7 @@`
`9`	`9`	`from sqlalchemy.orm import Session`
`10`	`10`
`11`	`11`	`from onyx.configs.app_configs import ENABLE_CONTEXTUAL_RAG`
	`12`	`+from onyx.configs.app_configs import NUM_FEDERATED_SECTIONS`
`12`	`13`	`from onyx.configs.app_configs import NUM_MAX_SLACK_QUERIES`
`13`	`14`	`from onyx.configs.app_configs import NUM_SLACK_SEARCH_DOCS`
`14`	`15`	`from onyx.configs.app_configs import SLACK_USER_TOKEN`
`@@ -284,4 +285,7 @@ def slack_retrieval(query: SearchQuery, db_session: Session) -> list[InferenceCh`
`284`	`285`	`updated_at=docid_to_message[document_id].timestamp,`
`285`	`286`	`)`
`286`	`287`	`)`
	`288`	`+ if len(top_chunks) >= NUM_FEDERATED_SECTIONS:`
	`289`	`+ break`
	`290`	`+`
`287`	`291`	`return top_chunks`