Skip to content

Commit fae5f75

Browse files
committed
feat(agent): Add usage type to handle python deployment
1 parent 2993581 commit fae5f75

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

libertai_client/commands/agent.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,11 @@ class AgentPythonPackageManager(str, Enum):
5252
pip = "pip"
5353

5454

55+
class AgentUsageType(str, Enum):
56+
fastapi = "fastapi"
57+
python = "python"
58+
59+
5560
@app.command()
5661
async def deploy(
5762
path: Annotated[
@@ -63,6 +68,9 @@ async def deploy(
6368
package_manager: Annotated[
6469
AgentPythonPackageManager, typer.Option(case_sensitive=False, prompt=True)
6570
] = AgentPythonPackageManager.pip.value, # type: ignore
71+
usage_type: Annotated[
72+
AgentUsageType, typer.Option(case_sensitive=False, prompt=True)
73+
] = AgentUsageType.fastapi.value, # type: ignore
6674
):
6775
"""
6876
Deploy or redeploy an agent
@@ -88,6 +96,7 @@ async def deploy(
8896
data.add_field("secret", libertai_config.secret)
8997
data.add_field("python_version", python_version)
9098
data.add_field("package_manager", package_manager.value)
99+
data.add_field("usage_type", usage_type.value)
91100
data.add_field("code", open(agent_zip_path, "rb"), filename="libertai-agent.zip")
92101

93102
async with aiohttp.ClientSession() as session:
@@ -98,6 +107,7 @@ async def deploy(
98107
) as response:
99108
if response.status == 200:
100109
response_data = UpdateAgentResponse(**json.loads(await response.text())) # noqa: F821
110+
# TODO: don't show /docs if deployed in python mode
101111
rich.print(
102112
f"[green]Agent successfully deployed on http://[{response_data.instance_ip}]:8000/docs"
103113
)

0 commit comments

Comments
 (0)