32 lines
893 B
Python
32 lines
893 B
Python
|
import os
|
|||
|
|
|
|||
|
|
import pandas as pd
|
|||
|
|
|
|||
|
|
from app.models.yandexgpt import YandexGPT
|
|||
|
|
from app.utils.submit import generate_submit
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
load_dotenv()
|
|||
|
|
|
|||
|
|
system_prompt = """
|
|||
|
|
Ты - профессиональный программист и ментор. Давай очень короткие ответы о синтаксических ошибках в коде, если они есть.
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
yandex_gpt = YandexGPT(
|
|||
|
|
token=os.environ["YANDEX_GPT_IAM_TOKEN"],
|
|||
|
|
folder_id=os.environ["YANDEX_GPT_FOLDER_ID"],
|
|||
|
|
system_prompt=system_prompt,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def predict(row: pd.Series) -> str:
|
|||
|
|
return yandex_gpt.ask(row["student_solution"])
|
|||
|
|
|
|||
|
|
|
|||
|
|
generate_submit(
|
|||
|
|
test_solutions_path="../data/raw/test/solutions.xlsx",
|
|||
|
|
predict_func=predict,
|
|||
|
|
save_path="../data/processed/submission.csv",
|
|||
|
|
use_tqdm=True,
|
|||
|
|
)
|