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

format

Browse source
This commit is contained in:
Lucas Oliveira 2025-09-24 13:10:51 -03:00
parent 7ec608b5c5
commit 36fbe26406
1 changed files with 40 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

63
src/api/settings.py
View file

@ -47,9 +47,6 @@ def get_docling_preset_configs():
} }
async def get_settings(request, session_manager): async def get_settings(request, session_manager):
"""Get application settings""" """Get application settings"""
try: try:
@ -222,8 +219,12 @@ async def update_settings(request, session_manager):
# Also update the flow with the new docling preset # Also update the flow with the new docling preset
try: try:
await _update_flow_docling_preset(body["doclingPresets"], preset_configs[body["doclingPresets"]]) await _update_flow_docling_preset(
logger.info(f"Successfully updated docling preset in flow to '{body['doclingPresets']}'") body["doclingPresets"], preset_configs[body["doclingPresets"]]
)
logger.info(
f"Successfully updated docling preset in flow to '{body['doclingPresets']}'"
)
except Exception as e: except Exception as e:
logger.error(f"Failed to update docling preset in flow: {str(e)}") logger.error(f"Failed to update docling preset in flow: {str(e)}")
# Don't fail the entire settings update if flow update fails # Don't fail the entire settings update if flow update fails
@ -535,7 +536,9 @@ async def _update_flow_docling_preset(preset: str, preset_config: dict):
) )
if response.status_code != 200: if response.status_code != 200:
raise Exception(f"Failed to get ingest flow: HTTP {response.status_code} - {response.text}") raise Exception(
f"Failed to get ingest flow: HTTP {response.status_code} - {response.text}"
)
flow_data = response.json() flow_data = response.json()
@ -551,13 +554,24 @@ async def _update_flow_docling_preset(preset: str, preset_config: dict):
break break
if target_node is None: if target_node is None:
raise Exception(f"Docling component '{DOCLING_COMPONENT_ID}' not found in ingest flow") raise Exception(
f"Docling component '{DOCLING_COMPONENT_ID}' not found in ingest flow"
)
# Update the docling_serve_opts value directly in the existing node # Update the docling_serve_opts value directly in the existing node
if (target_node.get("data", {}).get("node", {}).get("template", {}).get("docling_serve_opts")): if (
flow_data["data"]["nodes"][target_node_index]["data"]["node"]["template"]["docling_serve_opts"]["value"] = preset_config target_node.get("data", {})
.get("node", {})
.get("template", {})
.get("docling_serve_opts")
):
flow_data["data"]["nodes"][target_node_index]["data"]["node"]["template"][
"docling_serve_opts"
]["value"] = preset_config
else: else:
raise Exception(f"docling_serve_opts field not found in node '{DOCLING_COMPONENT_ID}'") raise Exception(
f"docling_serve_opts field not found in node '{DOCLING_COMPONENT_ID}'"
)
# Update the flow via PATCH request # Update the flow via PATCH request
patch_response = await clients.langflow_request( patch_response = await clients.langflow_request(
@ -565,7 +579,9 @@ async def _update_flow_docling_preset(preset: str, preset_config: dict):
) )
if patch_response.status_code != 200: if patch_response.status_code != 200:
raise Exception(f"Failed to update ingest flow: HTTP {patch_response.status_code} - {patch_response.text}") raise Exception(
f"Failed to update ingest flow: HTTP {patch_response.status_code} - {patch_response.text}"
)
async def update_docling_preset(request, session_manager): async def update_docling_preset(request, session_manager):
@ -577,8 +593,7 @@ async def update_docling_preset(request, session_manager):
# Validate preset parameter # Validate preset parameter
if "preset" not in body: if "preset" not in body:
return JSONResponse( return JSONResponse(
{"error": "preset parameter is required"}, {"error": "preset parameter is required"}, status_code=400
status_code=400
) )
preset = body["preset"] preset = body["preset"]
@ -587,8 +602,10 @@ async def update_docling_preset(request, session_manager):
if preset not in preset_configs: if preset not in preset_configs:
valid_presets = list(preset_configs.keys()) valid_presets = list(preset_configs.keys())
return JSONResponse( return JSONResponse(
{"error": f"Invalid preset '{preset}'. Valid presets: {', '.join(valid_presets)}"}, {
status_code=400 "error": f"Invalid preset '{preset}'. Valid presets: {', '.join(valid_presets)}"
},
status_code=400,
) )
# Get the preset configuration # Get the preset configuration
@ -599,16 +616,16 @@ async def update_docling_preset(request, session_manager):
logger.info(f"Successfully updated docling preset to '{preset}' in ingest flow") logger.info(f"Successfully updated docling preset to '{preset}' in ingest flow")
return JSONResponse({ return JSONResponse(
"message": f"Successfully updated docling preset to '{preset}'", {
"preset": preset, "message": f"Successfully updated docling preset to '{preset}'",
"preset_config": preset_config "preset": preset,
}) "preset_config": preset_config,
}
)
except Exception as e: except Exception as e:
logger.error("Failed to update docling preset", error=str(e)) logger.error("Failed to update docling preset", error=str(e))
return JSONResponse( return JSONResponse(
{"error": f"Failed to update docling preset: {str(e)}"}, {"error": f"Failed to update docling preset: {str(e)}"}, status_code=500
status_code=500
) )