Skip to content

Commit 2b616f4

Browse files
committed
add use of handleLlamaStackClientError to more lsd handlers where llamastack client errors can arise
1 parent 0f89e45 commit 2b616f4

3 files changed

Lines changed: 10 additions & 10 deletions

File tree

packages/gen-ai/bff/internal/api/lsd_files_handler.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ func (app *App) LlamaStackUploadFileHandler(w http.ResponseWriter, r *http.Reque
8282

8383
result, err := app.repositories.Files.UploadFile(ctx, uploadParams)
8484
if err != nil {
85-
app.serverErrorResponse(w, r, err)
85+
app.handleLlamaStackClientError(w, r, err)
8686
return
8787
}
8888

@@ -128,7 +128,7 @@ func (app *App) LlamaStackListFilesHandler(w http.ResponseWriter, r *http.Reques
128128

129129
result, err := app.repositories.Files.ListFiles(ctx, params)
130130
if err != nil {
131-
app.serverErrorResponse(w, r, err)
131+
app.handleLlamaStackClientError(w, r, err)
132132
return
133133
}
134134

@@ -156,7 +156,7 @@ func (app *App) LlamaStackDeleteFileHandler(w http.ResponseWriter, r *http.Reque
156156

157157
err := app.repositories.Files.DeleteFile(ctx, fileID)
158158
if err != nil {
159-
app.serverErrorResponse(w, r, err)
159+
app.handleLlamaStackClientError(w, r, err)
160160
return
161161
}
162162

packages/gen-ai/bff/internal/api/lsd_models_handler.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ func (app *App) LlamaStackModelsHandler(w http.ResponseWriter, r *http.Request,
1717

1818
models, err := app.repositories.Models.ListModels(ctx)
1919
if err != nil {
20-
app.serverErrorResponse(w, r, err)
20+
app.handleLlamaStackClientError(w, r, err)
2121
return
2222
}
2323

packages/gen-ai/bff/internal/api/lsd_vectorstores_handler.go

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ func (app *App) LlamaStackListVectorStoresHandler(w http.ResponseWriter, r *http
8888
// Get all vectorstores
8989
vectorStores, err := app.repositories.VectorStores.ListVectorStores(ctx, params)
9090
if err != nil {
91-
app.serverErrorResponse(w, r, err)
91+
app.handleLlamaStackClientError(w, r, err)
9292
return
9393
}
9494

@@ -117,7 +117,7 @@ func (app *App) LlamaStackListVectorStoresHandler(w http.ResponseWriter, r *http
117117

118118
newVectorStore, err := app.repositories.VectorStores.CreateVectorStore(ctx, createParams)
119119
if err != nil {
120-
app.serverErrorResponse(w, r, err)
120+
app.handleLlamaStackClientError(w, r, err)
121121
return
122122
}
123123

@@ -160,7 +160,7 @@ func (app *App) LlamaStackCreateVectorStoreHandler(w http.ResponseWriter, r *htt
160160

161161
vectorStore, err := app.repositories.VectorStores.CreateVectorStore(ctx, params)
162162
if err != nil {
163-
app.serverErrorResponse(w, r, err)
163+
app.handleLlamaStackClientError(w, r, err)
164164
return
165165
}
166166

@@ -187,7 +187,7 @@ func (app *App) LlamaStackDeleteVectorStoreHandler(w http.ResponseWriter, r *htt
187187

188188
err := app.repositories.VectorStores.DeleteVectorStore(ctx, vectorStoreID)
189189
if err != nil {
190-
app.serverErrorResponse(w, r, err)
190+
app.handleLlamaStackClientError(w, r, err)
191191
return
192192
}
193193

@@ -244,7 +244,7 @@ func (app *App) LlamaStackListVectorStoreFilesHandler(w http.ResponseWriter, r *
244244

245245
result, err := app.repositories.VectorStores.ListVectorStoreFiles(ctx, vectorStoreID, params)
246246
if err != nil {
247-
app.serverErrorResponse(w, r, err)
247+
app.handleLlamaStackClientError(w, r, err)
248248
return
249249
}
250250

@@ -317,7 +317,7 @@ func (app *App) LlamaStackDeleteVectorStoreFileHandler(w http.ResponseWriter, r
317317
// Step 1: Remove file from vectorstore
318318
err := app.repositories.VectorStores.DeleteVectorStoreFile(ctx, vectorStoreID, fileID)
319319
if err != nil {
320-
app.serverErrorResponse(w, r, err)
320+
app.handleLlamaStackClientError(w, r, err)
321321
return
322322
}
323323

0 commit comments

Comments
 (0)