@@ -16,29 +16,29 @@ TEST_F(HuggingFaceUtilTestSuite, TestGetModelRepositoryBranches) {
1616 EXPECT_EQ (branches.value ()[" gguf" ].ref , " refs/heads/gguf" );
1717}
1818
19- // temporary disable this test due to the model got removed
20- // TEST_F(HuggingFaceUtilTestSuite, TestGetHuggingFaceModelRepoInfoSuccessfully ) {
21- // auto model_info =
22- // huggingface_utils::GetHuggingFaceModelRepoInfo("cortexso", "tinyllama");
23-
24- // EXPECT_TRUE(model_info.has_value());
25- // EXPECT_EQ(model_info->id, "cortexso/tinyllama");
26- // EXPECT_EQ(model_info->modelId, "cortexso/tinyllama");
27- // EXPECT_EQ(model_info->author, "cortexso");
28- // EXPECT_EQ(model_info->disabled, false);
29- // EXPECT_EQ(model_info->gated, false);
30-
31- // auto tag_contains_gguf =
32- // std::find(model_info->tags.begin(), model_info->tags.end(), "gguf") !=
33- // model_info->tags.end();
34- // EXPECT_TRUE(tag_contains_gguf);
35-
36- // auto contain_gguf_info = model_info->gguf.has_value();
37- // EXPECT_TRUE(contain_gguf_info);
38-
39- // auto sibling_not_empty = !model_info->siblings.empty();
40- // EXPECT_TRUE(sibling_not_empty);
41- // }
19+
20+ TEST_F (HuggingFaceUtilTestSuite, DISABLED_TestGetHuggingFaceModelRepoInfoSuccessfully ) {
21+ auto model_info =
22+ huggingface_utils::GetHuggingFaceModelRepoInfo (" cortexso" , " tinyllama" );
23+
24+ EXPECT_TRUE (model_info.has_value ());
25+ EXPECT_EQ (model_info->id , " cortexso/tinyllama" );
26+ EXPECT_EQ (model_info->modelId , " cortexso/tinyllama" );
27+ EXPECT_EQ (model_info->author , " cortexso" );
28+ EXPECT_EQ (model_info->disabled , false );
29+ EXPECT_EQ (model_info->gated , false );
30+
31+ auto tag_contains_gguf =
32+ std::find (model_info->tags .begin (), model_info->tags .end (), " gguf" ) !=
33+ model_info->tags .end ();
34+ EXPECT_TRUE (tag_contains_gguf);
35+
36+ auto contain_gguf_info = model_info->gguf .has_value ();
37+ EXPECT_TRUE (contain_gguf_info);
38+
39+ auto sibling_not_empty = !model_info->siblings .empty ();
40+ EXPECT_TRUE (sibling_not_empty);
41+ }
4242
4343TEST_F (HuggingFaceUtilTestSuite,
4444 TestGetHuggingFaceModelRepoInfoReturnNullGgufInfoWhenNotAGgufModel) {
0 commit comments