github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2026-01-29 21:49:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(ChatKnowledge): Add custom text separators and refactor log configuration

Browse Source
This commit is contained in:
FangYin Cheng
2023-09-28 11:54:58 +08:00
parent 20bdddec51
commit 5dfe611478
52 changed files with 705 additions and 158 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pilot/openapi/api_v1/api_v1.py
View File

@@ -38,8 +38,6 @@ from pilot.server.knowledge.request.request import KnowledgeSpaceRequest
from pilot.scene.base_chat import BaseChat
from pilot.scene.base import ChatScene
from pilot.scene.chat_factory import ChatFactory
from pilot.configs.model_config import LOGDIR
from pilot.utils import build_logger
from pilot.common.schema import DBType
from pilot.memory.chat_history.duckdb_history import DuckdbHistoryMemory
from pilot.scene.message import OnceConversation
@@ -409,7 +407,7 @@ async def model_types(controller: BaseModelController = Depends(get_model_contro
@router.get("/v1/model/supports")
async def model_types(worker_manager: WorkerManager = Depends(get_worker_manager)):
async def model_supports(worker_manager: WorkerManager = Depends(get_worker_manager)):
logger.info(f"/controller/model/supports")
try:
models = await worker_manager.supported_models()

6
pilot/openapi/api_v1/editor/api_editor_v1.py
View File

@@ -6,12 +6,11 @@ from fastapi import (
)
from typing import List
import logging
from pilot.configs.config import Config
from pilot.scene.chat_factory import ChatFactory
from pilot.configs.model_config import LOGDIR
from pilot.utils import build_logger
from pilot.openapi.api_view_model import (
Result,
@@ -34,7 +33,8 @@ from pilot.scene.chat_db.data_loader import DbDataLoader
router = APIRouter()
CFG = Config()
CHAT_FACTORY = ChatFactory()
logger = build_logger("api_editor_v1", LOGDIR + "api_editor_v1.log")
logger = logging.getLogger(__name__)
@router.get("/v1/editor/db/tables", response_model=Result[DbTable])