run_sql_bigquery with invalid query fails with TypeError "exceptions must derive from BaseException"

[andreaskaltenbach]

Describe the bug
Executing run_sql_bigquery with an invalid SQL query results in a type error.

Example:

vn = VannaDefault(model='chinook', api_key=vanna.get_api_key('[email protected]'))
vn.connect_to_bigquery(cred_file_path="some.json", project_id="my_project")
vn.run_sql("SELECT xyz")

results in TypeError('exceptions must derive from BaseException')

def connect_to_bigquery(self, cred_file_path: str = None, project_id: str = None):
  ...
  try:
      job = conn.query(sql)
      df = job.result().to_dataframe()
      return df
  except GoogleAPIError as error:
      errors = []
      for error in error.errors:
          errors.append(error["message"]) 
      raise errors # errors is a list of strings

Expected behavior
GoogleAPIError should be handled properly and an exceptiobn should be raised instead of a list of strings.

Desktop (please complete the following information where):

• OS: MacOS
• Version: 14.4.1
• Python: 3.12.3
• Vanna: [2.8.0]

Additional context
Add any other context about the problem here.

[Molrn]

this was fixed in version 0.4.2, just upgrade your vanna version

[everdark]

I'm using v0.6.0 but still got this error.

[a-brohi]

same; im using vanna==0.6.0 and getting error TypeError: exceptions must derive from BaseException with below code. let me know if you require anything else.

vn = VannaDefault(model=vanna_model_name, api_key=api_key)

vn.connect_to_bigquery(project_id="project_id", cred_file_path="example.json")

_question= "What are the top 5 customers by spend?"
_sql = vn.generate_sql(question = "What are the top 5 customers by spend?")
_df = vn.run_sql(sql= _sql)