feat: add curated_topic.py to llm interface

Sefaria · Apr 1, 2024 · bba5c9f · bba5c9f
1 parent ce0ce92
commit bba5c9f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/app/llm_interface/sefaria_llm_interface/topic_source_curation/__init__.py b/app/llm_interface/sefaria_llm_interface/topic_source_curation/__init__.py
@@ -0,0 +1 @@
+from curated_topic import *
diff --git a/app/llm_interface/sefaria_llm_interface/topic_source_curation/curated_topic.py b/app/llm_interface/sefaria_llm_interface/topic_source_curation/curated_topic.py
@@ -0,0 +1,8 @@
+from dataclasses import dataclass
+from sefaria_llm_interface.common.topic import Topic
+from sefaria_llm_interface.topic_prompt import TopicPromptSource
+
+@dataclass
+class CuratedTopic:
+    topic: Topic
+    sources: list[TopicPromptSource]