Gage-Technologies/embedding-server

A (hopefully) performant implementation of a sentence-transformer embedding server. This repo is forked from https://github.com/huggingface/text-generation-inference

Rust

8

260 commits

updated Oct 4, 2023

See the code

README

Embedding Server

GitHub Repo stars License Swagger API documentation

A Rust, Python and gRPC server for sentence-transformer embeddings.

Overview

This is the beginning of the repo so we're pretty light on the info. More will come! The basic gist is that we intend to create the equivalent of Huggingface's Text Generation Inference API but for sentence-transformer embeddings. This repo is a fork of the text-generation-inference repo.

The current state of the repo is that we have forked the codebase to run the base Sentence-Transformer class from the sentence-transformers library. The server will run manually but we have not completed the dockerfile or actions to create a build pipeline. More to come!

Develop

make server-dev
make router-dev

Testing

# python
make python-server-tests
make python-client-tests
# or both server and client tests
make python-tests
# rust cargo tests
make rust-tests
# integration tests
make integration-tests

Contributors

OlivierDehaene

200 commits

Narsil

20 commits

sam-ulrich1

18 commits

njhill

12 commits

Gage-Technologies/embedding-server

A (hopefully) performant implementation of a sentence-transformer embedding server. This repo is forked from https://github.com/huggingface/text-generation-inference

Rust

8

260 commits

updated Oct 4, 2023

See the code

README

Embedding Server

GitHub Repo stars License Swagger API documentation

A Rust, Python and gRPC server for sentence-transformer embeddings.

Overview

This is the beginning of the repo so we're pretty light on the info. More will come! The basic gist is that we intend to create the equivalent of Huggingface's Text Generation Inference API but for sentence-transformer embeddings. This repo is a fork of the text-generation-inference repo.

The current state of the repo is that we have forked the codebase to run the base Sentence-Transformer class from the sentence-transformers library. The server will run manually but we have not completed the dockerfile or actions to create a build pipeline. More to come!

Develop

make server-dev
make router-dev

Testing

# python
make python-server-tests
make python-client-tests
# or both server and client tests
make python-tests
# rust cargo tests
make rust-tests
# integration tests
make integration-tests

Contributors

OlivierDehaene

200 commits

Narsil

20 commits

sam-ulrich1

18 commits

njhill

12 commits

Languages

Rust

50.4%

Python

43.2%

Go

2.4%

Dockerfile

1.7%

JavaScript

1.2%