mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-07-29 15:23:51 +00:00
Add more sample content
This commit is contained in:
parent
6cbc387851
commit
ef042e1605
21 changed files with 198 additions and 1 deletions
|
@ -47,6 +47,8 @@ class StackListProviders(Subcommand):
|
||||||
|
|
||||||
rows = []
|
rows = []
|
||||||
for spec in providers_for_api.values():
|
for spec in providers_for_api.values():
|
||||||
|
if spec.provider_id == "sample":
|
||||||
|
continue
|
||||||
rows.append(
|
rows.append(
|
||||||
[
|
[
|
||||||
spec.provider_id,
|
spec.provider_id,
|
||||||
|
|
|
@ -3,3 +3,13 @@
|
||||||
#
|
#
|
||||||
# This source code is licensed under the terms described in the LICENSE file in
|
# This source code is licensed under the terms described in the LICENSE file in
|
||||||
# the root directory of this source tree.
|
# the root directory of this source tree.
|
||||||
|
|
||||||
|
from .config import SampleConfig
|
||||||
|
|
||||||
|
|
||||||
|
async def get_adapter_impl(config: SampleConfig, _deps) -> Any:
|
||||||
|
from .sample import SampleAgentsImpl
|
||||||
|
|
||||||
|
impl = SampleAgentsImpl(config)
|
||||||
|
await impl.initialize()
|
||||||
|
return impl
|
||||||
|
|
|
@ -3,3 +3,10 @@
|
||||||
#
|
#
|
||||||
# This source code is licensed under the terms described in the LICENSE file in
|
# This source code is licensed under the terms described in the LICENSE file in
|
||||||
# the root directory of this source tree.
|
# the root directory of this source tree.
|
||||||
|
|
||||||
|
from pydantic import BaseModel
|
||||||
|
|
||||||
|
|
||||||
|
class SampleConfig(BaseModel):
|
||||||
|
host: str = "localhost"
|
||||||
|
port: int = 9999
|
||||||
|
|
|
@ -3,3 +3,16 @@
|
||||||
#
|
#
|
||||||
# This source code is licensed under the terms described in the LICENSE file in
|
# This source code is licensed under the terms described in the LICENSE file in
|
||||||
# the root directory of this source tree.
|
# the root directory of this source tree.
|
||||||
|
|
||||||
|
from .config import SampleConfig
|
||||||
|
|
||||||
|
|
||||||
|
from llama_stack.apis.agents import * # noqa: F403
|
||||||
|
|
||||||
|
|
||||||
|
class SampleAgentsImpl(Agents):
|
||||||
|
def __init__(self, config: SampleConfig):
|
||||||
|
self.config = config
|
||||||
|
|
||||||
|
async def initialize(self):
|
||||||
|
pass
|
||||||
|
|
|
@ -3,3 +3,13 @@
|
||||||
#
|
#
|
||||||
# This source code is licensed under the terms described in the LICENSE file in
|
# This source code is licensed under the terms described in the LICENSE file in
|
||||||
# the root directory of this source tree.
|
# the root directory of this source tree.
|
||||||
|
|
||||||
|
from .config import SampleConfig
|
||||||
|
|
||||||
|
|
||||||
|
async def get_adapter_impl(config: SampleConfig, _deps) -> Any:
|
||||||
|
from .sample import SampleInferenceImpl
|
||||||
|
|
||||||
|
impl = SampleInferenceImpl(config)
|
||||||
|
await impl.initialize()
|
||||||
|
return impl
|
||||||
|
|
|
@ -3,3 +3,10 @@
|
||||||
#
|
#
|
||||||
# This source code is licensed under the terms described in the LICENSE file in
|
# This source code is licensed under the terms described in the LICENSE file in
|
||||||
# the root directory of this source tree.
|
# the root directory of this source tree.
|
||||||
|
|
||||||
|
from pydantic import BaseModel
|
||||||
|
|
||||||
|
|
||||||
|
class SampleConfig(BaseModel):
|
||||||
|
host: str = "localhost"
|
||||||
|
port: int = 9999
|
||||||
|
|
|
@ -3,3 +3,16 @@
|
||||||
#
|
#
|
||||||
# This source code is licensed under the terms described in the LICENSE file in
|
# This source code is licensed under the terms described in the LICENSE file in
|
||||||
# the root directory of this source tree.
|
# the root directory of this source tree.
|
||||||
|
|
||||||
|
from .config import SampleConfig
|
||||||
|
|
||||||
|
|
||||||
|
from llama_stack.apis.inference import * # noqa: F403
|
||||||
|
|
||||||
|
|
||||||
|
class SampleInferenceImpl(Inference):
|
||||||
|
def __init__(self, config: SampleConfig):
|
||||||
|
self.config = config
|
||||||
|
|
||||||
|
async def initialize(self):
|
||||||
|
pass
|
||||||
|
|
|
@ -3,3 +3,13 @@
|
||||||
#
|
#
|
||||||
# This source code is licensed under the terms described in the LICENSE file in
|
# This source code is licensed under the terms described in the LICENSE file in
|
||||||
# the root directory of this source tree.
|
# the root directory of this source tree.
|
||||||
|
|
||||||
|
from .config import SampleConfig
|
||||||
|
|
||||||
|
|
||||||
|
async def get_adapter_impl(config: SampleConfig, _deps) -> Any:
|
||||||
|
from .sample import SampleMemoryImpl
|
||||||
|
|
||||||
|
impl = SampleMemoryImpl(config)
|
||||||
|
await impl.initialize()
|
||||||
|
return impl
|
||||||
|
|
|
@ -3,3 +3,10 @@
|
||||||
#
|
#
|
||||||
# This source code is licensed under the terms described in the LICENSE file in
|
# This source code is licensed under the terms described in the LICENSE file in
|
||||||
# the root directory of this source tree.
|
# the root directory of this source tree.
|
||||||
|
|
||||||
|
from pydantic import BaseModel
|
||||||
|
|
||||||
|
|
||||||
|
class SampleConfig(BaseModel):
|
||||||
|
host: str = "localhost"
|
||||||
|
port: int = 9999
|
||||||
|
|
|
@ -3,3 +3,16 @@
|
||||||
#
|
#
|
||||||
# This source code is licensed under the terms described in the LICENSE file in
|
# This source code is licensed under the terms described in the LICENSE file in
|
||||||
# the root directory of this source tree.
|
# the root directory of this source tree.
|
||||||
|
|
||||||
|
from .config import SampleConfig
|
||||||
|
|
||||||
|
|
||||||
|
from llama_stack.apis.memory import * # noqa: F403
|
||||||
|
|
||||||
|
|
||||||
|
class SampleMemoryImpl(Memory):
|
||||||
|
def __init__(self, config: SampleConfig):
|
||||||
|
self.config = config
|
||||||
|
|
||||||
|
async def initialize(self):
|
||||||
|
pass
|
||||||
|
|
|
@ -3,3 +3,13 @@
|
||||||
#
|
#
|
||||||
# This source code is licensed under the terms described in the LICENSE file in
|
# This source code is licensed under the terms described in the LICENSE file in
|
||||||
# the root directory of this source tree.
|
# the root directory of this source tree.
|
||||||
|
|
||||||
|
from .config import SampleConfig
|
||||||
|
|
||||||
|
|
||||||
|
async def get_adapter_impl(config: SampleConfig, _deps) -> Any:
|
||||||
|
from .sample import SampleSafetyImpl
|
||||||
|
|
||||||
|
impl = SampleSafetyImpl(config)
|
||||||
|
await impl.initialize()
|
||||||
|
return impl
|
||||||
|
|
|
@ -3,3 +3,10 @@
|
||||||
#
|
#
|
||||||
# This source code is licensed under the terms described in the LICENSE file in
|
# This source code is licensed under the terms described in the LICENSE file in
|
||||||
# the root directory of this source tree.
|
# the root directory of this source tree.
|
||||||
|
|
||||||
|
from pydantic import BaseModel
|
||||||
|
|
||||||
|
|
||||||
|
class SampleConfig(BaseModel):
|
||||||
|
host: str = "localhost"
|
||||||
|
port: int = 9999
|
||||||
|
|
|
@ -3,3 +3,16 @@
|
||||||
#
|
#
|
||||||
# This source code is licensed under the terms described in the LICENSE file in
|
# This source code is licensed under the terms described in the LICENSE file in
|
||||||
# the root directory of this source tree.
|
# the root directory of this source tree.
|
||||||
|
|
||||||
|
from .config import SampleConfig
|
||||||
|
|
||||||
|
|
||||||
|
from llama_stack.apis.safety import * # noqa: F403
|
||||||
|
|
||||||
|
|
||||||
|
class SampleSafetyImpl(Safety):
|
||||||
|
def __init__(self, config: SampleConfig):
|
||||||
|
self.config = config
|
||||||
|
|
||||||
|
async def initialize(self):
|
||||||
|
pass
|
||||||
|
|
|
@ -3,3 +3,13 @@
|
||||||
#
|
#
|
||||||
# This source code is licensed under the terms described in the LICENSE file in
|
# This source code is licensed under the terms described in the LICENSE file in
|
||||||
# the root directory of this source tree.
|
# the root directory of this source tree.
|
||||||
|
|
||||||
|
from .config import SampleConfig
|
||||||
|
|
||||||
|
|
||||||
|
async def get_adapter_impl(config: SampleConfig, _deps) -> Any:
|
||||||
|
from .sample import SampleTelemetryImpl
|
||||||
|
|
||||||
|
impl = SampleTelemetryImpl(config)
|
||||||
|
await impl.initialize()
|
||||||
|
return impl
|
||||||
|
|
|
@ -3,3 +3,10 @@
|
||||||
#
|
#
|
||||||
# This source code is licensed under the terms described in the LICENSE file in
|
# This source code is licensed under the terms described in the LICENSE file in
|
||||||
# the root directory of this source tree.
|
# the root directory of this source tree.
|
||||||
|
|
||||||
|
from pydantic import BaseModel
|
||||||
|
|
||||||
|
|
||||||
|
class SampleConfig(BaseModel):
|
||||||
|
host: str = "localhost"
|
||||||
|
port: int = 9999
|
||||||
|
|
|
@ -3,3 +3,16 @@
|
||||||
#
|
#
|
||||||
# This source code is licensed under the terms described in the LICENSE file in
|
# This source code is licensed under the terms described in the LICENSE file in
|
||||||
# the root directory of this source tree.
|
# the root directory of this source tree.
|
||||||
|
|
||||||
|
from .config import SampleConfig
|
||||||
|
|
||||||
|
|
||||||
|
from llama_stack.apis.telemetry import * # noqa: F403
|
||||||
|
|
||||||
|
|
||||||
|
class SampleTelemetryImpl(Telemetry):
|
||||||
|
def __init__(self, config: SampleConfig):
|
||||||
|
self.config = config
|
||||||
|
|
||||||
|
async def initialize(self):
|
||||||
|
pass
|
||||||
|
|
|
@ -6,7 +6,7 @@
|
||||||
|
|
||||||
from typing import List
|
from typing import List
|
||||||
|
|
||||||
from llama_stack.distribution.datatypes import Api, InlineProviderSpec, ProviderSpec
|
from llama_stack.distribution.datatypes import * # noqa: F403
|
||||||
from llama_stack.providers.utils.kvstore import kvstore_dependencies
|
from llama_stack.providers.utils.kvstore import kvstore_dependencies
|
||||||
|
|
||||||
|
|
||||||
|
@ -30,4 +30,13 @@ def available_providers() -> List[ProviderSpec]:
|
||||||
Api.memory,
|
Api.memory,
|
||||||
],
|
],
|
||||||
),
|
),
|
||||||
|
remote_provider_spec(
|
||||||
|
api=Api.agents,
|
||||||
|
adapter=AdapterSpec(
|
||||||
|
adapter_id="sample",
|
||||||
|
pip_packages=[],
|
||||||
|
module="llama_stack.providers.adapters.agents.sample",
|
||||||
|
config_class="llama_stack.providers.adapters.agents.sample.SampleConfig",
|
||||||
|
),
|
||||||
|
),
|
||||||
]
|
]
|
||||||
|
|
|
@ -26,6 +26,15 @@ def available_providers() -> List[ProviderSpec]:
|
||||||
module="llama_stack.providers.impls.meta_reference.inference",
|
module="llama_stack.providers.impls.meta_reference.inference",
|
||||||
config_class="llama_stack.providers.impls.meta_reference.inference.MetaReferenceImplConfig",
|
config_class="llama_stack.providers.impls.meta_reference.inference.MetaReferenceImplConfig",
|
||||||
),
|
),
|
||||||
|
remote_provider_spec(
|
||||||
|
api=Api.inference,
|
||||||
|
adapter=AdapterSpec(
|
||||||
|
adapter_id="sample",
|
||||||
|
pip_packages=[],
|
||||||
|
module="llama_stack.providers.adapters.inference.sample",
|
||||||
|
config_class="llama_stack.providers.adapters.inference.sample.SampleConfig",
|
||||||
|
),
|
||||||
|
),
|
||||||
remote_provider_spec(
|
remote_provider_spec(
|
||||||
api=Api.inference,
|
api=Api.inference,
|
||||||
adapter=AdapterSpec(
|
adapter=AdapterSpec(
|
||||||
|
|
|
@ -42,4 +42,13 @@ def available_providers() -> List[ProviderSpec]:
|
||||||
config_class="llama_stack.providers.adapters.memory.pgvector.PGVectorConfig",
|
config_class="llama_stack.providers.adapters.memory.pgvector.PGVectorConfig",
|
||||||
),
|
),
|
||||||
),
|
),
|
||||||
|
remote_provider_spec(
|
||||||
|
api=Api.memory,
|
||||||
|
adapter=AdapterSpec(
|
||||||
|
adapter_id="sample",
|
||||||
|
pip_packages=[],
|
||||||
|
module="llama_stack.providers.adapters.memory.sample",
|
||||||
|
config_class="llama_stack.providers.adapters.memory.sample.SampleConfig",
|
||||||
|
),
|
||||||
|
),
|
||||||
]
|
]
|
||||||
|
|
|
@ -23,4 +23,13 @@ def available_providers() -> List[ProviderSpec]:
|
||||||
module="llama_stack.providers.impls.meta_reference.safety",
|
module="llama_stack.providers.impls.meta_reference.safety",
|
||||||
config_class="llama_stack.providers.impls.meta_reference.safety.SafetyConfig",
|
config_class="llama_stack.providers.impls.meta_reference.safety.SafetyConfig",
|
||||||
),
|
),
|
||||||
|
remote_provider_spec(
|
||||||
|
api=Api.safety,
|
||||||
|
adapter=AdapterSpec(
|
||||||
|
adapter_id="sample",
|
||||||
|
pip_packages=[],
|
||||||
|
module="llama_stack.providers.adapters.safety.sample",
|
||||||
|
config_class="llama_stack.providers.adapters.safety.sample.SampleConfig",
|
||||||
|
),
|
||||||
|
),
|
||||||
]
|
]
|
||||||
|
|
|
@ -18,4 +18,13 @@ def available_providers() -> List[ProviderSpec]:
|
||||||
module="llama_stack.providers.impls.meta_reference.telemetry",
|
module="llama_stack.providers.impls.meta_reference.telemetry",
|
||||||
config_class="llama_stack.providers.impls.meta_reference.telemetry.ConsoleConfig",
|
config_class="llama_stack.providers.impls.meta_reference.telemetry.ConsoleConfig",
|
||||||
),
|
),
|
||||||
|
remote_provider_spec(
|
||||||
|
api=Api.telemetry,
|
||||||
|
adapter=AdapterSpec(
|
||||||
|
adapter_id="sample",
|
||||||
|
pip_packages=[],
|
||||||
|
module="llama_stack.providers.adapters.telemetry.sample",
|
||||||
|
config_class="llama_stack.providers.adapters.telemetry.sample.SampleConfig",
|
||||||
|
),
|
||||||
|
),
|
||||||
]
|
]
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue