phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-03 17:29:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

vector_store

Browse source
This commit is contained in:
Xi Yan 2024-12-26 16:54:33 -08:00
parent b7bc1c6297
commit 30fee82407
1 changed files with 8 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

13
llama_stack/providers/utils/memory/vector_store.py
View file

@ -15,14 +15,17 @@ from urllib.parse import unquote
import chardet import chardet
import httpx import httpx
import numpy as np import numpy as np
from llama_models.llama3.api.tokenizer import Tokenizer
from numpy.typing import NDArray from numpy.typing import NDArray
from pypdf import PdfReader from pypdf import PdfReader
from llama_models.llama3.api.datatypes import * # noqa: F403 from llama_stack.apis.common.content_types import (
from llama_models.llama3.api.tokenizer import Tokenizer InterleavedContent,
TextContentItem,
from llama_stack.apis.common.content_types import InterleavedContent, TextContentItem URL,
from llama_stack.apis.memory import * # noqa: F403 )
from llama_stack.apis.memory import Chunk, MemoryBankDocument, QueryDocumentsResponse
from llama_stack.apis.memory_banks import VectorMemoryBank from llama_stack.apis.memory_banks import VectorMemoryBank
from llama_stack.providers.datatypes import Api from llama_stack.providers.datatypes import Api
from llama_stack.providers.utils.inference.prompt_adapter import ( from llama_stack.providers.utils.inference.prompt_adapter import (