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

packages/dbgpt-client/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "dbgpt-client"
-version = "0.7.0"
+version = "0.7.1"
 description = "Add your description here"
 authors = [
     { name = "csunny", email = "cfqcsunny@gmail.com" }
@@ -9,7 +9,17 @@ license = "MIT"
 readme = "README.md"
 requires-python = ">= 3.10"
 
-dependencies = ["dbgpt-serve"]
+dependencies = [
+    "dbgpt[client,cli]",
+    "dbgpt_ext",
+    "shortuuid",
+    # 2.0.29 not support duckdb now
+    "SQLAlchemy>=2.0.25, <2.0.29",
+    # for cache
+    "msgpack",
+    # for AWEL operator serialization
+    "cloudpickle",
+]
 
 [project.urls]
 Homepage = "https://github.com/eosphoros-ai/DB-GPT"

packages/dbgpt-client/src/dbgpt_client/_version.py

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

packages/dbgpt-client/src/dbgpt_client/evaluation.py

@@ -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]: