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

update template when provider updates

Browse source
This commit is contained in:
Lucas Oliveira 2025-11-19 16:18:26 -03:00
parent 42e5e25f60
commit 141a7da339
1 changed files with 50 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

59
src/services/flows_service.py
View file

@ -816,7 +816,7 @@ class FlowsService:
if not DISABLE_INGEST_WITH_LANGFLOW and embedding_model: if not DISABLE_INGEST_WITH_LANGFLOW and embedding_model:
embedding_node, _ = self._find_node_in_flow(flow_data, display_name=OPENAI_EMBEDDING_COMPONENT_DISPLAY_NAME) embedding_node, _ = self._find_node_in_flow(flow_data, display_name=OPENAI_EMBEDDING_COMPONENT_DISPLAY_NAME)
if embedding_node: if embedding_node:
if self._update_component_fields( if await self._update_component_fields(
embedding_node, provider, embedding_model, endpoint embedding_node, provider, embedding_model, endpoint
): ):
updates_made.append(f"embedding model: {embedding_model}") updates_made.append(f"embedding model: {embedding_model}")
@ -825,14 +825,14 @@ class FlowsService:
if llm_model: if llm_model:
llm_node, _ = self._find_node_in_flow(flow_data, display_name=OPENAI_LLM_COMPONENT_DISPLAY_NAME) llm_node, _ = self._find_node_in_flow(flow_data, display_name=OPENAI_LLM_COMPONENT_DISPLAY_NAME)
if llm_node: if llm_node:
if self._update_component_fields( if await self._update_component_fields(
llm_node, provider, llm_model, endpoint llm_node, provider, llm_model, endpoint
): ):
updates_made.append(f"llm model: {llm_model}") updates_made.append(f"llm model: {llm_model}")
# Update LLM component (if exists in this flow) # Update LLM component (if exists in this flow)
agent_node, _ = self._find_node_in_flow(flow_data, display_name=AGENT_COMPONENT_DISPLAY_NAME) agent_node, _ = self._find_node_in_flow(flow_data, display_name=AGENT_COMPONENT_DISPLAY_NAME)
if agent_node: if agent_node:
if self._update_component_fields( if await self._update_component_fields(
agent_node, provider, llm_model, endpoint agent_node, provider, llm_model, endpoint
): ):
updates_made.append(f"agent model: {llm_model}") updates_made.append(f"agent model: {llm_model}")
@ -865,7 +865,7 @@ class FlowsService:
"flow_id": flow_id, "flow_id": flow_id,
} }
def _update_component_fields( async def _update_component_fields(
self, self,
component_node, component_node,
provider: str, provider: str,
@ -881,13 +881,54 @@ class FlowsService:
updated = False updated = False
provider_name = "IBM watsonx.ai" if provider == "watsonx" else "Ollama" if provider == "ollama" else "Anthropic" if provider == "anthropic" else "OpenAI" provider_name = "IBM watsonx.ai" if provider == "watsonx" else "Ollama" if provider == "ollama" else "Anthropic" if provider == "anthropic" else "OpenAI"
if "agent_llm" in template:
template["agent_llm"]["value"] = provider_name
updated = True
if "provider" in template: field_name = "provider" if "provider" in template else "agent_llm"
template["provider"]["value"] = provider_name
# Update provider field and call custom_component/update endpoint
if field_name in template:
# First, update the provider value
template[field_name]["value"] = provider_name
# Call custom_component/update endpoint to get updated template
# Only call if code field exists (custom components should have code)
if "code" in template and "value" in template["code"]:
code_value = template["code"]["value"]
field_value = provider_name
try:
update_payload = {
"code": code_value,
"template": template,
"field": field_name,
"field_value": field_value,
"tool_mode": False,
}
response = await clients.langflow_request(
"POST", "/api/v1/custom_component/update", json=update_payload
)
if response.status_code == 200:
response_data = response.json()
# Update template with the new template from response.data
if "template" in response_data:
# Update the template in component_node
component_node["data"]["node"]["template"] = response_data["template"]
# Update local template reference
template = response_data["template"]
logger.info(f"Successfully updated template via custom_component/update for provider: {provider_name}")
else:
logger.warning("Response from custom_component/update missing 'data' field")
else:
logger.warning(
f"Failed to call custom_component/update: HTTP {response.status_code} - {response.text}"
)
except Exception as e:
logger.error(f"Error calling custom_component/update: {str(e)}")
# Continue with manual updates even if API call fails
updated = True updated = True
# Update model_name field (common to all providers) # Update model_name field (common to all providers)
if "model" in template: if "model" in template: