statements.py

from openai_lib.api import Gpt4oMiniInterface
from pydantic import BaseModel
import shelve


class StatemtExtractor(BaseModel):
    statements: list[str]


api = Gpt4oMiniInterface()

instructions = """
You are a math professor teaching a class about logic, specifically about statements. 
To ilustrate the subject, you ask the students to give an example of a sentence, containg atomic propositions. 
You will, then, write the atomic propositions that can be stracted from the example. 
A atomic propositions is a declaration of a sigle fact. 
A sentence can contain multiple atomic propositions. 

The example given by the student will be inputed, output the atomic propositions that can be extracted from the example. 
"""

statement_resolver = api.of_format(StatemtExtractor, instructions)


def main():

    with shelve.open('extracted') as db:
        affirmative = db["affirmative"]

    extracted = statement_resolver(affirmative)

    with shelve.open('extracted') as db:
        db["statements"] = extracted.statements

    for s in extracted.statements:
        print(s)