fix(dspy): adding LM connection parameter update to docs, resolving ruff issues and failing tests

Author: sfc-gh-alherrera

docs/api/language_model_clients/Snowflake.md

@@ -7,6 +7,19 @@ sidebar_position:
 ### Usage
 
 ```python
+import dspy
+import os
+
+connection_parameters = {
+
+    "account": os.getenv('SNOWFLAKE_ACCOUNT'),
+    "user": os.getenv('SNOWFLAKE_USER'),
+    "password": os.getenv('SNOWFLAKE_PASSWORD'),
+    "role": os.getenv('SNOWFLAKE_ROLE'),
+    "warehouse": os.getenv('SNOWFLAKE_WAREHOUSE'),
+    "database": os.getenv('SNOWFLAKE_DATABASE'),
+    "schema": os.getenv('SNOWFLAKE_SCHEMA')}
+
 lm = dspy.Snowflake(model="mixtral-8x7b",credentials=connection_parameters)
 ```
 

dsp/modules/snowflake.py

@@ -46,6 +46,9 @@ def __init__(self, model: str = "mixtral-8x7b", credentials=None, **kwargs):
             Which pre-trained model from Snowflake to use?
             Choices are 'snowflake-arctic','mistral-large','reka-flash','mixtral-8x7b','llama2-70b-chat','mistral-7b','gemma-7b'
             Full list of supported models is available here: https://docs.snowflake.com/en/user-guide/snowflake-cortex/llm-functions#complete
+        credentials: dict
+            Snowflake credentials required to initialize the session. 
+            Full list of requirements can be found here: https://docs.snowflake.com/en/developer-guide/snowpark/reference/python/latest/api/snowflake.snowpark.Session
         **kwargs: dict
             Additional arguments to pass to the API provider.
         """

dspy/__init__.py

@@ -7,7 +7,6 @@
 from .signatures import *
 from .utils.logging import logger, set_log_output
 
-
 # Functional must be imported after primitives, predict and signatures
 from .functional import *  # isort: skip
 

dspy/retrieve/snowflake_rm.py

@@ -6,7 +6,7 @@
 try:
     from snowflake.snowpark import Session
     from snowflake.snowpark import functions as snow_fn
-    from snowflake.snowpark.functions import lit, function, col
+    from snowflake.snowpark.functions import col, function, lit
     from snowflake.snowpark.types import VectorType
 
 except ImportError: