gmakstutis/openrag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

modify reset_langflow to also put right embedding model

Browse source
This commit is contained in:
Lucas Oliveira 2025-09-25 17:15:47 -03:00
parent 606cd35e9a
commit fa07838f77
1 changed files with 29 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
src/services/flows_service.py
View file

@ -19,6 +19,7 @@ from config.settings import (
WATSONX_LLM_COMPONENT_ID,
OLLAMA_EMBEDDING_COMPONENT_ID,
OLLAMA_LLM_COMPONENT_ID,
get_openrag_config,
)
import json
import os
@ -152,6 +153,34 @@ class FlowsService:
flow_id=flow_id,
flow_file=os.path.basename(flow_path),
)
# Now update the flow with current configuration settings
try:
config = get_openrag_config()
# Check if configuration has been edited (onboarding completed)
if config.edited:
logger.info(f"Updating {flow_type} flow with current configuration settings")
# Update the flow with the current embedding and LLM models
update_result = await self.change_langflow_model_value(
provider=config.provider.model_provider.lower(),
embedding_model=config.knowledge.embedding_model,
llm_model=config.agent.llm_model,
endpoint=getattr(config.provider, 'endpoint', None)
)
if update_result.get("success"):
logger.info(f"Successfully updated {flow_type} flow with current configuration")
else:
logger.warning(f"Failed to update {flow_type} flow with current configuration: {update_result.get('error', 'Unknown error')}")
else:
logger.info(f"Configuration not yet edited (onboarding not completed), skipping model updates for {flow_type} flow")
except Exception as e:
logger.error(f"Error updating {flow_type} flow with current configuration", error=str(e))
# Don't fail the entire reset operation if configuration update fails
return {
"success": True,
"message": f"Successfully reset {flow_type} flow",