From 1a1feb9ff88d61b0262bec7a2acb88e11b7e7dde Mon Sep 17 00:00:00 2001
From: Jelle Zijlstra <jelle.zijlstra@gmail.com>
Date: Tue, 1 Aug 2023 13:03:39 -0700
Subject: [PATCH] Add new fields for API 1.2 (#5)

---
 pyproject.toml           |  2 +-
 src/fastapi_poe/types.py | 10 ++++++++++
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 3df3a78..cc5e1d8 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "fastapi_poe"
-version = "0.0.15"
+version = "0.0.16"
 authors = [
   { name="Lida Li", email="lli@quora.com" },
 ]
diff --git a/src/fastapi_poe/types.py b/src/fastapi_poe/types.py
index 3608b23..48adcee 100644
--- a/src/fastapi_poe/types.py
+++ b/src/fastapi_poe/types.py
@@ -15,6 +15,12 @@ class MessageFeedback(BaseModel):
     reason: Optional[str]
 
 
+class Attachment(BaseModel):
+    url: str
+    content_type: str
+    name: str
+
+
 class ProtocolMessage(BaseModel):
     """A message as used in the Poe protocol."""
 
@@ -24,6 +30,7 @@ class ProtocolMessage(BaseModel):
     timestamp: int = 0
     message_id: str = ""
     feedback: List[MessageFeedback] = Field(default_factory=list)
+    attachments: List[Attachment] = Field(default_factory=list)
 
 
 class BaseRequest(BaseModel):
@@ -40,6 +47,7 @@ class QueryRequest(BaseRequest):
     user_id: Identifier
     conversation_id: Identifier
     message_id: Identifier
+    metadata: Identifier = ""
     api_key: str = "<missing>"
     temperature: float = 0.7
     skip_system_prompt: bool = False
@@ -70,3 +78,5 @@ class ReportErrorRequest(BaseRequest):
 class SettingsResponse(BaseModel):
     context_clear_window_secs: Optional[int] = None
     allow_user_context_clear: bool = True
+    server_bot_dependencies: Dict[str, int] = Field(default_factory=dict)
+    allow_attachments: bool = False