Ver Fonte

Add headers

Rodrigo Agundez há 3 meses atrás
pai
commit
91df1c56b2
1 ficheiros alterados com 7 adições e 1 exclusões
  1. 7 1
      backend/open_webui/utils/oauth.py

+ 7 - 1
backend/open_webui/utils/oauth.py

@@ -261,8 +261,14 @@ class OAuthManager:
                 if picture_url:
                 if picture_url:
                     # Download the profile image into a base64 string
                     # Download the profile image into a base64 string
                     try:
                     try:
+                        access_token = token.get("access_token")
+                        get_kwargs = {}
+                        if access_token:
+                            get_kwargs["headers"] = {
+                                "Authorization": f"Bearer {access_token}",
+                            }
                         async with aiohttp.ClientSession() as session:
                         async with aiohttp.ClientSession() as session:
-                            async with session.get(picture_url) as resp:
+                            async with session.get(picture_url, **get_kwargs) as resp:
                                 picture = await resp.read()
                                 picture = await resp.read()
                                 base64_encoded_picture = base64.b64encode(
                                 base64_encoded_picture = base64.b64encode(
                                     picture
                                     picture