github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-05 10:29:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(graphrag): upgrade dbgpt-tugraph-plugins to version 0.1.1 (#2213)

Browse Source 
Co-authored-by: Fangyin Cheng <staneyffer@gmail.com>
This commit is contained in:
Florian 2024-12-17 18:33:27 +08:00 committed by GitHub
parent 38a8fc11d9
commit 3745d7411d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dbgpt/agent/core/base_agent.py
View File

@ -80,8 +80,8 @@ class ConversableAgent(Role, Agent):
or not self.resource.get_resource_by_type(action.resource_need) or not self.resource.get_resource_by_type(action.resource_need)
): ):
raise ValueError( raise ValueError(
f"{self.name}[{self.role}] Missing resources[{action.resource_need}] required for " f"{self.name}[{self.role}] Missing resources"
"runtime！" f"[{action.resource_need}] required for runtime！"
) )
else: else:
if not self.is_human and not self.is_team: if not self.is_human and not self.is_team:

4
dbgpt/app/scene/chat_db/auto_execute/out_parser.py
View File

@ -134,7 +134,9 @@ class DbChatOutputParser(BaseOutputParser):
if param["type"] == "response_vector_chart": if param["type"] == "response_vector_chart":
df, visualizable = self.parse_vector_data_with_pca(df) 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["sql"] = prompt_response.sql
param["data"] = json.loads( param["data"] = json.loads(

3
dbgpt/storage/graph_store/tugraph_store.py
View File

@ -139,8 +139,7 @@ class TuGraphStore(GraphStoreBase):
except ImportError: except ImportError:
logger.error( logger.error(
"dbgpt-tugraph-plugins is not installed, " "dbgpt-tugraph-plugins is not installed, "
"pip install dbgpt-tugraph-plugins==0.1.0rc1 -U -i " "pip install dbgpt-tugraph-plugins==0.1.1"
"https://pypi.org/simple"
) )
plugin_path = get_plugin_binary_path("leiden") # type: ignore plugin_path = get_plugin_binary_path("leiden") # type: ignore
with open(plugin_path, "rb") as f: with open(plugin_path, "rb") as f:

2
setup.py
View File

@ -547,7 +547,7 @@ def knowledge_requires():
setup_spec.extras["graph_rag"] = setup_spec.extras["rag"] + [ setup_spec.extras["graph_rag"] = setup_spec.extras["rag"] + [
"neo4j", "neo4j",
"dbgpt-tugraph-plugins>=0.1.0rc1", "dbgpt-tugraph-plugins>=0.1.1",
] ]