truefoundry 0.3.2__py3-none-any.whl → 0.4.0.dev0__py3-none-any.whl

truefoundry/ml/autogen/client/models/agent_app.py

@@ -0,0 +1,118 @@
+# coding: utf-8
+
+"""
+FastAPI
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 0.1.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+from typing import Optional
+
+from truefoundry.ml.autogen.client.models.agent_open_api_tool_with_fqn import (
+    AgentOpenAPIToolWithFQN,
+)
+from truefoundry.ml.autogen.client.models.agent_with_fqn import AgentWithFQN
+from truefoundry.pydantic_v1 import (
+    BaseModel,
+    Field,
+    StrictStr,
+    conlist,
+    constr,
+    validator,
+)
+
+
+class AgentApp(BaseModel):
+    """
+    AgentApp
+    """
+
+    type: Optional[StrictStr] = "agent-app"
+    tools: conlist(AgentOpenAPIToolWithFQN) = Field(...)
+    agents: conlist(AgentWithFQN) = Field(...)
+    root_agent: constr(strict=True, min_length=1) = Field(...)
+    __properties = ["type", "tools", "agents", "root_agent"]
+
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value is None:
+            return value
+
+        if value not in ("agent-app"):
+            raise ValueError("must be one of enum values ('agent-app')")
+        return value
+
+    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) -> AgentApp:
+        """Create an instance of AgentApp 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)
+        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of each item in tools (list)
+        _items = []
+        if self.tools:
+            for _item in self.tools:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict["tools"] = _items
+        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of each item in agents (list)
+        _items = []
+        if self.agents:
+            for _item in self.agents:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict["agents"] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> AgentApp:
+        """Create an instance of AgentApp from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return AgentApp.parse_obj(obj)
+
+        _obj = AgentApp.parse_obj(
+            {
+                "type": obj.get("type") if obj.get("type") is not None else "agent-app",
+                "tools": [
+                    AgentOpenAPIToolWithFQN.from_dict(_item)
+                    for _item in obj.get("tools")
+                ]
+                if obj.get("tools") is not None
+                else None,
+                "agents": [AgentWithFQN.from_dict(_item) for _item in obj.get("agents")]
+                if obj.get("agents") is not None
+                else None,
+                "root_agent": obj.get("root_agent"),
+            }
+        )
+        return _obj

truefoundry/ml/autogen/client/models/agent_open_api_tool.py

@@ -0,0 +1,143 @@
+# coding: utf-8
+
+"""
+FastAPI
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 0.1.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+from typing import Dict, Optional
+
+from truefoundry.ml.autogen.client.models.method import Method
+from truefoundry.ml.autogen.client.models.openapi_spec import OpenapiSpec
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, constr, validator
+
+
+class AgentOpenAPITool(BaseModel):
+    """
+    OpenAPI Tool artifact.  # noqa: E501
+    """
+
+    type: Optional[StrictStr] = "openapi-tool"
+    name: constr(strict=True) = Field(
+        default=..., description="Names cannot repeat accross tools +uiType=Hidden"
+    )
+    description: constr(strict=True, max_length=128, min_length=1) = Field(
+        default=..., description="+sort=10 +uiType=TextArea"
+    )
+    openapi_spec: OpenapiSpec = Field(...)
+    base_url: constr(strict=True) = Field(default=..., description="+sort=30")
+    path: constr(strict=True, min_length=1) = Field(
+        default=...,
+        description="+sort=40 +uiType=MethodPathSelector +label=Method + Path +usage=Select one Method and Path for this tool",
+    )
+    method: Method = Field(default=..., description="+sort=50 +uiType=Hidden")
+    headers: Optional[Dict[str, StrictStr]] = Field(
+        default=None,
+        description='+sort=60 +uiType=KV +uiProps={"allowSecrets":true,"secretConfig":{"enableNew":true,"hideOptions":true}}',
+    )
+    __properties = [
+        "type",
+        "name",
+        "description",
+        "openapi_spec",
+        "base_url",
+        "path",
+        "method",
+        "headers",
+    ]
+
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value is None:
+            return value
+
+        if value not in ("openapi-tool"):
+            raise ValueError("must be one of enum values ('openapi-tool')")
+        return value
+
+    @validator("name")
+    def name_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if not re.match(r"^[a-zA-Z][a-zA-Z0-9\-]{1,30}[a-zA-Z0-9]$", value):
+            raise ValueError(
+                r"must validate the regular expression /^[a-zA-Z][a-zA-Z0-9\-]{1,30}[a-zA-Z0-9]$/"
+            )
+        return value
+
+    @validator("base_url")
+    def base_url_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if not re.match(
+            r"^(https?:\/\/)?(?:www\.)?[-a-zA-Z0-9@:%._\+~#=]{1,256}\.[a-zA-Z0-9()]{1,6}(?:[-a-zA-Z0-9()@:%_\+.~#?&\/=]*)$",
+            value,
+        ):
+            raise ValueError(
+                r"must validate the regular expression /^(https?:\/\/)?(?:www\.)?[-a-zA-Z0-9@:%._\+~#=]{1,256}\.[a-zA-Z0-9()]{1,6}(?:[-a-zA-Z0-9()@:%_\+.~#?&\/=]*)$/"
+            )
+        return value
+
+    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) -> AgentOpenAPITool:
+        """Create an instance of AgentOpenAPITool 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)
+        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of openapi_spec
+        if self.openapi_spec:
+            _dict["openapi_spec"] = self.openapi_spec.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> AgentOpenAPITool:
+        """Create an instance of AgentOpenAPITool from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return AgentOpenAPITool.parse_obj(obj)
+
+        _obj = AgentOpenAPITool.parse_obj(
+            {
+                "type": obj.get("type")
+                if obj.get("type") is not None
+                else "openapi-tool",
+                "name": obj.get("name"),
+                "description": obj.get("description"),
+                "openapi_spec": OpenapiSpec.from_dict(obj.get("openapi_spec"))
+                if obj.get("openapi_spec") is not None
+                else None,
+                "base_url": obj.get("base_url"),
+                "path": obj.get("path"),
+                "method": obj.get("method"),
+                "headers": obj.get("headers"),
+            }
+        )
+        return _obj

