diff --git a/server/retriever/splitter.py b/server/retriever/splitter.py index 3ac892e..7c9c5c8 100644 --- a/server/retriever/splitter.py +++ b/server/retriever/splitter.py @@ -57,8 +57,7 @@ def __init__( every document """ if chunk_overlap > chunk_size: - raise ValueError(f"Got a larger chunk overlap ({ - chunk_overlap}) than chunk size ({chunk_size}), should be smaller.") + raise ValueError(f"Got a larger chunk overlap ({chunk_overlap}) than chunk size ({chunk_size}), should be smaller.") self._chunk_size = chunk_size self._chunk_overlap = chunk_overlap self._length_function = length_function diff --git a/server/server.py b/server/server.py index a7d1206..f457a3d 100644 --- a/server/server.py +++ b/server/server.py @@ -87,8 +87,11 @@ def format_messages(messages: List[Dict], indexed_files: Optional[str], instruct 'personalization', '').strip().replace('\n', '; ') response = instructions.get('response', '').strip().replace('\n', '; ') - context = f"with background knowledge of { - indexed_files.strip().replace('\n', '; ')}" if indexed_files else '' + bits = indexed_files.strip().replace('\n', '; ') + if indexed_files: + context = f"with background knowledge of {bits}" + else: + context = '' audience = personalization if personalization else 'general' style = response if response else 'technical, accurate, and professional' diff --git a/server/utils.py b/server/utils.py index c939fc1..c7b8081 100644 --- a/server/utils.py +++ b/server/utils.py @@ -162,8 +162,7 @@ def sample(logits: mx.array) -> Tuple[mx.array, float]: repetition_penalty < 0 or not isinstance(repetition_penalty, float) ): raise ValueError( - f"repetition_penalty must be a non-negative float, got { - repetition_penalty}" + f"repetition_penalty must be a non-negative float, got {repetition_penalty}" ) y = prompt