From 30fee82407bcfa2d1fcc598b08e654e467ba11de Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Thu, 26 Dec 2024 16:54:33 -0800 Subject: [PATCH] vector_store --- llama_stack/providers/utils/memory/vector_store.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/llama_stack/providers/utils/memory/vector_store.py b/llama_stack/providers/utils/memory/vector_store.py index 072a8ae30..c97633558 100644 --- a/llama_stack/providers/utils/memory/vector_store.py +++ b/llama_stack/providers/utils/memory/vector_store.py @@ -15,14 +15,17 @@ from urllib.parse import unquote import chardet import httpx import numpy as np + +from llama_models.llama3.api.tokenizer import Tokenizer from numpy.typing import NDArray from pypdf import PdfReader -from llama_models.llama3.api.datatypes import * # noqa: F403 -from llama_models.llama3.api.tokenizer import Tokenizer - -from llama_stack.apis.common.content_types import InterleavedContent, TextContentItem -from llama_stack.apis.memory import * # noqa: F403 +from llama_stack.apis.common.content_types import ( + InterleavedContent, + TextContentItem, + URL, +) +from llama_stack.apis.memory import Chunk, MemoryBankDocument, QueryDocumentsResponse from llama_stack.apis.memory_banks import VectorMemoryBank from llama_stack.providers.datatypes import Api from llama_stack.providers.utils.inference.prompt_adapter import (