consult_query.py

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


class QueryGenerator(BaseModel):
    query: str


api = Gpt4oInterface()

instructions = """
You and your team are working with the prolog programing language and you need to create a script to solve logical puzzles. 
Your team already created a script that solve a specific puzzle and your job is to test it. 
The script and the puzzle will be inputed separately and you must generate a consult query to feed the prolog terminal and test the script.
Output only the query ending in a period, do not include the prolog terminal sign '?-'.
"""

extractor_resolver = api.of_format(QueryGenerator, instructions)


def main():
    
    with open("prolog/reasoning.pl", "r") as f:
        prolog_script = f.read()

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

    extracted = extractor_resolver([prolog_script, prompt])
    print(extracted.query)

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