| json_response = response.json() | json_response = response.json() | ||||
| completion = json_response["responses"][0] | completion = json_response["responses"][0] | ||||
| if completion: | |||||
| completion = completion[len(prompt):] | |||||
| if stop is not None: | if stop is not None: | ||||
| completion = enforce_stop_tokens(completion, stop) | completion = enforce_stop_tokens(completion, stop) | ||||