fuck noembeddingphrase
This commit is contained in:
1 parent
15a06521b3
commit
ed01ec414e
2 files changed
+29
-29
No files matched your search
@@ -50,10 +50,10 @@ class GetSimilarWordRequest(ModifiedBaseModel):
|
||||
embedding: Embedding
|
||||
|
||||
class GetAllPhrasesResponse(ModifiedBaseModel):
|
||||
phrases: Optional[List[NoEmbeddingPhrase]]
|
||||
phrases: Optional[List[Phrase]]
|
||||
|
||||
class GetFullPhrasesRequest(ModifiedBaseModel):
|
||||
phrases: List[NoEmbeddingPhrase]
|
||||
phrases: List[Phrase]
|
||||
|
||||
class GetFullPhrasesResponse(ModifiedBaseModel):
|
||||
phrases: Optional[List[Phrase]]
|
||||
|
||||
@@ -8,24 +8,24 @@ from .embedding import Embedding
|
||||
import numpy as np
|
||||
|
||||
|
||||
class NoEmbeddingPhrase(Lexeme):
|
||||
def __init__(self, id: str, content: str,original_content: str):
|
||||
#initialised with empty embedding
|
||||
super().__init__(id=id, content=content, embedding=Embedding(data=torch.tensor([])), original_content=original_content)
|
||||
# class NoEmbeddingPhrase(Lexeme):
|
||||
# def __init__(self, id: str, content: str,original_content: str):
|
||||
# #initialised with empty embedding
|
||||
# super().__init__(id=id, content=content, embedding=Embedding(data=torch.tensor([])), original_content=original_content)
|
||||
|
||||
@classmethod
|
||||
def from_dict(cls, data: dict) -> "NoEmbeddingPhrase":
|
||||
return cls(
|
||||
id=data["id"],
|
||||
content=data["content"],
|
||||
original_content=data["original_content"],
|
||||
)
|
||||
@model_validator(mode="before")
|
||||
@classmethod
|
||||
def ignore_embedding(cls, values):
|
||||
# Remove embedding from incoming data
|
||||
values.pop("embedding", None)
|
||||
return values
|
||||
# @classmethod
|
||||
# def from_dict(cls, data: dict) -> "NoEmbeddingPhrase":
|
||||
# return cls(
|
||||
# id=data["id"],
|
||||
# content=data["content"],
|
||||
# original_content=data["original_content"],
|
||||
# )
|
||||
# @model_validator(mode="before")
|
||||
# @classmethod
|
||||
# def ignore_embedding(cls, values):
|
||||
# # Remove embedding from incoming data
|
||||
# values.pop("embedding", None)
|
||||
# return values
|
||||
|
||||
class Phrase(Lexeme):
|
||||
|
||||
@@ -38,13 +38,13 @@ class Phrase(Lexeme):
|
||||
embedding=Embedding.from_dict(data["embedding"]),
|
||||
)
|
||||
|
||||
@classmethod
|
||||
def from_noembeddingphrase(
|
||||
cls, phrase: NoEmbeddingPhrase, embedding: Embedding
|
||||
) -> "Phrase":
|
||||
return cls(
|
||||
id=phrase.id,
|
||||
content=phrase.content,
|
||||
original_content=phrase.original_content,
|
||||
embedding=embedding,
|
||||
)
|
||||
# @classmethod
|
||||
# def from_noembeddingphrase(
|
||||
# cls, phrase: NoEmbeddingPhrase, embedding: Embedding
|
||||
# ) -> "Phrase":
|
||||
# return cls(
|
||||
# id=phrase.id,
|
||||
# content=phrase.content,
|
||||
# original_content=phrase.original_content,
|
||||
# embedding=embedding,
|
||||
# )
|
||||
Reference in new issue
Block a user