From 2d9505466638dd4a6d78fba3d40e8bd16da8dc6f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=97=8D+85CD?= <50108258+kwaa@users.noreply.github.com>
Date: Thu, 21 Nov 2024 21:37:11 +0800
Subject: [PATCH] refactor(generate-text): return finish_reason

---
 src/lib/generate-text.ts       | 7 ++++---
 src/lib/types/finish-reason.ts | 1 +
 2 files changed, 5 insertions(+), 3 deletions(-)
 create mode 100644 src/lib/types/finish-reason.ts

diff --git a/src/lib/generate-text.ts b/src/lib/generate-text.ts
index 27d7fe0..14e2ca4 100644
--- a/src/lib/generate-text.ts
+++ b/src/lib/generate-text.ts
@@ -1,11 +1,10 @@
-import type { Message } from './types/message'
+import type { FinishReason } from './types/finish-reason'
 import type { GenerationModel } from './types/model'
 
 import { clean } from '../utils/clean'
 import { base, type CommonRequestOptions } from './common'
 
 export interface GenerateTextOptions extends CommonRequestOptions {
-  messages?: Message[]
   model: GenerationModel
   /** @default `completions` */
   path?: 'completions' | ({} & string)
@@ -14,7 +13,7 @@ export interface GenerateTextOptions extends CommonRequestOptions {
 
 export interface GenerateTextResponse {
   choices: {
-    finish_reason: 'content_filter' | 'length' | 'stop'
+    finish_reason: FinishReason
     index: number
     text: string
   }[]
@@ -33,6 +32,7 @@ export interface GenerateTextResponseUsage {
 }
 
 export interface GenerateTextResult {
+  finishReason: FinishReason
   request: Request
   response: Response
   text: string
@@ -55,6 +55,7 @@ export const generateText = async (options: GenerateTextOptions): Promise<Genera
   const json = await response.json() as GenerateTextResponse
 
   return {
+    finishReason: json.choices[0].finish_reason,
     request,
     response,
     text: json.choices[0].text,
diff --git a/src/lib/types/finish-reason.ts b/src/lib/types/finish-reason.ts
new file mode 100644
index 0000000..42865ee
--- /dev/null
+++ b/src/lib/types/finish-reason.ts
@@ -0,0 +1 @@
+export type FinishReason = 'content_filter' | 'error' | 'length' | 'other' | 'stop' | 'tool-calls' | ({} & string)