Skip to content

feat(azure-cosmosdb): expose query request charges - #876

Open
Saurish (nos-redacted) wants to merge 2 commits into
langchain-ai:mainfrom
nos-redacted:feat/cosmosdb-request-charge
Open

feat(azure-cosmosdb): expose query request charges#876
Saurish (nos-redacted) wants to merge 2 commits into
langchain-ai:mainfrom
nos-redacted:feat/cosmosdb-request-charge

Conversation

@nos-redacted

Copy link
Copy Markdown
Contributor

Summary

  • add an optional callback that reports aggregated Cosmos DB request charges once per logical vector query
  • support sync and async query pagination while preserving existing return values and document metadata
  • expose a public immutable request-charge event and document the reporting contract
  • raise the azure-cosmos minimum to 4.14.0, the first stable release without the stale initial query response hook

Testing

  • 256 unit tests passed
  • Ruff lint and format checks passed
  • mypy passed across 50 files
  • uv lock --check passed
  • Cosmos authentication and Azure OpenAI embeddings were verified independently

Integration note

The credential-gated sync and async integration assertions are included. The complete live pytest case did not finish within the local VS Code terminal timeout, so this PR does not claim a successful end-to-end integration run.

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