Skip to content

Commit ce95a3a

Browse files
authored
feat: metadata brain core
1 parent 7dba0dc commit ce95a3a

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

core/quivr_core/brain/brain.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -585,6 +585,7 @@ async def aask(
585585
"""
586586
# question_language = detect_language(question) -- Commented until we use it
587587
full_answer = ""
588+
metadata = None
588589

589590
async for response in self.ask_streaming(
590591
run_id=run_id,
@@ -597,8 +598,10 @@ async def aask(
597598
**input_kwargs,
598599
):
599600
full_answer += response.answer
601+
if response.metadata:
602+
metadata = response.metadata
600603

601-
return ParsedRAGResponse(answer=full_answer)
604+
return ParsedRAGResponse(answer=full_answer, metadata=metadata)
602605

603606
def ask(
604607
self,

0 commit comments

Comments
 (0)