k

Author: pablonyx

backend/onyx/auth/users.py

@@ -105,6 +105,7 @@
 from onyx.utils.telemetry import create_milestone_and_report
 from onyx.utils.telemetry import optional_telemetry
 from onyx.utils.telemetry import RecordType
+from onyx.utils.timing import log_function_time
 from onyx.utils.url import add_url_params
 from onyx.utils.variable_functionality import fetch_ee_implementation_or_noop
 from onyx.utils.variable_functionality import fetch_versioned_implementation
@@ -363,6 +364,7 @@ async def validate_password(self, password: str, _: schemas.UC | models.UP) -> N
             )
         return
 
+    @log_function_time(print_only=True)
     async def oauth_callback(
         self,
         oauth_name: str,
@@ -609,6 +611,7 @@ async def on_after_request_verify(
             user.email, token, new_organization=user_count == 1
         )
 
+    @log_function_time(print_only=True)
     async def authenticate(
         self, credentials: OAuth2PasswordRequestForm
     ) -> Optional[User]:
@@ -1235,6 +1238,7 @@ async def authorize(
 
         return OAuth2AuthorizeResponse(authorization_url=authorization_url)
 
+    @log_function_time(print_only=True)
     @router.get(
         "/callback",
         name=callback_route_name,

backend/onyx/chat/process_message.py

@@ -914,6 +914,7 @@ def stream_chat_message_objects(
                     retrieval_options.filters.user_folder_ids = user_folder_ids
 
                     # Create override kwargs for the search tool
+
                     override_kwargs = SearchToolOverrideKwargs(
                         force_no_rerank=search_for_ordering_only,  # Skip reranking for ordering-only
                         alternate_db_session=None,
@@ -1113,9 +1114,6 @@ def stream_chat_message_objects(
                         logger.info(
                             f"ORDERING: Processing search results for ordering {len(user_files)} user files"
                         )
-                        import time
-
-                        ordering_start = time.time()
 
                         # Extract document order from search results
                         doc_order = []
@@ -1151,8 +1149,6 @@ def stream_chat_message_objects(
                             if f_id in file_id_to_user_file
                         ]
 
-                        time.time() - ordering_start
-
                         yield UserKnowledgeFilePacket(
                             user_files=[
                                 FileDescriptor(

backend/onyx/server/query_and_chat/chat_backend.py

@@ -418,6 +418,7 @@ def handle_new_chat_message(
     """
     tenant_id = get_current_tenant_id()
     logger.debug(f"Received new chat message: {chat_message_req.message}")
+    start_time = time.time()
 
     if (
         not chat_message_req.message
@@ -448,6 +449,10 @@ def stream_generator() -> Generator[str, None, None]:
                 ),
                 is_connected=is_connected_func,
             ):
+                if isinstance(packet, dict) and "top_documents" in packet:
+                    document_retrieval_latency = time.time() - start_time
+                    logger.debug(f"First doc time: {document_retrieval_latency}")
+
                 yield json.dumps(packet) if isinstance(packet, dict) else packet
 
         except Exception as e: