@@ -30,7 +30,7 @@ class Models : public drogon::HttpController<Models, false> {
30
30
31
31
ADD_METHOD_TO (Models::PullModel, " /v1/models/pull" , Options, Post);
32
32
ADD_METHOD_TO (Models::AbortPullModel, " /v1/models/pull" , Options, Delete);
33
- ADD_METHOD_TO (Models::ListModel, " /v1/models?author={author} " , Get);
33
+ ADD_METHOD_TO (Models::ListModel, " /v1/models" , Get);
34
34
ADD_METHOD_TO (Models::GetModel, " /v1/models/{1}" , Get);
35
35
ADD_METHOD_TO (Models::UpdateModel, " /v1/models/{1}" , Options, Patch);
36
36
ADD_METHOD_TO (Models::ImportModel, " /v1/models/import" , Options, Post);
@@ -44,6 +44,8 @@ class Models : public drogon::HttpController<Models, false> {
44
44
ADD_METHOD_TO (Models::DeleteModelSource, " /v1/models/sources" , Delete);
45
45
ADD_METHOD_TO (Models::GetModelSources, " /v1/models/sources" , Get);
46
46
ADD_METHOD_TO (Models::GetModelSource, " /v1/models/sources/{src}" , Get);
47
+ ADD_METHOD_TO (Models::GetRepositoryList, " /v1/models/hub?author={author}" ,
48
+ Get);
47
49
METHOD_LIST_END
48
50
49
51
explicit Models (std::shared_ptr<DatabaseService> db_service,
@@ -63,8 +65,7 @@ class Models : public drogon::HttpController<Models, false> {
63
65
void AbortPullModel (const HttpRequestPtr& req,
64
66
std::function<void (const HttpResponsePtr&)>&& callback);
65
67
void ListModel (const HttpRequestPtr& req,
66
- std::function<void (const HttpResponsePtr&)>&& callback,
67
- std::optional<std::string> author) const ;
68
+ std::function<void (const HttpResponsePtr&)>&& callback) const ;
68
69
void GetModel (const HttpRequestPtr& req,
69
70
std::function<void (const HttpResponsePtr&)>&& callback,
70
71
const std::string& model_id) const ;
@@ -112,6 +113,10 @@ class Models : public drogon::HttpController<Models, false> {
112
113
std::function<void (const HttpResponsePtr&)>&& callback,
113
114
const std::string& src);
114
115
116
+ void GetRepositoryList (const HttpRequestPtr& req,
117
+ std::function<void (const HttpResponsePtr&)>&& callback,
118
+ std::optional<std::string> author);
119
+
115
120
private:
116
121
std::shared_ptr<DatabaseService> db_service_;
117
122
std::shared_ptr<ModelService> model_service_;
0 commit comments