diff --git a/superagi/controllers/models_controller.py b/superagi/controllers/models_controller.py index f4f018e9a..e8415e45a 100644 --- 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 index 5de459d3f..815697298 100644 --- 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