A few mistakes in the "Chat with your Snowflake database using Gemini" documentation (https://vanna.ai/docs/configure/gemini/snowflake)

[badgiojuni]

I have noticed two code errors on this documentation page.

Configure Your Agent Memory section

Code is currently:

# Import agent memory tools
from vanna.tools.agent_memory import SaveQuestionToolArgsTool, SearchSavedCorrectToolUsesTool, SaveTextMemoryTool
from vanna.integrations.local.agent_memory import DemoAgentMemory

# Set up agent memory for learning from questions and SQL
agent_memory = DemoAgentMemory(max_items=1000)

# Register memory tools (they access agent_memory via ToolContext)
tools.register_local_tool(SaveQuestionToolArgsTool(), access_groups=['admin'])
tools.register_local_tool(SearchSavedCorrectToolUsesTool(), access_groups=['admin', 'user'])
tools.register_local_tool(SaveTextMemoryTool(), access_groups=['admin', 'user'])

To avoid a NameError because tools is not defined, code should be:

# Import agent memory tools
from vanna.tools.agent_memory import SaveQuestionToolArgsTool, SearchSavedCorrectToolUsesTool, SaveTextMemoryTool
from vanna.integrations.local.agent_memory import DemoAgentMemory
from vanna.core.registry import ToolRegistry ##################################################### line to add

# Set up agent memory for learning from questions and SQL
agent_memory = DemoAgentMemory(max_items=1000)

# Register memory tools (they access agent_memory via ToolContext)
tools = ToolRegistry()  ##################################################### line to add
tools.register_local_tool(SaveQuestionToolArgsTool(), access_groups=['admin'])
tools.register_local_tool(SearchSavedCorrectToolUsesTool(), access_groups=['admin', 'user'])
tools.register_local_tool(SaveTextMemoryTool(), access_groups=['admin', 'user'])

Configure Your Database section

Select Snowflake as the database option.
In the code section ``# Option 2: RSA Key-Pair Authentication (recommended for production) ```, code is currently:

db_tool = RunSqlTool(
    sql_runner=SnowflakeRunner(
        account="your-account",
        username="your-username",
        private_key_path="/path/to/rsa_key.p8",
        private_key_passphrase="your-passphrase",  # Optional if key is encrypted
        database="your-database",
        warehouse="your-warehouse"
    )
)

if you execute it, you'll get the following error:
ProgrammingError: 251006: Password is empty

Two things happen:

• first, one should mention they want to use the RSA Key-Pair Authentification by adding the authenticator="SNOWFLAKE_JWT" argument in the SnowflakeRunner call above. If not, authenticator="SNOWFLAKE" by default and it falls back to the deprecated user/password identification method.
• second, private_key_path argument does not provide the path to your private RSA key when connection to snowflake is setting up so one would get the following error: TypeError : Expected bytes or RSAPrivateKey, got <class 'NoneType'>.
  Instead, use the private_file_path argument, it works like a charm.

Thereby this code section should be:

db_tool = RunSqlTool(
    sql_runner=SnowflakeRunner(
        account="your-account",
        username="your-username",
        private_key_file="/path/to/rsa_key.p8",  ############################################## here we replace 'private_key_path' by 'private_key_file'
        private_key_passphrase="your-passphrase",  # Optional if key is encrypted
        database="your-database",
        warehouse="your-warehouse",
        authenticator ="SNOWFLAKE_JWT" ############################################## add the authenticator argument
    )
)