Add error handling for response validation in providers

TarsLab · Nov 4, 2024 · d1ed8a3 · d1ed8a3
1 parent 401a924
commit d1ed8a3
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 0 deletions.
diff --git a/src/providers/claude.ts b/src/providers/claude.ts
@@ -69,6 +69,14 @@ const sendRequestFunc = (settings: ClaudeOptions): SendRequest =>
 			agent: proxyUrl ? new HttpsProxyAgent(proxyUrl) : undefined
 		})
 
+		if (!response || !response.body) {
+			throw new Error('No response')
+		}
+		if (!response.ok) {
+			console.error('response', response)
+			throw new Error(`Unexpected response status: ${response.status} ${response.statusText}`)
+		}
+
 		// @ts-ignore
 		const stream = Stream.fromSSEResponse<PartialEvent>(response, new AbortController())
 

diff --git a/src/providers/doubao.ts b/src/providers/doubao.ts
@@ -28,6 +28,10 @@ const sendRequestFunc = (settings: BaseOptions): SendRequest =>
 		if (!response || !response.body) {
 			throw new Error('No response')
 		}
+		if (!response.ok) {
+			console.error('response', response)
+			throw new Error(`Unexpected response status: ${response.status} ${response.statusText}`)
+		}
 		const decoder = new TextDecoder('utf-8')
 		for await (const chunk of response.body) {
 			const lines = decoder.decode(Buffer.from(chunk))

diff --git a/src/providers/qianFan.ts b/src/providers/qianFan.ts
@@ -107,6 +107,11 @@ const sendRequestFunc = (settings: QianFanOptions): SendRequest =>
 		if (!response || !response.body) {
 			throw new Error('No response')
 		}
+		if (!response.ok) {
+			console.error('response', response)
+			throw new Error(`Unexpected response status: ${response.status} ${response.statusText}`)
+		}
+
 		const buffer: string[] = []
 		const decoder = new TextDecoder('utf-8')
 		for await (const chunk of response.body) {