kopia lustrzana https://github.com/biobootloader/wolverine
Decrement instead of increment, nb_retry default to VALIDATE_JSON_RETRY
rodzic
3f333a4df7
commit
5c91c4ccba
|
@ -45,14 +45,14 @@ def run_script(script_name: str, script_args: List) -> str:
|
||||||
return result.decode("utf-8"), 0
|
return result.decode("utf-8"), 0
|
||||||
|
|
||||||
|
|
||||||
def json_validated_response(model: str, messages: List[Dict], nb_retry: int = 0) -> Dict:
|
def json_validated_response(model: str, messages: List[Dict], nb_retry: int = VALIDATE_JSON_RETRY) -> Dict:
|
||||||
"""
|
"""
|
||||||
This function is needed because the API can return a non-json response.
|
This function is needed because the API can return a non-json response.
|
||||||
This will run recursively VALIDATE_JSON_RETRY times.
|
This will run recursively VALIDATE_JSON_RETRY times.
|
||||||
If VALIDATE_JSON_RETRY is -1, it will run recursively until a valid json response is returned.
|
If VALIDATE_JSON_RETRY is -1, it will run recursively until a valid json response is returned.
|
||||||
"""
|
"""
|
||||||
json_response = {}
|
json_response = {}
|
||||||
if VALIDATE_JSON_RETRY == -1 or nb_retry < VALIDATE_JSON_RETRY:
|
if nb_retry != 0:
|
||||||
response = openai.ChatCompletion.create(
|
response = openai.ChatCompletion.create(
|
||||||
model=model,
|
model=model,
|
||||||
messages=messages,
|
messages=messages,
|
||||||
|
@ -80,8 +80,8 @@ def json_validated_response(model: str, messages: List[Dict], nb_retry: int = 0)
|
||||||
"content": "Your response could not be parsed by json.loads. Please restate your last message as pure JSON.",
|
"content": "Your response could not be parsed by json.loads. Please restate your last message as pure JSON.",
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
# inc nb_retry
|
# dec nb_retry
|
||||||
nb_retry+=1
|
nb_retry-=1
|
||||||
# rerun the api call
|
# rerun the api call
|
||||||
return json_validated_response(model, messages, nb_retry)
|
return json_validated_response(model, messages, nb_retry)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
|
Ładowanie…
Reference in New Issue