Simple API No Longer Require Specify Prompt (#40)

This commit is contained in:
Yuhong Sun
2024-02-23 20:46:09 -08:00
committed by Chris Weaver
parent 6b46a71cb5
commit c1862e961b
2 changed files with 3 additions and 6 deletions

View File

@@ -57,9 +57,9 @@ def handle_simplified_chat_message(
db_session: Session = Depends(get_session),
) -> ChatBasicResponse:
"""This is a Non-Streaming version that only gives back a minimal set of information"""
logger.info(f"Received new chat message: {chat_message_req.message}")
logger.info(f"Received new simple api chat message: {chat_message_req.message}")
if not chat_message_req.message and chat_message_req.prompt_id is not None:
if not chat_message_req.message:
raise HTTPException(status_code=400, detail="Empty chat message is invalid")
try:
@@ -86,7 +86,7 @@ def handle_simplified_chat_message(
chat_session_id=chat_message_req.chat_session_id,
parent_message_id=parent_message.id,
message=chat_message_req.message,
prompt_id=chat_message_req.prompt_id,
prompt_id=None,
search_doc_ids=chat_message_req.search_doc_ids,
retrieval_options=retrieval_options,
query_override=chat_message_req.query_override,

View File

@@ -17,9 +17,6 @@ class BasicCreateChatMessageRequest(BaseModel):
# Allows the caller to specify the exact search query they want to use
# will disable Query Rewording if specified
query_override: str | None = None
# If no prompt provided, provide canned retrieval answer, no actually LLM flow
# Use prompt_id 0 to use the system default prompt which is Answer-Question
prompt_id: int | None = 0
# If search_doc_ids provided, then retrieval options are unused
search_doc_ids: list[int] | None = None