test/dify
1
0
Fork 0
mirror of https://gitee.com/dify_ai/dify.git synced 2024-12-04 20:28:12 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix: _convert_prompt_message_to_dict parameters err (#5716)

Browse Source
This commit is contained in:
takatost 2024-06-28 21:00:00 +08:00 committed by GitHub
parent 8e5569f773
commit 0bf4817474
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
api/core/app/task_pipeline/workflow_cycle_manage.py
View File

@ -167,6 +167,7 @@ class WorkflowCycleManage(WorkflowIterationCycleManage):
db.session.refresh(workflow_run) db.session.refresh(workflow_run)
db.session.close() db.session.close()
if trace_manager:
trace_manager.add_trace_task( trace_manager.add_trace_task(
TraceTask( TraceTask(
TraceTaskName.WORKFLOW_TRACE, TraceTaskName.WORKFLOW_TRACE,

2
api/core/model_runtime/model_providers/moonshot/llm/llm.py
View File

@ -93,7 +93,7 @@ class MoonshotLargeLanguageModel(OAIAPICompatLargeLanguageModel):
}.intersection(model_schema.features or []): }.intersection(model_schema.features or []):
credentials['function_calling_type'] = 'tool_call' credentials['function_calling_type'] = 'tool_call'
def _convert_prompt_message_to_dict(self, message: PromptMessage) -> dict: def _convert_prompt_message_to_dict(self, message: PromptMessage, credentials: Optional[dict] = None) -> dict:
""" """
Convert PromptMessage to dict for OpenAI API format Convert PromptMessage to dict for OpenAI API format
""" """

2
api/core/model_runtime/model_providers/nvidia/llm/llm.py
View File

@ -200,7 +200,7 @@ class NVIDIALargeLanguageModel(OAIAPICompatLargeLanguageModel):
endpoint_url = str(URL(endpoint_url) / 'chat' / 'completions') endpoint_url = str(URL(endpoint_url) / 'chat' / 'completions')
elif 'server_url' in credentials: elif 'server_url' in credentials:
endpoint_url = server_url endpoint_url = server_url
data['messages'] = [self._convert_prompt_message_to_dict(m) for m in prompt_messages] data['messages'] = [self._convert_prompt_message_to_dict(m, credentials) for m in prompt_messages]
elif completion_type is LLMMode.COMPLETION: elif completion_type is LLMMode.COMPLETION:
data['prompt'] = 'ping' data['prompt'] = 'ping'
if 'endpoint_url' in credentials: if 'endpoint_url' in credentials:

2
api/core/model_runtime/model_providers/openai_api_compatible/llm/llm.py
View File

@ -582,7 +582,7 @@ class OAIAPICompatLargeLanguageModel(_CommonOAI_API_Compat, LargeLanguageModel):
return result return result
def _convert_prompt_message_to_dict(self, message: PromptMessage, credentials: dict = None) -> dict: def _convert_prompt_message_to_dict(self, message: PromptMessage, credentials: Optional[dict] = None) -> dict:
""" """
Convert PromptMessage to dict for OpenAI API format Convert PromptMessage to dict for OpenAI API format
""" """