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

fix: Solve the problem of invalid subgraph naming when creating tugraph (#1731)

Browse Source
This commit is contained in:
A-Salty-Fish 2024-07-22 19:13:52 +08:00 committed by GitHub
parent f889fa3775
commit 84988b89fe
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
dbgpt/app/knowledge/service.py
View File

@ -1,5 +1,6 @@
import json import json
import logging import logging
import re
from datetime import datetime from datetime import datetime
from dbgpt._private.config import Config from dbgpt._private.config import Config
@ -79,6 +80,10 @@ class KnowledgeService:
) )
if request.vector_type == "VectorStore": if request.vector_type == "VectorStore":
request.vector_type = CFG.VECTOR_STORE_TYPE request.vector_type = CFG.VECTOR_STORE_TYPE
if request.vector_type == "KnowledgeGraph":
knowledge_space_name_pattern = r"^[a-zA-Z0-9\u4e00-\u9fa5]+$"
if not re.match(knowledge_space_name_pattern, request.name):
raise Exception(f"space name:{request.name} invalid")
spaces = knowledge_space_dao.get_knowledge_space(query) spaces = knowledge_space_dao.get_knowledge_space(query)
if len(spaces) > 0: if len(spaces) > 0:
raise Exception(f"space name:{request.name} have already named") raise Exception(f"space name:{request.name} have already named")