Back out collection name

dewy/documents/models.py

@@ -5,9 +5,6 @@
 
 
 class CreateRequest(BaseModel):
-    collection: Optional[str] = None
-    """The name of the collection the document should be added to. Either `collection` or `collection_id` must be provided"""
-
     collection_id: Optional[int] = None
     """The id of the collection the document should be added to. Either `collection` or `collection_id` must be provided"""
 

dewy/documents/router.py

@@ -25,25 +25,15 @@ async def add_document(
 ) -> Document:
     """Add a document."""
 
-    collection_id = req.collection_id    
     async with pg_pool.acquire() as conn:
-        if collection_id is None:
-            collection_id = await conn.fetchval(
-                """
-                SELECT collection_id
-                FROM document
-                WHERE name = $1
-            """,
-            req.collection
-            )
         row = None
         row = await conn.fetchrow(
             """
         INSERT INTO document (collection_id, url, ingest_state)
         VALUES ($1, $2, 'pending')
         RETURNING id, collection_id, url, ingest_state, ingest_error
         """,
-            collection_id,
+            req.collection_id,
             req.url,
         )