Add endpoint to submit tool output

Author: adinapoli

openai-api-servant/src/OpenAI/Api.hs

@@ -155,6 +155,11 @@ type ThreadsApi =
     :<|> OpenAIAuth :> AzureAPIVer
                     :> Capture "thread_id" ThreadId :> "runs"
                     :> Capture "run_id" RunId :> Get '[JSON] Run
+    :<|> OpenAIAuth :> AzureAPIVer
+                    :> Capture "thread_id" ThreadId :> "runs"
+                    :> Capture "run_id" RunId :> "submit_tool_outputs"
+                    :> ReqBody '[JSON] ToolOutputs
+                    :> Post '[JSON] Run
     :<|> OpenAIAuth :> AzureAPIVer
                     :> Capture "thread_id" ThreadId :> "runs"
                     :> ReqBody '[JSON] RunCreate :> Post '[JSON] Run

openai-api-servant/src/OpenAI/Resources.hs

@@ -135,6 +135,8 @@ module OpenAI.Resources
     RunStatus(..),
     RunRequiredAction(..),
     SubmitToolOutputs(..),
+    ToolOutput(..),
+    ToolOutputs(..),
     ThreadAndRunCreate(..),
 
     -- * Vector Stores
@@ -1353,6 +1355,23 @@ data RunCreate = RunCreate
 
 $(deriveJSON (jsonOpts 3) ''RunCreate)
 
+data ToolOutput = ToolOutput
+  { toToolCallId :: Maybe T.Text
+  , toOutput     :: Maybe T.Text
+  }
+  deriving stock (Show, Eq, Generic)
+  deriving anyclass NFData
+
+$(deriveJSON (jsonOpts 2) ''ToolOutput)
+
+data ToolOutputs = ToolOutputs
+  { tosToolOutputs :: [ToolOutput]
+  }
+  deriving stock (Show, Eq, Generic)
+  deriving anyclass NFData
+
+$(deriveJSON (jsonOpts 3) ''ToolOutputs)
+
 -- | Create a thread and run it in one request.
 data ThreadAndRunCreate = ThreadAndRunCreate
   { tarcAssistantId  :: AssistantId

openai-api/src/OpenAI/Client.hs

@@ -142,10 +142,13 @@ module OpenAI.Client
     RunStatus(..),
     RunRequiredAction(..),
     SubmitToolOutputs(..),
+    ToolOutput(..),
+    ToolOutputs(..),
     ThreadAndRunCreate(..),
     createRun,
     createThreadAndRun,
     getRun,
+    submitToolOutputs,
     cancelRun,
 
     -- * Fine tunes (out of date)
@@ -346,6 +349,7 @@ EP3 (createRun, Maybe String, ThreadId, RunCreate, Run)
 EP3 (cancelRun, Maybe String, ThreadId, RunId, Run)
 EP2 (createThreadAndRun, Maybe String, ThreadAndRunCreate, Run)
 EP3 (getRun, Maybe String, ThreadId, RunId, Run)
+EP4 (submitToolOutputs, Maybe String, ThreadId, RunId, ToolOutputs, Run)
 
 EP2 (createVectorStore, Maybe String, VectorStoreCreate, VectorStore)
 EP2 (deleteVectorStore, Maybe String, VectorStoreId, DeleteConfirmation)
@@ -384,6 +388,7 @@ completeChatStreaming' :: Token -> ChatCompletionRequest -> Maybe String -> Clie
     :<|> ( createThread'
             :<|> createThreadAndRun'
             :<|> getRun'
+            :<|> submitToolOutputs'
             :<|> createRun'
             :<|> cancelRun'
             :<|> deleteThread'