feat: allow "url" source_type for pdf in google memory adapter (#108)

riccardotornesello · web-flow · commit 499c7c160fca · 2026-02-02T12:02:19.000+01:00
diff --git a/datapizza-ai-clients/datapizza-ai-clients-google/datapizza/clients/google/memory_adapter.py b/datapizza-ai-clients/datapizza-ai-clients-google/datapizza/clients/google/memory_adapter.py
@@ -102,6 +102,11 @@ def _process_pdf_block(self, block: MediaBlock) -> types.Part | dict:
                         "data": pdf_bytes,
                     }
                 }
+            case "url":
+                return types.Part.from_uri(
+                    file_uri=block.media.source,
+                    mime_type="application/pdf",
+                )
 
             case _:
                 raise NotImplementedError(

Original file line number	Diff line number	Diff line change
`@@ -102,6 +102,11 @@ def _process_pdf_block(self, block: MediaBlock) -> types.Part \| dict:`
`102`	`102`	`"data": pdf_bytes,`
`103`	`103`	`}`
`104`	`104`	`}`
	`105`	`+ case "url":`
	`106`	`+ return types.Part.from_uri(`
	`107`	`+ file_uri=block.media.source,`
	`108`	`+ mime_type="application/pdf",`
	`109`	`+ )`
`105`	`110`
`106`	`111`	`case _:`
`107`	`112`	`raise NotImplementedError(`