affirmative_interrogative.py

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

'''
Classify the statements as affirmative or interrogative (prolog propositions and questions).
'''

class AffimativeInterrogative(BaseModel):
    affirmative: str
    interrogative: str


api = Gpt4oMiniInterface()

instructions = """
You are a english teacher explaining to your students the difference between affirmative and interrogative sentences.
You will recieve a short paregraph that states facts and in then asks a question.
You must extract the affirmative and interrogative sentences from the paragraph without changing the original text.
"""

extractor_resolver = api.of_format(AffimativeInterrogative, instructions)

def main():

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

    extracted = extractor_resolver(prompt)

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

    print(extracted.affirmative)
    print(extracted.interrogative)