refactor: reorganize imports and fix datasource endpoint URL

feat/datasource
Yeuoly 1 year ago
parent a15bf8e8fe
commit 6123f1ab21

@ -8,7 +8,6 @@ from flask_restful.inputs import int_range # type: ignore
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
from werkzeug.exceptions import Forbidden, InternalServerError, NotFound from werkzeug.exceptions import Forbidden, InternalServerError, NotFound
from models.model import EndUser
import services import services
from configs import dify_config from configs import dify_config
from controllers.console import api from controllers.console import api
@ -40,6 +39,7 @@ from libs.helper import TimestampField, uuid_value
from libs.login import current_user, login_required from libs.login import current_user, login_required
from models.account import Account from models.account import Account
from models.dataset import Pipeline from models.dataset import Pipeline
from models.model import EndUser
from services.errors.app import WorkflowHashNotEqualError from services.errors.app import WorkflowHashNotEqualError
from services.errors.llm import InvokeRateLimitError from services.errors.llm import InvokeRateLimitError
from services.rag_pipeline.pipeline_generate_service import PipelineGenerateService from services.rag_pipeline.pipeline_generate_service import PipelineGenerateService

@ -1,4 +1,5 @@
from typing import Any, Mapping from collections.abc import Mapping
from typing import Any
from core.datasource.entities.api_entities import DatasourceProviderApiEntity from core.datasource.entities.api_entities import DatasourceProviderApiEntity
from core.datasource.entities.datasource_entities import ( from core.datasource.entities.datasource_entities import (
@ -63,7 +64,7 @@ class PluginDatasourceManager(BasePluginClient):
response = self._request_with_plugin_daemon_response( response = self._request_with_plugin_daemon_response(
"GET", "GET",
f"plugin/{tenant_id}/management/datasources", f"plugin/{tenant_id}/management/datasource",
PluginDatasourceProviderEntity, PluginDatasourceProviderEntity,
params={"provider": tool_provider_id.provider_name, "plugin_id": tool_provider_id.plugin_id}, params={"provider": tool_provider_id.provider_name, "plugin_id": tool_provider_id.plugin_id},
transformer=transformer, transformer=transformer,

@ -12,7 +12,11 @@ from sqlalchemy.orm import Session
import contexts import contexts
from configs import dify_config from configs import dify_config
from core.datasource.entities.datasource_entities import DatasourceProviderType, GetOnlineDocumentPagesRequest, GetOnlineDocumentPagesResponse, GetWebsiteCrawlRequest, GetWebsiteCrawlResponse from core.datasource.entities.datasource_entities import (
DatasourceProviderType,
GetOnlineDocumentPagesResponse,
GetWebsiteCrawlResponse,
)
from core.datasource.online_document.online_document_plugin import OnlineDocumentDatasourcePlugin from core.datasource.online_document.online_document_plugin import OnlineDocumentDatasourcePlugin
from core.datasource.website_crawl.website_crawl_plugin import WebsiteCrawlDatasourcePlugin from core.datasource.website_crawl.website_crawl_plugin import WebsiteCrawlDatasourcePlugin
from core.model_runtime.utils.encoders import jsonable_encoder from core.model_runtime.utils.encoders import jsonable_encoder

Loading…
Cancel
Save