Update ollama.go

henomis · Jun 28, 2024 · 7ffad0c · 7ffad0c
1 parent 650fb59
commit 7ffad0c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/embedder/ollama/ollama.go b/embedder/ollama/ollama.go
@@ -17,11 +17,11 @@ const (
 	defaultEndpoint = "http://localhost:11434/api"
 )
 
-type ErrorOllamaEmbed struct {
+type OllamaEmbedError struct {
 	Err error
 }
 
-func (e *ErrorOllamaEmbed) Error() string {
+func (e *OllamaEmbedError) Error() string {
 	return fmt.Sprintf("Error embedding text: %v", e.Err)
 }
 
@@ -100,7 +100,7 @@ func (e *Embedder) embed(ctx context.Context, text string) (embedder.Embedding,
 	}
 
 	if resp.HTTPStatusCode >= http.StatusBadRequest {
-		return nil, &ErrorOllamaEmbed{
+		return nil, &OllamaEmbedError{
 			Err: errors.New(string(resp.RawBody)),
 		}
 	}