Skip to content

Commit 87f523b

Browse files
authored
Merge pull request #55 from ls1intum/refinements
Adjust limits and thresholds
2 parents eca612a + e5645f9 commit 87f523b

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

rag/app/data/database_requests.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@ class DatabaseDocument(BaseModel):
66
link: Optional[str] = None
77
study_programs: List[str]
88
content: str
9-
title: Optional[str] = None
109
org_id: int
10+
title: Optional[str] = None
1111

1212
class DatabaseDocumentMetadata(BaseModel):
1313
link: Optional[str] = None

rag/app/managers/request_handler.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,10 +20,10 @@ def __init__(self, weaviate_manager: WeaviateManager, reranker: Reranker, format
2020
self.text_formatter = formatter
2121
self.response_evaluator = response_evaluator
2222

23-
GEN_THRESH = 0.25
24-
SPEC_THRESH = 0.35
25-
MAX_GENERAL = 8
26-
MAX_SPECIFIC = 8
23+
GEN_THRESH = 0.35
24+
SPEC_THRESH = 0.4
25+
MAX_GENERAL = 6
26+
MAX_SPECIFIC = 6
2727

2828
def handle_question(self, question: str, classification: str, language: str, org_id: int):
2929
"""Handles the question by fetching relevant documents and generating an answer."""

0 commit comments

Comments
 (0)