feat: add llm_endpoint, llm_api_key, llm_model to QueryRequest
Browse files
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
|