From 040e7b7c5511f9c3c9fa28071c269bd8566a359f Mon Sep 17 00:00:00 2001 From: Wesley Stewart Date: Wed, 20 Mar 2024 21:20:42 +0000 Subject: [PATCH] Updated code for mypy --- private_gpt/server/chat/chat_service.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/private_gpt/server/chat/chat_service.py b/private_gpt/server/chat/chat_service.py index dcd0514d..ea57f2c0 100644 --- a/private_gpt/server/chat/chat_service.py +++ b/private_gpt/server/chat/chat_service.py @@ -109,7 +109,9 @@ class ChatService: settings = self.settings if use_context: vector_index_retriever = self.vector_store_component.get_retriever( - index=self.index, context_filter=context_filter, similarity_top_k=self.settings.rag.similarity_top_k, + index=self.index, + context_filter=context_filter, + similarity_top_k=self.settings.rag.similarity_top_k, ) return ContextChatEngine.from_defaults( system_prompt=system_prompt,