mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-19 06:54:22 +01:00
90 lines
2.5 KiB
Python
90 lines
2.5 KiB
Python
# coding: utf-8
|
|
|
|
"""
|
|
Agent Communication Protocol
|
|
|
|
Specification of the API protocol for communication with an agent. # noqa: E501
|
|
|
|
The version of the OpenAPI document: v1
|
|
Generated by OpenAPI Generator (https://openapi-generator.tech)
|
|
|
|
Do not edit the class manually.
|
|
"""
|
|
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import pprint
|
|
import re # noqa: F401
|
|
from typing import Any, Optional
|
|
|
|
from pydantic import BaseModel, Field, StrictBool, conlist
|
|
|
|
|
|
class StepResult(BaseModel):
|
|
"""
|
|
Result of the task step.
|
|
"""
|
|
|
|
output: Optional[Any] = Field(
|
|
None,
|
|
description="Output that the task step has produced. Any value is allowed.",
|
|
)
|
|
artifacts: conlist(Any) = Field(
|
|
..., description="A list of artifacts that the step has produced."
|
|
)
|
|
is_last: Optional[StrictBool] = Field(
|
|
False, description="Whether this is the last step in the task."
|
|
)
|
|
__properties = ["output", "artifacts", "is_last"]
|
|
|
|
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) -> StepResult:
|
|
"""Create an instance of StepResult 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)
|
|
# set to None if output (nullable) is None
|
|
# and __fields_set__ contains the field
|
|
if self.output is None and "output" in self.__fields_set__:
|
|
_dict["output"] = None
|
|
|
|
return _dict
|
|
|
|
@classmethod
|
|
def from_dict(cls, obj: dict) -> StepResult:
|
|
"""Create an instance of StepResult from a dict"""
|
|
if obj is None:
|
|
return None
|
|
|
|
if not isinstance(obj, dict):
|
|
return StepResult.parse_obj(obj)
|
|
|
|
_obj = StepResult.parse_obj(
|
|
{
|
|
"output": obj.get("output"),
|
|
"artifacts": obj.get("artifacts"),
|
|
"is_last": obj.get("is_last")
|
|
if obj.get("is_last") is not None
|
|
else False,
|
|
}
|
|
)
|
|
return _obj
|