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,
|
||
)
|