Langchain
Vector store
import os
import getpass
os.environ["LLM_RAILS_API_KEY"] = getpass.getpass("LLMRails API Key:")
os.environ["LLM_RAILS_DATASTORE_ID"] = getpass.getpass("LLMRails Datastore Id:")
vectorstore = LLMRails(api_key=llm_rails_api_key, datastore_id=datastore_id)
# Similarity Search
query = "What do you plan to do about national security?"
found_docs = vectorstore.similarity_search(query, k=5)
print(found_docs[0].page_content)
# Similarity search with score
query = "What is your approach to national defense"
found_docs = llm_rails.similarity_search_with_score(query, k=5)
document, score = found_docs[0]
print(document.page_content)
print(f"\nScore: {score}")