x

langserve/server.py

@@ -516,7 +516,6 @@ def __init__(
                 dictionary form. Note that only keys in `config_keys` will be
                 modifiable by this function.
         """
-        self.runnable = runnable
 
         try:
             from sse_starlette import EventSourceResponse
@@ -558,6 +557,8 @@ def __init__(
         if with_types:
             runnable = runnable.with_types(**with_types)
 
+        self.runnable = runnable
+
         model_namespace = _replace_non_alphanumeric_with_underscores(path.strip("/"))
 
         input_type_ = _resolve_model(
@@ -655,7 +656,7 @@ async def invoke(
 
         event_aggregator = AsyncEventAggregatorCallback()
         config["callbacks"] = [event_aggregator]
-        output = await self.runnable.ainvoke(input_, config)
+        output = await self.runnable.ainvoke(input_, config=config)
 
         if self.include_callback_events:
             callback_events = [