github/privateGPT
1
0
Fork 1
mirror of https://github.com/imartinez/privateGPT.git synced 2025-08-23 17:49:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Updated settings in vector and chat service per Ivans request

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

1
private_gpt/components/vector_store/vector_store_component.py
View File

@ -135,7 +135,6 @@ class VectorStoreComponent:
similarity_top_k: int = 2, similarity_top_k: int = 2,
) -> VectorIndexRetriever: ) -> VectorIndexRetriever:
# This way we support qdrant (using doc_ids) and the rest (using filters) # This way we support qdrant (using doc_ids) and the rest (using filters)
similarity_top_k = self.settings.rag.similarity_top_k
return VectorIndexRetriever( return VectorIndexRetriever(
index=index, index=index,
similarity_top_k=similarity_top_k, similarity_top_k=similarity_top_k,

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

@ -109,7 +109,7 @@ 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 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,