From ca4b87aa05555e991ff6f0a429fa882adc491c5a Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Thu, 19 Sep 2024 09:29:40 -0700 Subject: [PATCH] fix memory client --- llama_stack/apis/memory/client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/llama_stack/apis/memory/client.py b/llama_stack/apis/memory/client.py index 7b61cd830..0cddf0d0e 100644 --- a/llama_stack/apis/memory/client.py +++ b/llama_stack/apis/memory/client.py @@ -13,9 +13,9 @@ from typing import Any, Dict, List, Optional import fire import httpx -from termcolor import cprint from llama_stack.distribution.datatypes import RemoteProviderConfig +from termcolor import cprint from llama_stack.apis.memory import * # noqa: F403 from llama_stack.providers.utils.memory.file_utils import data_url_from_file @@ -151,7 +151,7 @@ async def run_main(host: str, port: int, stream: bool): ] this_dir = os.path.dirname(__file__) - files = [Path(this_dir).parent.parent / "CONTRIBUTING.md"] + files = [Path(this_dir).parent.parent.parent / "CONTRIBUTING.md"] documents += [ MemoryBankDocument( document_id=f"num-{i}",