Merge pull request #33 from eifrach/create_default_warehouse

NO-ISSUE: setting the default warehouse

Author: eifrach

.flake8

@@ -0,0 +1,4 @@
+[flake8]
+max-line-length = 120
+exclude = .git,__pycache__,docs/source/conf.py,old,build,dist,venv,.venv
+ignore = E501,W503

.gitignore

@@ -1,2 +1,4 @@
 Snowflake_CSV/
 .venv/
+**/*.pvc
+__pycache__/

src/config.py

@@ -15,6 +15,7 @@
 SNOWFLAKE_BASE_URL = os.environ.get("SNOWFLAKE_BASE_URL")
 SNOWFLAKE_DATABASE = os.environ.get("SNOWFLAKE_DATABASE")
 SNOWFLAKE_SCHEMA = os.environ.get("SNOWFLAKE_SCHEMA")
+SNOWFLAKE_WAREHOUSE = os.environ.get("SNOWFLAKE_WAREHOUSE", "DEFAULT")
 
 # Snowflake token handling - for stdio transport, get from environment
 # For other transports, it will be retrieved from request context in tools layer

src/database.py

@@ -9,7 +9,8 @@
     SNOWFLAKE_BASE_URL,
     SNOWFLAKE_DATABASE,
     SNOWFLAKE_SCHEMA,
-    SNOWFLAKE_TOKEN
+    SNOWFLAKE_TOKEN,
+    SNOWFLAKE_WAREHOUSE
 )
 from metrics import track_snowflake_query
 
@@ -85,7 +86,8 @@ async def execute_snowflake_query(sql: str, snowflake_token: Optional[str] = Non
             "statement": sql,
             "timeout": 60,
             "database": SNOWFLAKE_DATABASE,
-            "schema": SNOWFLAKE_SCHEMA
+            "schema": SNOWFLAKE_SCHEMA,
+            "warehouse": SNOWFLAKE_WAREHOUSE,
         }
 
         logger.info(f"Executing Snowflake query: {sql[:100]}...")  # Log first 100 chars of query