feat(ChatSnowflake): Add tool calling support (& move to new API) (#98)

* wip add tool calling support to ChatSnowflake()

* Get the non-streaming case working

* Handle additional chat parameters

* Add tests

* Undo outdated change

* Change model default; update changelog

* Add password

* cleanup

Author: cpsievert

.github/workflows/test.yml

@@ -48,6 +48,24 @@ jobs:
       - name: 📦 Install the project
         run: uv sync --python ${{ matrix.config.python-version }} --all-extras
 
+      - name: Create Snowflake connections.toml file
+        run: |
+          mkdir -p ~/.ssh
+          echo '${{ secrets.SNOWFLAKE_SSH_PUBLIC_KEY }}' > ~/.ssh/snowflake_key.pub
+          echo '${{ secrets.SNOWFLAKE_SSH_PRIVATE_KEY }}' > ~/.ssh/snowflake_key.p8
+          chmod 644 ~/.ssh/snowflake_key.pub
+          chmod 600 ~/.ssh/snowflake_key.p8
+          chmod 700 ~/.ssh
+
+          mkdir -p ~/.snowflake
+          cat > ~/.snowflake/connections.toml << 'EOF'
+          [posit]
+          account="duloftf-posit-software-pbc-dev"
+          user="carson@posit.co"
+          private_key_file="/home/runner/.ssh/snowflake_key.p8"
+          private_key_file_pwd="${{ secrets.SNOWFLAKE_SSH_PRIVATE_KEY_PASSWORD }}"
+          EOF
+
       - name: 🧪 Check tests
         run: make check-tests
 

CHANGELOG.md

@@ -9,15 +9,17 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [UNRELEASED]
 
-### Changes
-
-* `ChatDatabricks()`'s `model` now defaults to `databricks-claude-3-7-sonnet` instead of `databricks-dbrx-instruct`. (#95)
-
 ### Improvements
 
-* The `CHATLAS_LOG` environment variable will now enable logs for the relevant model provider. It now also supports a lovel of `debug` in addition to `info`. (#97)
+* The `CHATLAS_LOG` environment variable nows enable logs for the relevant model provider. It now also supports a lovel of `debug` in addition to `info`. (#97)
+* `ChatSnowflake()` now supports tool calling. (#98)
 * `Chat` instances can now be deep copied, which is useful for forking the chat session. (#96)
 
+### Changes
+
+* `ChatDatabricks()`'s `model` now defaults to `databricks-claude-3-7-sonnet` instead of `databricks-dbrx-instruct`. (#95)
+* `ChatSnowflake()`'s `model` now defaults to `claude-3-7-sonnet` instead of `llama3.1-70b`. (#98)
+
 ### Bug fixes
 
 * Fixed an issue where `ChatDatabricks()` with an Anthropic `model` wasn't handling empty-string responses gracefully. (#95)