| 118 | ) |
| 119 | |
| 120 | def invoke( |
| 121 | self, |
| 122 | input: Dict[str, Any], |
| 123 | config: Optional[RunnableConfig] = None, |
| 124 | **kwargs: Any, |
| 125 | ) -> Dict[str, Any]: |
| 126 | config = ensure_config(config) |
| 127 | callbacks = config.get("callbacks") |
| 128 | tags = config.get("tags") |
| 129 | metadata = config.get("metadata") |
| 130 | run_name = config.get("run_name") or self.get_name() |
| 131 | run_id = config.get("run_id") |
| 132 | include_run_info = kwargs.get("include_run_info", False) |
| 133 | return_only_outputs = kwargs.get("return_only_outputs", False) |
| 134 | |
| 135 | inputs = self.prep_inputs(input) |
| 136 | callback_manager = CallbackManager.configure( |
| 137 | callbacks, |
| 138 | self.callbacks, |
| 139 | self.verbose, |
| 140 | tags, |
| 141 | self.tags, |
| 142 | metadata, |
| 143 | self.metadata, |
| 144 | ) |
| 145 | new_arg_supported = inspect.signature(self._call).parameters.get("run_manager") |
| 146 | |
| 147 | run_manager = callback_manager.on_chain_start( |
| 148 | dumpd(self), |
| 149 | inputs, |
| 150 | run_id, |
| 151 | name=run_name, |
| 152 | ) |
| 153 | try: |
| 154 | self._validate_inputs(inputs) |
| 155 | outputs = ( |
| 156 | self._call(inputs, run_manager=run_manager) |
| 157 | if new_arg_supported |
| 158 | else self._call(inputs) |
| 159 | ) |
| 160 | |
| 161 | final_outputs: Dict[str, Any] = self.prep_outputs( |
| 162 | inputs, outputs, return_only_outputs |
| 163 | ) |
| 164 | except BaseException as e: |
| 165 | run_manager.on_chain_error(e) |
| 166 | raise e |
| 167 | run_manager.on_chain_end(outputs) |
| 168 | |
| 169 | if include_run_info: |
| 170 | final_outputs[RUN_KEY] = RunInfo(run_id=run_manager.run_id) |
| 171 | return final_outputs |
| 172 | |
| 173 | async def ainvoke( |
| 174 | self, |