@@ -395,7 +395,7 @@ impl DiclConfig {
395395 /// The second message is an Assistant message with the output as native output blocks
396396 /// - For chat messages, this is a simple vector of ContentBlocks
397397 /// - For JSON messages, this is a single JSON output block (as Text)
398- fn prepare_message ( & self , example : & Example ) -> Result < Vec < RequestMessage > , Error > {
398+ fn prepare_message ( example : & Example ) -> Result < Vec < RequestMessage > , Error > {
399399 let mut messages = Vec :: new ( ) ;
400400 let input = match example {
401401 Example :: Chat ( chat_example) => chat_example. input . clone ( ) ,
@@ -437,7 +437,7 @@ impl DiclConfig {
437437 Ok ( messages)
438438 }
439439
440- fn prepare_input_message ( & self , input : & ResolvedInput ) -> Result < RequestMessage , Error > {
440+ fn prepare_input_message ( input : & ResolvedInput ) -> Result < RequestMessage , Error > {
441441 let content = vec ! [ serde_json:: to_string( & input)
442442 . map_err( |e| {
443443 Error :: new( ErrorDetails :: Serialization {
@@ -489,11 +489,11 @@ impl DiclConfig {
489489 }
490490 let messages = examples
491491 . iter ( )
492- . map ( |example| self . prepare_message ( example ) )
492+ . map ( Self :: prepare_message)
493493 . collect :: < Result < Vec < Vec < RequestMessage > > , _ > > ( ) ?
494494 . into_iter ( )
495495 . flatten ( )
496- . chain ( std:: iter:: once ( self . prepare_input_message ( input) ?) )
496+ . chain ( std:: iter:: once ( Self :: prepare_input_message ( input) ?) )
497497 . collect :: < Vec < _ > > ( ) ;
498498
499499 let system = Some ( self . system_instructions . clone ( ) ) ;
@@ -658,9 +658,6 @@ mod tests {
658658
659659 #[ test]
660660 fn test_prepare_message ( ) {
661- // Create an instance of DiclConfig (assuming default implementation is available)
662- let dicl_config = DiclConfig :: default ( ) ;
663-
664661 // ---------- Test with ChatExample ----------
665662
666663 // Mock Input data
@@ -701,7 +698,7 @@ mod tests {
701698 output : chat_output. clone ( ) ,
702699 } ) ;
703700
704- let chat_messages = dicl_config . prepare_message ( & chat_example) . unwrap ( ) ;
701+ let chat_messages = DiclConfig :: prepare_message ( & chat_example) . unwrap ( ) ;
705702
706703 assert_eq ! ( chat_messages. len( ) , 2 ) ;
707704
@@ -737,7 +734,7 @@ mod tests {
737734 output : json_output. clone ( ) ,
738735 } ) ;
739736
740- let json_messages = dicl_config . prepare_message ( & json_example) . unwrap ( ) ;
737+ let json_messages = DiclConfig :: prepare_message ( & json_example) . unwrap ( ) ;
741738
742739 // Assertions for JsonExample
743740 assert_eq ! ( json_messages. len( ) , 2 ) ;
@@ -762,9 +759,6 @@ mod tests {
762759
763760 #[ test]
764761 fn test_prepare_input_message ( ) {
765- // Create an instance of DiclConfig (assuming default implementation is available)
766- let dicl_config = DiclConfig :: default ( ) ;
767-
768762 // Mock Input data
769763 let input_data = ResolvedInput {
770764 system : Some ( json ! ( { "assistant_name" : "Dr. Mehta" } ) ) ,
@@ -792,7 +786,7 @@ mod tests {
792786 } ;
793787
794788 // Call the prepare_input_message function
795- let request_message = dicl_config . prepare_input_message ( & input_data) . unwrap ( ) ;
789+ let request_message = DiclConfig :: prepare_input_message ( & input_data) . unwrap ( ) ;
796790
797791 // The role should be User
798792 assert_eq ! ( request_message. role, Role :: User ) ;
0 commit comments