Fix more lint in libs/oci

Author: paxiaatucsdedu

libs/oci/langchain_oci/chat_models/oci_data_science.py

@@ -31,12 +31,7 @@
     agenerate_from_stream,
     generate_from_stream,
 )
-from langchain_core.messages import (
-    AIMessage,
-    AIMessageChunk,
-    BaseMessage,
-    BaseMessageChunk,
-)
+from langchain_core.messages import AIMessageChunk, BaseMessage, BaseMessageChunk
 from langchain_core.output_parsers import (
     JsonOutputParser,
     PydanticOutputParser,
@@ -772,7 +767,7 @@ def bind_tools(
         self,
         tools: Sequence[Union[Dict[str, Any], Type[BaseModel], Callable, BaseTool]],
         **kwargs: Any,
-    ) -> Runnable[LanguageModelInput, AIMessage]:
+    ) -> Runnable[LanguageModelInput, BaseMessage]:
         formatted_tools = [convert_to_openai_tool(tool) for tool in tools]
         return super().bind(tools=formatted_tools, **kwargs)
 

libs/oci/langchain_oci/chat_models/oci_generative_ai.py

@@ -924,8 +924,8 @@ def convert_to_oci_tool(
                     "required": parameters.get("required", []),
                 },
             )
-        elif isinstance(tool, BaseTool):
-            return self.oci_function_definition(
+        elif isinstance(tool, BaseTool):  # type: ignore[unreachable]
+            return self.oci_function_definition(  # type: ignore[unreachable]
                 name=tool.name,
                 description=OCIUtils.remove_signature_from_tool_description(
                     tool.name, tool.description
@@ -1205,7 +1205,7 @@ def bind_tools(
             Union[dict, str, Literal["auto", "none", "required", "any"], bool]
         ] = None,
         **kwargs: Any,
-    ) -> Runnable[LanguageModelInput, AIMessage]:
+    ) -> Runnable[LanguageModelInput, BaseMessage]:
         """Bind tool-like objects to this chat model.
 
         Assumes model is compatible with Meta's tool-calling API.

libs/oci/tests/unit_tests/chat_models/test_oci_data_science.py

@@ -155,7 +155,7 @@ def test_stream_vllm(*args: Any) -> None:
         if output is None:
             output = chunk
         else:
-            output += chunk
+            output += chunk  # type: ignore[assignment]
         count += 1
     assert count == 5
     assert output is not None