22import hashlib
33import json
44import logging
5+ import time
56from typing import Optional
67
78import aiohttp
@@ -488,6 +489,7 @@ async def get_filtered_models(models, user):
488489)
489490async def get_all_models (request : Request , user : UserModel ) -> dict [str , list ]:
490491 log .info ("get_all_models()" )
492+ t1 = time .time ()
491493
492494 if not request .app .state .config .ENABLE_OPENAI_API :
493495 return {"data" : []}
@@ -548,6 +550,9 @@ def get_merged_models(model_lists):
548550 models = get_merged_models (map (extract_data , responses ))
549551 log .debug (f"models: { models } " )
550552
553+ t2 = time .time ()
554+ delta = round (t2 - t1 , 2 )
555+ log .debug (f"OpenAI get_all_models() took { delta } s" )
551556 request .app .state .OPENAI_MODELS = models
552557 return {"data" : list (models .values ())}
553558
@@ -939,6 +944,7 @@ async def generate_chat_completion(
939944 trust_env = True , timeout = aiohttp .ClientTimeout (total = AIOHTTP_CLIENT_TIMEOUT )
940945 )
941946
947+ t1 = time .time ()
942948 r = await session .request (
943949 method = "POST" ,
944950 url = request_url ,
@@ -947,6 +953,9 @@ async def generate_chat_completion(
947953 cookies = cookies ,
948954 ssl = AIOHTTP_CLIENT_SESSION_SSL ,
949955 )
956+ t2 = time .time ()
957+ delta = round (t2 - t1 , 2 )
958+ log .debug (f"OpenAI request session.request() took { delta } s" )
950959
951960 # Check if response is SSE
952961 if "text/event-stream" in r .headers .get ("Content-Type" , "" ):
0 commit comments