This is an automated email from the ASF dual-hosted git repository.
rahulvats pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new 22d7ba8f676 Fix Pinecone Cohere Integration System Test (#51290)
22d7ba8f676 is described below
commit 22d7ba8f6764dab5d669d7de618ff6fcbc55c761
Author: Atul Singh <[email protected]>
AuthorDate: Mon Jun 2 14:57:27 2025 +0530
Fix Pinecone Cohere Integration System Test (#51290)
Co-authored-by: Atul Singh <[email protected]>
---
providers/pinecone/tests/system/pinecone/example_pinecone_cohere.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git
a/providers/pinecone/tests/system/pinecone/example_pinecone_cohere.py
b/providers/pinecone/tests/system/pinecone/example_pinecone_cohere.py
index 8813479d61a..496c0a40154 100644
--- a/providers/pinecone/tests/system/pinecone/example_pinecone_cohere.py
+++ b/providers/pinecone/tests/system/pinecone/example_pinecone_cohere.py
@@ -44,7 +44,7 @@ with DAG(
hook = PineconeHook()
pod_spec = hook.get_pod_spec_obj()
- hook.create_index(index_name=index_name, dimension=768, spec=pod_spec)
+ hook.create_index(index_name=index_name, dimension=1024, spec=pod_spec)
embed_task = CohereEmbeddingOperator(
task_id="embed_task",
@@ -54,7 +54,7 @@ with DAG(
@task
def transform_output(embedding_output) -> list[dict]:
# Convert each embedding to a map with an ID and the embedding vector
- return [dict(id=str(i), values=embedding) for i, embedding in
enumerate(embedding_output)]
+ return [dict(id=str(i), values=embedding) for i, embedding in
enumerate(embedding_output.float_)]
transformed_output = transform_output(embed_task.output)
