fix(anthropic_text.py): fix linting error

This commit is contained in:
Krrish Dholakia 2024-05-11 20:01:50 -07:00
parent 8592a73881
commit cca89c2061

View file

@ -100,7 +100,7 @@ class AnthropicTextCompletion(BaseLLM):
def __init__(self) -> None: def __init__(self) -> None:
super().__init__() super().__init__()
def process_response( def _process_response(
self, model_response: ModelResponse, response, encoding, prompt: str, model: str self, model_response: ModelResponse, response, encoding, prompt: str, model: str
): ):
## RESPONSE OBJECT ## RESPONSE OBJECT
@ -171,7 +171,7 @@ class AnthropicTextCompletion(BaseLLM):
additional_args={"complete_input_dict": data}, additional_args={"complete_input_dict": data},
) )
response = self.process_response( response = self._process_response(
model_response=model_response, model_response=model_response,
response=response, response=response,
encoding=encoding, encoding=encoding,
@ -330,7 +330,7 @@ class AnthropicTextCompletion(BaseLLM):
) )
print_verbose(f"raw model_response: {response.text}") print_verbose(f"raw model_response: {response.text}")
response = self.process_response( response = self._process_response(
model_response=model_response, model_response=model_response,
response=response, response=response,
encoding=encoding, encoding=encoding,