Skip to content
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: Add YandexGPT Encoder #331

Open
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

KirillKukharev
Copy link

Pull Request Description:

This pull request adds a new encoder class, YandexGPTEncoder, for generating text embeddings using the YandexGPT API. The encoder integrates with the semantic_router.

Changes Introduced:

  • New Encoder Class: YandexGPTEncoder for YandexGPT API integration.
  • Initialization: Requires YandexGPT API key and catalog ID, which can be provided directly or via environment variables.
  • Unit tests and docs for YandexGPT encoder

Method Implementation:

  • _initialize_client for setting up the API client.
  • get_headers for preparing the request headers.
  • call for generating embeddings by making API requests.

* add yandexgpt encoder

* add unit tests for YandexGPT

* Change encoder type name
@KirillKukharev KirillKukharev changed the title Add YandexGPT Encoder feat: Add YandexGPT Encoder Jun 19, 2024
@KirillKukharev
Copy link
Author

@jamescalam could you review this PR please?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

1 participant