sjyangkevin commented on issue #50599: URL: https://github.com/apache/airflow/issues/50599#issuecomment-2896565869
> > Not related to [#50470](https://github.com/apache/airflow/pull/50470), the system test (airflow/providers/cohere/tests/system/cohere/example_cohere_embedding_operator.py), the operator is having issue to push embedding to XCOM, resulting in TypeError: cannot serialize object of type <class 'cohere.types.embed_by_type_response_embeddings.EmbedByTypeResponseEmbeddings'> > > [@sjyangkevin](https://github.com/sjyangkevin) this is happening because for airflow 3, we use `from airflow.serialization.serde import serialize` module to serialize the data that is being sent to xcoms. The case seems to be that the `EmbedByTypeResponseEmbeddings` is a pydantic model which our serialize library doesn't know how to handle. Verify the issue and the return type is changed in Cohere 1.4.2. Issue created https://github.com/apache/airflow/issues/50867 -- 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]
