@@ -74,12 +74,12 @@ def setUpClass(cls):
7474 cls .mock_get_knowledge_list .return_value = [
7575 {
7676 "index_name" : "kb1" ,
77- "model_name " : "embedding-model-1" ,
77+ "embedding_model_name " : "embedding-model-1" ,
7878 "knowledge_sources" : ["source1" , "source2" ]
7979 },
8080 {
8181 "index_name" : "kb2" ,
82- "model_name " : "embedding-model-2" ,
82+ "embedding_model_name " : "embedding-model-2" ,
8383 "knowledge_sources" : ["source3" ]
8484 }
8585 ]
@@ -102,12 +102,12 @@ def setUp(self):
102102 self .mock_get_knowledge_list .return_value = [
103103 {
104104 "index_name" : "kb1" ,
105- "model_name " : "embedding-model-1" ,
105+ "embedding_model_name " : "embedding-model-1" ,
106106 "knowledge_sources" : ["source1" , "source2" ]
107107 },
108108 {
109109 "index_name" : "kb2" ,
110- "model_name " : "embedding-model-2" ,
110+ "embedding_model_name " : "embedding-model-2" ,
111111 "knowledge_sources" : ["source3" ]
112112 }
113113 ]
@@ -181,7 +181,7 @@ def test_load_knowledge_list_missing_model_name(self):
181181 {
182182 "index_name" : "kb1" ,
183183 "knowledge_sources" : ["source1" ]
184- # Missing model_name field
184+ # Missing embedding_model_name field
185185 }
186186 ]
187187
0 commit comments