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

fix(model): Fix apiserver error (#2605)

Browse Source
This commit is contained in:
Fangyin Cheng
2025-04-10 10:23:49 +08:00
committed by GitHub
parent babc484261
commit 5ddd9e5bf8
39 changed files with 1300 additions and 243 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
packages/dbgpt-client/src/dbgpt_client/_version.py
View File

@@ -1 +1 @@
version = "0.7.0"
version = "0.7.1"

31
packages/dbgpt-client/src/dbgpt_client/evaluation.py
View File

@@ -1,14 +1,41 @@
"""Evaluation."""
from typing import List
from typing import List, Optional
from dbgpt._private.pydantic import BaseModel, Field
from dbgpt.core.interface.evaluation import EvaluationResult
from dbgpt.core.schema.api import Result
from dbgpt_serve.evaluate.api.schemas import EvaluateServeRequest
from .client import Client, ClientException
class EvaluateServeRequest(BaseModel):
evaluate_code: Optional[str] = Field(None, description="evaluation code")
scene_key: Optional[str] = Field(None, description="evaluation scene key")
scene_value: Optional[str] = Field(None, description="evaluation scene value")
datasets_name: Optional[str] = Field(None, description="evaluation datasets name")
datasets: Optional[List[dict]] = Field(None, description="datasets")
evaluate_metrics: Optional[List[str]] = Field(
None, description="evaluation metrics"
)
context: Optional[dict] = Field(None, description="The context of the evaluate")
user_name: Optional[str] = Field(None, description="user name")
user_id: Optional[str] = Field(None, description="user id")
sys_code: Optional[str] = Field(None, description="system code")
parallel_num: Optional[int] = Field(None, description="system code")
state: Optional[str] = Field(None, description="evaluation state")
result: Optional[str] = Field(None, description="evaluation result")
storage_type: Optional[str] = Field(None, comment="datasets storage type")
average_score: Optional[str] = Field(None, description="evaluation average score")
log_info: Optional[str] = Field(None, description="evaluation log_info")
gmt_create: Optional[str] = Field(None, description="create time")
gmt_modified: Optional[str] = Field(None, description="create time")
class EvaluateServeResponse(EvaluateServeRequest):
"""EvaluateServeResponse."""
async def run_evaluation(
client: Client, request: EvaluateServeRequest
) -> List[EvaluationResult]: