fix:chromadb version

pilot/connections/conn_spark.py

@@ -1,6 +1,4 @@
 from typing import Optional, Any
-from pyspark.sql import SparkSession, DataFrame
-from sqlalchemy import text
 
 from pilot.connections.base import BaseConnect
 
@@ -19,6 +17,7 @@ class SparkConnect(BaseConnect):
     driver: str = "spark"
     """db dialect"""
     dialect: str = "sparksql"
+    from pyspark.sql import SparkSession, DataFrame
 
     def __init__(
         self,
@@ -30,6 +29,7 @@ class SparkConnect(BaseConnect):
         """Initialize the Spark DataFrame from Datasource path
         return: Spark DataFrame
         """
+        from pyspark.sql import SparkSession
         self.spark_session = (
             spark_session or SparkSession.builder.appName("dbgpt_spark").getOrCreate()
         )

setup.py

@@ -281,7 +281,7 @@ def core_requires():
         "importlib-resources==5.12.0",
         "psutil==5.9.4",
         "python-dotenv==1.0.0",
-        "colorama==0.4.10",
+        "colorama==0.4.6",
         "prettytable",
         "cachetools",
     ]
@@ -312,7 +312,7 @@ def knowledge_requires():
     setup_spec.extras["knowledge"] = [
         "spacy==3.5.3",
         # "chromadb==0.3.22",
-        "chromadb",
+        "chromadb==0.4.10",
         "markdown",
         "bs4",
         "python-pptx",