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

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

View File

@@ -1,7 +1,8 @@
from abc import ABC, abstractmethod
from typing import Generic
from typing import Generic, Optional
from dbgpt.component import BaseComponent
from dbgpt.component import BaseComponent, SystemApp
from dbgpt.core.awel.dag.dag_manager import DAGManager
from dbgpt.serve.core.config import BaseServeConfig
from dbgpt.storage.metadata._base_dao import REQ, RES, BaseDao, T
@@ -9,8 +10,20 @@ from dbgpt.storage.metadata._base_dao import REQ, RES, BaseDao, T
class BaseService(BaseComponent, Generic[T, REQ, RES], ABC):
name = "dbgpt_serve_base_service"
_dag_manager: Optional[DAGManager] = None
_system_app: Optional[SystemApp] = None
def __init__(self, system_app):
super().__init__(system_app)
self._system_app = system_app
def init_app(self, system_app: SystemApp):
"""Initialize the service
Args:
system_app (SystemApp): The system app
"""
self._system_app = system_app
@property
@abstractmethod
@@ -32,3 +45,14 @@ class BaseService(BaseComponent, Generic[T, REQ, RES], ABC):
RES: The response
"""
return self.dao.create(request)
@property
def dag_manager(self) -> DAGManager:
if self._dag_manager is None:
raise ValueError("DAGManager is not initialized")
return self._dag_manager
def before_start(self):
"""Execute before the application starts"""
# if not self._system_app
self._dag_manager = DAGManager.get_instance(self._system_app)