feat: enhance conversation saving and response streaming with source handling

This commit is contained in:
Siddhant Rai
2025-03-10 14:19:43 +05:30
parent 46d32b4072
commit faa583864d
2 changed files with 88 additions and 34 deletions

View File

@@ -107,6 +107,7 @@ class ClassicAgent(BaseAgent):
if isinstance(line, str):
yield {"answer": line}
yield {"sources": retrieved_data}
yield {"tool_calls": self.tool_calls.copy()}
def _retriever_search(self, retriever, query, log_context):