jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: type num of variable converted to str (#3758)

Browse source
This commit is contained in:
takatost 2024-04-24 15:07:56 +08:00 • committed by GitHub
commit 2ea8c73cd8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 17 additions and 47 deletions
Download patch file Download diff file Expand all files Collapse all files

2
api/core/prompt/advanced_prompt_transform.py
View file

@ -32,6 +32,8 @@ class AdvancedPromptTransform(PromptTransform):
memory_config: Optional[MemoryConfig],
memory: Optional[TokenBufferMemory],
model_config: ModelConfigWithCredentialsEntity) -> list[PromptMessage]:
inputs = {key: str(value) for key, value in inputs.items()}
prompt_messages = []
model_mode = ModelMode.value_of(model_config.mode)

2
api/core/prompt/simple_prompt_transform.py
View file

@ -55,6 +55,8 @@ class SimplePromptTransform(PromptTransform):
memory: Optional[TokenBufferMemory],
model_config: ModelConfigWithCredentialsEntity) -> \
tuple[list[PromptMessage], Optional[list[str]]]:
inputs = {key: str(value) for key, value in inputs.items()}
model_mode = ModelMode.value_of(model_config.mode)
if model_mode == ModelMode.CHAT:
prompt_messages, stops = self._get_chat_model_prompt_messages(