Skip to content
Snippets Groups Projects
Unverified Commit 99b2d075 authored by James Briggs's avatar James Briggs
Browse files

threshold applies to encoder

parent 545a4c8b
No related branches found
No related tags found
No related merge requests found
...@@ -18,6 +18,7 @@ class ConsecutiveSimSplitter(BaseSplitter): ...@@ -18,6 +18,7 @@ class ConsecutiveSimSplitter(BaseSplitter):
score_threshold: float = 0.45, score_threshold: float = 0.45,
): ):
super().__init__(name=name, score_threshold=score_threshold, encoder=encoder) super().__init__(name=name, score_threshold=score_threshold, encoder=encoder)
encoder.score_threshold = score_threshold
def __call__(self, docs: List[str]): def __call__(self, docs: List[str]):
# Check if there's only a single document # Check if there's only a single document
......
...@@ -18,6 +18,7 @@ class CumulativeSimSplitter(BaseSplitter): ...@@ -18,6 +18,7 @@ class CumulativeSimSplitter(BaseSplitter):
score_threshold: float = 0.45, score_threshold: float = 0.45,
): ):
super().__init__(name=name, score_threshold=score_threshold, encoder=encoder) super().__init__(name=name, score_threshold=score_threshold, encoder=encoder)
encoder.score_threshold = score_threshold
def __call__(self, docs: List[str]): def __call__(self, docs: List[str]):
total_docs = len(docs) total_docs = len(docs)
......
...@@ -50,6 +50,11 @@ class Conversation(BaseModel): ...@@ -50,6 +50,11 @@ class Conversation(BaseModel):
def add_new_messages(self, new_messages: List[Message]): def add_new_messages(self, new_messages: List[Message]):
"""Adds new messages to the conversation.
:param messages: The new messages to be added to the conversation.
:type messages: List[Message]
"""
self.messages.extend(new_messages) self.messages.extend(new_messages)
def remove_topics(self): def remove_topics(self):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment