Unit Test Fix (#1177)

TransformerOptimus · Sep 4, 2023 · 0e5abbc · 0e5abbc
1 parent e5fe0a5
commit 0e5abbc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/superagi/controllers/models_controller.py b/superagi/controllers/models_controller.py
@@ -122,7 +122,9 @@ def get_knowledge_list(page: int = 0, organisation=Depends(get_user_organisation
 
 @router.get("/marketplace/list/{page}", status_code=200)
 def get_marketplace_knowledge_list(page: int = 0):
-    organisation_id = int(get_config("MARKETPLACE_ORGANISATION_ID"))
+    organisation_id = get_config("MARKETPLACE_ORGANISATION_ID")
+    if organisation_id is not None:
+        organisation_id = int(organisation_id)
     page_size = 16
 
     query = db.session.query(Models).filter(Models.org_id == organisation_id)