remove tei embeddings

pattonjp · Dec 8, 2023 · a9142b9 · a9142b9
1 parent 2a45fb2
commit a9142b9
Show file tree

Hide file tree

Showing 9 changed files with 14 additions and 416 deletions.
diff --git a/embeddings/tei/doc.go b/embeddings/tei/doc.go
diff --git a/embeddings/tei/options.go b/embeddings/tei/options.go
diff --git a/embeddings/tei/text_embeddings_inference..go b/embeddings/tei/text_embeddings_inference..go
diff --git a/embeddings/tei_test.go b/embeddings/tei_test.go
diff --git a/examples/huggingface-milvus-vectorstore-example/milvus_vectorstore_example.go b/examples/huggingface-milvus-vectorstore-example/milvus_vectorstore_example.go
@@ -9,7 +9,7 @@ import (
 	"github.com/milvus-io/milvus-sdk-go/v2/client"
 	"github.com/milvus-io/milvus-sdk-go/v2/entity"
 	"github.com/tmc/langchaingo/embeddings"
-	"github.com/tmc/langchaingo/llms/tei"
+	"github.com/tmc/langchaingo/llms/openai"
 	"github.com/tmc/langchaingo/schema"
 
 	"github.com/tmc/langchaingo/vectorstores"
@@ -18,6 +18,7 @@ import (
 
 const (
 	poolSize = 5
+	baseURL  = "http://localhost:5500"
 )
 
 func main() {
@@ -29,10 +30,7 @@ func main() {
 }
 
 func newStore() (vectorstores.VectorStore, error) {
-	llm, err := tei.New(
-		tei.WithAPIBaseURL("http://localhost:5500"),
-		tei.WithPoolSize(poolSize),
-	)
+	llm, err := openai.New(openai.WithBaseURL(baseURL))
 	if err != nil {
 		log.Fatal(err)
 	}

diff --git a/llms/tei/doc.go b/llms/tei/doc.go
diff --git a/llms/tei/options.go b/llms/tei/options.go