File size: 593 Bytes
b09edd3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
from flask import Flask
from flask import request
from pinecone import Pinecone, ServerlessSpec

import os


PINECONE_DB_API = os.environ.get('PINECONE_DB_API')
pc = Pinecone(api_key=PINECONE_DB_API)
index = pc.Index('fury-actor-py')       # NOTE: FOR NOW

app = Flask(__name__)

@app.route("/query", methods=['POST'])
def query():
    """
    {
        embeddings: <embeddings_list>
    }
    """

    embeddings = request.get_json()['embeddings']

    matches = index.query(
        vector = embeddings,
        top_k = 3,
        include_metadata = True
    )

    return matches.to_dict()