diff --git a/src/vanna/__init__.py b/src/vanna/__init__.py index 55177031..e0d62bf4 100644 --- a/src/vanna/__init__.py +++ b/src/vanna/__init__.py @@ -233,18 +233,18 @@ def __rpc_call(method, params): raise ImproperlyConfigured( "model not set. Use vn.set_model(...) to set the model to use." ) - - if method != "list_orgs": + + if method == "list_orgs": headers = { "Content-Type": "application/json", "Vanna-Key": api_key, - "Vanna-Org": __org, + "Vanna-Org": "demo-tpc-h", } else: headers = { "Content-Type": "application/json", "Vanna-Key": api_key, - "Vanna-Org": "demo-tpc-h", + "Vanna-Org": __org, } data = {"method": method, "params": [__dataclass_to_dict(obj) for obj in params]}