From 9b2b7d4e75a67cd722cf2041ab5d96c51e9c1eac Mon Sep 17 00:00:00 2001 From: Jakub Novak Date: Tue, 8 Aug 2023 15:25:39 +0200 Subject: [PATCH] Add missing file --- .../agent_protocol_client/models/artifact.py | 77 +++++++++++++++++++ agent_client/python/pyproject.toml | 2 +- 2 files changed, 78 insertions(+), 1 deletion(-) create mode 100644 agent_client/python/agent_protocol_client/models/artifact.py diff --git a/agent_client/python/agent_protocol_client/models/artifact.py b/agent_client/python/agent_protocol_client/models/artifact.py new file mode 100644 index 0000000..75f6e9a --- /dev/null +++ b/agent_client/python/agent_protocol_client/models/artifact.py @@ -0,0 +1,77 @@ +# coding: utf-8 + +""" + Agent Communication Protocol + + Specification of the API protocol for communication with an agent. # noqa: E501 + + The version of the OpenAPI document: v0.2 + Generated by OpenAPI Generator (https://openapi-generator.tech) + + Do not edit the class manually. +""" + + +from __future__ import annotations +import pprint +import re # noqa: F401 +import json + + +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"] + + 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"), + } + ) + return _obj diff --git a/agent_client/python/pyproject.toml b/agent_client/python/pyproject.toml index 95d5b66..bebbc13 100644 --- a/agent_client/python/pyproject.toml +++ b/agent_client/python/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "agent-protocol-client" -version = "0.2.1" +version = "0.2.2" description = "Agent Communication Protocol Client" authors = ["e2b "] license = "MIT"