phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-04 20:14:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

api build works for conda now

Browse source
This commit is contained in:
Ashwin Bharambe 2024-08-27 21:40:43 -07:00
parent c4fe72c3a3
commit 2076d2b6db
5 changed files with 28 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

19
llama_toolchain/cli/api/build.py
View file

@ -5,6 +5,7 @@
# the root directory of this source tree.
import argparse
import json
import os
import random
import string
@ -137,6 +138,7 @@ class ApiBuild(Subcommand):
def _run_api_build_command(self, args: argparse.Namespace) -> None:
from llama_toolchain.common.exec import run_with_pty
from llama_toolchain.distribution.distribution import api_providers
from llama_toolchain.common.serialize import EnumEncoder
os.makedirs(BUILDS_BASE_DIR, exist_ok=True)
all_providers = api_providers()
@ -174,7 +176,7 @@ class ApiBuild(Subcommand):
}
with open(package_file, "w") as f:
c = PackageConfig(
built_at=datetime.now(),
built_at=str(datetime.now()),
package_name=package_name,
docker_image=(
package_name if args.type == BuildType.container.value else None
@ -184,7 +186,8 @@ class ApiBuild(Subcommand):
),
providers=stub_config,
)
f.write(yaml.dump(c.dict(), sort_keys=False))
to_write = json.loads(json.dumps(c.dict(), cls=EnumEncoder))
f.write(yaml.dump(to_write, sort_keys=False))
if args.type == BuildType.container.value:
script = pkg_resources.resource_filename(
@ -209,10 +212,14 @@ class ApiBuild(Subcommand):
]
return_code = run_with_pty(args)
assert return_code == 0, cprint(
f"Failed to build target {package_name}", color="red"
)
if return_code != 0:
cprint(
f"Failed to build target {package_name} with return code {return_code}",
color="red",
)
return
cprint(
f"Target `{target_name}` built with configuration at {str(package_file)}",
f"Target `{package_name}` built with configuration at {str(package_file)}",
color="green",
)

5
llama_toolchain/cli/api/configure.py
View file

@ -81,13 +81,12 @@ def configure_llama_provider(config_file: Path) -> None:
provider_spec = providers[provider_id]
cprint(f"Configuring API surface: {api}", "white", attrs=["bold"])
config_type = instantiate_class_type(provider_spec.config_class)
print(f"Config type: {config_type}")
provider_config = prompt_for_config(
config_type,
)
print("")
provider_configs[api.value] = {
provider_configs[api] = {
"provider_id": provider_id,
**provider_config.dict(),
}
@ -97,4 +96,4 @@ def configure_llama_provider(config_file: Path) -> None:
to_write = json.loads(json.dumps(config.dict(), cls=EnumEncoder))
fp.write(yaml.dump(to_write, sort_keys=False))
print(f"YAML configuration has been written to {config_path}")
print(f"YAML configuration has been written to {config_file}")