Skip to content
This repository was archived by the owner on Nov 3, 2023. It is now read-only.

[RAG] Handle Self Chat #4207

Merged
merged 1 commit into from
Nov 22, 2021
Merged

[RAG] Handle Self Chat #4207

merged 1 commit into from
Nov 22, 2021

Conversation

klshuster
Copy link
Contributor

Patch description
Update RAG to ensure it handles self-chat. This primarily involves ensuring that the query_key is in the observation before accessing it.

Addresses #4202

Testing steps
Added test, confirmed it failed pre-fix and works post-fix

@jxmsML
Copy link
Contributor

jxmsML commented Nov 22, 2021

btw, I've notice the KeyError when running interactive (in human-model chat, not selfchat) before, but unable to reproduce.

Copy link
Contributor

@jxmsML jxmsML left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@klshuster
Copy link
Contributor Author

btw, I've notice the KeyError when running interactive (in human-model chat, not selfchat) before, but unable to reproduce.

indeed it's a related error

@klshuster klshuster merged commit 5d15e28 into main Nov 22, 2021
@klshuster klshuster deleted the fix_rag branch November 22, 2021 23:03
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants