Modify LLM prompts

[saurabhsbora]

Is it possible to modify the llm prompts to add any custom instructions ?

[a29874750a]

Yes, you can go to vanna package and find base.py
the path is vanna/base/base.py
There's a initial_prompt and you can modify it.

[saurabhsbora]

yeah did that eventually, but wanted to check if there is any parameter exposed to pass it

[FatK1234]

Correct me if I am wrong, but I think we can modify the prompt directly by tweaking the initial_prompt within config? Below is an example.

from vanna.ollama import Ollama
from vanna.chromadb import ChromaDB_VectorStore

class MyVanna(ChromaDB_VectorStore, Ollama):
    def __init__(self, config=None):
        ChromaDB_VectorStore.__init__(self, config=config)
        Ollama.__init__(self, config=config)

vn = MyVanna(config={'model': 'gemma3:latest', 'initial_prompt': f"You are a T-SQL / Microsoft SQL Server expert. " + \
            "Generate a SQL query to answer the question."})