From c64d9d26d2f810e11e0bfd35f8077a8e9d7377d2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=BA=90=E6=96=87=E9=9B=A8?=
 <41315874+fumiama@users.noreply.github.com>
Date: Sat, 22 Feb 2025 21:38:07 +0900
Subject: [PATCH] feat: add error status to Request

---
 api.go   | 10 ++++++++++
 model.go | 15 ---------------
 2 files changed, 10 insertions(+), 15 deletions(-)

diff --git a/api.go b/api.go
index 2646be6..2543afe 100644
--- a/api.go
+++ b/api.go
@@ -1,7 +1,10 @@
 package deepinfra
 
 import (
+	"errors"
+	"io"
 	"net/http"
+	"strings"
 )
 
 const (
@@ -29,6 +32,13 @@ func (api *API) Request(model Model) (string, error) {
 		return "", err
 	}
 	defer resp.Body.Close()
+	if resp.StatusCode != http.StatusOK {
+		sb := strings.Builder{}
+		sb.WriteString(resp.Status)
+		sb.WriteByte(' ')
+		_, _ = io.Copy(&sb, resp.Body)
+		return "", errors.New(sb.String())
+	}
 	err = model.Parse(resp.Body)
 	if err != nil {
 		return "", err
diff --git a/model.go b/model.go
index 26e7c03..35d13f8 100644
--- a/model.go
+++ b/model.go
@@ -1,8 +1,6 @@
 package deepinfra
 
 import (
-	"fmt"
-
 	"github.com/fumiama/deepinfra/model"
 )
 
@@ -10,16 +8,3 @@ type Model interface {
 	model.Inputer
 	model.Outputer
 }
-
-func init() {
-	api := NewAPI(APIDeepInfra, "PUT YOUR API KEY HERE")
-	txt, err := api.Request(model.NewOpenAI("fumiama/ninus", "", 0.7, 0.9, 1024).
-		System("你正在QQ群与用户聊天，用户发送了消息。按自己的心情简短思考后，条理清晰地回应**一句话**，禁止回应多句。").
-		User("总不能什么都查吧").User("后面DOGE就成恶龙了 很常见的场景"),
-	)
-	if err != nil {
-		panic(err)
-	}
-	fmt.Println(txt)
-	// 要不我给你查一下？
-}