From f6606f0dc7d9ed6b313844bbcd27544eb82679e5 Mon Sep 17 00:00:00 2001 From: I748376 Date: Tue, 2 Jul 2024 13:03:14 +0000 Subject: [PATCH] adds error handling for too long diffs --- prospector/llm/llm_service.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/prospector/llm/llm_service.py b/prospector/llm/llm_service.py index 6140f874d..1134794e3 100644 --- a/prospector/llm/llm_service.py +++ b/prospector/llm/llm_service.py @@ -3,6 +3,7 @@ import validators from langchain_core.language_models.llms import LLM from langchain_core.output_parsers import StrOutputParser +from requests import HTTPError from datamodel.commit import Commit from llm.instantiation import create_model_instance @@ -103,6 +104,12 @@ def classify_commit( ) logger.info(f"LLM returned is_relevant={is_relevant}") + except HTTPError as e: + # if the diff is too big, a 400 error is returned -> silently ignore by returning False for this commit + status_code = e.response.status_code + if status_code == 400: + return False + raise RuntimeError(f"Prompt-model chain could not be invoked: {e}") except Exception as e: raise RuntimeError(f"Prompt-model chain could not be invoked: {e}")