Skip to content

Conversation

@dmsmith25
Copy link

  • Add --pinecone_multi_namespace flag to enable benchmarking across all namespaces
  • Auto-discover all populated namespaces in existing indexes
  • Distribute requests evenly across namespaces using round-robin
  • Add validation to ensure multi-namespace mode only works with existing populated indexes
  • Require --skip_populate when using multi-namespace mode
  • Aggregate metrics across all namespaces (same format as single-namespace mode)
  • Update Pinecone README with multi-namespace documentation

Problem

Describe the purpose of this change. What problem is being solved and why?

Enables benchmarking for existing, populated, multi-namespace Pinecone indexes.

Solution

Describe the approach you took. Link to any relevant bugs, issues, docs, or other resources.

Workload is spread evenly across namespaces. Results are aggregated for metrics.

Type of Change

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to not work as expected)
  • This change requires a documentation update
  • Infrastructure change (CI configs, etc)
  • Non-code change (docs, etc)
  • None of the above: (explain here)

Test Plan

Describe specific steps for validating this change.

Tested on existing test suite

- Add --pinecone_multi_namespace flag to enable benchmarking across all namespaces
- Auto-discover all populated namespaces in existing indexes
- Distribute requests evenly across namespaces using round-robin
- Add validation to ensure multi-namespace mode only works with existing populated indexes
- Require --skip_populate when using multi-namespace mode
- Aggregate metrics across all namespaces (same format as single-namespace mode)
- Update Pinecone README with multi-namespace documentation
…namespaces

This fixes test timeouts by using a direct API call to check namespace
existence instead of iterating through all namespaces. The previous
implementation was inefficient and caused timeouts in CI tests.

- Changed check_namespace_exists() to use describe_namespace() API
- Handles NotFoundException when namespace doesn't exist
- Much faster and more efficient for single namespace checks
- No impact on multi-namespace functionality (uses _discover_all_namespaces)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants