push

Author: hassiebp

langfuse/api/prompts/client.py

@@ -173,7 +173,6 @@ def create(
         from langfuse import LangfuseAPI
         from langfuse.prompts import (
             ChatMessage,
-            ChatMessageType,
             CreateChatPromptRequest,
             CreateChatPromptType,
         )
@@ -193,12 +192,10 @@ def create(
                     ChatMessage(
                         role="role",
                         content="content",
-                        type=ChatMessageType.CHATMESSAGE,
                     ),
                     ChatMessage(
                         role="role",
                         content="content",
-                        type=ChatMessageType.CHATMESSAGE,
                     ),
                 ],
                 type=CreateChatPromptType.CHAT,
@@ -438,7 +435,6 @@ async def create(
         from langfuse import AsyncLangfuseAPI
         from langfuse.prompts import (
             ChatMessage,
-            ChatMessageType,
             CreateChatPromptRequest,
             CreateChatPromptType,
         )
@@ -461,12 +457,10 @@ async def main() -> None:
                         ChatMessage(
                             role="role",
                             content="content",
-                            type=ChatMessageType.CHATMESSAGE,
                         ),
                         ChatMessage(
                             role="role",
                             content="content",
-                            type=ChatMessageType.CHATMESSAGE,
                         ),
                     ],
                     type=CreateChatPromptType.CHAT,

langfuse/api/prompts/types/chat_message.py

@@ -10,7 +10,7 @@
 class ChatMessage(UniversalBaseModel):
     role: str
     content: str
-    type: ChatMessageType
+    type: typing.Optional[ChatMessageType] = None
 
     model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
         extra="allow", frozen=True

langfuse/api/prompts/types/placeholder_message.py

@@ -9,7 +9,7 @@
 
 class PlaceholderMessage(UniversalBaseModel):
     name: str
-    type: PlaceholderMessageType
+    type: typing.Optional[PlaceholderMessageType] = None
 
     model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
         extra="allow", frozen=True