set org

vanna-ai · Jul 7, 2023 · 97932c5 · 97932c5
1 parent 23c22d3
commit 97932c5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/vanna/__init__.py b/src/vanna/__init__.py
@@ -200,9 +200,11 @@ def set_org(org: str) -> None:
 
         if create.lower() == 'y':
             db_type = input("What type of database would you like to use? (Snowflake, BigQuery, Postgres, etc.): ")
+            __org = 'demo-sales'
             if create_org(org=org, db_type=db_type):
                 __org = org
             else:
+                __org = None
                 raise Exception("Failed to create organization")
     else:
         __org = org