jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

✨ (record.py): add __eq__ method to compare Record instances based on their data attribute

Browse source
This commit is contained in:
ogabrielluiz 2024-06-10 14:12:24 -03:00
commit f8aa1274fc
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/base/langflow/schema/record.py
View file

@ -1,12 +1,12 @@
import copy import copy
import json import json
from typing import cast, Optional from typing import Optional, cast
from langchain_core.documents import Document from langchain_core.documents import Document
from langchain_core.messages import AIMessage, BaseMessage, HumanMessage, SystemMessage from langchain_core.messages import AIMessage, BaseMessage, HumanMessage, SystemMessage
from langchain_core.prompt_values import ImagePromptValue
from langchain_core.prompts.image import ImagePromptTemplate from langchain_core.prompts.image import ImagePromptTemplate
from pydantic import BaseModel, model_serializer, model_validator from pydantic import BaseModel, model_serializer, model_validator
from langchain_core.prompt_values import ImagePromptValue
class Record(BaseModel): class Record(BaseModel):
@ -200,3 +200,6 @@ class Record(BaseModel):
def __contains__(self, key): def __contains__(self, key):
return key in self.data return key in self.data
def __eq__(self, other):
return isinstance(other, Record) and self.data == other.data