Skip to content

Commit

Permalink
Merge pull request #13 from ls1intum/add-maximum-question-length
Browse files Browse the repository at this point in the history
Add max question length
  • Loading branch information
ninori9 authored Oct 29, 2024
2 parents bba76d4 + f0e107a commit 1fde5bb
Show file tree
Hide file tree
Showing 7 changed files with 21 additions and 7 deletions.
17 changes: 15 additions & 2 deletions app/api/question_router.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,17 +17,22 @@ class UserRequest(BaseModel):

router = APIRouter(prefix="/api/v1/question", tags=["response"])


@router.post("/ask")
async def ask(request: UserRequest):
question = request.message
classification = request.study_program
language = request.language
if not question or not classification:
raise HTTPException(status_code=400, detail="No question or classification provided")

if len(question) > config.MAX_MESSAGE_LENGTH:
raise HTTPException(
status_code=400,
detail=f"Question length exceeds the allowed limit of {config.MAX_MESSAGE_LENGTH} characters"
)

logging.info(f"Received question: {question} with classification: {classification}")

if config.TEST_MODE:
answer, used_tokens, general_context, specific_context = request_handler.handle_question_test_mode(question,
classification,
Expand All @@ -44,6 +49,14 @@ async def chat(request: UserChat):
messages = request.messages
if not messages:
raise HTTPException(status_code=400, detail="No messages have been provided")

last_message = messages[-1].message
if len(last_message) > config.MAX_MESSAGE_LENGTH:
raise HTTPException(
status_code=400,
detail=f"Message length exceeds the allowed limit of {config.MAX_MESSAGE_LENGTH} characters"
)

logging.info(f"Received messages.")
answer = request_handler.handle_chat(messages, study_program=request.study_program)
return {"answer": answer}
Expand Down
1 change: 1 addition & 0 deletions app/utils/environment.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@


class Config:
MAX_MESSAGE_LENGTH = 3000
# Weaviate Database
WEAVIATE_URL = os.getenv("WEAVIATE_URL", "localhost")
WEAVIATE_PORT = os.getenv("WEAVIATE_PORT", "8001")
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
{
"orderKey": 1,
"sender": "AA",
"message": "Bitte wenden Sie sich hierfür an die Sekretärin Ihres Prüfungsausschusses."
"message": "Bitte wenden Sie sich hierfür an die Schriftführung Ihres Studiengangs."
}
]
}
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
{
"orderKey": 1,
"sender": "AA",
"message": "Bitte wenden Sie sich hierzu an die Sekretärin des für Ihren Studiengang zuständigen Prüfungsausschusses. Die Kontaktdaten finden Sie hier: cit.tum.de/cit/kontakt-studium-informatik/"
"message": "Bitte wenden Sie sich hierzu an die für Ihren Studiengang zuständige Schriftführung. Die Kontaktdaten finden Sie hier: cit.tum.de/cit/kontakt-studium-informatik/"
}
]
}
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
{
"orderKey": 1,
"sender": "AA",
"message": "Bitte wenden Sie sich bei Fragen zur Kursanmeldung an die Sekretärin des Prüfungsausschusses."
"message": "Bitte wenden Sie sich bei Fragen zur Kursanmeldung an die Schriftführung Ihres Studiengangs."
}
]
}
2 changes: 1 addition & 1 deletion knowledge/sample-correspondences/Praktikum_de.json
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
{
"orderKey": 1,
"sender": "AA",
"message": "Bitte wenden Sie sich hierfür an die Sekretärin Ihres Prüfungsausschusses."
"message": "Bitte wenden Sie sich hierfür an die Schriftführung Ihres Studiengangs."
}
]
}
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
{
"orderKey": 1,
"sender": "AA",
"message": "Bitte wenden Sie sich für Unterstützung bei der Aktualisierung Ihrer Daten an die Sekretärin Ihres Prüfungsausschusses."
"message": "Bitte wenden Sie sich für Unterstützung bei der Aktualisierung Ihrer Daten an die zuständige Schriftführung."
}
]
}

0 comments on commit 1fde5bb

Please sign in to comment.