local_qna_pair.py 554 B

12345678910111213141516171819
  1. from embedchain.helper_classes.json_serializable import register_deserializable
  2. from embedchain.loaders.base_loader import BaseLoader
  3. @register_deserializable
  4. class LocalQnaPairLoader(BaseLoader):
  5. def load_data(self, content):
  6. """Load data from a local QnA pair."""
  7. question, answer = content
  8. content = f"Q: {question}\nA: {answer}"
  9. meta_data = {
  10. "url": "local",
  11. }
  12. return [
  13. {
  14. "content": content,
  15. "meta_data": meta_data,
  16. }
  17. ]