truefoundry/ml/autogen/client/models/agent_open_api_tool_with_fqn.py

@@ -0,0 +1,144 @@
+# coding: utf-8
+
+"""
+FastAPI
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 0.1.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+from typing import Dict, Optional
+
+from truefoundry.ml.autogen.client.models.method import Method
+from truefoundry.ml.autogen.client.models.openapi_spec import OpenapiSpec
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, constr, validator
+
+
+class AgentOpenAPIToolWithFQN(BaseModel):
+    """
+    AgentOpenAPIToolWithFQN
+    """
+
+    type: StrictStr = Field(default=..., description="+value=openapi-tool")
+    name: constr(strict=True) = Field(
+        default=..., description="Names cannot repeat accross tools +uiType=Hidden"
+    )
+    description: constr(strict=True, max_length=128, min_length=1) = Field(
+        default=..., description="+sort=10 +uiType=TextArea"
+    )
+    openapi_spec: OpenapiSpec = Field(...)
+    base_url: constr(strict=True) = Field(default=..., description="+sort=30")
+    path: constr(strict=True, min_length=1) = Field(
+        default=...,
+        description="+sort=40 +uiType=MethodPathSelector +label=Method + Path +usage=Select one Method and Path for this tool",
+    )
+    method: Method = Field(default=..., description="+sort=50 +uiType=Hidden")
+    headers: Optional[Dict[str, StrictStr]] = Field(
+        default=None,
+        description='+sort=60 +uiType=KV +uiProps={"allowSecrets":true,"secretConfig":{"enableNew":true,"hideOptions":true}}',
+    )
+    id: StrictStr = Field(...)
+    fqn: StrictStr = Field(...)
+    __properties = [
+        "type",
+        "name",
+        "description",
+        "openapi_spec",
+        "base_url",
+        "path",
+        "method",
+        "headers",
+        "id",
+        "fqn",
+    ]
+
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in ("openapi-tool"):
+            raise ValueError("must be one of enum values ('openapi-tool')")
+        return value
+
+    @validator("name")
+    def name_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if not re.match(r"^[a-zA-Z][a-zA-Z0-9\-]{1,30}[a-zA-Z0-9]$", value):
+            raise ValueError(
+                r"must validate the regular expression /^[a-zA-Z][a-zA-Z0-9\-]{1,30}[a-zA-Z0-9]$/"
+            )
+        return value
+
+    @validator("base_url")
+    def base_url_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if not re.match(
+            r"^(https?:\/\/)?(?:www\.)?[-a-zA-Z0-9@:%._\+~#=]{1,256}\.[a-zA-Z0-9()]{1,6}(?:[-a-zA-Z0-9()@:%_\+.~#?&\/=]*)$",
+            value,
+        ):
+            raise ValueError(
+                r"must validate the regular expression /^(https?:\/\/)?(?:www\.)?[-a-zA-Z0-9@:%._\+~#=]{1,256}\.[a-zA-Z0-9()]{1,6}(?:[-a-zA-Z0-9()@:%_\+.~#?&\/=]*)$/"
+            )
+        return value
+
+    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) -> AgentOpenAPIToolWithFQN:
+        """Create an instance of AgentOpenAPIToolWithFQN 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)
+        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of openapi_spec
+        if self.openapi_spec:
+            _dict["openapi_spec"] = self.openapi_spec.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> AgentOpenAPIToolWithFQN:
+        """Create an instance of AgentOpenAPIToolWithFQN from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return AgentOpenAPIToolWithFQN.parse_obj(obj)
+
+        _obj = AgentOpenAPIToolWithFQN.parse_obj(
+            {
+                "type": obj.get("type"),
+                "name": obj.get("name"),
+                "description": obj.get("description"),
+                "openapi_spec": OpenapiSpec.from_dict(obj.get("openapi_spec"))
+                if obj.get("openapi_spec") is not None
+                else None,
+                "base_url": obj.get("base_url"),
+                "path": obj.get("path"),
+                "method": obj.get("method"),
+                "headers": obj.get("headers"),
+                "id": obj.get("id"),
+                "fqn": obj.get("fqn"),
+            }
+        )
+        return _obj

truefoundry/ml/autogen/client/models/agent_with_fqn.py

@@ -0,0 +1,127 @@
+# coding: utf-8
+
+"""
+FastAPI
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 0.1.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+
+from truefoundry.pydantic_v1 import (
+    BaseModel,
+    Field,
+    StrictStr,
+    conlist,
+    constr,
+    validator,
+)
+
+
+class AgentWithFQN(BaseModel):
+    """
+    AgentWithFQN
+    """
+
+    type: StrictStr = Field(default=..., description="+value=agent")
+    name: constr(strict=True) = Field(
+        default=..., description="Names cannot repeat accross agent +uiType=Hidden"
+    )
+    available_tools: conlist(constr(strict=True, min_length=1)) = Field(
+        default=..., description="+sort=30 +uiType=Hidden"
+    )
+    goal: constr(strict=True, max_length=128, min_length=1) = Field(
+        default=..., description="+sort=10 +uiType=TextArea"
+    )
+    instruction: constr(strict=True, max_length=2620, min_length=1) = Field(
+        default=...,
+        description='`instruction` is the system prompt for now. (2.5 * 1024) +sort=20 +uiType=AgentInstructions +uiProps={"helpText":"Use the syntax ${Tool FQN} to reference a tool, and ${AGENT FQN} to reference another agent"}',
+    )
+    model_id: constr(strict=True, min_length=1) = Field(
+        default=...,
+        description='+sort=40 +uiType=EnabledModelSelector +uiProps={"searchable":true,"modelType":"chat","providerType":"openai"}',
+    )
+    id: StrictStr = Field(...)
+    fqn: StrictStr = Field(...)
+    __properties = [
+        "type",
+        "name",
+        "available_tools",
+        "goal",
+        "instruction",
+        "model_id",
+        "id",
+        "fqn",
+    ]
+
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in ("agent"):
+            raise ValueError("must be one of enum values ('agent')")
+        return value
+
+    @validator("name")
+    def name_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if not re.match(r"^[a-zA-Z][a-zA-Z0-9\-]{1,30}[a-zA-Z0-9]$", value):
+            raise ValueError(
+                r"must validate the regular expression /^[a-zA-Z][a-zA-Z0-9\-]{1,30}[a-zA-Z0-9]$/"
+            )
+        return value
+
+    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) -> AgentWithFQN:
+        """Create an instance of AgentWithFQN 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) -> AgentWithFQN:
+        """Create an instance of AgentWithFQN from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return AgentWithFQN.parse_obj(obj)
+
+        _obj = AgentWithFQN.parse_obj(
+            {
+                "type": obj.get("type"),
+                "name": obj.get("name"),
+                "available_tools": obj.get("available_tools"),
+                "goal": obj.get("goal"),
+                "instruction": obj.get("instruction"),
+                "model_id": obj.get("model_id"),
+                "id": obj.get("id"),
+                "fqn": obj.get("fqn"),
+            }
+        )
+        return _obj

