From 973bf1cff9d564292a1e593edafccfc82ecae2ae Mon Sep 17 00:00:00 2001 From: Nate Sesti Date: Mon, 26 Jun 2023 13:54:26 -0700 Subject: catch error from proxy server --- continuedev/src/continuedev/core/autopilot.py | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'continuedev/src') diff --git a/continuedev/src/continuedev/core/autopilot.py b/continuedev/src/continuedev/core/autopilot.py index 16fc5dcb..b9308409 100644 --- a/continuedev/src/continuedev/core/autopilot.py +++ b/continuedev/src/continuedev/core/autopilot.py @@ -2,6 +2,8 @@ from functools import cached_property import traceback import time from typing import Any, Callable, Coroutine, Dict, List + +from aiohttp import ClientPayloadError from ..models.filesystem_edit import FileEditWithFullContents from ..libs.llm import LLM from .observation import Observation, InternalErrorObservation @@ -27,6 +29,8 @@ def get_error_title(e: Exception) -> str: return "OpenAI timed out. Please try again." elif isinstance(e, openai_errors.InvalidRequestError) and e.code == "context_length_exceeded": return e._message + elif isinstance(e, ClientPayloadError): + return "The request to OpenAI failed. Please try again." return e.__repr__() -- cgit v1.2.3-70-g09d2