github/privateGPT
1
0
Fork 1
mirror of https://github.com/imartinez/privateGPT.git synced 2025-08-08 02:53:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Updated code for mypy

Browse Source
This commit is contained in:
Wesley Stewart 2024-03-20 21:20:42 +00:00
parent 335df17cb8
commit 040e7b7c55
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
private_gpt/server/chat/chat_service.py
View File

@ -109,7 +109,9 @@ class ChatService:
settings = self.settings settings = self.settings
if use_context: if use_context:
vector_index_retriever = self.vector_store_component.get_retriever( 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( return ContextChatEngine.from_defaults(
system_prompt=system_prompt, system_prompt=system_prompt,