chore: reconstruct cli

Author: yaozheng-fang

config.yaml.full

@@ -98,7 +98,7 @@ database:
 
 
 # [optional] for prompt optimization in cli/app
-agent_pilot:
+prompt_pilot:
   api_key:
 
 

pyproject.toml

@@ -21,11 +21,10 @@ dependencies = [
     "opentelemetry-exporter-otlp>=1.35.0",
     "opentelemetry-instrumentation-logging>=0.56b0",
     "wrapt>=1.17.2", # For patching built-in functions
-    "typer>=0.16.0", # For command-line implementation
 ]
 
 [project.scripts]
-veadk = "veadk.cli.main:app"
+veadk = "veadk.cli.cli:veadk"
 
 [project.optional-dependencies]
 database = [

veadk/cli/cli.py

@@ -0,0 +1,19 @@
+import click
+
+from veadk.cli.cli_deploy import deploy
+from veadk.cli.cli_init import init
+from veadk.cli.cli_prompt import prompt
+
+
+@click.group()
+def veadk():
+    """Volcengine ADK command line tools"""
+    pass
+
+
+veadk.add_command(deploy)
+veadk.add_command(prompt)
+veadk.add_command(init)
+
+if __name__ == "__main__":
+    veadk()

veadk/cli/cli_deploy.py

@@ -0,0 +1,32 @@
+import click
+
+
+@click.command()
+@click.option(
+    "--access-key",
+    default=None,
+    help="Volcengine access key",
+)
+@click.option(
+    "--secret-key",
+    default=None,
+    help="Volcengine secret key",
+)
+@click.option("--name", help="Expected Volcengine FaaS application name")
+@click.option("--path", default=".", help="Local project path")
+def deploy(access_key: str, secret_key: str, name: str, path: str) -> None:
+    """Deploy a user project to Volcengine FaaS application."""
+    from pathlib import Path
+
+    from veadk.config import getenv
+    from veadk.integrations.ve_faas.ve_faas import VeFaaS
+
+    if not access_key:
+        access_key = getenv("VOLCENGINE_ACCESS_KEY")
+    if not secret_key:
+        secret_key = getenv("VOLCENGINE_SECRET_KEY")
+
+    user_proj_abs_path = Path(path).resolve()
+
+    ve_faas = VeFaaS(access_key=access_key, secret_key=secret_key)
+    ve_faas.deploy(name=name, path=str(user_proj_abs_path))

veadk/cli/cli_init.py

@@ -0,0 +1,96 @@
+from typing import Any
+
+import click
+
+
+def _set_variable_in_file(file_path: str, setting_values: dict):
+    import ast
+
+    with open(file_path, "r", encoding="utf-8") as f:
+        source_code = f.read()
+
+    tree = ast.parse(source_code)
+
+    class VariableTransformer(ast.NodeTransformer):
+        def visit_Assign(self, node: ast.Assign):
+            for target in node.targets:
+                if isinstance(target, ast.Name) and target.id in setting_values:
+                    node.value = ast.Constant(value=setting_values[target.id])
+            return node
+
+    transformer = VariableTransformer()
+    new_tree = transformer.visit(tree)
+    ast.fix_missing_locations(new_tree)
+    new_source_code = ast.unparse(new_tree)
+
+    with open(file_path, "w", encoding="utf-8") as f:
+        f.write(new_source_code)
+
+    click.echo("Your project has beed created.")
+
+
+def _render_prompts() -> dict[str, Any]:
+    vefaas_application_name = click.prompt(
+        "Volcengine FaaS application name", default="veadk-cloud-agent"
+    )
+
+    gateway_name = click.prompt(
+        "Volcengine gateway instance name", default="", show_default=True
+    )
+
+    gateway_service_name = click.prompt(
+        "Volcengine gateway service name", default="", show_default=True
+    )
+
+    gateway_upstream_name = click.prompt(
+        "Volcengine gateway upstream name", default="", show_default=True
+    )
+
+    deploy_mode_options = {
+        "1": "A2A/MCP Server",
+        "2": "VeADK Studio",
+        "3": "VeADK Web / Google ADK Web",
+    }
+
+    click.echo("Choose a deploy mode:")
+    for key, value in deploy_mode_options.items():
+        click.echo(f"  {key}. {value}")
+
+    deploy_mode = click.prompt(
+        "Enter your choice", type=click.Choice(deploy_mode_options.keys())
+    )
+
+    return {
+        "VEFAAS_APPLICATION_NAME": vefaas_application_name,
+        "GATEWAY_NAME": gateway_name,
+        "GATEWAY_SERVICE_NAME": gateway_service_name,
+        "GATEWAY_UPSTREAM_NAME": gateway_upstream_name,
+        "USE_STUDIO": deploy_mode == deploy_mode_options["2"],
+        "USE_ADK_WEB": deploy_mode == deploy_mode_options["3"],
+    }
+
+
+@click.command()
+def init() -> None:
+    """Init a veadk project that can be deployed to Volcengine VeFaaS."""
+    import shutil
+    from pathlib import Path
+
+    import veadk.integrations.ve_faas as vefaas
+
+    cwd = Path.cwd()
+    local_dir_name = click.prompt("Directory name", default="veadk-cloud-proj")
+    target_dir_path = cwd / local_dir_name
+
+    if target_dir_path.exists():
+        click.confirm(
+            f"Directory '{target_dir_path}' already exists, do you want to overwrite it",
+            abort=True,
+        )
+        shutil.rmtree(target_dir_path)
+
+    setting_values = _render_prompts()
+
+    template_dir_path = Path(vefaas.__file__).parent / "template"
+    shutil.copytree(template_dir_path, target_dir_path)
+    _set_variable_in_file(target_dir_path / "deploy.py", setting_values)

veadk/cli/cli_prompt.py

@@ -0,0 +1,50 @@
+import click
+
+
+@click.command()
+@click.option(
+    "--path", default=".", help="Agent file path with global variable `agent=...`"
+)
+@click.option("--feedback", default=None, help="Suggestions for prompt optimization")
+@click.option("--api-key", default=None, help="API Key of PromptPilot")
+@click.option(
+    "--model-name",
+    default="doubao-1.5-pro-32k-250115",
+    help="Model name for prompt optimization",
+)
+def prompt(path: str, feedback: str, api_key: str, model_name: str) -> None:
+    """Optimize agent system prompt from a local file."""
+    from pathlib import Path
+
+    from veadk.agent import Agent
+    from veadk.config import getenv
+    from veadk.integrations.ve_prompt_pilot.ve_prompt_pilot import VePromptPilot
+    from veadk.utils.misc import load_module_from_file
+
+    module_name = "agents_for_prompt_pilot"
+    module_abs_path = Path(path).resolve()
+
+    module = load_module_from_file(
+        module_name=module_name, file_path=str(module_abs_path)
+    )
+
+    # get all global variables from module
+    globals_in_module = vars(module)
+
+    agents = []
+    for global_variable_name, global_variable_value in globals_in_module.items():
+        if isinstance(global_variable_value, Agent):
+            agent = global_variable_value
+            agents.append(agent)
+
+    if len(agents) > 0:
+        click.echo(f"Found {len(agents)} agents in {module_abs_path}")
+
+        if not api_key:
+            api_key = getenv("PROMPT_PILOT_API_KEY")
+        ve_prompt_pilot = VePromptPilot(api_key)
+        ve_prompt_pilot.optimize(
+            agents=agents, feedback=feedback, model_name=model_name
+        )
+    else:
+        click.echo(f"No agents found in {module_abs_path}")

veadk/cli/cli_web.py

@@ -0,0 +1,2 @@
+def web():
+    pass