estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix: change model mode. (#1520)

Browse Source
This commit is contained in:
Garfield Dai
2023-11-13 23:13:01 +08:00
committed by GitHub
parent a80d8286c2
commit 8835435558
2 changed files with 1 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/prompt/prompt_transform.py
View File

@@ -27,7 +27,6 @@ class AppMode(enum.Enum):
class PromptTransform:
def get_prompt(self,
app_mode: str,
app_model_config: AppModelConfig,
pre_prompt: str,
inputs: dict,
query: str,
@@ -36,10 +35,9 @@ class PromptTransform:
memory: Optional[BaseChatMemory],
model_instance: BaseLLM) -> \
Tuple[List[PromptMessage], Optional[List[str]]]:
model_mode = app_model_config.model_dict['mode']
app_mode_enum = AppMode(app_mode)
model_mode_enum = ModelMode(model_mode)
model_mode_enum = model_instance.model_mode
prompt_rules = self._read_prompt_rules_from_file(self._prompt_file_name(app_mode, model_instance))