Skip to content

Commit 4f0fb6f

Browse files
authored
fix(brain): pass missing run_id (#3631)
# Description Please include a summary of the changes and the related issue. Please also include relevant motivation and context. ## Checklist before requesting a review Please delete options that are not relevant. - [ ] My code follows the style guidelines of this project - [ ] I have performed a self-review of my code - [ ] I have commented hard-to-understand areas - [ ] I have ideally added tests that prove my fix is effective or that my feature works - [ ] New and existing unit tests pass locally with my changes - [ ] Any dependent changes have been merged ## Screenshots (if appropriate):
1 parent 2898a5d commit 4f0fb6f

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

core/quivr_core/brain/brain.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -496,6 +496,7 @@ def add_file(self) -> None:
496496
async def ask_streaming(
497497
self,
498498
question: str,
499+
run_id: UUID,
499500
system_prompt: str | None = None,
500501
retrieval_config: RetrievalConfig | None = None,
501502
rag_pipeline: Type[Union[QuivrQARAG, QuivrQARAGLangGraph]] | None = None,
@@ -542,6 +543,7 @@ async def ask_streaming(
542543
"langfuse_session_id": str(self.chat_id),
543544
}
544545
async for response in rag_instance.answer_astream(
546+
run_id=run_id,
545547
question=question,
546548
system_prompt=system_prompt or None,
547549
history=chat_history,
@@ -585,6 +587,7 @@ async def aask(
585587
full_answer = ""
586588

587589
async for response in self.ask_streaming(
590+
run_id=run_id,
588591
question=question,
589592
system_prompt=system_prompt,
590593
retrieval_config=retrieval_config,

0 commit comments

Comments
 (0)