fix(dspy): adding language for recently released snowflake retriever model and updating snowflake session init to also include session log

Author: sfc-gh-alherrera

docs/api/retrieval_model_clients/SnowflakeRM.md

@@ -6,7 +6,7 @@ sidebar_position:
 
 ### Constructor
 
-Initialize an instance of the `SnowflakeRM` class, with the option to use `e5-base-v2` embeddings or any Snowflake Cortex supported embeddings model.
+Initialize an instance of the `SnowflakeRM` class, with the option to use `e5-base-v2` or `snowflake-arctic-embed-m` embeddings or any other Snowflake Cortex supported embeddings model.
 
 ```python
 SnowflakeRM(

dspy/retrieve/snowflake_rm.py

@@ -42,7 +42,7 @@ def __init__(
         self.embeddings_field = embeddings_field
         self.embeddings_text_field = embeddings_text_field
         self.embeddings_model = embeddings_model
-        self.client = Session.builder.configs(snowflake_credentials).create()
+        self.client = self._init_cortex(credentials=snowflake_credentials)
 
         super().__init__(k=k)
 
@@ -99,6 +99,14 @@ def _top_k_similar_chunks(self, query_embeddings, k):
 
         return top_k.select(doc_table_key).to_pandas().values
 
+    @classmethod
+    def _init_cortex(cls, credentials: dict) -> None:
+        
+        session = Session.builder.configs(credentials).create()
+        session.query_tag = {"origin":"sf_sit", "name":"dspy", "version":{"major":1, "minor":0}}
+
+        return session
+
     def _get_embeddings(self, query: str) -> list[float]:
         # create embeddings for the query
         embed = snow_fn.builtin("snowflake.cortex.embed_text")