Bug fixes for memory

togethercomputer · Sep 19, 2024 · 9ab27e8 · 9ab27e8
1 parent 8cdc2f0
commit 9ab27e8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/llama_stack/apis/memory/client.py b/llama_stack/apis/memory/client.py
@@ -13,12 +13,12 @@
 
 import fire
 import httpx
+from termcolor import cprint
 
 from llama_stack.distribution.datatypes import RemoteProviderConfig
-from termcolor import cprint
 
-from .memory import *  # noqa: F403
-from .common.file_utils import data_url_from_file
+from llama_stack.apis.memory import *  # noqa: F403
+from llama_stack.providers.utils.memory.file_utils import data_url_from_file
 
 
 async def get_client_impl(config: RemoteProviderConfig, _deps: Any) -> Memory:

diff --git a/llama_stack/cli/stack/build.py b/llama_stack/cli/stack/build.py
@@ -212,7 +212,7 @@ def _run_stack_build_command(self, args: argparse.Namespace) -> None:
                 providers_for_api = all_providers[api]
 
                 api_provider = prompt(
-                    "> Enter the API provider for the {} API: (default=meta-reference): ".format(
+                    "> Enter provider for the {} API: (default=meta-reference): ".format(
                         api.value
                     ),
                     validator=Validator.from_callable(