fix(cohere): pass toolResults only when needed to cohere endpoint type (

#1382)
huggingface · Aug 2, 2024 · 6de97af · 6de97af
1 parent e5c2b96
commit 6de97af
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/src/lib/server/endpoints/cohere/endpointCohere.ts b/src/lib/server/endpoints/cohere/endpointCohere.ts
@@ -91,12 +91,15 @@ export async function endpointCohere(
 						stopSequences: parameters?.stop,
 						frequencyPenalty: parameters?.frequency_penalty,
 						tools,
-						toolResults: toolResults?.map((toolResult) => {
-							if (toolResult.status === ToolResultStatus.Error) {
-								return { call: toolResult.call, outputs: [{ error: toolResult.message }] };
-							}
-							return { call: toolResult.call, outputs: toolResult.outputs };
-						}),
+						toolResults:
+							toolResults?.length && toolResults?.length > 0
+								? toolResults?.map((toolResult) => {
+										if (toolResult.status === ToolResultStatus.Error) {
+											return { call: toolResult.call, outputs: [{ error: toolResult.message }] };
+										}
+										return { call: toolResult.call, outputs: toolResult.outputs };
+								  })
+								: undefined,
 					})
 					.catch(async (err) => {
 						if (!err.body) throw err;