github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-08 20:39:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix:chromadb version

Browse Source
This commit is contained in:
aries_ckt
2023-09-27 12:03:05 +08:00
parent 6843fc7376
commit 1d3ea2acd5
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pilot/connections/conn_spark.py
View File

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

4
setup.py
View File

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