diff --git a/OAI/types/common.py b/OAI/types/common.py
index 7ba44c4..9c5ebbd 100644
--- a/OAI/types/common.py
+++ b/OAI/types/common.py
@@ -9,9 +9,9 @@ class LogProbs(BaseModel):
     """Represents log probabilities."""
 
     text_offset: List[int] = Field(default_factory=list)
-    token_logprobs: List[float] = Field(default_factory=list)
+    token_logprobs: List[Optional[float]] = Field(default_factory=list)
     tokens: List[str] = Field(default_factory=list)
-    top_logprobs: List[Dict[str, float]] = Field(default_factory=list)
+    top_logprobs: List[Optional[Dict[str, float]]] = Field(default_factory=list)
 
 
 class UsageStats(BaseModel):