changed signature

This commit is contained in:
Lucas Oliveira 2025-09-29 16:43:35 -03:00
parent cb95473243
commit 1eeabeb370

View file

@ -7,7 +7,6 @@ from config.settings import (
LANGFLOW_CHAT_FLOW_ID, LANGFLOW_CHAT_FLOW_ID,
LANGFLOW_INGEST_FLOW_ID, LANGFLOW_INGEST_FLOW_ID,
LANGFLOW_PUBLIC_URL, LANGFLOW_PUBLIC_URL,
DOCLING_COMPONENT_ID,
clients, clients,
get_openrag_config, get_openrag_config,
config_manager, config_manager,
@ -203,7 +202,7 @@ async def update_settings(request, session_manager):
# Also update the chat flow with the new model # Also update the chat flow with the new model
try: try:
flows_service = _get_flows_service() flows_service = _get_flows_service()
await flows_service.update_chat_flow_model(body["llm_model"]) await flows_service.update_chat_flow_model(body["llm_model"], current_config.provider.model_provider.lower())
logger.info( logger.info(
f"Successfully updated chat flow model to '{body['llm_model']}'" f"Successfully updated chat flow model to '{body['llm_model']}'"
) )
@ -220,7 +219,8 @@ async def update_settings(request, session_manager):
try: try:
flows_service = _get_flows_service() flows_service = _get_flows_service()
await flows_service.update_chat_flow_system_prompt( await flows_service.update_chat_flow_system_prompt(
body["system_prompt"] body["system_prompt"],
current_config.provider.model_provider.lower()
) )
logger.info(f"Successfully updated chat flow system prompt") logger.info(f"Successfully updated chat flow system prompt")
except Exception as e: except Exception as e:
@ -245,7 +245,8 @@ async def update_settings(request, session_manager):
try: try:
flows_service = _get_flows_service() flows_service = _get_flows_service()
await flows_service.update_ingest_flow_embedding_model( await flows_service.update_ingest_flow_embedding_model(
body["embedding_model"].strip() body["embedding_model"].strip(),
current_config.provider.model_provider.lower()
) )
logger.info( logger.info(
f"Successfully updated ingest flow embedding model to '{body['embedding_model'].strip()}'" f"Successfully updated ingest flow embedding model to '{body['embedding_model'].strip()}'"