truefoundry/ml/autogen/client/models/artifact_dto.py

@@ -0,0 +1,115 @@
+# coding: utf-8
+
+"""
+FastAPI
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 0.1.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+from datetime import datetime
+from typing import Optional
+
+from truefoundry.ml.autogen.client.models.artifact_type import ArtifactType
+from truefoundry.ml.autogen.client.models.artifact_version_dto import ArtifactVersionDto
+from truefoundry.pydantic_v1 import BaseModel, StrictInt, StrictStr, conlist
+
+
+class ArtifactDto(BaseModel):
+    """
+    ArtifactDto
+    """
+
+    id: Optional[StrictStr] = None
+    experiment_id: Optional[StrictStr] = None
+    type: Optional[ArtifactType] = None
+    name: Optional[StrictStr] = None
+    fqn: Optional[StrictStr] = None
+    description: Optional[StrictStr] = None
+    created_by: Optional[StrictStr] = None
+    created_at: Optional[datetime] = None
+    updated_at: Optional[datetime] = None
+    artifact_storage_root: Optional[StrictStr] = None
+    latest_version: Optional[ArtifactVersionDto] = None
+    run_steps: Optional[conlist(StrictInt)] = None
+    __properties = [
+        "id",
+        "experiment_id",
+        "type",
+        "name",
+        "fqn",
+        "description",
+        "created_by",
+        "created_at",
+        "updated_at",
+        "artifact_storage_root",
+        "latest_version",
+        "run_steps",
+    ]
+
+    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) -> ArtifactDto:
+        """Create an instance of ArtifactDto 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)
+        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of latest_version
+        if self.latest_version:
+            _dict["latest_version"] = self.latest_version.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> ArtifactDto:
+        """Create an instance of ArtifactDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return ArtifactDto.parse_obj(obj)
+
+        _obj = ArtifactDto.parse_obj(
+            {
+                "id": obj.get("id"),
+                "experiment_id": obj.get("experiment_id"),
+                "type": obj.get("type"),
+                "name": obj.get("name"),
+                "fqn": obj.get("fqn"),
+                "description": obj.get("description"),
+                "created_by": obj.get("created_by"),
+                "created_at": obj.get("created_at"),
+                "updated_at": obj.get("updated_at"),
+                "artifact_storage_root": obj.get("artifact_storage_root"),
+                "latest_version": ArtifactVersionDto.from_dict(
+                    obj.get("latest_version")
+                )
+                if obj.get("latest_version") is not None
+                else None,
+                "run_steps": obj.get("run_steps"),
+            }
+        )
+        return _obj