Spaces:
Runtime error
Runtime error
File size: 3,509 Bytes
b115d50 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 |
from __future__ import annotations
import json
from typing import Any, Dict, List, Optional, Type
from pydantic import BaseModel, Field
from steamship.base import Task
from steamship.base.client import Client
from steamship.base.model import CamelModel
from steamship.base.request import Request
from steamship.base.response import Response
from steamship.data.plugin import HostingMemory, HostingTimeout
class CreatePluginVersionRequest(Request):
plugin_id: str = None
handle: str = None
hosting_memory: Optional[HostingMemory] = None
hosting_timeout: Optional[HostingTimeout] = None
hosting_handler: str = None
is_public: bool = None
is_default: bool = None
type: str = "file"
# Note: this is a Dict[str, Any] but should be transmitted to the Engine as a JSON string
config_template: str = None
class ListPluginVersionsRequest(Request):
handle: str
plugin_id: str
class ListPluginVersionsResponse(Response):
plugins: List[PluginVersion]
class PluginVersion(CamelModel):
client: Client = Field(None, exclude=True)
id: str = None
plugin_id: str = None
handle: str = None
hosting_memory: Optional[HostingMemory] = None
hosting_timeout: Optional[HostingTimeout] = None
hosting_handler: str = None
is_public: bool = None
is_default: bool = None
config_template: Dict[str, Any] = None
@classmethod
def parse_obj(cls: Type[BaseModel], obj: Any) -> BaseModel:
# TODO (enias): This needs to be solved at the engine side
obj = obj["pluginVersion"] if "pluginVersion" in obj else obj
return super().parse_obj(obj)
@staticmethod
def create(
client: Client,
handle: str,
plugin_id: str = None,
filename: str = None,
filebytes: bytes = None,
hosting_memory: Optional[HostingMemory] = None,
hosting_timeout: Optional[HostingTimeout] = None,
hosting_handler: str = None,
is_public: bool = None,
is_default: bool = None,
config_template: Dict[str, Any] = None,
) -> Task[PluginVersion]:
if filename is None and filebytes is None:
raise Exception("Either filename or filebytes must be provided.")
if filename is not None and filebytes is not None:
raise Exception("Only either filename or filebytes should be provided.")
if filename is not None:
with open(filename, "rb") as f:
filebytes = f.read()
req = CreatePluginVersionRequest(
handle=handle,
plugin_id=plugin_id,
hosting_memory=hosting_memory,
hosting_timeout=hosting_timeout,
hosting_handler=hosting_handler,
is_public=is_public,
is_default=is_default,
config_template=json.dumps(config_template or {}),
)
task = client.post(
"plugin/version/create",
payload=req,
file=("plugin.zip", filebytes, "multipart/form-data"),
expect=PluginVersion,
)
task.wait()
return task.output
@staticmethod
def list(
client: Client, plugin_id: str = None, handle: str = None, public: bool = True
) -> ListPluginVersionsResponse:
return client.post(
f"plugin/version/{'public' if public else 'private'}",
ListPluginVersionsRequest(handle=handle, plugin_id=plugin_id),
expect=ListPluginVersionsResponse,
)
|