Task: Handle http errors better

zya · Oct 25, 2023 · 71cd932 · 71cd932
1 parent e0e8813
commit 71cd932
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/handlers/ollama.ts b/src/handlers/ollama.ts
@@ -114,6 +114,12 @@ export async function OllamaHandler(
 
   const res = await getOllamaResponse(model, prompt, baseUrl);
 
+  if (!res.ok) {
+    throw new Error(
+      `Recieved an error with code ${res.status} from Ollama API.`,
+    );
+  }
+
   if (params.stream) {
     return iterateResponse(res, model, prompt);
   }

diff --git a/src/handlers/ollamaEmbedding.ts b/src/handlers/ollamaEmbedding.ts
@@ -30,6 +30,12 @@ export async function OllamaEmbeddingHandler(
   const model = params.model.split('ollama/')[1];
   const baseUrl = params.baseUrl ?? 'http://127.0.0.1:11434';
   const response = await getOllamaResponse(model, params.input, baseUrl);
+
+  if (!response.ok) {
+    throw new Error(
+      `Recieved an error with code ${response.status} from Ollama API.`,
+    );
+  }
   const body = (await response.json()) as OllamaEmbeddingsResponseChunk;
 
   return {