Skip to content

Commit

Permalink
Add missing file
Browse files Browse the repository at this point in the history
  • Loading branch information
jakubno committed Aug 8, 2023
1 parent 2f4e7cb commit 9b2b7d4
Show file tree
Hide file tree
Showing 2 changed files with 78 additions and 1 deletion.
77 changes: 77 additions & 0 deletions agent_client/python/agent_protocol_client/models/artifact.py
Original file line number Diff line number Diff line change
@@ -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
2 changes: 1 addition & 1 deletion agent_client/python/pyproject.toml
Original file line number Diff line number Diff line change
@@ -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 <[email protected]>"]
license = "MIT"
Expand Down

0 comments on commit 9b2b7d4

Please sign in to comment.