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

ApiSurface -> Api

Browse source
This commit is contained in:
Ashwin Bharambe 2024-08-05 12:44:56 -07:00
parent 7890921e5c
commit 125fdb1b2a
13 changed files with 76 additions and 87 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_toolchain/agentic_system/adapters.py
View file

@ -6,13 +6,13 @@
from typing import List
from llama_toolchain.distribution.datatypes import Adapter, ApiSurface, SourceAdapter
from llama_toolchain.distribution.datatypes import Adapter, Api, SourceAdapter
def available_agentic_system_adapters() -> List[Adapter]:
return [
SourceAdapter(
api_surface=ApiSurface.agentic_system,
api=Api.agentic_system,
adapter_id="meta-reference",
pip_packages=[
"codeshield",
@ -22,8 +22,8 @@ def available_agentic_system_adapters() -> List[Adapter]:
module="llama_toolchain.agentic_system.agentic_system",
config_class="llama_toolchain.agentic_system.config.AgenticSystemConfig",
adapter_dependencies=[
ApiSurface.inference,
ApiSurface.safety,
Api.inference,
Api.safety,
],
),
]

10
llama_toolchain/agentic_system/agentic_system.py
View file

@ -7,7 +7,7 @@
from llama_toolchain.agentic_system.api import AgenticSystem
from llama_toolchain.distribution.datatypes import Adapter, ApiSurface
from llama_toolchain.distribution.datatypes import Adapter, Api
from llama_toolchain.inference.api import Inference
from llama_toolchain.safety.api import Safety
@ -44,16 +44,14 @@ logger = logging.getLogger()
logger.setLevel(logging.INFO)
async def get_adapter_impl(
config: AgenticSystemConfig, deps: Dict[ApiSurface, Adapter]
):
async def get_adapter_impl(config: AgenticSystemConfig, deps: Dict[Api, Adapter]):
assert isinstance(
config, AgenticSystemConfig
), f"Unexpected config type: {type(config)}"
impl = MetaReferenceAgenticSystemImpl(
deps[ApiSurface.inference],
deps[ApiSurface.safety],
deps[Api.inference],
deps[Api.safety],
)
await impl.initialize()
return impl