highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-04-09 20:39:29 +02:00
Code Issues Packages Projects Releases Wiki Activity

[1802] adjust the code to support a different db schemas (#1803)

Browse Source
This commit is contained in:
Andres Jose Sebastian Rincon Gonzalez 2024-09-08 16:16:54 -05:00 committed by GitHub
parent be4b6189d2
commit 3fa9676478
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
backend/danswer/db/models.py
View File

@ -61,7 +61,7 @@ from shared_configs.enums import RerankerProvider
class Base(DeclarativeBase):
pass
__abstract__ = True
class EncryptedString(TypeDecorator):
@ -448,7 +448,7 @@ class Document(Base):
)
tags = relationship(
"Tag",
secondary="document__tag",
secondary=Document__Tag.__table__,
back_populates="documents",
)
@ -465,7 +465,7 @@ class Tag(Base):
documents = relationship(
"Document",
secondary="document__tag",
secondary=Document__Tag.__table__,
back_populates="tags",
)
@ -814,7 +814,7 @@ class SearchDoc(Base):
chat_messages = relationship(
"ChatMessage",
secondary="chat_message__search_doc",
secondary=ChatMessage__SearchDoc.__table__,
back_populates="search_docs",
)
@ -957,7 +957,7 @@ class ChatMessage(Base):
)
search_docs: Mapped[list["SearchDoc"]] = relationship(
"SearchDoc",
secondary="chat_message__search_doc",
secondary=ChatMessage__SearchDoc.__table__,
back_populates="chat_messages",
)
# NOTE: Should always be attached to the `assistant` message.