wol
03f2c49b7e
Some checks failed
Remove old artifacts / remove-old-artifacts (push) Has been cancelled
54 lines
2.0 KiB
Python
54 lines
2.0 KiB
Python
import uuid
|
|
from typing import Any, Dict
|
|
from camel.toolkits import WebDeployToolkit as BaseWebDeployToolkit
|
|
|
|
from app.service.task import Agents
|
|
from app.utils.listen.toolkit_listen import listen_toolkit
|
|
from app.utils.toolkit.abstract_toolkit import AbstractToolkit
|
|
|
|
|
|
class WebDeployToolkit(BaseWebDeployToolkit, AbstractToolkit):
|
|
agent_name: str = Agents.developer_agent
|
|
|
|
def __init__(
|
|
self,
|
|
api_task_id: str,
|
|
timeout: float | None = None,
|
|
add_branding_tag: bool = True,
|
|
logo_path: str = "../../../../public/favicon.png",
|
|
tag_text: str = "Created by Eigent",
|
|
tag_url: str = "https://main.eigent.ai/",
|
|
remote_server_ip: str | None = "space.eigent.ai",
|
|
remote_server_port: int = 8080,
|
|
):
|
|
self.api_task_id = api_task_id
|
|
super().__init__(timeout, add_branding_tag, logo_path, tag_text, tag_url, remote_server_ip, remote_server_port)
|
|
|
|
@listen_toolkit(BaseWebDeployToolkit.deploy_html_content)
|
|
def deploy_html_content(
|
|
self,
|
|
html_content: str | None = None,
|
|
html_file_path: str | None = None,
|
|
file_name: str = "index.html",
|
|
port: int = 8080,
|
|
domain: str | None = None,
|
|
subdirectory: str | None = None,
|
|
) -> Dict[str, Any]:
|
|
subdirectory = str(uuid.uuid4())
|
|
return super().deploy_html_content(html_content, html_file_path, file_name, port, domain, subdirectory)
|
|
|
|
@listen_toolkit(BaseWebDeployToolkit.deploy_folder)
|
|
def deploy_folder(
|
|
self, folder_path: str, port: int = 8080, domain: str | None = None, subdirectory: str | None = None
|
|
) -> Dict[str, Any]:
|
|
subdirectory = str(uuid.uuid4())
|
|
return super().deploy_folder(folder_path, port, domain, subdirectory)
|
|
|
|
@listen_toolkit(BaseWebDeployToolkit.stop_server)
|
|
def stop_server(self, port: int) -> Dict[str, Any]:
|
|
return super().stop_server(port)
|
|
|
|
@listen_toolkit(BaseWebDeployToolkit.list_running_servers)
|
|
def list_running_servers(self) -> Dict[str, Any]:
|
|
return super().list_running_servers()
|