16
16
17
17
from approaches .approach import (
18
18
Approach ,
19
- DataPoints ,
20
19
ExtraInfo ,
21
20
ThoughtStep ,
22
21
)
@@ -284,9 +283,11 @@ async def run_search_approach(
284
283
minimum_reranker_score = overrides .get ("minimum_reranker_score" , 0.0 )
285
284
search_index_filter = self .build_filter (overrides , auth_claims )
286
285
send_text_sources = overrides .get ("send_text_sources" , True )
287
- send_image_sources = overrides .get ("send_image_sources" , True )
286
+ send_image_sources = overrides .get ("send_image_sources" , self . multimodal_enabled ) and self . multimodal_enabled
288
287
search_text_embeddings = overrides .get ("search_text_embeddings" , True )
289
- search_image_embeddings = overrides .get ("search_image_embeddings" , self .multimodal_enabled )
288
+ search_image_embeddings = (
289
+ overrides .get ("search_image_embeddings" , self .multimodal_enabled ) and self .multimodal_enabled
290
+ )
290
291
291
292
original_user_query = messages [- 1 ]["content" ]
292
293
if not isinstance (original_user_query , str ):
@@ -342,11 +343,12 @@ async def run_search_approach(
342
343
343
344
# STEP 3: Generate a contextual and content specific answer using the search results and chat history
344
345
data_points = await self .get_sources_content (
345
- results , use_semantic_captions , download_image_sources = send_image_sources , user_oid = auth_claims .get ("oid" )
346
+ results ,
347
+ use_semantic_captions ,
348
+ include_text_sources = send_text_sources ,
349
+ download_image_sources = send_image_sources ,
350
+ user_oid = auth_claims .get ("oid" ),
346
351
)
347
- if not send_text_sources :
348
- data_points = DataPoints (text = [], images = data_points .images , citations = data_points .citations )
349
-
350
352
extra_info = ExtraInfo (
351
353
data_points ,
352
354
thoughts = [
@@ -396,7 +398,7 @@ async def run_agentic_retrieval_approach(
396
398
# 50 is the amount of documents that the reranker can process per query
397
399
max_docs_for_reranker = max_subqueries * 50
398
400
send_text_sources = overrides .get ("send_text_sources" , True )
399
- send_image_sources = overrides .get ("send_image_sources" , True )
401
+ send_image_sources = overrides .get ("send_image_sources" , self . multimodal_enabled ) and self . multimodal_enabled
400
402
401
403
response , results = await self .run_agentic_retrieval (
402
404
messages = messages ,
@@ -412,12 +414,10 @@ async def run_agentic_retrieval_approach(
412
414
data_points = await self .get_sources_content (
413
415
results ,
414
416
use_semantic_captions = False ,
417
+ include_text_sources = send_text_sources ,
415
418
download_image_sources = send_image_sources ,
416
419
user_oid = auth_claims .get ("oid" ),
417
420
)
418
- if not send_text_sources :
419
- data_points = DataPoints (text = [], images = data_points .images , citations = data_points .citations )
420
-
421
421
extra_info = ExtraInfo (
422
422
data_points ,
423
423
thoughts = [
0 commit comments