github/privateGPT
1
0
Fork 1
mirror of https://github.com/imartinez/privateGPT.git synced 2025-07-01 17:52:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update files for ruff

Browse Source
This commit is contained in:
Wesley Stewart 2024-03-12 19:44:37 +00:00
parent 6bb4d18ade
commit eab8799174
1 changed files with 16 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -7,9 +7,11 @@ from llama_index.core.chat_engine.types import (
)
from llama_index.core.indices import VectorStoreIndex
from llama_index.core.indices.postprocessor import MetadataReplacementPostProcessor
from llama_index.core.postprocessor import SimilarityPostprocessor
from llama_index.core.postprocessor import KeywordNodePostprocessor
from llama_index.core.llms import ChatMessage, MessageRole
from llama_index.core.postprocessor import (
KeywordNodePostprocessor,
SimilarityPostprocessor,
)
from llama_index.core.storage import StorageContext
from llama_index.core.types import TokenGen
from pydantic import BaseModel
@ -24,6 +26,7 @@ from private_gpt.open_ai.extensions.context_filter import ContextFilter
from private_gpt.server.chunks.chunks_service import Chunk
from private_gpt.settings.settings import settings
class Completion(BaseModel):
response: str
sources: list[Chunk] | None = None
@ -105,7 +108,7 @@ class ChatService:
index=self.index, context_filter=context_filter
)
#To use the keyword search, you must install spacy:
# To use the keyword search, you must install spacy:
# pip install spacy
if settings().llm.keywords_include or settings().llm.keywords_exclude:
return ContextChatEngine.from_defaults(
@ -114,8 +117,13 @@ class ChatService:
llm=self.llm_component.llm, # Takes no effect at the moment
node_postprocessors=[
MetadataReplacementPostProcessor(target_metadata_key="window"),
SimilarityPostprocessor(similarity_cutoff=settings().llm.similarity_value),
KeywordNodePostprocessor(required_keywords=settings().llm.keywords_include, exclude_keywords=settings().llm.keywords_exclude),
SimilarityPostprocessor(
similarity_cutoff=settings().llm.similarity_value
),
KeywordNodePostprocessor(
required_keywords=settings().llm.keywords_include,
exclude_keywords=settings().llm.keywords_exclude,
),
],
)
else:
@ -125,7 +133,9 @@ class ChatService:
llm=self.llm_component.llm, # Takes no effect at the moment
node_postprocessors=[
MetadataReplacementPostProcessor(target_metadata_key="window"),
SimilarityPostprocessor(similarity_cutoff=settings().llm.similarity_value),
SimilarityPostprocessor(
similarity_cutoff=settings().llm.similarity_value
),
],
)