community: add reka chat model integration #27379
Conversation
|
The latest updates on your projects. Learn more about Vercel for Git ↗︎
|
dosubot
bot
added
size:XXL
|
closing as duplicate of #26405 - let me know if they're different |
|
nvm - got it - personal branch |
|
looks like CI might still be failing though |
|
HI @vbarda we made some updates and have addressed all of your comments (including notebook, token count, duplicated class, system role). Added a patch in the |
Thanks! Looks like there are a few things failing (due to missing |
|
Updated to address the token count comment. thanks for the feedback. I have ran make lint, make lint and make test, and make extended_tests which passed. |
| @@ -329,11 +330,29 @@ async def _agenerate( | |||
|
|
|||
| return ChatResult(generations=[ChatGeneration(message=message)]) | |||
|
|
|||
| def get_num_tokens(self, text: str) -> int: | |||
| def get_num_tokens(self, input: Union[str, BaseMessage, List[BaseMessage]]) -> int: | |||
There was a problem hiding this comment.
looks like the input type here is not matching the one in the token_counter callable (failing linter)
community: add Reka chat model integration
Description: This PR adds support for the Reka chat model to the LangChain community package. It includes the implementation of ChatReka class, which allows users to interact with Reka's API for chat completions. The integration supports both text and image inputs, streaming responses, and async operations.
Dependencies: This integration inside langchain-community requires the reka-api package to be installed. Users will need to install it separately with pip install reka-api.
Added features:
ChatReka class for interacting with Reka chat models
Support for text and image inputs
Streaming and non-streaming response handling
Synchronous and asynchronous methods
Unit and integration tests
Example notebook demonstrating usage
Testing:
Unit tests have been added in tests/unit_tests/chat_models/test_reka.py
Integration tests have been added in tests/integration_tests/chat_models/test_reka.py
An example notebook has been created in docs/docs/integrations/chat/reka.ipynb, including tool use with search.
I have run make format, make lint, and make test from the root of the community package, and all checks pass.
X/Twitter handle: @RekaAILabs