From 8a984a6a5245e6b3daf75ba4c13474a244f812cf Mon Sep 17 00:00:00 2001
From: Blake Rosenthal <blakepdx@protonmail.com>
Date: Tue, 11 Jun 2024 12:18:01 -0700
Subject: [PATCH] don't keep sources

---
 ragna/deploy/_api/core.py    |  4 +---
 ragna/deploy/_api/schemas.py | 19 +------------------
 2 files changed, 2 insertions(+), 21 deletions(-)

diff --git a/ragna/deploy/_api/core.py b/ragna/deploy/_api/core.py
index a0c7f8c3..2a95ae96 100644
--- a/ragna/deploy/_api/core.py
+++ b/ragna/deploy/_api/core.py
@@ -233,9 +233,7 @@ def schema_to_core_chat(
             chat_name=chat.metadata.name,
             **chat.metadata.params,
         )
-        core_chat._messages = [
-            message.to_core() for message in chat.messages
-        ]
+        core_chat._messages = [message.to_core() for message in chat.messages]
         core_chat._prepared = chat.prepared
 
         return core_chat
diff --git a/ragna/deploy/_api/schemas.py b/ragna/deploy/_api/schemas.py
index 73acd717..e01d5953 100644
--- a/ragna/deploy/_api/schemas.py
+++ b/ragna/deploy/_api/schemas.py
@@ -26,13 +26,6 @@ def from_core(cls, document: ragna.core.Document) -> Document:
             name=document.name,
         )
 
-    @classmethod
-    def to_core(cls, document: Document) -> ragna.core.Document:
-        return ragna.core.Document(
-            id=document.id,
-            name=document.name,
-        )
-
 
 class DocumentUpload(BaseModel):
     parameters: ragna.core.DocumentUploadParameters
@@ -57,16 +50,6 @@ def from_core(cls, source: ragna.core.Source) -> Source:
             num_tokens=source.num_tokens,
         )
 
-    @classmethod
-    def to_core(cls, source: Source) -> ragna.core.Source:
-        return ragna.core.Source(
-            id=source.id,
-            document=Document.to_core(source.document),
-            location=source.location,
-            content=source.content,
-            num_tokens=source.num_tokens,
-        )
-
 
 class Message(BaseModel):
     id: uuid.UUID = Field(default_factory=uuid.uuid4)
@@ -87,7 +70,7 @@ def to_core(self) -> ragna.core.Message:
         return ragna.core.Message(
             content=self.content,
             role=self.role,
-            sources=[source.to_core() for source in self.sources],
+            sources=[],
         )