Hi Everyone, Currently I’m solving the use case where you can chat with your bigquery database. I’m using deepseek-coder-6.7B-instruct-GGUF model.
Here’s the snippet of my implementation
from llama_index.indices.struct_store import SQLTableRetrieverQueryEngine
from llama_index import VectorStoreIndex
obj_index = ObjectIndex.from_objects(
table_schema_objs,
table_node_mapping,
VectorStoreIndex,
service_context=service_context
)
query_engine = SQLTableRetrieverQueryEngine(
sql_database, obj_index.as_retriever(similarity_top_k=1), service_context=service_context,
)
response = query_engine.query("List all the unique countries present in item master.")
Add a reply
Sign up and join the conversation on Discord