在 Pinecone 中查询向量数据

After setting up your index and adding data, the next crucial step is to retrieve relevant information. This is where querying comes in!

Querying in Pinecone means finding vectors in your index that are most similar to a given "query vector." It's how you perform semantic search, recommendations, and more.

Just like the data you stored, your search query also needs to be converted into a vector. This "query vector" is then compared against all vectors in your Pinecone index.

• Embedding Model: You use the same embedding model that generated your stored vectors to create your query vector.
• Similarity: Pinecone calculates the distance or similarity between your query vector and indexed vectors.

Before you can query Pinecone, you need an embedding for your search term. Let's say you want to find documents similar to "machine learning models."

You'd pass "machine learning models" through your chosen embedding model (e.g., OpenAI's text-embedding-ada-002) to get a vector representation.

Pinecone's client provides a straightforward method for querying: index.query(). This method is your gateway to finding similar vectors.

Key parameters you'll often use:

• vector: The embedding of your query.
• top_k: How many similar results you want.
• include_metadata: Whether to return associated metadata.
• include_values: Whether to return the raw vector values.

Let's perform a simple query. We'll use a placeholder vector for now, assuming it's already generated. Remember to replace YOUR_API_KEY and YOUR_ENVIRONMENT.

The query_results object contains a list of matches. Each match represents a similar vector found in your index.

• id: The unique identifier of the matched vector.
• score: A numerical value indicating similarity. Higher scores (closer to 1 for cosine, closer to 0 for Euclidean) mean higher similarity.
• values: The raw vector (if include_values=True).
• metadata: Any associated metadata (if include_metadata=True).

The top_k parameter is crucial for controlling how many results Pinecone returns. It specifies the number of nearest neighbors you want to retrieve.

• If top_k=1, you get only the single most similar vector.
• If top_k=10, you get the top 10 most similar vectors.

Choose top_k based on how many relevant items your application needs.

Often, you don't just want the ID and score; you need the original content or other properties associated with the vector. This is where include_metadata comes in.

• Set include_metadata=True to retrieve the dictionary of metadata stored with each vector.
• You can also set include_values=True to get the actual vector array of the matched item, though this is less common for basic retrieval.

Pinecone allows you to refine your similarity searches by adding filters based on the metadata you stored with your vectors.

For example, you could search for similar items only within a specific category or by a certain author.

We'll dive deeper into powerful metadata filtering in a later lesson, but know that it's a key feature for precise searches.

Let's expand our previous example to include metadata in the results. For this to work, we'd need to have upserted data with metadata in a previous step.

This example assumes an index with vectors and associated metadata (e.g., {"genre": "sci-fi"}).

You want to retrieve the 5 most similar vectors from your Pinecone index. You also need to see the original metadata associated with each matched vector.

Which combination of parameters should you use in your index.query() call?

Great job! You've learned how to query your Pinecone index to find similar vectors.

• Queries use a query vector, typically generated by the same embedding model.
• The index.query() method is used, with key parameters like vector, top_k, include_metadata, and include_values.
• Results include id and a score indicating similarity.

Next, we'll explore more advanced ways to refine your searches!