formatting improvements

This commit is contained in:
ishaan-jaff 2023-08-28 09:20:50 -07:00
parent 70b323e0f5
commit b713acb0a4
17 changed files with 464 additions and 323 deletions

View file

@ -94,7 +94,10 @@ class AnthropicLLM:
## COMPLETION CALL
if "stream" in optional_params and optional_params["stream"] == True:
response = requests.post(
self.completion_url, headers=self.headers, data=json.dumps(data), stream=optional_params["stream"]
self.completion_url,
headers=self.headers,
data=json.dumps(data),
stream=optional_params["stream"],
)
return response.iter_lines()
else:
@ -142,4 +145,3 @@ class AnthropicLLM:
self,
): # logic for parsing in - calling - parsing out model embedding calls
pass