yumpyy commited on
Commit
cb695d8
·
unverified ·
1 Parent(s): 5400ea0

feat(api): update endpoint to support new parameter

Browse files

Update the API server to support the new parameter from the core library (PR #1032).

lightrag/api/routers/query_routes.py CHANGED
@@ -78,6 +78,11 @@ class QueryRequest(BaseModel):
78
  description="Number of complete conversation turns (user-assistant pairs) to consider in the response context.",
79
  )
80
 
 
 
 
 
 
81
  user_prompt: Optional[str] = Field(
82
  default=None,
83
  description="User-provided prompt for the query. If provided, this will be used instead of the default value from prompt template.",
 
78
  description="Number of complete conversation turns (user-assistant pairs) to consider in the response context.",
79
  )
80
 
81
+ ids: list[str] | None = Field(
82
+ default=None,
83
+ description="List of ids to filter the results."
84
+ )
85
+
86
  user_prompt: Optional[str] = Field(
87
  default=None,
88
  description="User-provided prompt for the query. If provided, this will be used instead of the default value from prompt template.",