Update oai_base.py

Attempt at a resolution for #14 . After replicating the issue today

assistant_manager/oai_base.py

@@ -3,7 +3,7 @@
 from typing import List, Optional
 from openai import OpenAI
 from openai._types import NotGiven, NOT_GIVEN
-from openai.types.beta.threads import ThreadMessage
+from openai.types.beta.threads import Message as ThreadMessage
 from assistant_manager.interface_base import InterfaceBase
 
 
@@ -580,4 +580,4 @@ def submit_tool_outputs(self, thread_id, run_id, tool_outputs):
         )
         self.logger.debug(f"Submitted tool outputs for run {run_id}")
 
-        return run
+        return run