diff --git a/lang_agent/pipeline.py b/lang_agent/pipeline.py index b062a2d..582f0b2 100644 --- a/lang_agent/pipeline.py +++ b/lang_agent/pipeline.py @@ -76,6 +76,9 @@ class Pipeline: def invoke(self, *nargs, as_stream:bool=False, **kwargs): + """ + as_stream (bool): for debug only, gets the agent to print its thoughts + """ if as_stream: for step in self.agent.stream(*nargs, stream_mode="values", **kwargs): @@ -116,9 +119,6 @@ class Pipeline: def chat(self, inp:str, as_stream:bool=False): - """ - as_stream (bool): for debug only, gets the agent to print its thoughts - """ inp = {"messages":[HumanMessage(inp)]}, {"configurable": {"thread_id": 3}} out = self.invoke(*inp, as_stream=as_stream)