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

Moved some stuff out of common/; re-generated OpenAPI spec

Browse source
This commit is contained in:
Ashwin Bharambe 2024-09-16 21:21:11 -07:00
parent 76b354a081
commit 6665d31cdf
12 changed files with 238 additions and 236 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/apis/agents/agents.py
View file

@ -14,7 +14,7 @@ from pydantic import BaseModel, ConfigDict, Field
from typing_extensions import Annotated
from llama_models.llama3.api.datatypes import * # noqa: F403
from llama_stack.common.deployment_types import * # noqa: F403
from llama_stack.apis.common.deployment_types import * # noqa: F403
from llama_stack.apis.inference import * # noqa: F403
from llama_stack.apis.safety import * # noqa: F403
from llama_stack.apis.memory import * # noqa: F403

5
llama_stack/apis/common/__init__.py Normal file
View file

@ -0,0 +1,5 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.

0
llama_stack/common/deployment_types.py → llama_stack/apis/common/deployment_types.py
View file

0
llama_stack/common/training_types.py → llama_stack/apis/common/training_types.py
View file

2
llama_stack/apis/evals/evals.py
View file

@ -13,7 +13,7 @@ from pydantic import BaseModel
from llama_models.llama3.api.datatypes import * # noqa: F403
from llama_stack.apis.dataset import * # noqa: F403
from llama_stack.common.training_types import * # noqa: F403
from llama_stack.apis.common.training_types import * # noqa: F403
class TextGenerationMetric(Enum):

2
llama_stack/apis/post_training/post_training.py
View file

@ -15,7 +15,7 @@ from pydantic import BaseModel, Field
from llama_models.llama3.api.datatypes import * # noqa: F403
from llama_stack.apis.dataset import * # noqa: F403
from llama_stack.common.training_types import * # noqa: F403
from llama_stack.apis.common.training_types import * # noqa: F403
class OptimizerType(Enum):

2
llama_stack/apis/safety/safety.py
View file

@ -11,7 +11,7 @@ from llama_models.schema_utils import json_schema_type, webmethod
from pydantic import BaseModel, validator
from llama_models.llama3.api.datatypes import * # noqa: F403
from llama_stack.common.deployment_types import RestAPIExecutionConfig
from llama_stack.apis.common.deployment_types import RestAPIExecutionConfig
@json_schema_type

2
llama_stack/core/datatypes.py
View file

@ -151,7 +151,7 @@ as being "Llama Stack compatible"
def module(self) -> str:
if self.adapter:
return self.adapter.module
return f"llama_stack.{self.api.value}.client"
return f"llama_stack.apis.{self.api.value}.client"
@property
def pip_packages(self) -> List[str]:

2
llama_stack/core/server.py
View file

@ -309,7 +309,7 @@ async def resolve_impls(
specs[api] = RouterProviderSpec(
api=api,
module=f"llama_stack.{api.value.lower()}.router",
module=f"llama_stack.providers.routers.{api.value.lower()}",
api_dependencies=[],
inner_specs=inner_specs,
)

2
llama_stack/stack.py
View file

@ -7,7 +7,7 @@
from llama_models.llama3.api.datatypes import * # noqa: F403
from llama_stack.apis.agents import * # noqa: F403
from llama_stack.apis.dataset import * # noqa: F403
from llama_stack.evaluations.api import * # noqa: F403
from llama_stack.apis.evals import * # noqa: F403
from llama_stack.apis.inference import * # noqa: F403
from llama_stack.apis.batch_inference import * # noqa: F403
from llama_stack.apis.memory import * # noqa: F403