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

enable import of subcommands from llama-agentic-system

Browse source
This commit is contained in:
Hardik Shah 2024-07-21 12:54:38 -07:00
parent 67f0510edd
commit 6bcd826b32
1 changed files with 11 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
toolchain/cli/llama.py
View file

@ -23,6 +23,17 @@ class LlamaCLIParser:
Download.create(subparsers) Download.create(subparsers)
InferenceParser.create(subparsers) InferenceParser.create(subparsers)
# Import sub-commands from agentic_system if they exist
try:
from agentic_system.cli.subcommand_modules import (
SUBCOMMAND_MODULES,
)
for module in SUBCOMMAND_MODULES:
module.create(subparsers)
except ImportError:
pass
def parse_args(self) -> argparse.Namespace: def parse_args(self) -> argparse.Namespace:
return self.parser.parse_args() return self.parser.parse_args()