lawliet/CS224-knowledge-discovery

1import pinecone2from .encoder import TextEncoder3import os4 5 6pinecone.init(api_key=os.environ["PINECONE_API_KEY"])7index = pinecone.Index("nlu-background")8 9 10async def get_pinecone_results(_q: str, k=3):11    encoder = TextEncoder()12    query_vec, usage = await encoder.encode_text([_q])13    query_vec = query_vec[0]14    query_response = index.query(15        namespace="nlu-background-cs224n",16        top_k=k,17        include_values=True,18        include_metadata=True,19        vector=query_vec,20        filter={},21    )22    query_response_dict = {23        "matches": query_response["matches"],24    }25    return query_response_dict, usage26