fix(bedrock.py): adding support for cohere embeddings

This commit is contained in:
Krrish Dholakia 2023-12-06 13:24:49 -08:00
parent 9703afdef1
commit a18bdb3f2e
4 changed files with 85 additions and 38 deletions

View file

@ -843,6 +843,7 @@ async def chat_completion(request: Request, model: Optional[str] = None, user_ap
data["user"] = user_api_key_dict.user_id
if "metadata" in data:
print(f'received metadata: {data["metadata"]}')
data["metadata"]["user_api_key"] = user_api_key_dict.api_key
data["metadata"]["headers"] = request.headers
else: