diff options
Diffstat (limited to 'benchmark/agbenchmark/agent_protocol_client/models/artifact.py')
-rw-r--r-- | benchmark/agbenchmark/agent_protocol_client/models/artifact.py | 72 |
1 files changed, 0 insertions, 72 deletions
diff --git a/benchmark/agbenchmark/agent_protocol_client/models/artifact.py b/benchmark/agbenchmark/agent_protocol_client/models/artifact.py deleted file mode 100644 index d2e7c101b..000000000 --- a/benchmark/agbenchmark/agent_protocol_client/models/artifact.py +++ /dev/null @@ -1,72 +0,0 @@ -# coding: utf-8 - - -from __future__ import annotations - -import json -import pprint -import re # noqa: F401 -from typing import Optional - -from pydantic import BaseModel, Field, StrictStr - - -class Artifact(BaseModel): - """ - Artifact that the task has produced. - """ - - artifact_id: StrictStr = Field(..., description="ID of the artifact.") - file_name: StrictStr = Field(..., description="Filename of the artifact.") - relative_path: Optional[StrictStr] = Field( - None, description="Relative path of the artifact in the agent's workspace." - ) - __properties = ["artifact_id", "file_name", "relative_path"] - created_at: StrictStr = Field(..., description="Creation date of the artifact.") - # modified_at: StrictStr = Field(..., description="Modification date of the artifact.") - agent_created: bool = Field(..., description="True if created by the agent") - - class Config: - """Pydantic configuration""" - - allow_population_by_field_name = True - validate_assignment = True - - def to_str(self) -> str: - """Returns the string representation of the model using alias""" - return pprint.pformat(self.dict(by_alias=True)) - - def to_json(self) -> str: - """Returns the JSON representation of the model using alias""" - return json.dumps(self.to_dict()) - - @classmethod - def from_json(cls, json_str: str) -> Artifact: - """Create an instance of Artifact from a JSON string""" - return cls.from_dict(json.loads(json_str)) - - def to_dict(self): - """Returns the dictionary representation of the model using alias""" - _dict = self.dict(by_alias=True, exclude={}, exclude_none=True) - return _dict - - @classmethod - def from_dict(cls, obj: dict) -> Artifact: - """Create an instance of Artifact from a dict""" - if obj is None: - return None - - if not isinstance(obj, dict): - return Artifact.parse_obj(obj) - - _obj = Artifact.parse_obj( - { - "artifact_id": obj.get("artifact_id"), - "file_name": obj.get("file_name"), - "relative_path": obj.get("relative_path"), - "created_at": obj.get("created_at"), - "modified_at": obj.get("modified_at"), - "agent_created": obj.get("agent_created"), - } - ) - return _obj |