Update snowflake docs

griptape-ai · Aug 30, 2023 · 41a0224 · 41a0224
1 parent 663c0dc
commit 41a0224
Showing 1 changed file with 9 additions and 12 deletions.
diff --git a/docs/griptape-framework/data/sql-drivers.md b/docs/griptape-framework/data/sql-drivers.md
@@ -51,20 +51,17 @@ import snowflake.connector
 from snowflake.connector import SnowflakeConnection
 from griptape.drivers import SnowflakeSqlDriver
 
-PWD = os.environ.get('PWD')
-
 def get_snowflake_connection() -> SnowflakeConnection:
     return snowflake.connector.connect(
-        account = 'account',
-        user = 'user',
-        password = PWD,
-        database = 'database',
-        schema = 'schema',
+        account=os.environ['SNOWFLAKE_ACCOUNT'],
+        user=os.environ['SNOWFLAKE_USER'],
+        password=os.environ['SNOWFLAKE_PASSWORD'],
+        database=os.environ['SNOWFLAKE_DATABASE'],
+        schema=os.environ['SNOWFLAKE_SCHEMA'],
+        warehouse=os.environ['SNOWFLAKE_WAREHOUSE']
     )
 
-driver=SnowflakeSqlDriver(
-    connection_func=get_snowflake_connection
-)
+driver = SnowflakeSqlDriver(connection_func=get_snowflake_connection)
 
-driver.execute_query("select * from users;")
-```
+driver.execute_query("SELECT * from PEOPLE;")
+```