Skip to content

Commit a7f56a7

Browse files
committed
simplify code
1 parent 46096ba commit a7f56a7

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

templates/components/engines/python/agent/tools/query_engine.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
import os
22
from typing import Any, Dict, List, Optional, Sequence
33

4-
from llama_index.core import get_response_synthesizer
54
from llama_index.core.base.base_query_engine import BaseQueryEngine
65
from llama_index.core.base.response.schema import RESPONSE_TYPE, Response
76
from llama_index.core.multi_modal_llms import MultiModalLLM
@@ -42,11 +41,10 @@ def create_query_engine(index, **kwargs) -> BaseQueryEngine:
4241
# If index is index is LlamaCloudIndex
4342
# use auto_routed mode for better query results
4443
if index.__class__.__name__ == "LlamaCloudIndex":
45-
retrieval_mode = kwargs.get("retrieval_mode")
46-
if retrieval_mode is None:
44+
if kwargs.get("retrieval_mode") is None:
4745
kwargs["retrieval_mode"] = "auto_routed"
48-
if multimodal_llm:
49-
kwargs["retrieve_image_nodes"] = True
46+
if multimodal_llm:
47+
kwargs["retrieve_image_nodes"] = True
5048
return index.as_query_engine(**kwargs)
5149

5250

@@ -86,14 +84,14 @@ class MultiModalSynthesizer(BaseSynthesizer):
8684
def __init__(
8785
self,
8886
multimodal_model: MultiModalLLM,
89-
response_synthesizer: Optional[BaseSynthesizer] = None,
87+
response_synthesizer: Optional[BaseSynthesizer],
9088
text_qa_template: Optional[BasePromptTemplate] = None,
9189
*args,
9290
**kwargs,
9391
):
9492
super().__init__(*args, **kwargs)
9593
self._multi_modal_llm = multimodal_model
96-
self._response_synthesizer = response_synthesizer or get_response_synthesizer()
94+
self._response_synthesizer = response_synthesizer
9795
self._text_qa_template = text_qa_template or DEFAULT_TEXT_QA_PROMPT_SEL
9896

9997
def _get_prompts(self, **kwargs) -> Dict[str, Any]:

0 commit comments

Comments
 (0)