mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-04 18:40:10 +00:00
feat: Add dbgpt client and add api v2
This commit is contained in:
28
dbgpt/serve/rag/config.py
Normal file
28
dbgpt/serve/rag/config.py
Normal file
@@ -0,0 +1,28 @@
|
||||
from dataclasses import dataclass, field
|
||||
from typing import Optional
|
||||
|
||||
from dbgpt.serve.core import BaseServeConfig
|
||||
|
||||
APP_NAME = "rag"
|
||||
SERVE_APP_NAME = "dbgpt_rag"
|
||||
SERVE_APP_NAME_HUMP = "dbgpt_rag"
|
||||
SERVE_CONFIG_KEY_PREFIX = "dbgpt_rag"
|
||||
SERVE_SERVICE_COMPONENT_NAME = f"{SERVE_APP_NAME}_service"
|
||||
|
||||
|
||||
@dataclass
|
||||
class ServeConfig(BaseServeConfig):
|
||||
"""Parameters for the serve command"""
|
||||
|
||||
api_keys: Optional[str] = field(
|
||||
default=None, metadata={"help": "API keys for the endpoint, if None, allow all"}
|
||||
)
|
||||
|
||||
default_user: Optional[str] = field(
|
||||
default=None,
|
||||
metadata={"help": "Default user name for prompt"},
|
||||
)
|
||||
default_sys_code: Optional[str] = field(
|
||||
default=None,
|
||||
metadata={"help": "Default system code for prompt"},
|
||||
)
|
Reference in New Issue
Block a user