diff --git a/dbgpt/agent/core/base_agent.py b/dbgpt/agent/core/base_agent.py index 123eb68f4..a1362461f 100644 --- a/dbgpt/agent/core/base_agent.py +++ b/dbgpt/agent/core/base_agent.py @@ -80,8 +80,8 @@ def check_available(self) -> None: or not self.resource.get_resource_by_type(action.resource_need) ): raise ValueError( - f"{self.name}[{self.role}] Missing resources[{action.resource_need}] required for " - "runtimeļ¼" + f"{self.name}[{self.role}] Missing resources" + f"[{action.resource_need}] required for runtimeļ¼" ) else: if not self.is_human and not self.is_team: diff --git a/dbgpt/app/scene/chat_db/auto_execute/out_parser.py b/dbgpt/app/scene/chat_db/auto_execute/out_parser.py index 557ea71d3..52df40358 100644 --- a/dbgpt/app/scene/chat_db/auto_execute/out_parser.py +++ b/dbgpt/app/scene/chat_db/auto_execute/out_parser.py @@ -134,7 +134,9 @@ def parse_view_response(self, speak, data, prompt_response) -> str: if param["type"] == "response_vector_chart": df, visualizable = self.parse_vector_data_with_pca(df) - param["type"] = "response_scatter_chart" if visualizable else "response_table" + param["type"] = ( + "response_scatter_chart" if visualizable else "response_table" + ) param["sql"] = prompt_response.sql param["data"] = json.loads( diff --git a/dbgpt/storage/graph_store/tugraph_store.py b/dbgpt/storage/graph_store/tugraph_store.py index 4f8437245..4838906a8 100644 --- a/dbgpt/storage/graph_store/tugraph_store.py +++ b/dbgpt/storage/graph_store/tugraph_store.py @@ -139,8 +139,7 @@ def _upload_plugin(self): except ImportError: logger.error( "dbgpt-tugraph-plugins is not installed, " - "pip install dbgpt-tugraph-plugins==0.1.0rc1 -U -i " - "https://pypi.org/simple" + "pip install dbgpt-tugraph-plugins==0.1.1" ) plugin_path = get_plugin_binary_path("leiden") # type: ignore with open(plugin_path, "rb") as f: diff --git a/setup.py b/setup.py index df82b7035..b330c6c2b 100644 --- a/setup.py +++ b/setup.py @@ -547,7 +547,7 @@ def knowledge_requires(): setup_spec.extras["graph_rag"] = setup_spec.extras["rag"] + [ "neo4j", - "dbgpt-tugraph-plugins>=0.1.0rc1", + "dbgpt-tugraph-plugins>=0.1.1", ]