Fixed import errors for experimental JSONalyzeQueryEngine (#17228)

run-llama · Dec 10, 2024 · 9a6f30d · 9a6f30d
1 parent 278bde0
commit 9a6f30d
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/llama-index-experimental/llama_index/experimental/query_engine/__init__.py b/llama-index-experimental/llama_index/experimental/query_engine/__init__.py
@@ -4,5 +4,8 @@
 from llama_index.experimental.query_engine.pandas.output_parser import (
     PandasInstructionParser,
 )
+from llama_index.experimental.query_engine.jsonalyze.jsonalyze_query_engine import (
+    JSONalyzeQueryEngine,
+)
 
-__all__ = ["PandasQueryEngine", "PandasInstructionParser"]
+__all__ = ["PandasQueryEngine", "PandasInstructionParser", "JSONalyzeQueryEngine"]
diff --git a/llama-index-experimental/llama_index/experimental/query_engine/jsonalyze/__init__.py b/llama-index-experimental/llama_index/experimental/query_engine/jsonalyze/__init__.py
@@ -1,6 +1,6 @@
 """Init file."""
 
-from llama_index.core.query_engine.jsonalyze.jsonalyze_query_engine import (
+from llama_index.experimental.query_engine.jsonalyze.jsonalyze_query_engine import (
     JSONalyzeQueryEngine,
 )
 

diff --git a/llama-index-experimental/pyproject.toml b/llama-index-experimental/pyproject.toml
@@ -25,7 +25,7 @@ exclude = ["**/BUILD"]
 license = "MIT"
 name = "llama-index-experimental"
 readme = "README.md"
-version = "0.5.1"
+version = "0.5.2"
 
 [tool.poetry.dependencies]
 python = ">=3.10,<4.0"