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

feat(ChatKnowledge): Support Financial Report Analysis (#1702)

Browse Source 
Co-authored-by: hzh97 <2976151305@qq.com>
Co-authored-by: Fangyin Cheng <staneyffer@gmail.com>
Co-authored-by: licunxing <864255598@qq.com>
This commit is contained in:
Aries-ckt
2024-07-26 13:40:54 +08:00
committed by GitHub
parent 22e0680a6a
commit 167d972093
160 changed files with 89339 additions and 795 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
dbgpt/serve/rag/models/models.py
View File

@@ -13,6 +13,7 @@ class KnowledgeSpaceEntity(Model):
id = Column(Integer, primary_key=True)
name = Column(String(100))
vector_type = Column(String(100))
domain_type = Column(String(100))
desc = Column(String(100))
owner = Column(String(100))
context = Column(Text)
@@ -20,7 +21,7 @@ class KnowledgeSpaceEntity(Model):
gmt_modified = Column(DateTime)
def __repr__(self):
return f"KnowledgeSpaceEntity(id={self.id}, name='{self.name}', vector_type='{self.vector_type}', desc='{self.desc}', owner='{self.owner}' context='{self.context}', gmt_created='{self.gmt_created}', gmt_modified='{self.gmt_modified}')"
return f"KnowledgeSpaceEntity(id={self.id}, name='{self.name}', vector_type='{self.vector_type}', domain_type='{self.domain_type}', desc='{self.desc}', owner='{self.owner}' context='{self.context}', gmt_created='{self.gmt_created}', gmt_modified='{self.gmt_modified}')"
class KnowledgeSpaceDao(BaseDao):
@@ -30,6 +31,7 @@ class KnowledgeSpaceDao(BaseDao):
knowledge_space = KnowledgeSpaceEntity(
name=space.name,
vector_type=space.vector_type,
domain_type=space.domain_type,
desc=space.desc,
owner=space.owner,
gmt_created=datetime.now(),
@@ -57,6 +59,10 @@ class KnowledgeSpaceDao(BaseDao):
knowledge_spaces = knowledge_spaces.filter(
KnowledgeSpaceEntity.vector_type == query.vector_type
)
if query.domain_type is not None:
knowledge_spaces = knowledge_spaces.filter(
KnowledgeSpaceEntity.domain_type == query.domain_type
)
if query.desc is not None:
knowledge_spaces = knowledge_spaces.filter(
KnowledgeSpaceEntity.desc == query.desc
@@ -159,4 +165,5 @@ class KnowledgeSpaceDao(BaseDao):
desc=entity.desc,
owner=entity.owner,
context=entity.context,
domain_type=entity.domain_type,
)