File size: 622 Bytes
4c91492
 
 
 
 
 
 
 
 
 
3653851
4c91492
3653851
 
 
 
 
4c91492
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
from pprint import pprint
from langchain_openai import OpenAIEmbeddings
from langchain_core.vectorstores import InMemoryVectorStore


embeddings = OpenAIEmbeddings(model="text-embedding-3-large")
vector_store = InMemoryVectorStore.load("/code/data/vectorstore.json", embedding=embeddings)

# query = "ryan brown"
# query = "defensive midfielder"
query = "* FC Everglade"
results = vector_store.similarity_search(query, k=3)
results = vector_store.similarity_search(
    query,
    k=3,
    filter=lambda doc: doc.metadata.get("type") == "player",
)

for result in results:
    pprint(result.page_content)
    print("---")