From 8adc914b1f8419dcaadda6778fa0a8ee55f08bad Mon Sep 17 00:00:00 2001
From: hananel <hananel.hadad@accenture.com>
Date: Thu, 11 Jan 2024 19:09:33 +0200
Subject: [PATCH] remove hints that failed mypy
---
semantic_router/schema.py | 16 ++++++++--------
1 file changed, 8 insertions(+), 8 deletions(-)
diff --git a/semantic_router/schema.py b/semantic_router/schema.py
index 8d479ec9..c7912fa1 100644
--- a/semantic_router/schema.py
+++ b/semantic_router/schema.py
@@ -55,12 +55,12 @@ class Message(BaseModel):
role: str
content: str
- def to_openai(self) -> dict[str, str]:
+ def to_openai(self):
if self.role.lower() not in ["user", "assistant", "system"]:
raise ValueError("Role must be either 'user', 'assistant' or 'system'")
return {"role": self.role, "content": self.content}
- def to_cohere(self) -> dict[str, str]:
+ def to_cohere(self):
return {"role": self.role, "message": self.content}
@@ -68,12 +68,12 @@ class Conversation(BaseModel):
messages: list[Message]
def split_by_topic(
- self,
- encoder: BaseEncoder,
- threshold: float = 0.5,
- split_method: Literal[
- "consecutive_similarity_drop", "cumulative_similarity_drop"
- ] = "consecutive_similarity_drop",
+ self,
+ encoder: BaseEncoder,
+ threshold: float = 0.5,
+ split_method: Literal[
+ "consecutive_similarity_drop", "cumulative_similarity_drop"
+ ] = "consecutive_similarity_drop",
) -> dict[str, list[str]]:
docs = [f"{m.role}: {m.content}" for m in self.messages]
return semantic_splitter(
--
GitLab