mirror of
https://github.com/langgenius/dify.git
synced 2026-04-30 13:37:24 +08:00
fix preview and recommend plugins
This commit is contained in:
parent
fc4bc08796
commit
343f1a375f
@ -17,8 +17,8 @@ class DatabasePipelineTemplateRetrieval(PipelineTemplateRetrievalBase):
|
|||||||
result = self.fetch_pipeline_templates_from_db(language)
|
result = self.fetch_pipeline_templates_from_db(language)
|
||||||
return result
|
return result
|
||||||
|
|
||||||
def get_pipeline_template_detail(self, pipeline_id: str):
|
def get_pipeline_template_detail(self, template_id: str):
|
||||||
result = self.fetch_pipeline_template_detail_from_db(pipeline_id)
|
result = self.fetch_pipeline_template_detail_from_db(template_id)
|
||||||
return result
|
return result
|
||||||
|
|
||||||
def get_type(self) -> str:
|
def get_type(self) -> str:
|
||||||
@ -53,7 +53,7 @@ class DatabasePipelineTemplateRetrieval(PipelineTemplateRetrievalBase):
|
|||||||
return {"pipeline_templates": recommended_pipelines_results}
|
return {"pipeline_templates": recommended_pipelines_results}
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def fetch_pipeline_template_detail_from_db(cls, pipeline_id: str) -> Optional[dict]:
|
def fetch_pipeline_template_detail_from_db(cls, template_id: str) -> Optional[dict]:
|
||||||
"""
|
"""
|
||||||
Fetch pipeline template detail from db.
|
Fetch pipeline template detail from db.
|
||||||
:param pipeline_id: Pipeline ID
|
:param pipeline_id: Pipeline ID
|
||||||
@ -61,7 +61,7 @@ class DatabasePipelineTemplateRetrieval(PipelineTemplateRetrievalBase):
|
|||||||
"""
|
"""
|
||||||
# is in public recommended list
|
# is in public recommended list
|
||||||
pipeline_template = (
|
pipeline_template = (
|
||||||
db.session.query(PipelineBuiltInTemplate).filter(PipelineBuiltInTemplate.id == pipeline_id).first()
|
db.session.query(PipelineBuiltInTemplate).filter(PipelineBuiltInTemplate.id == template_id).first()
|
||||||
)
|
)
|
||||||
|
|
||||||
if not pipeline_template:
|
if not pipeline_template:
|
||||||
|
|||||||
@ -16,12 +16,12 @@ class RemotePipelineTemplateRetrieval(PipelineTemplateRetrievalBase):
|
|||||||
Retrieval recommended app from dify official
|
Retrieval recommended app from dify official
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def get_pipeline_template_detail(self, pipeline_id: str):
|
def get_pipeline_template_detail(self, template_id: str):
|
||||||
try:
|
try:
|
||||||
result = self.fetch_pipeline_template_detail_from_dify_official(pipeline_id)
|
result = self.fetch_pipeline_template_detail_from_dify_official(template_id)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.warning("fetch recommended app detail from dify official failed: %r, switch to built-in.", e)
|
logger.warning("fetch recommended app detail from dify official failed: %r, switch to built-in.", e)
|
||||||
result = BuildInRecommendAppRetrieval.fetch_recommended_app_detail_from_builtin(pipeline_id)
|
result = BuildInRecommendAppRetrieval.fetch_recommended_app_detail_from_builtin(template_id)
|
||||||
return result
|
return result
|
||||||
|
|
||||||
def get_pipeline_templates(self, language: str) -> dict:
|
def get_pipeline_templates(self, language: str) -> dict:
|
||||||
@ -36,14 +36,14 @@ class RemotePipelineTemplateRetrieval(PipelineTemplateRetrievalBase):
|
|||||||
return PipelineTemplateType.REMOTE
|
return PipelineTemplateType.REMOTE
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def fetch_pipeline_template_detail_from_dify_official(cls, pipeline_id: str) -> Optional[dict]:
|
def fetch_pipeline_template_detail_from_dify_official(cls, template_id: str) -> Optional[dict]:
|
||||||
"""
|
"""
|
||||||
Fetch pipeline template detail from dify official.
|
Fetch pipeline template detail from dify official.
|
||||||
:param pipeline_id: Pipeline ID
|
:param template_id: Pipeline ID
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
domain = dify_config.HOSTED_FETCH_PIPELINE_TEMPLATES_REMOTE_DOMAIN
|
domain = dify_config.HOSTED_FETCH_PIPELINE_TEMPLATES_REMOTE_DOMAIN
|
||||||
url = f"{domain}/pipelines/{pipeline_id}"
|
url = f"{domain}/pipeline-templates/{template_id}"
|
||||||
response = requests.get(url, timeout=(3, 10))
|
response = requests.get(url, timeout=(3, 10))
|
||||||
if response.status_code != 200:
|
if response.status_code != 200:
|
||||||
return None
|
return None
|
||||||
@ -58,7 +58,7 @@ class RemotePipelineTemplateRetrieval(PipelineTemplateRetrievalBase):
|
|||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
domain = dify_config.HOSTED_FETCH_PIPELINE_TEMPLATES_REMOTE_DOMAIN
|
domain = dify_config.HOSTED_FETCH_PIPELINE_TEMPLATES_REMOTE_DOMAIN
|
||||||
url = f"{domain}/pipelines?language={language}"
|
url = f"{domain}/pipeline-templates?language={language}"
|
||||||
response = requests.get(url, timeout=(3, 10))
|
response = requests.get(url, timeout=(3, 10))
|
||||||
if response.status_code != 200:
|
if response.status_code != 200:
|
||||||
raise ValueError(f"fetch pipeline templates failed, status code: {response.status_code}")
|
raise ValueError(f"fetch pipeline templates failed, status code: {response.status_code}")
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user