From 6597f1745f3a27f67471bfc04bdef4f6505b52f6 Mon Sep 17 00:00:00 2001
From: aandrew-me <aandrew.me@proton.me>
Date: Sat, 7 Dec 2024 19:55:54 +0300
Subject: [PATCH] Test fixes

---
 providers/koboldai/koboldai_test.go | 18 ++++++++++++++----
 providers/phind/phind_test.go       | 10 ++++++++++
 2 files changed, 24 insertions(+), 4 deletions(-)

diff --git a/providers/koboldai/koboldai_test.go b/providers/koboldai/koboldai_test.go
index 6ce1555..f3f7a29 100644
--- a/providers/koboldai/koboldai_test.go
+++ b/providers/koboldai/koboldai_test.go
@@ -11,11 +11,21 @@ import (
 
 func TestRequest(t *testing.T) {
 	resp, err := NewRequest("What is 1+1", structs.Params{
-		Provider: "koboldai",
+		Provider:    "koboldai",
 		Temperature: "0.5",
-		Top_p: "0.5",
-		Max_length: "300",
+		Top_p:       "0.5",
+		Max_length:  "300",
 	})
+
+	if err != nil {
+        t.Fatalf("NewRequest returned an error: %v", err)
+
+	}
+
+	if resp == nil {
+        t.Fatalf("NewRequest returned a nil response")
+	}
+
 	body, _ := io.ReadAll(resp.Body)
 
 	assert := assert.New(t)
@@ -25,4 +35,4 @@ func TestRequest(t *testing.T) {
 
 	assert.Nil(err, "Error should be Nil")
 	assert.Equal(200, resp.StatusCode, "Response status code should be 200")
-}
\ No newline at end of file
+}
diff --git a/providers/phind/phind_test.go b/providers/phind/phind_test.go
index fa39b71..5b290ca 100644
--- a/providers/phind/phind_test.go
+++ b/providers/phind/phind_test.go
@@ -11,6 +11,16 @@ import (
 
 func TestRequest(t *testing.T) {
 	resp, err := NewRequest("What is 1+1", structs.Params{})
+
+	if err != nil {
+        t.Fatalf("NewRequest returned an error: %v", err)
+
+	}
+
+	if resp == nil {
+        t.Fatalf("NewRequest returned a nil response")
+	}
+	
 	body, _ := io.ReadAll(resp.Body)
 
 	assert := assert.New(t)