YAshhh29 opened a new pull request, #71989: URL: https://github.com/apache/airflow/pull/71989
## Why The OpenAI Embeddings API accepts batches, but `OpenAIHook.create_embeddings()` returned only `response.data[0]`. Every successful batch result after the first item was silently discarded. ## What changed - Preserve the existing single-input return shape. - Return every batch embedding ordered by the response `index`. - Update the operator return type and user documentation. - Add regressions for string and token-array batches. ## Tests - Focused batch regression test against this branch: 2 passed. - The same regression test against the unmodified implementation: 2 failed. - `ruff format` and `ruff check --fix`: passed. - `git diff --check`: passed. The full provider suite was not run locally because Airflow does not support native Windows; CI will provide the Linux suite result. --- ##### Was generative AI tooling used to co-author this PR? - [X] Yes — GitHub Copilot Generated-by: GitHub Copilot following [the guidelines](https://github.com/apache/airflow/blob/main/contributing-docs/05_pull_requests.rst#gen-ai-assisted-contributions) -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
