check for any Runnable in call_api()

Signed-off-by: Ishaan Mehta <45380942+ishaan-mehta@users.noreply.github.com>

mlflow/langchain/api_request_parallel_processor.py

@@ -179,8 +179,7 @@ def call_api(
         Calls the LangChain API and stores results.
         """
         from langchain.schema import BaseRetriever
-
-        from mlflow.langchain.utils import lc_runnables_types
+        from langchain_core.runnables import Runnable
 
         _logger.debug(f"Request #{self.index} started with payload: {self.request_json}")
 
@@ -191,7 +190,7 @@ def call_api(
                 response = [
                     {"page_content": doc.page_content, "metadata": doc.metadata} for doc in docs
                 ]
-            elif isinstance(self.lc_model, lc_runnables_types()):
+            elif isinstance(self.lc_model, Runnable):
                 if isinstance(self.request_json, dict):
                     # This is a temporary fix for the case when spark_udf converts
                     # input into pandas dataframe with column name, while the model

pyproject.toml

@@ -162,10 +162,6 @@ extend-exclude = [
   "tests/protos",
 ]
 
-[tool.ruff.format]
-docstring-code-format = true
-docstring-code-line-length = 88
-
 [tool.ruff.lint]
 dummy-variable-rgx = "^(_+|(_+[a-zA-Z0-9_]*[a-zA-Z0-9]+?))$"
 select = [