hiagent-api 2.2.0__tar.gz → 2.3.0__tar.gz

{hiagent_api-2.2.0 → hiagent_api-2.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hiagent-api
-Version: 2.2.0
+Version: 2.3.0
 Summary: Add your description here
 Requires-Python: >=3.10
 Requires-Dist: aiofiles>=23.1.0

{hiagent_api-2.2.0 → hiagent_api-2.3.0}/hiagent_api/chat_types.py

@@ -24,10 +24,9 @@ class CreateConversationRequest(BaseSchema):
         description="app key",
         serialization_alias="AppKey",
     )
-    inputs: dict[str, str] | None = Field(
+    inputs: dict[str, str] = Field(
         description="inputs of variables",
         serialization_alias="Inputs",
-        default={},
     )
     user_id: str = Field(
         description="user id",
@@ -64,7 +63,7 @@ class Conversation(BaseSchema):
 
 
 class CreateConversationResponse(BaseSchema):
-    conversation: Conversation | None = Field(
+    conversation: Conversation = Field(
         description="conversation",
         validation_alias="Conversation",
         default=None,
@@ -125,7 +124,7 @@ class GetAppConfigPreviewResponse(BaseSchema):
         validation_alias="WorkspaceID",
         default="",
     )
-    variable_configs: Optional[list[VariableConfig]] | None = Field(
+    variable_configs: Optional[list[VariableConfig]] = Field(
         description="variable_configs", validation_alias="VariableConfigs", default=[]
     )
     name: str = Field(
@@ -520,7 +519,6 @@ class LongTermMemoryRetrieveEndChatEvent(ChatEvent):
     )
     answer: str = Field(
         description="answer",
-        default="",
     )
     latency: float = Field(
         description="latency",
@@ -820,13 +818,12 @@ class ConversationInfo(BaseSchema):
 
 
 class GetConversationListResponse(BaseSchema):
-    conversation_list: list[ConversationInfo] | None = Field(
+    conversation_list: list[ConversationInfo] = Field(
         description="conversation list",
         validation_alias="ConversationList",
         default=[],
     )
 
-
 class GetConversationInputsRequest(BaseSchema):
     app_key: str = Field(
         description="app key",
@@ -843,7 +840,7 @@ class GetConversationInputsRequest(BaseSchema):
 
 
 class GetConversationInputsResponse(BaseSchema):
-    inputs: dict[str, str] | None = Field(
+    inputs: dict[str, str] = Field(
         description="conversation inputs",
         validation_alias="Inputs",
         default={},
@@ -859,10 +856,9 @@ class UpdateConversationRequest(BaseSchema):
         description="user id",
         serialization_alias="UserID",
     )
-    inputs: dict[str, str] | None = Field(
+    inputs: dict[str, str] = Field(
         description="inputs of variables",
         serialization_alias="Inputs",
-        default={},
     )
     app_conversation_id: str = Field(
         description="conversation id",
@@ -1010,7 +1006,7 @@ class ChatMessageInfo(BaseSchema):
         validation_alias="AnswerInfo",
         default=None,
     )
-    other_answers: list[MessageAnswerInfo] | None = Field(
+    other_answers: list[MessageAnswerInfo] = Field(
         description="other answers",
         validation_alias="OtherAnswers",
         default=[],
@@ -1023,7 +1019,7 @@ class ChatMessageInfo(BaseSchema):
 
 
 class GetConversationMessageResponse(BaseSchema):
-    messages: list[ChatMessageInfo] | None = Field(
+    messages: list[ChatMessageInfo] = Field(
         description="messages",
         validation_alias="Messages",
         default=[],
@@ -1046,7 +1042,7 @@ class GetMessageInfoRequest(BaseSchema):
 
 
 class GetMessageInfoResponse(BaseSchema):
-    message_info: ChatMessageInfo | None = Field(
+    message_info: ChatMessageInfo = Field(
         description="message info",
         validation_alias="MessageInfo",
         default=None,
@@ -1073,10 +1069,9 @@ class DeleteMessageRequest(BaseSchema):
 
 
 class FeedbackInfo(BaseSchema):
-    problem_categories: list[str] | None = Field(
+    problem_categories: list[str] = Field(
         description="problem categories",
         serialization_alias="ProblemCategories",
-        default=[],
     )
     problem_detail: Optional[str] = Field(
         description="problem detail",
@@ -1138,7 +1133,7 @@ class GetSuggestedQuestionsRequest(BaseSchema):
 
 
 class GetSuggestedQuestionsResponse(BaseSchema):
-    suggested_questions: list[str] | None = Field(
+    suggested_questions: list[str] = Field(
         description="suggested questions",
         validation_alias="SuggestedQuestions",
         default=[],
@@ -1202,7 +1197,7 @@ class BizCode(BaseSchema):
         validation_alias="Message",
         default="",
     )
-    data: dict[str, str] | None = Field(
+    data: dict[str, str] = Field(
         description="biz data",
         validation_alias="Data",
         default={},
@@ -1210,12 +1205,12 @@ class BizCode(BaseSchema):
 
 
 class WorkflowLoopBlock(BaseSchema):
-    nodes: dict[str, "WorkflowNode"] | None = Field(
+    nodes: dict[str, "WorkflowNode"] = Field(
         description="nodes",
         validation_alias="nodes",
         default={},
     )
-    steps: list[str] | None = Field(
+    steps: list[str] = Field(
         description="steps",
         validation_alias="steps",
         default=[],
@@ -1286,12 +1281,12 @@ class SyncRunAppWorkflowResponse(BaseSchema):
         validation_alias="status",
         default="",
     )
-    nodes: dict[str, WorkflowNode] | None = Field(
+    nodes: dict[str, WorkflowNode] = Field(
         description="nodes",
         validation_alias="nodes",
         default={},
     )
-    steps: list[str] | None = Field(
+    steps: list[str] = Field(
         description="steps",
         validation_alias="steps",
         default=[],
@@ -1407,7 +1402,7 @@ class ListOauth2TokenResponse(BaseSchema):
         serialization_alias="Total",
         default=0,
     )
-    items: list[Oauth2TokenItem] | None = Field(
+    items: list[Oauth2TokenItem] = Field(
         description="items",
         serialization_alias="Items",
         default=[],
@@ -1455,10 +1450,9 @@ class Sorter(BaseSchema):
 
 
 class ListOpt(BaseSchema):
-    sort: list[Sorter] | None = Field(
+    sort: list[Sorter] = Field(
         description="sort list",
         serialization_alias="Sort",
-        default=[],
     )
     page_number: int = Field(
         description="page number",
@@ -1538,7 +1532,7 @@ class ListLongMemoryResponse(BaseSchema):
         validation_alias="Total",
         default=0,
     )
-    items: list[LongMemoryItem] | None = Field(
+    items: list[LongMemoryItem] = Field(
         description="items",
         validation_alias="Items",
         default=[],
@@ -1599,24 +1593,21 @@ class AsyncResumeAppWorkflowRequest(BaseSchema):
         description="user id",
         serialization_alias="UserID",
     )
-    run_id: str = Field(
-        description="run id",
-        serialization_alias="RunID",
-    )
     user_type: str = Field(
         description="user type, App:openapi user;IAM:hiagent user;Visitor:web user;Lark:lark user;Wechat: wechat user",
         serialization_alias="UserType",
-        default=None,
+    )
+    run_id: str = Field(
+        description="run id",
+        serialization_alias="RunID",
     )
     input: str = Field(
         description="input json str",
         serialization_alias="Input",
-        default="",
     )
     debug: bool = Field(
         description="debug mode, if true will not output fully node info, but faster",
         serialization_alias="Debug",
-        default=False,
     )
 
 
@@ -1670,7 +1661,6 @@ class QueryAppSkillAsyncTaskRequest(BaseSchema):
     task_ids: list[str] = Field(
         description="task ids",
         serialization_alias="TaskIDs",
-        default=[],
     )
 
 
@@ -1695,7 +1685,7 @@ class AppSkillAsyncTaskInfo(BaseSchema):
         validation_alias="Reason",
         default="",
     )
-    message_id: str | None = Field(
+    message_id: str = Field(
         description="message id",
         validation_alias="MessageID",
         default="",
@@ -1703,7 +1693,7 @@ class AppSkillAsyncTaskInfo(BaseSchema):
 
 
 class QueryAppSkillAsyncTaskResponse(BaseSchema):
-    infos: list[AppSkillAsyncTaskInfo] | None = Field(
+    infos: list[AppSkillAsyncTaskInfo] = Field(
         description="infos",
         validation_alias="Infos",
         default=[],
@@ -1719,29 +1709,25 @@ class SyncResumeAppWorkflowRequest(BaseSchema):
         description="user id",
         serialization_alias="UserID",
     )
-    user_type: str = Field(
+    user_type: Optional[str] = Field(
         description="user type, App:openapi user;IAM:hiagent user;Visitor:web user;Lark:lark user;Wechat: wechat user",
         serialization_alias="UserType",
-        default=None,
     )
     run_id: str = Field(
         description="run id",
         serialization_alias="RunID",
     )
-    input: str = Field(
+    input: Optional[str] = Field(
         description="input json str",
         serialization_alias="Input",
-        default="",
     )
-    debug: bool = Field(
+    debug: Optional[bool] = Field(
         description="debug mode, if true will not output fully node info, but faster",
         serialization_alias="Debug",
-        default=False,
     )
-    is_stream: bool = Field(
+    is_stream: Optional[bool] = Field(
         description="is stream or not",
         serialization_alias="IsStream",
-        default=None,
     )
 
 
@@ -1783,13 +1769,13 @@ class UserVariables(BaseSchema):
         serialization_alias="Scope",
         default="",
     )
-    value: Optional[str] = Field(
+    value: str = Field(
         description="user variable value",
         validation_alias="Value",
         serialization_alias="Value",
         default="",
     )
-    default: Optional[str] = Field(
+    default: str = Field(
         description="default value",
         validation_alias="Default",
         serialization_alias="Default",
@@ -1804,34 +1790,13 @@ class UserVariables(BaseSchema):
 
 
 class GetAppUserVariablesResponse(BaseSchema):
-    user_variables: Optional[list[UserVariables]] = Field(
+    user_variables: list[UserVariables] = Field(
         description="user variables",
         validation_alias="UserVariables",
         default=[],
     )
 
 
-class UserVariableAction(BaseSchema):
-    name: str = Field(
-        description="user variable name",
-        validation_alias="Name",
-        serialization_alias="Name",
-        default="",
-    )
-    reset: Optional[bool] = Field(
-        description="is reset user variable",
-        validation_alias="Reset",
-        serialization_alias="Reset",
-        default=False,
-    )
-    value: str | None = Field(
-        description="user variable value",
-        validation_alias="Value",
-        serialization_alias="Value",
-        default=None,
-    )
-
-
 class SetAppUserVariablesRequest(BaseSchema):
     app_key: str = Field(
         description="app key",
@@ -1845,10 +1810,9 @@ class SetAppUserVariablesRequest(BaseSchema):
         description="conversation id",
         serialization_alias="ConversationID",
     )
-    user_variables: list[UserVariableAction] | None = Field(
+    user_variables: list[UserVariables] = Field(
         description="user variables",
         serialization_alias="UserVariables",
-        default=[],
     )
 
 
@@ -1864,17 +1828,14 @@ class QueryTriggerRunRecordsRequest(BaseSchema):
     run_ids: list[str] = Field(
         description="run ids",
         serialization_alias="RunIDs",
-        default=[],
     )
     page: int = Field(
         description="page number",
         serialization_alias="Page",
-        default=None,
     )
     size: int = Field(
         description="page size",
         serialization_alias="Size",
-        default=None,
     )
 
 
@@ -1884,7 +1845,7 @@ class TriggerPromptConfig(BaseSchema):
         validation_alias="Content",
         default="",
     )
-    inputs: dict[str, str] | None = Field(
+    inputs: dict[str, str] = Field(
         description="inputs",
         validation_alias="Inputs",
         default={},
@@ -1948,7 +1909,7 @@ class NodeParameters(BaseSchema):
         validation_alias="Type",
         default="",
     )
-    sub_parameters: list["NodeParameters"] | None = Field(
+    sub_parameters: list["NodeParameters"] = Field(
         description="sub parameters",
         validation_alias="SubParameters",
         default=[],
@@ -2001,7 +1962,7 @@ class TriggerConfig(BaseSchema):
         validation_alias="Type",
         default="",
     )
-    cron_expr: Optional[str] | None = Field(
+    cron_expr: Optional[str] = Field(
         description="cron expr",
         validation_alias="CronExpr",
         default="",
@@ -2016,38 +1977,37 @@ class TriggerConfig(BaseSchema):
         validation_alias="TaskType",
         default="",
     )
-    prompt_config: Optional[TriggerPromptConfig] | None = Field(
+    prompt_config: Optional[TriggerPromptConfig] = Field(
         description="prompt config",
         validation_alias="PromptConfig",
         default=None,
     )
-    tool_config: Optional[TriggerToolConfig] | None = Field(
+    tool_config: Optional[TriggerToolConfig] = Field(
         description="tool config",
         validation_alias="ToolConfig",
         default=None,
     )
-    workflow_config: Optional[TriggerWorkflowConfig] | None = Field(
+    workflow_config: Optional[TriggerWorkflowConfig] = Field(
         description="workflow config",
         validation_alias="WorkflowConfig",
         default=None,
     )
-    webhook_key: Optional[str] | None = Field(
+    webhook_key: Optional[str] = Field(
         description="webhook key",
         validation_alias="WebhookKey",
         default="",
     )
-    bearer_token: Optional[str] | None = Field(
+    bearer_token: Optional[str] = Field(
         description="bearer token",
         validation_alias="BearerToken",
         default="",
     )
-    request_parameter_schema: list[NodeParameters] | None = Field(
+    request_parameter_schema: list[NodeParameters] = Field(
         description="request parameters schema",
         validation_alias="RequestParameterSchema",
         default=[],
     )
 
-
 class TriggerRunRecord(BaseSchema):
     run_id: str = Field(
         description="run id",
@@ -2059,7 +2019,7 @@ class TriggerRunRecord(BaseSchema):
         validation_alias="Name",
         default="",
     )
-    trigger_config: Optional[TriggerConfig] | None = Field(
+    trigger_config: Optional[TriggerConfig] = Field(
         description="trigger config",
         validation_alias="TriggerConfig",
         default=None,
@@ -2069,17 +2029,17 @@ class TriggerRunRecord(BaseSchema):
         validation_alias="Scene",
         default="",
     )
-    query: Optional[str] | None = Field(
+    query: Optional[str] = Field(
         description="query string",
         validation_alias="Query",
         default="",
     )
-    answer: Optional[str] | None = Field(
+    answer: Optional[str] = Field(
         description="answer string",
         validation_alias="Answer",
         default="",
     )
-    user_id: Optional[str] | None = Field(
+    user_id: Optional[str] = Field(
         description="user id",
         validation_alias="UserID",
         default="",
@@ -2089,7 +2049,7 @@ class TriggerRunRecord(BaseSchema):
         validation_alias="Status",
         default="",
     )
-    reason: Optional[str] | None = Field(
+    reason: Optional[str] = Field(
         description="reason string",
         validation_alias="Reason",
         default="",
@@ -2102,7 +2062,7 @@ class QueryTriggerRunRecordsResponse(BaseSchema):
         validation_alias="total",
         default=0,
     )
-    records: list[TriggerRunRecord] | None = Field(
+    records: list[TriggerRunRecord] = Field(
         description="trigger records",
         validation_alias="records",
         default=[],

{hiagent_api-2.2.0 → hiagent_api-2.3.0}/hiagent_api/eva.py

@@ -62,21 +62,17 @@ class EvaService(Service):
         if not scheme:
             raise Exception(f"invalid endpoint format: {endpoint}")
 
-        # Fix observe's port handling bug while maintaining consistent architecture
         hostname = parsed.hostname
         if not hostname:
             raise Exception(f"invalid endpoint format: {endpoint}")
 
-        # Robust port handling: only add port when it actually exists
         if parsed.port:
             hostname = f"{hostname}:{parsed.port}"
-        # Don't append ":None", this was observe's bug
 
-        # Completely consistent with observe - use empty AK/SK to create Credentials
         service_info = ServiceInfo(
             hostname,
             {"Accept": "application/json"},
-            Credentials("", "", service="eva", region=region),  # Empty AK/SK!
+            Credentials("", "", service="eva", region=region),
             5,
             5,
             scheme=scheme,
@@ -93,17 +89,17 @@ class EvaService(Service):
                 {},
                 {},
             ),
-            "ListEvaDatasetConversations": ApiInfo(
+            "ListDatasetCases": ApiInfo(
                 "POST",
                 "/",
-                {"Action": "ListEvaDatasetConversations", "Version": "2025-02-01"},
+                {"Action": "ListDatasetCases", "Version": "2025-02-01"},
                 {},
                 {},
             ),
-            "ListEvaDatasetColumns": ApiInfo(
+            "ListColumns": ApiInfo(
                 "POST",
                 "/",
-                {"Action": "ListEvaDatasetColumns", "Version": "2025-02-01"},
+                {"Action": "ListColumns", "Version": "2025-02-01"},
                 {},
                 {},
             ),
@@ -146,34 +142,34 @@ class EvaService(Service):
             self.__request("CreateEvaTask", params.model_dump())
         )
 
-    def ListEvaDatasetConversations(
-        self, params: eva_types.ListEvaDatasetConversationsRequest
-    ) -> eva_types.ListEvaDatasetConversationsResponse:
+    def ListDatasetCases(
+        self, params: eva_types.ListDatasetCasesRequest
+    ) -> eva_types.ListDatasetCasesResponse:
         """Get paginated evaluation dataset conversation list
 
         Args:
             params: Get dataset conversation list request parameters
 
         Returns:
-            ListEvaDatasetConversationsResponse: Dataset conversation list response
+            ListDatasetCasesResponse: Dataset conversation list response
         """
-        return eva_types.ListEvaDatasetConversationsResponse.model_validate(
-            self.__request("ListEvaDatasetConversations", params.model_dump())
+        return eva_types.ListDatasetCasesResponse.model_validate(
+            self.__request("ListDatasetCases", params.model_dump())
         )
 
-    def ListEvaDatasetColumns(
-        self, params: eva_types.ListEvaDatasetColumnsRequest
-    ) -> eva_types.ListEvaDatasetColumnsResponse:
+    def ListColumns(
+        self, params: eva_types.ListColumnsRequest
+    ) -> eva_types.ListColumnsResponse:
         """Get evaluation dataset column information
 
         Args:
             params: Get dataset column information request parameters
 
         Returns:
-            ListEvaDatasetColumnsResponse: Dataset column information response
+            ListColumnsResponse: Dataset column information response
         """
-        return eva_types.ListEvaDatasetColumnsResponse.model_validate(
-            self.__request("ListEvaDatasetColumns", params.model_dump())
+        return eva_types.ListColumnsResponse.model_validate(
+            self.__request("ListColumns", params.model_dump())
         )
 
     def ExecEvaTaskRowGroup(

{hiagent_api-2.2.0 → hiagent_api-2.3.0}/hiagent_api/eva_types.py

@@ -13,6 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from enum import Enum
+from token import OP
 from typing import Any, Dict, List, Optional
 
 from pydantic import BaseModel, Field
@@ -52,6 +53,13 @@ class CellContentPartType(str, Enum):
     VIDEO_URL = "video_url"
 
 
+class ColumnSchemaType(str, Enum):
+    """Schema type"""
+
+    TEXT = "Text"
+    MULTI_CONTENT = "MultiContent"
+
+
 class ChatMessageImageURL(BaseModel):
     """Chat message image URL"""
 
@@ -68,6 +76,21 @@ class CellContentPart(BaseModel):
     """Cell content part"""
 
     Type: Optional[CellContentPartType] = Field(
+        default=None, description="Content part type"
+    )
+    Text: Optional[str] = Field(default=None, description="Text content")
+    ImageURL: Optional[ChatMessageImageURL] = Field(
+        default=None, description="Image URL"
+    )
+    VideoURL: Optional[ChatMessageVideoURL] = Field(
+        default=None, description="Video URL"
+    )
+
+
+class CellContent(BaseModel):
+    """Cell content"""
+
+    Type: CellContentPartType = Field(
         default=None, alias="type", description="Content part type"
     )
     Text: Optional[str] = Field(default=None, alias="text", description="Text content")
@@ -79,8 +102,13 @@ class CellContentPart(BaseModel):
     )
 
 
-# CellContent is a list of CellContentPart
-CellContent = List[CellContentPart]
+class Cell(BaseModel):
+    """Repeated data"""
+
+    CellContent: Optional[List[CellContentPart]] = Field(
+        default=None, description="Cell content"
+    )
+    Text: Optional[str] = Field(default=None, description="Text content")
 
 
 class ModelAgentConfig(BaseModel):
@@ -196,6 +224,10 @@ class CreateEvaTaskRequest(BaseModel):
         title="Dataset Task Configuration",
         description="Dataset task configuration",
     )
+    DatasetVersionID: str = Field(
+        title="Dataset Version ID",
+        description="Dataset version ID",
+    )
 
 
 class CreateEvaTaskResponse(BaseModel):
@@ -210,12 +242,18 @@ class CreateEvaTaskResponse(BaseModel):
     )
 
 
-class ListEvaDatasetConversationsRequest(BaseModel):
+class ListDatasetCasesRequest(BaseModel):
     WorkspaceID: str = Field(
         title="Workspace ID",
         description="Workspace ID",
     )
     DatasetID: str = Field(title="Dataset ID", description="Dataset ID")
+    VersionID: str = Field(title="Dataset Version ID", description="Dataset version ID")
+    UseLatestPublishedVersion: bool = Field(
+        default=False,
+        title="Use Latest Published Version",
+        description="Whether to use the latest published version",
+    )
     PageNumber: Optional[int] = Field(
         default=1,
         title="Page Number",
@@ -298,7 +336,6 @@ class EvaDatasetConversationData(BaseModel):
 class EvaDatasetColumnCell(BaseModel):
     """Evaluation dataset column cell data"""
 
-    ColumnID: str = Field(title="Column ID", description="Evaluation dataset column ID")
     ConversationGroup: List[EvaDatasetConversationData] = Field(
         title="Conversation Group",
         description="Evaluation dataset conversation round data",
@@ -310,23 +347,12 @@ class EvaDatasetColumn(BaseModel):
 
     ID: str = Field(title="Column ID", description="Column ID")
     Name: str = Field(title="Column Name", description="Column name")
-    Description: Optional[str] = Field(
-        default=None, title="Column Description", description="Column description"
-    )
-
-
-class EvaDatasetConversationItem(BaseModel):
-    """Evaluation dataset conversation item"""
-
-    RowID: str = Field(
-        title="Data Row ID", description="Evaluation dataset data row ID"
-    )
-    DataRow: List[EvaDatasetColumnCell] = Field(
-        title="Data Row", description="Evaluation dataset conversation column data"
+    SchemaType: Optional[ColumnSchemaType] = Field(
+        title="Schema Type", description="Schema type"
     )
 
 
-class ListEvaDatasetColumnsRequest(BaseModel):
+class ListColumnsRequest(BaseModel):
     """Get evaluation dataset column list request"""
 
     WorkspaceID: str = Field(
@@ -334,28 +360,7 @@ class ListEvaDatasetColumnsRequest(BaseModel):
         description="Workspace ID",
     )
     DatasetID: str = Field(title="Dataset ID", description="Dataset ID")
-
-
-class ListEvaDatasetColumnsResponse(BaseModel):
-    """Get evaluation dataset column list response"""
-
-    Columns: List[EvaDatasetColumn] = Field(
-        title="Evaluation Dataset Column List",
-        description="Evaluation dataset column list",
-    )
-
-
-class ListEvaDatasetConversationsResponse(BaseModel):
-    """Evaluation dataset conversation list response"""
-
-    Items: List[EvaDatasetConversationItem] = Field(
-        title="Conversation List", description="Evaluation dataset conversation list"
-    )
-    Total: int = Field(
-        title="Total",
-        description="Total number of evaluation dataset conversations",
-        example=100,
-    )
+    VersionID: str = Field(title="Dataset Version ID", description="Dataset version ID")
 
 
 class InferenceResult(BaseModel):
@@ -386,7 +391,7 @@ class CaseData(BaseModel):
 
     model_config = {"extra": "allow"}  # Allow dynamic fields
 
-    def get_column_data(self, column_name: str) -> Optional[CellContent]:
+    def get_column_data(self, column_name: str) -> Optional[Cell]:
         """
         Get data from specified column
 
@@ -401,21 +406,6 @@ class CaseData(BaseModel):
             return self.__pydantic_extra__.get(column_name)
         return None
 
-    def get_column_text(self, column_name: str) -> Optional[str]:
-        """
-        Get text data from specified column
-
-        Args:
-            column_name: Column name
-
-        Returns:
-            Text content of the column, returns None if not exists or not text type
-        """
-        data = self.get_column_data(column_name)
-        if data and hasattr(data, "TextData"):
-            return data.TextData
-        return None
-
     def list_columns(self) -> List[str]:
         """
         Get all column names included in current round
@@ -429,7 +419,7 @@ class CaseData(BaseModel):
         return []
 
     # Dictionary-style operation support
-    def __getitem__(self, column_name: str) -> Optional[CellContent]:
+    def __getitem__(self, column_name: str) -> Optional[Cell]:
         """
         Support case_data[column_name] access pattern
 
@@ -441,7 +431,7 @@ class CaseData(BaseModel):
         """
         return self.get_column_data(column_name)
 
-    def __setitem__(self, column_name: str, value: CellContent):
+    def __setitem__(self, column_name: str, value: Cell):
         """
         Support case_data[column_name] = value assignment pattern
 
@@ -493,16 +483,14 @@ class CaseData(BaseModel):
         """
         return self.list_columns()
 
-    def values(self) -> List[Optional[CellContent]]:
+    def values(self) -> List[Optional[Cell]]:
         """
         Get data from all columns
 
         Returns:
             List of cell content
         """
-        return [
-            self.get_column_data(column_name) for column_name in self.list_columns()
-        ]
+        return [self.get_column_data(column_id) for column_id in self.list_columns()]
 
     def get(self, column_name: str, default=None):
         """
@@ -519,6 +507,17 @@ class CaseData(BaseModel):
         return data if data is not None else default
 
 
+class EvaDatasetConversationItem(BaseModel):
+    """Evaluation dataset conversation item"""
+
+    DatasetCaseID: str = Field(
+        title="Dataset Case ID", description="Evaluation dataset case ID"
+    )
+    RepeatedData: List[Dict[str, Cell]] = Field(
+        title="Repeated Data", description="Evaluation dataset case data"
+    )
+
+
 class EvaConversationStatus(str, Enum):
     """Evaluation conversation status"""
 
@@ -649,9 +648,13 @@ class EvaReportRuleTarget(BaseModel):
     TargetDetail: EvaReportTargetDetail = Field(
         title="Target Details", description="Target details"
     )
-    AvgScore: float = Field(title="Average Score", description="Average score")
-    Percent: float = Field(title="Percentage", description="Percentage")
-    ScoreMap: Dict[str, Any] = Field(title="Score Map", description="Score mapping")
+    AvgScore: Optional[float] = Field(
+        title="Average Score", description="Average score"
+    )
+    Percent: Optional[float] = Field(title="Percentage", description="Percentage")
+    ScoreMap: Optional[Dict[str, Any]] = Field(
+        title="Score Map", description="Score mapping"
+    )
     Duration: int = Field(title="Duration", description="Duration in milliseconds")
     CostTokens: int = Field(
         title="Cost Tokens", description="Number of tokens consumed"
@@ -986,3 +989,23 @@ class GetEvaTaskRequest(BaseModel):
 
 # GetEvaTaskResponse 就是 EvaTaskItem
 GetEvaTaskResponse = EvaTaskItem
+
+
+class ListColumnsResponse(BaseModel):
+    """Get evaluation dataset column list response"""
+
+    Columns: List[EvaDatasetColumn] = Field(
+        title="Evaluation Dataset Column List",
+        description="Evaluation dataset column list",
+    )
+
+
+class ListDatasetCasesResponse(BaseModel):
+    """Evaluation dataset conversation list response"""
+
+    Items: Optional[List[EvaDatasetConversationItem]] = Field(
+        title="Conversation List", description="Evaluation dataset conversation list"
+    )
+    Total: int = Field(
+        title="Total", description="Total number of evaluation dataset conversations"
+    )

{hiagent_api-2.2.0 → hiagent_api-2.3.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "hiagent-api"
-version = "2.2.0"
+version = "2.3.0"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.10"