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

Fix Linting errors

Browse source
This commit is contained in:
NeelM0906 2025-10-09 16:52:22 -04:00
parent 9f44e89de7
commit f6d1fb98ac
5 changed files with 15 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
lightrag/api/routers/graph_routes.py
View file

@ -284,7 +284,9 @@ def create_graph_routes(rag, api_key: Optional[str] = None):
"data": entity_data, "data": entity_data,
} }
except ValueError as ve: except ValueError as ve:
logger.error(f"Validation error creating entity '{request.entity_name}': {str(ve)}") logger.error(
f"Validation error creating entity '{request.entity_name}': {str(ve)}"
)
raise HTTPException(status_code=400, detail=str(ve)) raise HTTPException(status_code=400, detail=str(ve))
except Exception as e: except Exception as e:
logger.error(f"Error creating entity '{request.entity_name}': {str(e)}") logger.error(f"Error creating entity '{request.entity_name}': {str(e)}")
@ -328,9 +330,13 @@ def create_graph_routes(rag, api_key: Optional[str] = None):
) )
if not source_exists: if not source_exists:
raise ValueError(f"Source entity '{request.source_entity}' does not exist") raise ValueError(
f"Source entity '{request.source_entity}' does not exist"
)
if not target_exists: if not target_exists:
raise ValueError(f"Target entity '{request.target_entity}' does not exist") raise ValueError(
f"Target entity '{request.target_entity}' does not exist"
)
# Create the relationship # Create the relationship
await rag.chunk_entity_relation_graph.upsert_edge( await rag.chunk_entity_relation_graph.upsert_edge(

4
lightrag/kg/postgres_impl.py
View file

@ -828,8 +828,8 @@ class PostgreSQLDB:
# Execute the migration # Execute the migration
alter_sql = f""" alter_sql = f"""
ALTER TABLE {migration['table']} ALTER TABLE {migration["table"]}
ALTER COLUMN {migration['column']} TYPE {migration['new_type']} ALTER COLUMN {migration["column"]} TYPE {migration["new_type"]}
""" """
await self.execute(alter_sql) await self.execute(alter_sql)

3
lightrag/utils.py
View file

@ -1472,8 +1472,7 @@ async def aexport_data(
else: else:
raise ValueError( raise ValueError(
f"Unsupported file format: {file_format}. " f"Unsupported file format: {file_format}. Choose from: csv, excel, md, txt"
f"Choose from: csv, excel, md, txt"
) )
if file_format is not None: if file_format is not None:
print(f"Data exported to: {output_path} with format: {file_format}") print(f"Data exported to: {output_path} with format: {file_format}")