Merge pull request #65 from thedadams/add-tool-cat

chore: add ToolCategory
gptscript-ai · Jun 27, 2024 · 0a63991 · 0a63991
2 parents 954d73d + 45c34d9
commit 0a63991
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/gptscript.ts b/src/gptscript.ts
@@ -710,6 +710,15 @@ export enum RunState {
 	Error = "error"
 }
 
+export enum ToolCategory {
+	ProviderToolCategory = "provider",
+	CredentialToolCategory = "credential",
+	ContextToolCategory = "context",
+	InputToolCategory = "input",
+	OutputToolCategory = "output",
+	NoCategory = ""
+}
+
 export interface RunFrame {
 	id: string
 	type: RunEventType.RunStart | RunEventType.RunFinish
@@ -750,7 +759,7 @@ export interface CallFrame {
 	agentGroup?: ToolReference[]
 	displayText?: string
 	inputContext: InputContext[]
-	toolCategory?: string
+	toolCategory?: ToolCategory
 	toolName: string
 	parentID?: string
 	type: RunEventType.CallStart | RunEventType.CallChat | RunEventType.CallConfirm | RunEventType.CallContinue | RunEventType.CallSubCalls | RunEventType.CallProgress | RunEventType.CallFinish