Skip to content

Commit 3745d74

Browse files
chore(graphrag): upgrade dbgpt-tugraph-plugins to version 0.1.1 (#2213)
Co-authored-by: Fangyin Cheng <staneyffer@gmail.com>
1 parent 38a8fc1 commit 3745d74

File tree

4 files changed

+7
-6
lines changed

4 files changed

+7
-6
lines changed

dbgpt/agent/core/base_agent.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -80,8 +80,8 @@ def check_available(self) -> None:
8080
or not self.resource.get_resource_by_type(action.resource_need)
8181
):
8282
raise ValueError(
83-
f"{self.name}[{self.role}] Missing resources[{action.resource_need}] required for "
84-
"runtime!"
83+
f"{self.name}[{self.role}] Missing resources"
84+
f"[{action.resource_need}] required for runtime!"
8585
)
8686
else:
8787
if not self.is_human and not self.is_team:

dbgpt/app/scene/chat_db/auto_execute/out_parser.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,9 @@ def parse_view_response(self, speak, data, prompt_response) -> str:
134134

135135
if param["type"] == "response_vector_chart":
136136
df, visualizable = self.parse_vector_data_with_pca(df)
137-
param["type"] = "response_scatter_chart" if visualizable else "response_table"
137+
param["type"] = (
138+
"response_scatter_chart" if visualizable else "response_table"
139+
)
138140

139141
param["sql"] = prompt_response.sql
140142
param["data"] = json.loads(

dbgpt/storage/graph_store/tugraph_store.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -139,8 +139,7 @@ def _upload_plugin(self):
139139
except ImportError:
140140
logger.error(
141141
"dbgpt-tugraph-plugins is not installed, "
142-
"pip install dbgpt-tugraph-plugins==0.1.0rc1 -U -i "
143-
"https://pypi.org/simple"
142+
"pip install dbgpt-tugraph-plugins==0.1.1"
144143
)
145144
plugin_path = get_plugin_binary_path("leiden") # type: ignore
146145
with open(plugin_path, "rb") as f:

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -547,7 +547,7 @@ def knowledge_requires():
547547

548548
setup_spec.extras["graph_rag"] = setup_spec.extras["rag"] + [
549549
"neo4j",
550-
"dbgpt-tugraph-plugins>=0.1.0rc1",
550+
"dbgpt-tugraph-plugins>=0.1.1",
551551
]
552552

553553

0 commit comments

Comments
 (0)