get as_raw

This commit is contained in:
2025-10-24 16:14:45 +08:00
parent 2056dc3d75
commit 0c89f8cac2

View File

@@ -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