community: Add Snowflake model session handling via keyfile and external sessions

[zaramzamzam]

Description

• Implement session creation using keyfile for secure access without password

  • Keyfiles can be encrypted, with passwords stored in SNOWFLAKE_KEYFILE_PASSWORD

• Add session as a parameter for external session management

  • Enables running langchain inside Snowflake with existing session handling
  • External sessions passed to the model are not closed on completion

• Added comprehensive tests:

  • Session creation tests (keyfile, parameter-passed sessions)
  • Tests for varied environment variables and session configurations

• Updated the documentation at snowflake.ipynb

[vercel]

The latest updates on your projects. Learn more about Vercel for Git ↗︎

Name	Status	Preview	Comments	Updated (UTC)
langchain	✅ Ready (Inspect)	Visit Preview	💬 Add feedback	Nov 12, 2024 8:16pm

[zaramzamzam]

Hi,

I added some more options to connect to the Snowflake models, but I can only integration test the variant of adding the Session directly and connecting via a keyfile.

[efriis]

cc @sfc-gh-nmoiseyev - I think this should be done as part of the move of the snowflake chat models to the integration package https://github.com/langchain-ai/langchain-snowflake

[zaramzamzam]

Hi,

I hope the latest commit fixes the errors in the CI/CD pipeline.

For the unittests I mock the package import now which is a bit ugly. Maybe someone has a better idea.

For the integrationtests I removed the option to pass a Session to the model.

• this removes the need to have snowpark installed
• this results in not testing this passing of a session directly

[zaramzamzam]

I think now all checks should succeed 🤞

[zaramzamzam]

Looks like I forgot to run make format again.

Is there a possibility to have make format, make lint run before commit?
I could not see any pre-commit config.

[efriis]

believe some of the session mechanics were superceded by #27753 - if you want to reopen with just the changes related to aliasing feel free!