From 0b4b6c3af972e9f77494975895dc7472f896eb0d Mon Sep 17 00:00:00 2001 From: Logan Markewich Date: Thu, 29 Feb 2024 22:36:26 -0600 Subject: [PATCH] fix jsonalyze query engine --- .../core/query_engine/jsonalyze_query_engine.py | 8 ++++---- llama-index-core/pyproject.toml | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/llama-index-core/llama_index/core/query_engine/jsonalyze_query_engine.py b/llama-index-core/llama_index/core/query_engine/jsonalyze_query_engine.py index e6edc6f8670bc..c2b81afbf9e1e 100644 --- a/llama-index-core/llama_index/core/query_engine/jsonalyze_query_engine.py +++ b/llama-index-core/llama_index/core/query_engine/jsonalyze_query_engine.py @@ -80,7 +80,7 @@ def default_jsonalyzer( try: # Load list of dictionaries into SQLite database db[table_name].insert_all(list_of_dict) - except sqlite_utils.db_exceptions.IntegrityError as exc: + except sqlite_utils.utils.sqlite3.IntegrityError as exc: print_text(f"Error inserting into table {table_name}, expected format:") print_text("[{col1: val1, col2: val2, ...}, ...]") raise ValueError("Invalid list_of_dict") from exc @@ -105,7 +105,7 @@ def default_jsonalyzer( try: # Execute the SQL query results = list(db.query(sql_query)) - except sqlite_utils.db_exceptions.OperationalError as exc: + except sqlite_utils.utils.sqlite3.OperationalError as exc: print_text(f"Error executing query: {sql_query}") raise ValueError("Invalid query") from exc @@ -148,7 +148,7 @@ async def async_default_jsonalyzer( try: # Load list of dictionaries into SQLite database db[table_name].insert_all(list_of_dict) - except sqlite_utils.db_exceptions.IntegrityError as exc: + except sqlite_utils.utils.sqlite3.IntegrityError as exc: print_text(f"Error inserting into table {table_name}, expected format:") print_text("[{col1: val1, col2: val2, ...}, ...]") raise ValueError("Invalid list_of_dict") from exc @@ -173,7 +173,7 @@ async def async_default_jsonalyzer( try: # Execute the SQL query results = list(db.query(sql_query)) - except sqlite_utils.db_exceptions.OperationalError as exc: + except sqlite_utils.utils.sqlite3.OperationalError as exc: print_text(f"Error executing query: {sql_query}") raise ValueError("Invalid query") from exc diff --git a/llama-index-core/pyproject.toml b/llama-index-core/pyproject.toml index ffa6a94e30e90..c95f5551d1af4 100644 --- a/llama-index-core/pyproject.toml +++ b/llama-index-core/pyproject.toml @@ -43,7 +43,7 @@ name = "llama-index-core" packages = [{include = "llama_index"}] readme = "README.md" repository = "https://github.com/run-llama/llama_index" -version = "0.10.14" +version = "0.10.14.post1" [tool.poetry.dependencies] SQLAlchemy = {extras = ["asyncio"], version = ">=1.4.49"}