fix lint

Author: nimarb

langfuse/model.py

@@ -38,9 +38,6 @@
     CreateDatasetRequest,
 )
 from langfuse.api.resources.prompts import ChatMessage, Prompt, Prompt_Chat, Prompt_Text
-from langfuse.api.resources.prompts.types.chat_message_with_placeholders import (
-    ChatMessageWithPlaceholders,
-)
 
 
 class ModelUsage(TypedDict):
@@ -321,6 +318,23 @@ def __init__(self, prompt: Prompt_Chat, is_fallback: bool = False):
                         content=p.content,
                     ),
                 )
+            # Handle plain dictionaries (fallback case)
+            elif isinstance(p, dict):
+                if p.get("type") == "placeholder" and "name" in p:
+                    self.prompt.append(
+                        ChatMessageWithPlaceholdersDict_Placeholder(
+                            type="placeholder",
+                            name=p["name"],
+                        ),
+                    )
+                elif "role" in p and "content" in p:
+                    self.prompt.append(
+                        ChatMessageWithPlaceholdersDict_Message(
+                            type="message",
+                            role=p["role"],
+                            content=p["content"],
+                        ),
+                    )
 
     def compile(
         self, **kwargs