For customizing the framework code

[RajLaxmiHota]

Hii Team,
Please update the classes as per my custom requirement.
sql_join_query_engine.py - SQLJoinQueryEngine
sql_vector_query_engine.py - SQLAutoVectorQueryEngine

sql_query_tool: QueryEngineTool,
vector_query_tool: QueryEngineTool,

In the current setup it is only supporting for single query tool of each, but in my present requirement it should support for multiple query tools of each.
Can you please update the required frame work code for me?

[dosubot[bot]]

🤖

I understand that you want to modify the SQLJoinQueryEngine and SQLAutoVectorQueryEngine classes to support multiple query tools of each type. However, as a maintainer of the LlamaIndex repository, I'm unable to make direct changes to the codebase or create pull requests.

That said, I can certainly guide you on how you might approach this modification.

Currently, the SQLJoinQueryEngine and SQLAutoVectorQueryEngine classes are designed to work with a single SQL query tool and a single other query tool. If you want to support multiple query tools, you would need to modify these classes to accept and manage a list of query tools instead of a single one.

Here's a high-level idea of how you might modify the SQLJoinQueryEngine class:

class SQLJoinQueryEngine(BaseQueryEngine):
    def __init__(self, sql_query_tools, other_query_tools, selector=None, service_context=None, sql_join_synthesis_prompt=None, sql_augment_query_transform=None, use_sql_join_synthesis=True, callback_manager=None, verbose=True):
        # Initialization code...
        self.sql_query_tools = sql_query_tools
        self.other_query_tools = other_query_tools
        # Rest of the code...

In the above code, sql_query_tools and other_query_tools are lists of query tools. You would need to modify the _query_sql_other and _query methods to iterate over these lists and perform the necessary queries.

Please note that this is a high-level suggestion and the actual implementation might require additional changes based on your specific requirements and the existing codebase.

You can refer to the source code of these classes in the LlamaIndex repository for more details:

• SQLJoinQueryEngine
• SQLAutoVectorQueryEngine

I hope this helps! If you have any further questions, feel free to ask.

Dosu

Sources

• llama_index/query_engine/sql_vector_query_engine.py
• llama_index/query_engine/sql_join_query_engine.py

---

This response is meant to be useful and save you time. It is not meant to be a precise solution, but rather a starting point for your own research.

Help me be more useful! Please leave a 👍 if this is helpful and 👎 if it is irrelevant.

If you want to continue the conversation, start your reply with @dosu-bot.