Adding attachment URL to post_message_attachment response. (#96)

poe-platform · Apr 23, 2024 · f471849 · f471849
1 parent c29520d
commit f471849
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/fastapi_poe/base.py b/src/fastapi_poe/base.py
@@ -429,8 +429,10 @@ async def _make_file_attachment_request(
                         f"{response.status_code} {response.reason_phrase}: {''.join(error_pieces)}"
                     )
 
+                response_data = response.json()
                 return AttachmentUploadResponse(
-                    inline_ref=response.json().get("inline_ref")
+                    inline_ref=response_data.get("inline_ref"),
+                    attachment_url=response_data.get("attachment_url"),
                 )
 
             except httpx.HTTPError:

diff --git a/src/fastapi_poe/types.py b/src/fastapi_poe/types.py
@@ -193,6 +193,7 @@ class SettingsResponse(BaseModel):
 
 class AttachmentUploadResponse(BaseModel):
     inline_ref: Optional[str]
+    attachment_url: Optional[str]
 
 
 class PartialResponse(BaseModel):