Skip to content

Commit efdec24

Browse files
committed
update init
1 parent d05f3a5 commit efdec24

File tree

15 files changed

+69
-497
lines changed

15 files changed

+69
-497
lines changed

veadk/cli/cli_init.py

Lines changed: 58 additions & 45 deletions
Original file line numberDiff line numberDiff line change
@@ -24,12 +24,7 @@
2424
)
2525

2626

27-
def _render_prompts(attributes: list[str]) -> dict[str, Any]:
28-
for attr in attributes:
29-
pass
30-
31-
32-
def _render_prompts_legacy() -> dict[str, Any]:
27+
def _render_prompts() -> dict[str, Any]:
3328
vefaas_application_name = click.prompt(
3429
"Volcengine FaaS application name", default="veadk-cloud-agent"
3530
)
@@ -86,48 +81,66 @@ def init(
8681
from cookiecutter.main import cookiecutter
8782

8883
import veadk.integrations.ve_faas as vefaas
89-
90-
# copy to local
91-
# generate deploy.yaml
84+
from veadk.configs.deploy_config import (
85+
VeDeployConfig,
86+
_VeADKConfig,
87+
_VeApigConfig,
88+
_VeFaaSConfig,
89+
)
9290

9391
if vefaas_template_type == "web_template":
9492
click.echo(
9593
"Welcome use VeADK to create your project. We will generate a `simple-blog` web application for you."
9694
)
97-
cwd = Path.cwd()
98-
local_dir_name = click.prompt(
99-
"Local directory name", default="veadk-cloud-proj"
95+
else:
96+
click.echo(
97+
"Welcome use VeADK to create your project. We will generate a `weather-reporter` application for you."
10098
)
101-
target_dir_path = cwd / local_dir_name
102-
103-
if target_dir_path.exists():
104-
click.confirm(
105-
f"Directory '{target_dir_path}' already exists, do you want to overwrite it",
106-
abort=True,
107-
)
108-
shutil.rmtree(target_dir_path)
109-
110-
settings = _render_prompts_legacy()
111-
settings["local_dir_name"] = local_dir_name
112-
113-
if not vefaas_template_type:
114-
vefaas_template_type = "template"
115-
116-
template_dir_path = Path(vefaas.__file__).parent / vefaas_template_type
117-
118-
cookiecutter(
119-
template=str(template_dir_path),
120-
output_dir=str(cwd),
121-
extra_context=settings,
122-
no_input=True,
123-
)
124-
125-
click.echo(f"Template project has been generated at {target_dir_path}")
126-
click.echo(f"Edit {target_dir_path / 'src/'} to define your agents")
127-
click.echo(
128-
f"Edit {target_dir_path / 'deploy.py'} to define your deployment attributes"
129-
)
130-
click.echo(
131-
"Run python `deploy.py` for deployment on Volcengine FaaS platform."
132-
)
133-
return
99+
100+
cwd = Path.cwd()
101+
local_dir_name = click.prompt("Local directory name", default="veadk-cloud-proj")
102+
target_dir_path = cwd / local_dir_name
103+
104+
if target_dir_path.exists():
105+
click.confirm(
106+
f"Directory '{target_dir_path}' already exists, do you want to overwrite it",
107+
abort=True,
108+
)
109+
shutil.rmtree(target_dir_path)
110+
111+
settings = _render_prompts()
112+
settings["local_dir_name"] = local_dir_name
113+
114+
if not vefaas_template_type:
115+
vefaas_template_type = "template"
116+
117+
template_dir_path = Path(vefaas.__file__).parent / vefaas_template_type
118+
119+
cookiecutter(
120+
template=str(template_dir_path),
121+
output_dir=str(cwd),
122+
extra_context=settings,
123+
no_input=True,
124+
)
125+
126+
ve_deploy_config = VeDeployConfig(
127+
vefaas=_VeFaaSConfig(
128+
region="cn-beijing",
129+
application_name=settings["vefaas_application_name"],
130+
),
131+
veapig=_VeApigConfig(
132+
instance_name=settings["veapig_instance_name"],
133+
service_name=settings["veapig_service_name"],
134+
upstream_name=settings["veapig_upstream_name"],
135+
),
136+
veadk=_VeADKConfig(
137+
deploy_mode="WEB" if settings["use_adk_web"] else "A2A/MCP",
138+
entrypoint_agent="weather_reporter.agent:root_agent",
139+
),
140+
)
141+
142+
ve_deploy_config.to_yaml_file(target_dir_path / "deploy.yaml")
143+
144+
click.echo(f"Template project has been generated at {target_dir_path}")
145+
146+
click.echo("Run `veadk deploy` for deployment on Volcengine FaaS platform.")

veadk/configs/deploy_config.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -8,11 +8,11 @@
88

99

1010
class _VeFaaSConfig(BaseModel):
11-
region: Optional[str] = None
11+
region: Optional[str] = ""
1212

13-
application_name: Optional[str] = None
13+
application_name: Optional[str] = ""
1414

15-
function_name: Optional[str] = None
15+
function_name: Optional[str] = ""
1616

1717
enable_key_auth: bool = True
1818

@@ -22,15 +22,15 @@ class _VeFaaSConfig(BaseModel):
2222

2323

2424
class _VeApigConfig(BaseModel):
25-
instance_name: Optional[str] = None
25+
instance_name: Optional[str] = ""
2626

27-
service_name: Optional[str] = None
27+
service_name: Optional[str] = ""
2828

29-
upstream_name: Optional[str] = None
29+
upstream_name: Optional[str] = ""
3030

3131

3232
class _VeADKConfig(BaseModel):
33-
entrypoint_agent: Optional[str] = None
33+
entrypoint_agent: Optional[str] = ""
3434

3535
envs: list[dict[str, Any]] = Field(
3636
default_factory=lambda: [
Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,4 @@
11
{
22
"local_dir_name": "veadk_vefaas_proj",
3-
"app_name": "weather_report",
4-
"agent_module_name": "weather_agent",
5-
"short_term_memory_backend": "local",
6-
"vefaas_application_name": "weather-reporter",
7-
"veapig_instance_name": "",
8-
"veapig_service_name": "",
9-
"veapig_upstream_name": "",
10-
"use_adk_web": false,
113
"veadk_version": ""
124
}

veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/__init__.py

Lines changed: 0 additions & 13 deletions
This file was deleted.

veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/clean.py

Lines changed: 0 additions & 23 deletions
This file was deleted.
Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,3 @@
1-
model:
2-
agent:
3-
provider: openai
4-
name: doubao-1-5-pro-256k-250115
5-
api_base: https://ark.cn-beijing.volces.com/api/v3/
6-
api_key:
1+
volcengine:
2+
access_key:
3+
secret_key:

veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/deploy.py

Lines changed: 0 additions & 103 deletions
This file was deleted.
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
veadk-python=={{ cookiecutter.veadk_version }}

veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/src/__init__.py

Lines changed: 0 additions & 13 deletions
This file was deleted.

veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/src/agent.py

Lines changed: 0 additions & 25 deletions
This file was deleted.

0 commit comments

Comments
 (0)