diff --git a/lang_agent/pipeline.py b/lang_agent/pipeline.py index 7a8afa9..9c62035 100644 --- a/lang_agent/pipeline.py +++ b/lang_agent/pipeline.py @@ -105,7 +105,7 @@ class Pipeline: else: logger.info(f"show graph not supported for {type(self.graph)}") - def invoke(self, *nargs, **kwargs): + def invoke(self, *nargs, **kwargs)->str: out = self.graph.invoke(*nargs, **kwargs) # If streaming, yield chunks from the generator