chore:v0.5.8 tag (#1632)

eosphoros-ai · Jun 13, 2024 · 57f7b6d · 57f7b6d
1 parent 0541d14
commit 57f7b6d
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 4 deletions.
diff --git a/assets/wechat.jpg b/assets/wechat.jpg
diff --git a/dbgpt/_version.py b/dbgpt/_version.py
@@ -1 +1 @@
-version = "0.5.7"
+version = "0.5.8"
diff --git a/dbgpt/rag/embedding/embeddings.py b/dbgpt/rag/embedding/embeddings.py
@@ -294,7 +294,7 @@ def __init__(self, **kwargs: Any):
         kwargs["client"] = sentence_transformers.SentenceTransformer(
             kwargs.get("model_name"),
             cache_folder=kwargs.get("cache_folder"),
-            **kwargs.get("model_kwargs"),
+            **(kwargs.get("model_kwargs") or {}),
         )
 
         super().__init__(**kwargs)

diff --git a/dbgpt/rag/embedding/rerank.py b/dbgpt/rag/embedding/rerank.py
@@ -17,7 +17,7 @@ class CrossEncoderRerankEmbeddings(BaseModel, RerankEmbeddings):
 
     client: Any  #: :meta private:
     model_name: str = "BAAI/bge-reranker-base"
-    max_length: int
+    max_length: int = None    # type: ignore
     """Max length for input sequences. Longer sequences will be truncated. If None, max
         length of the model will be used"""
     """Model name to use."""

diff --git a/setup.py b/setup.py
@@ -19,7 +19,7 @@
 IS_DEV_MODE = os.getenv("IS_DEV_MODE", "true").lower() == "true"
 # If you modify the version, please modify the version in the following files:
 # dbgpt/_version.py
-DB_GPT_VERSION = os.getenv("DB_GPT_VERSION", "0.5.7")
+DB_GPT_VERSION = os.getenv("DB_GPT_VERSION", "0.5.8")
 
 BUILD_NO_CACHE = os.getenv("BUILD_NO_CACHE", "true").lower() == "true"
 LLAMA_CPP_GPU_ACCELERATION = (