Updating tests to handle new params in logs for BYOK.

Author: sriroopar

ols/utils/token_handler.py

@@ -121,7 +121,9 @@ def truncate_rag_context(
             list of `RagChunk` objects, available tokens after context usage
         """
         rag_chunks = []
-        logger.info("Processing %d retrieved nodes for RAG context", len(retrieved_nodes))
+        logger.info(
+            "Processing %d retrieved nodes for RAG context", len(retrieved_nodes)
+        )
 
         for idx, node in enumerate(retrieved_nodes):
             score = float(node.get_score(raise_error=False))
@@ -190,7 +192,9 @@ def truncate_rag_context(
 
             max_tokens -= available_tokens
 
-        logger.info("Final selection: %d documents chosen for RAG context", len(rag_chunks))
+        logger.info(
+            "Final selection: %d documents chosen for RAG context", len(rag_chunks)
+        )
         return rag_chunks, max_tokens
 
     def limit_conversation_history(

tests/unit/app/endpoints/test_ols.py

@@ -1047,8 +1047,24 @@ def test_store_transcript(transcripts_location):
         "query_is_valid": query_is_valid,
         "llm_response": response,
         "rag_chunks": [
-            {"text": "text1", "doc_url": "url1", "doc_title": "title1"},
-            {"text": "text2", "doc_url": "url2", "doc_title": "title2"},
+            {
+                "text": "text1",
+                "doc_url": "url1",
+                "doc_title": "title1",
+                "index_id": "",
+                "index_origin": "",
+                "similarity_score": 0.0,
+                "token_count": 0,
+            },
+            {
+                "text": "text2",
+                "doc_url": "url2",
+                "doc_title": "title2",
+                "index_id": "",
+                "index_origin": "",
+                "similarity_score": 0.0,
+                "token_count": 0,
+            },
         ],
         "truncated": truncated,
         "tool_calls": [