Spaces:
Sleeping
Sleeping
File size: 613 Bytes
b87de8f 15d7e65 b87de8f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from llama_index import download_loader, VectorStoreIndex, SimpleDirectoryReader, ServiceContext
from llama_index.tools import QueryEngineTool, ToolMetadata
from llama_index.query_engine import SubQuestionQueryEngine
from llama_index.callbacks import CallbackManager, LlamaDebugHandler
from llama_index.llms import LlamaCPP
from utils import completion_to_prompt, messages_to_prompt
def build_service_context():
return ServiceContext.from_defaults(embed_model='local')
def build_index(documents, context):
index = VectorStoreIndex.from_documents(documents, service_context=context)
return index
|