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

Relax conversation history role validation requirements

Browse source 
• Remove strict role value checking
• Allow any non-empty string roles
This commit is contained in:
yangdx 2025-09-29 13:10:15 +08:00
parent ba216787c1
commit df43afc89b
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
lightrag/api/routers/query_routes.py
View file

@ -111,10 +111,10 @@ class QueryRequest(BaseModel):
if conversation_history is None: if conversation_history is None:
return None return None
for msg in conversation_history: for msg in conversation_history:
if "role" not in msg or msg["role"] not in {"user", "assistant"}: if "role" not in msg:
raise ValueError( raise ValueError("Each message must have a 'role' key.")
"Each message must have a 'role' key with value 'user' or 'assistant'." if not isinstance(msg["role"], str) or not msg["role"].strip():
) raise ValueError("Each message 'role' must be a non-empty string.")
return conversation_history return conversation_history
def to_query_params(self, is_stream: bool) -> "QueryParam": def to_query_params(self, is_stream: bool) -> "QueryParam":