Add OpenAI Provider #35023
Merged
Add OpenAI Provider #35023
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Taragolis
reviewed
Oct 18, 2023
phanikumv
reviewed
Oct 18, 2023
mpgreg
reviewed
Oct 19, 2023
mpgreg
reviewed
Oct 19, 2023
mpgreg
reviewed
Oct 19, 2023
mpgreg
reviewed
Oct 19, 2023
mpgreg
reviewed
Oct 19, 2023
mpgreg
reviewed
Oct 19, 2023
mpgreg
reviewed
Oct 19, 2023
mpgreg
reviewed
Oct 19, 2023
b4dec95 to
dd03c66
Compare
1839139 to
9496fbb
Compare
utkarsharma2
commented
Oct 30, 2023
utkarsharma2
commented
Oct 30, 2023
Merged
75b57a9 to
969fb1f
Compare
pankajastro
reviewed
Nov 4, 2023
Co-authored-by: Phani Kumar <94376113+phanikumv@users.noreply.github.com>
Since the upstream API accepts str ot list of tokens, we accept the similar inputs from user.
Because of OpenAI SDK not being production ready
Co-authored-by: Pankaj Singh <98807258+pankajastro@users.noreply.github.com>
Co-authored-by: Pankaj Singh <98807258+pankajastro@users.noreply.github.com>
Co-authored-by: Pankaj Singh <98807258+pankajastro@users.noreply.github.com>
cd048bb to
5791deb
Compare
60 tasks
romsharon98
pushed a commit
to romsharon98/airflow
that referenced
this pull request
Nov 10, 2023
* Add OpenAI Provider * Apply suggestions from code review Co-authored-by: Phani Kumar <94376113+phanikumv@users.noreply.github.com> * Remove create_completions method from hook * Change type of input_text param Since the upstream API accepts str ot list of tokens, we accept the similar inputs from user. * Updated min-airflow version to 2.5.0 * Updated the interface and fix docs and static files * Fix tests * Fix tests * Change the version Because of OpenAI SDK not being production ready * Add embedding_kwargs as a param to operator * Update tests/providers/openai/hooks/test_openai.py Co-authored-by: Pankaj Singh <98807258+pankajastro@users.noreply.github.com> * Remove unwanted params in docstring * Update Changelog * Add security.rst file * Update docs/apache-airflow-providers-openai/index.rst Co-authored-by: Pankaj Singh <98807258+pankajastro@users.noreply.github.com> * Add host field for connections * Update docs/apache-airflow-providers-openai/index.rst Co-authored-by: Pankaj Singh <98807258+pankajastro@users.noreply.github.com> * Add changelog.rst file to docs * Change version to 1.0.0 * Resolve conflicts * Fix tests * Fixed tests * Fix test * Resolve Conflict --------- Co-authored-by: Pankaj Koti <pankajkoti699@gmail.com> Co-authored-by: Phani Kumar <94376113+phanikumv@users.noreply.github.com> Co-authored-by: Pankaj Singh <98807258+pankajastro@users.noreply.github.com>
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
11 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This PR is part of our larger effort to add first-class integrations to support LLMOps that was presented at Airflow Summit.
This PR adds explicitly the OpenAI Provider. OpenAI is a leading American artificial intelligence organization, which offers one of the most used LLM - ChatGPT and offers embedding models.
The primary objective of this Provider is to present users with an alternative embedding model. This allows them to generate vectors for their proprietary data, a pivotal step towards establishing integrations with LLM models like ChatGPT.
Example DAG:
The
OpenAIEmbeddingOperatorcan accept either a string or a callable returning a list of strings.Email Discussion related to the effort can be found here - https://lists.apache.org/thread/0d669fmy4hn29h5c0wj0ottdskd77ktp