-
Notifications
You must be signed in to change notification settings - Fork 2k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
feat: CohereGenerator
#6395
feat: CohereGenerator
#6395
Conversation
Signed-off-by: sunilkumardash9 <sunilkumardash9@gmail.com>
2. removed commented files in test-cohere_generators 3. removed unused imports Signed-off-by: sunilkumardash9 <sunilkumardash9@gmail.com>
2. remove dict casting of metadata in run Signed-off-by: sunilkumardash9 <sunilkumardash9@gmail.com>
Signed-off-by: sunilkumardash9 <sunilkumardash9@gmail.com>
2. small change in doc string Signed-off-by: sunilkumardash9 <sunilkumardash9@gmail.com>
2. changed api key env var from CO_API_KEY to COHERE_API_KEY Signed-off-by: sunilkumardash9 <sunilkumardash9@gmail.com>
Signed-off-by: sunilkumardash9 <sunilkumardash9@gmail.com>
2. Added kwargs doc strings for CohereGenerator 3. removed type hints for metadata and replies 4. use COHERE_API_URL instead of hard coded URL. Signed-off-by: sunilkumardash9 <sunilkumardash9@gmail.com>
Co-authored-by: Daria Fokina <daria.f93@gmail.com>
Co-authored-by: Daria Fokina <daria.f93@gmail.com>
Co-authored-by: Daria Fokina <daria.f93@gmail.com>
Co-authored-by: Daria Fokina <daria.f93@gmail.com>
Co-authored-by: Daria Fokina <daria.f93@gmail.com>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Couple of smaller change requests, otherwise looks good to me
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM! 👍
Wow, didn't know about this @ZanSara and @julian-risch Great work @sunilkumardash9 Any chance you can add the Chat version, I think they also support it if I'm not mistaken. |
@vblagoje in another PR please 🙏 |
@vblagoje Yes, I was initially working on the chat endpoint instead of the Generate. I can work on it. |
Yes, another PR @ZanSara @sunilkumardash9 chat would be a fantastic contribution. Try to match the format of TGI and OpenAI chat and non-chat generators as much as possible. We would like these to be completely interchangeable. When done, feel free to assign me for chat review! |
Related Issues
CohereGenerator
#5984Note:
This PR is a workaround to my lack of access to @sunilkumardash9's original PR here: #6034 Please refer to the original PR.