mark _summarize_prompt() private

TheDataGuild · Sep 28, 2023 · 2f45bc2 · 2f45bc2
1 parent 03f678c
commit 2f45bc2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/mind_palace/welcome.py b/mind_palace/welcome.py
@@ -13,7 +13,7 @@ def parse_abstracts(nodes) -> List[str]:
     ]
 
 
-def summarize_prompt(abstracts: List[str]):
+def _summarize_prompt(abstracts: List[str]):
     bullet_points = "\n".join([f"* {text}" for text in abstracts])
     return {
         "system": (
@@ -27,7 +27,7 @@ def summarize_prompt(abstracts: List[str]):
 
 
 def summarize(gpt_model, texts: List[str]):
-    prompt = summarize_prompt(texts)
+    prompt = _summarize_prompt(texts)
     messages = [
         ChatMessage(role="system", content=prompt["system"]),
         ChatMessage(role="user", content=prompt["user"]),

diff --git a/tests/unit/test_welcome.py b/tests/unit/test_welcome.py
@@ -25,7 +25,7 @@ def test_parse_abstracts():
 
 def test_summarize_prompt():
     abstracts = ["this is abstract", "second abstract"]
-    prompt = w.summarize_prompt(abstracts)
+    prompt = w._summarize_prompt(abstracts)
     assert isinstance(prompt, dict)
     assert isinstance(prompt["system"], str)
     assert re.search(r"'''\* this is abstract\n\* second abstract'''", prompt["user"])