trcreatives/go-ollama
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

go-ollama: changed development to dev container and added streaming capabilities to methods v0.1.1

Browse Source
This commit is contained in:
Timo Riegebauer
2026-02-23 21:37:17 +00:00
parent 69172f6d5f
commit 6b758d1096
14 changed files with 448 additions and 93 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
generate_embeddings.go
View File

@@ -36,15 +36,15 @@ type GenerateEmbeddingsResponse struct {
PromptEvalCount int `json:"prompt_eval_count"`
}
func (o Ollama) GenerateEmbeddings(reqBody GenerateEmbeddingsRequest) (GenerateEmbeddingsResponse, int, error) {
func (o Ollama) GenerateEmbeddings(reqBody GenerateEmbeddingsRequest) (GenerateEmbeddingsResponse, error) {
reqBodyBytes, err := json.Marshal(reqBody)
if err != nil {
return GenerateEmbeddingsResponse{}, -1, err
return GenerateEmbeddingsResponse{}, err
}
req, err := http.NewRequest(http.MethodPost, fmt.Sprintf("%s/embed", o.baseUrl), bytes.NewReader(reqBodyBytes))
if err != nil {
return GenerateEmbeddingsResponse{}, -1, err
return GenerateEmbeddingsResponse{}, err
}
for key, val := range o.customHeaders {
@@ -54,17 +54,17 @@ func (o Ollama) GenerateEmbeddings(reqBody GenerateEmbeddingsRequest) (GenerateE
resp, err := http.DefaultClient.Do(req)
if err != nil {
return GenerateEmbeddingsResponse{}, -1, err
return GenerateEmbeddingsResponse{}, err
}
defer resp.Body.Close()
if resp.StatusCode != 200 {
return GenerateEmbeddingsResponse{}, resp.StatusCode, errors.New("status code is not 200")
if resp.StatusCode != http.StatusOK {
return GenerateEmbeddingsResponse{}, errors.New("status code is not 200")
}
var respBody GenerateEmbeddingsResponse
if err := json.NewDecoder(resp.Body).Decode(&respBody); err != nil {
return GenerateEmbeddingsResponse{}, -1, err
return GenerateEmbeddingsResponse{}, err
}
return respBody, resp.StatusCode, nil
return respBody, nil
}