Nottybro commited on
Commit
b7470ec
·
verified ·
1 Parent(s): cfb2fb8

feat: add llm_endpoint, llm_api_key, llm_model to QueryRequest

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -35,6 +35,9 @@ class QueryRequest(BaseModel):
35
  top_k: Optional[int] = 5
36
  rerank: Optional[bool] = True
37
  use_web: Optional[bool] = False
 
 
 
38
 
39
  class QueryResponse(BaseModel):
40
  answer: str
 
35
  top_k: Optional[int] = 5
36
  rerank: Optional[bool] = True
37
  use_web: Optional[bool] = False
38
+ llm_endpoint: Optional[str] = None
39
+ llm_api_key: Optional[str] = None
40
+ llm_model: Optional[str] = None
41
 
42
  class QueryResponse(BaseModel):
43
  answer: str