fix memory client

This commit is contained in:
Xi Yan 2024-09-19 09:29:40 -07:00
parent 6302a1ee90
commit ca4b87aa05

View file

@ -13,9 +13,9 @@ from typing import Any, Dict, List, Optional
import fire import fire
import httpx import httpx
from termcolor import cprint
from llama_stack.distribution.datatypes import RemoteProviderConfig from llama_stack.distribution.datatypes import RemoteProviderConfig
from termcolor import cprint
from llama_stack.apis.memory import * # noqa: F403 from llama_stack.apis.memory import * # noqa: F403
from llama_stack.providers.utils.memory.file_utils import data_url_from_file 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__) this_dir = os.path.dirname(__file__)
files = [Path(this_dir).parent.parent / "CONTRIBUTING.md"] files = [Path(this_dir).parent.parent.parent / "CONTRIBUTING.md"]
documents += [ documents += [
MemoryBankDocument( MemoryBankDocument(
document_id=f"num-{i}", document_id=f"num-{i}",