LCORE-2648: Updated types

Author: tisnik

src/utils/endpoints.py

@@ -476,7 +476,7 @@ def _add_additional_metadata_docs(
 
 
 def _process_rag_chunks_for_documents(
-    rag_chunks: list,
+    rag_chunks: list[ReferencedDocument],
     metadata_map: Optional[dict[str, Any]] = None,
 ) -> list[tuple[Optional[AnyUrl], str]]:
     """
@@ -532,7 +532,7 @@ def _process_rag_chunks_for_documents(
 
 
 def create_referenced_documents(
-    rag_chunks: list,
+    rag_chunks: list[ReferencedDocument],
     metadata_map: Optional[dict[str, Any]] = None,
     return_dict_format: bool = False,
 ) -> list[ReferencedDocument] | list[dict[str, Optional[str]]]:
@@ -601,7 +601,7 @@ def create_referenced_documents_with_metadata(
 
 
 def create_referenced_documents_from_chunks(
-    rag_chunks: list,
+    rag_chunks: list[ReferencedDocument],
 ) -> list[ReferencedDocument]:
     """
     Create referenced documents from RAG chunks for query endpoint.