Refactor imports and update llm_provider in bible_verses.py example

examples/bible_verses.py

@@ -6,7 +6,7 @@ from rich.text import Text
 
 console = Console()
 
-claude = sm.Session(llm_provider="anthropic")
+claude = sm.Session(llm_provider="openai")
 
 
 class BibleVerse(BaseModel):
@@ -17,53 +17,82 @@ class BibleVerse(BaseModel):
     translation: str
 
 
+class BiblePassage(BaseModel):
+    book: str
+    chapter: int
+    verses: list[BibleVerse]
+    translation: str
+
+
 class CrossReference(BaseModel):
-    verse: BibleVerse
-    notes: str
+    passage: BiblePassage
+    notes: list[str]
     origin_verse: BibleVerse
 
 
-def get_random_bible_verse() -> BibleVerse:
-    verse = claude.generate_data(
-        prompt="Get a random Bible verse", response_model=BibleVerse
+def get_passage(book: str, chapter: int, translation: str = "ESV") -> BiblePassage:
+    passage = claude.generate_data(
+        prompt=f"""Return {book} chapter {chapter} from the {translation} translation.
+Format each verse as plain text without any special characters or formatting.
+For example:
+- "Love is patient, love is kind."
+- "It does not envy, it does not boast"
+
+Return only the biblical text, formatted as a BiblePassage object.""",
+        response_model=BiblePassage,
+        max_tokens=8000,
     )
-
-    return verse
+    return passage
 
 
-def get_cross_reference(verse: BibleVerse) -> CrossReference:
+def get_cross_reference(passage: BiblePassage) -> CrossReference:
+    verses_text = "\n".join([f"Verse {v.verse}: {v.text}" for v in passage.verses])
     ref = claude.generate_data(
-        prompt=f"Get a cross reference for the verse: {verse.text}",
+        prompt=f"""Find a thematically related Bible passage that connects with this text:
+{verses_text}
+
+Return a CrossReference object with:
+1. A related passage (using plain text without special characters)
+2. A list of clear, specific notes explaining the thematic connections
+3. The original passage included""",
         response_model=CrossReference,
     )
-
     return ref
 
 
 def pretty_print_reference(ref: CrossReference):
-    # Create origin verse panel
+    # Create origin passage panel
     origin_text = Text()
     origin_text.append(
-        f"{ref.origin_verse.book} {ref.origin_verse.chapter}:{ref.origin_verse.verse}\n",
+        f"{ref.origin_verse.book} {ref.origin_verse.chapter}\n",
         style="bold blue",
     )
+    origin_text.append(f"{ref.origin_verse.verse}. ", style="blue")
     origin_text.append(f"{ref.origin_verse.text}\n", style="italic")
-    origin_text.append(f"({ref.origin_verse.translation})", style="dim")
+    origin_text.append(f"\n({ref.origin_verse.translation})", style="dim")
 
-    origin_panel = Panel(origin_text, title="Original Verse", border_style="blue")
+    origin_panel = Panel(origin_text, title="Original Passage", border_style="blue")
 
     # Create cross reference panel
     ref_text = Text()
     ref_text.append(
-        f"{ref.verse.book} {ref.verse.chapter}:{ref.verse.verse}\n", style="bold green"
+        f"{ref.passage.book} {ref.passage.chapter}\n",
+        style="bold green",
     )
-    ref_text.append(f"{ref.verse.text}\n", style="italic")
-    ref_text.append(f"({ref.verse.translation})", style="dim")
+    for verse in ref.passage.verses:
+        ref_text.append(f"{verse.verse}. ", style="green")
+        ref_text.append(f"{verse.text}\n", style="italic")
+    ref_text.append(f"\n({ref.passage.translation})", style="dim")
 
     ref_panel = Panel(ref_text, title="Cross Reference", border_style="green")
 
-    # Create notes panel
-    notes_panel = Panel(ref.notes, title="Connection Notes", border_style="yellow")
+    # Create notes panel with bullet points
+    notes_text = Text()
+    for note in ref.notes:
+        notes_text.append("• ", style="yellow")
+        notes_text.append(f"{note}\n")
+
+    notes_panel = Panel(notes_text, title="Thematic Connections", border_style="yellow")
 
     # Print all panels
     console.print(origin_panel)
@@ -72,6 +101,7 @@ def pretty_print_reference(ref: CrossReference):
 
 
 if __name__ == "__main__":
-    verse = get_random_bible_verse()
-    ref = get_cross_reference(verse)
+    # Get 1 Corinthians 13 (The Love Chapter)
+    passage = get_passage("1 Corinthians", 13)
+    ref = get_cross_reference(passage)
     pretty_print_reference(ref)