Try catches, ensuring SuccessStatusCode

Services/ApiService.cs

@@ -1,6 +1,3 @@
-using System.Net.Http.Headers;
-using PaperlessAI.Models.LMStudio;
-
 namespace PaperlessAI.Services;
 
 public class ApiService(HttpClient client)

Services/DocumentProcessingService.cs

@@ -84,31 +84,54 @@ public class DocumentProcessingService(IServiceScopeFactory scopeFactory, ILogge
 
     private async Task UnloadModel()
     {
-        await using var scope = scopeFactory.CreateAsyncScope();
-        var agentService = scope.ServiceProvider.GetRequiredService<LLMAgentService>();
+        try
+        {
+            isWorking = true;
+            await using var scope = scopeFactory.CreateAsyncScope();
+            var agentService = scope.ServiceProvider.GetRequiredService<LLMAgentService>();
 
-        logger.LogInformation($"Unloading Model");
-        isWorking = true;
-        await agentService.UnloadModel();
-        isWorking = false;
+            logger.LogInformation($"Unloading Model");
+            await agentService.UnloadModel();
+        }
+        catch (Exception e)
+        {
+            logger.LogError(e.ToString());
+            throw;
+        }
+        finally
+        {
+            isWorking = false;
+        }
+        
     }
 
     private async Task<LmStudioLlmResponse> ProcessDocument(PaperlessDocument docToProcess)
     {
-        await using var scope = scopeFactory.CreateAsyncScope();
-        var agentService = scope.ServiceProvider.GetRequiredService<LLMAgentService>();
-        var repository = scope.ServiceProvider.GetRequiredService<PaperlessDocumentRepository>();
+        var llmResponse = new LmStudioLlmResponse();
+        try
+        {
+            isWorking = true;
 
-        isWorking = true;
+            await using var scope = scopeFactory.CreateAsyncScope();
+            var agentService = scope.ServiceProvider.GetRequiredService<LLMAgentService>();
+            var repository = scope.ServiceProvider.GetRequiredService<PaperlessDocumentRepository>();
 
-        logger.LogInformation($"Processing: {docToProcess}");
+            logger.LogInformation($"Processing: {docToProcess}");
 
-        docToProcess.DocumentState = EDocumentState.Started;
-        await repository.Update(docToProcess);
-        
-        var llmResponse = await agentService.ProcessDocumentWithAI(docToProcess);
-        
-        isWorking = false;
+            docToProcess.DocumentState = EDocumentState.Started;
+            await repository.Update(docToProcess);
+            
+            llmResponse = await agentService.ProcessDocumentWithAI(docToProcess);
+        }
+        catch (Exception e)
+        {
+            logger.LogError(e.ToString());
+            throw;
+        }
+        finally
+        {
+            isWorking = false;
+        }
         return llmResponse ?? new LmStudioLlmResponse();
     }
 }

Services/LLMAgentService.cs

@@ -24,6 +24,7 @@ public class LLMAgentService(ApiService apiService, Configuration config, ILogge
                 Reasoning = config.Reasoning
             };
         var response = await apiService.Post("chat", content);
+        response.EnsureSuccessStatusCode();
 
         // Response
         var contentMessage = await response.Content.ReadAsStringAsync();
@@ -55,6 +56,7 @@ public class LLMAgentService(ApiService apiService, Configuration config, ILogge
         };
 
         var response = await apiService.Post("models/unload", content);
+        response.EnsureSuccessStatusCode();
         var message = await response.Content.ReadAsStringAsync();
         logger.LogInformation(message);
     }