opensource
/
dify
mirror of https://github.com/langgenius/dify.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: change model mode. (#1520)

This commit is contained in:
Garfield Dai 2023-11-13 23:13:01 +08:00 committed by GitHub
parent a80d8286c2
commit 8835435558
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/completion.py
View File

@ -277,7 +277,6 @@ class Completion:
if app_model_config.prompt_type == 'simple':
prompt_messages, stop_words = prompt_transform.get_prompt(
app_mode=mode,
app_model_config=app_model_config,
pre_prompt=app_model_config.pre_prompt,
inputs=inputs,
query=query,
@ -366,7 +365,6 @@ class Completion:
if app_model_config.prompt_type == 'simple':
prompt_messages, _ = prompt_transform.get_prompt(
app_mode=mode,
app_model_config=app_model_config,
pre_prompt=app_model_config.pre_prompt,
inputs=inputs,
query=query,

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))