From a2470aae1186b0990e6fb2d7345c282db3480de0 Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Fri, 30 Aug 2024 10:40:17 -0700 Subject: [PATCH] Fix api dependencies not getting added to configuration --- llama_toolchain/core/package.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/llama_toolchain/core/package.py b/llama_toolchain/core/package.py index 7ef68bcc9..c6c42c0bf 100644 --- a/llama_toolchain/core/package.py +++ b/llama_toolchain/core/package.py @@ -112,6 +112,9 @@ def build_package( package_deps.pip_packages.extend(deps.pip_packages) stub_config[api.value] = {"provider_id": api_input.provider} + for dep_api, dep_spec in api_input.dependencies.items(): + if dep_api not in stub_config: + stub_config[dep_api] = {"provider_id": dep_spec.provider_id} if package_file.exists(): cprint(