marketplace changes (#1167)

Enabling Marketplace for models
TransformerOptimus · Sep 1, 2023 · e5fe0a5 · e5fe0a5
1 parent 48e68cf
commit e5fe0a5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/superagi/controllers/models_controller.py b/superagi/controllers/models_controller.py
@@ -4,6 +4,7 @@
 from superagi.apm.call_log_helper import CallLogHelper
 from superagi.models.models import Models
 from superagi.models.models_config import ModelsConfig
+from superagi.config.config import get_config
 from fastapi_sqlalchemy import db
 import logging
 from pydantic import BaseModel
@@ -121,7 +122,7 @@ 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 = 2
+    organisation_id = int(get_config("MARKETPLACE_ORGANISATION_ID"))
     page_size = 16
 
     query = db.session.query(Models).filter(Models.org_id == organisation_id)

diff --git a/superagi/models/models.py b/superagi/models/models.py
@@ -6,8 +6,10 @@
 from superagi.helper.encyption_helper import decrypt_data
 import requests, logging
 
-# marketplace_url = "https://app.superagi.com/api"
-marketplace_url = "http://localhost:8001"
+marketplace_url = "https://app.superagi.com/api"
+# marketplace_url = "http://localhost:8001"
+
+
 class Models(DBBaseModel):
     """
     Represents a Model record in the database