add function calling to OpenAI ChatCompletion API (#28)

Co-authored-by: bklynate <[email protected]>
Econify · May 31, 2024 · b6c9643 · b6c9643
1 parent 7a584db
commit b6c9643
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/packages/core/src/apis/openai/openAiChatApi.ts b/packages/core/src/apis/openai/openAiChatApi.ts
@@ -53,6 +53,7 @@ const templateSource = `{
   <% if (typeof tools !== 'undefined') { %>, "tools": <%= JSON.stringify(tools) %><% } %>
   <% if (typeof tool_choice !== 'undefined') { %>, "tool_choice": <%= JSON.stringify(tool_choice) %><% } %>
   <% if (typeof user !== 'undefined') { %>, "user": "<%= user %>"<% } %>
+  <% if (typeof function_call !== 'undefined') { %>, "function_call": "<%= function_call %>"<% } %>
 }`;
 
 export interface OpenAiChatOptions
@@ -97,7 +98,7 @@ export interface OpenAiChatOptions
         };
       };
   user?: string;
-  // function_call
+  function_call: string;
   // functions
 }
 
@@ -118,7 +119,13 @@ const OpenAiChatResponseCodec = t.type({
         role: t.string,
         content: t.string,
         // tool_calls
-        // function_call
+        function_call: t.union([
+          t.undefined,
+          t.type({
+            name: t.string,
+            args: t.string,
+          }),
+        ]),
       }),
       // logprobs: t.type({
       //   content: t.array(