Unverified Commit 83105d0d authored by Garfield Dai's avatar Garfield Dai Committed by GitHub

fix: dataset and moderation. (#2042)

parent 7b0818b8
...@@ -146,7 +146,7 @@ class BasicApplicationRunner(AppRunner): ...@@ -146,7 +146,7 @@ class BasicApplicationRunner(AppRunner):
# get context from datasets # get context from datasets
context = None context = None
if app_orchestration_config.dataset: if app_orchestration_config.dataset and app_orchestration_config.dataset.dataset_ids:
context = self.retrieve_dataset_context( context = self.retrieve_dataset_context(
tenant_id=app_record.tenant_id, tenant_id=app_record.tenant_id,
app_record=app_record, app_record=app_record,
......
...@@ -30,7 +30,7 @@ class KeywordsModeration(Moderation): ...@@ -30,7 +30,7 @@ class KeywordsModeration(Moderation):
if query: if query:
inputs['query__'] = query inputs['query__'] = query
keywords_list = self.config['keywords'].split('\n') keywords_list = [keyword for keyword in self.config['keywords'].split('\n') if keyword]
flagged = self._is_violated(inputs, keywords_list) flagged = self._is_violated(inputs, keywords_list)
return ModerationInputsResult(flagged=flagged, action=ModerationAction.DIRECT_OUTPUT, preset_response=preset_response) return ModerationInputsResult(flagged=flagged, action=ModerationAction.DIRECT_OUTPUT, preset_response=preset_response)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment