test case

Author: anthonyjgrove

src/helpers/__tests__/openAi.test.ts

@@ -1,3 +1,5 @@
+import { ChatCompletionFunctionMessageParam } from 'openai/resources'
+
 import { createPromptConfigurationFixture } from '../../test/__fixtures__/promptConfiguration'
 import { createPromptMessageFixture } from '../../test/__fixtures__/promptMessage'
 import { createPromptToolFixture } from '../../test/__fixtures__/promptTool'
@@ -81,6 +83,17 @@ describe('openAi helpers', () => {
     it('should correctly map system messages', () => {
       expect(mapOpenAIMessagesToMessages(mapMessagesToOpenAIMessages([mockMessages[3]]))).toEqual([mockMessages[3]])
     })
+
+    it('should support function calls', () => {
+      const message: ChatCompletionFunctionMessageParam = {
+        content: '{}',
+        role: 'function',
+        name: 'workflowResponse'
+      }
+
+      const result = mapOpenAIMessagesToMessages([message])
+      expect(result[0].role).toEqual('TOOL')
+    })
   })
 
   describe('mapPromptToOpenAIConfig', () => {

src/helpers/openAi.ts

@@ -77,7 +77,7 @@ function isToolMessage(message: ChatCompletionMessageParam): message is ChatComp
 }
 
 function isFunctionMessage(message: ChatCompletionMessageParam): message is ChatCompletionFunctionMessageParam {
-  return message.role === 'tool'
+  return message.role === 'function'
 }
 
 function isAssistantMessage(message: ChatCompletionMessageParam): message is ChatCompletionAssistantMessageParam {