diff --git a/lang_agent/pipeline.py b/lang_agent/pipeline.py index 12c72bc..566ba88 100644 --- a/lang_agent/pipeline.py +++ b/lang_agent/pipeline.py @@ -77,6 +77,10 @@ class Pipeline: def invoke(self, *nargs, **kwargs)->str: out = self.graph.invoke(*nargs, **kwargs) + + if kwargs.get("as_raw"): + return out + if isinstance(out, SystemMessage) or isinstance(out, HumanMessage): return out.content