push

Author: hassiebp

langfuse/api/__init__.py

@@ -231,6 +231,7 @@
     from .prompts import (
         BasePrompt,
         ChatMessage,
+        ChatMessageType,
         ChatMessageWithPlaceholders,
         ChatPrompt,
         CreateChatPromptRequest,
@@ -239,6 +240,7 @@
         CreateTextPromptRequest,
         CreateTextPromptType,
         PlaceholderMessage,
+        PlaceholderMessageType,
         Prompt,
         PromptMeta,
         PromptMetaListResponse,
@@ -315,6 +317,7 @@
     "CategoricalScore": ".commons",
     "CategoricalScoreV1": ".commons",
     "ChatMessage": ".prompts",
+    "ChatMessageType": ".prompts",
     "ChatMessageWithPlaceholders": ".prompts",
     "ChatPrompt": ".prompts",
     "Comment": ".commons",
@@ -446,6 +449,7 @@
     "PaginatedSessions": ".sessions",
     "PatchMediaBody": ".media",
     "PlaceholderMessage": ".prompts",
+    "PlaceholderMessageType": ".prompts",
     "PricingTier": ".commons",
     "PricingTierCondition": ".commons",
     "PricingTierInput": ".commons",
@@ -600,6 +604,7 @@ def __dir__():
     "CategoricalScore",
     "CategoricalScoreV1",
     "ChatMessage",
+    "ChatMessageType",
     "ChatMessageWithPlaceholders",
     "ChatPrompt",
     "Comment",
@@ -734,6 +739,7 @@ def __dir__():
     "PaginatedSessions",
     "PatchMediaBody",
     "PlaceholderMessage",
+    "PlaceholderMessageType",
     "PricingTier",
     "PricingTierCondition",
     "PricingTierInput",

langfuse/api/prompts/__init__.py

@@ -9,6 +9,7 @@
     from .types import (
         BasePrompt,
         ChatMessage,
+        ChatMessageType,
         ChatMessageWithPlaceholders,
         ChatPrompt,
         CreateChatPromptRequest,
@@ -17,6 +18,7 @@
         CreateTextPromptRequest,
         CreateTextPromptType,
         PlaceholderMessage,
+        PlaceholderMessageType,
         Prompt,
         PromptMeta,
         PromptMetaListResponse,
@@ -28,6 +30,7 @@
 _dynamic_imports: typing.Dict[str, str] = {
     "BasePrompt": ".types",
     "ChatMessage": ".types",
+    "ChatMessageType": ".types",
     "ChatMessageWithPlaceholders": ".types",
     "ChatPrompt": ".types",
     "CreateChatPromptRequest": ".types",
@@ -36,6 +39,7 @@
     "CreateTextPromptRequest": ".types",
     "CreateTextPromptType": ".types",
     "PlaceholderMessage": ".types",
+    "PlaceholderMessageType": ".types",
     "Prompt": ".types",
     "PromptMeta": ".types",
     "PromptMetaListResponse": ".types",
@@ -76,6 +80,7 @@ def __dir__():
 __all__ = [
     "BasePrompt",
     "ChatMessage",
+    "ChatMessageType",
     "ChatMessageWithPlaceholders",
     "ChatPrompt",
     "CreateChatPromptRequest",
@@ -84,6 +89,7 @@ def __dir__():
     "CreateTextPromptRequest",
     "CreateTextPromptType",
     "PlaceholderMessage",
+    "PlaceholderMessageType",
     "Prompt",
     "PromptMeta",
     "PromptMetaListResponse",

langfuse/api/prompts/client.py

@@ -173,6 +173,7 @@ def create(
         from langfuse import LangfuseAPI
         from langfuse.prompts import (
             ChatMessage,
+            ChatMessageType,
             CreateChatPromptRequest,
             CreateChatPromptType,
         )
@@ -192,10 +193,12 @@ def create(
                     ChatMessage(
                         role="role",
                         content="content",
+                        type=ChatMessageType.CHATMESSAGE,
                     ),
                     ChatMessage(
                         role="role",
                         content="content",
+                        type=ChatMessageType.CHATMESSAGE,
                     ),
                 ],
                 type=CreateChatPromptType.CHAT,
@@ -435,6 +438,7 @@ async def create(
         from langfuse import AsyncLangfuseAPI
         from langfuse.prompts import (
             ChatMessage,
+            ChatMessageType,
             CreateChatPromptRequest,
             CreateChatPromptType,
         )
@@ -457,10 +461,12 @@ 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/__init__.py

@@ -8,6 +8,7 @@
 if typing.TYPE_CHECKING:
     from .base_prompt import BasePrompt
     from .chat_message import ChatMessage
+    from .chat_message_type import ChatMessageType
     from .chat_message_with_placeholders import ChatMessageWithPlaceholders
     from .chat_prompt import ChatPrompt
     from .create_chat_prompt_request import CreateChatPromptRequest
@@ -16,6 +17,7 @@
     from .create_text_prompt_request import CreateTextPromptRequest
     from .create_text_prompt_type import CreateTextPromptType
     from .placeholder_message import PlaceholderMessage
+    from .placeholder_message_type import PlaceholderMessageType
     from .prompt import Prompt, Prompt_Chat, Prompt_Text
     from .prompt_meta import PromptMeta
     from .prompt_meta_list_response import PromptMetaListResponse
@@ -24,6 +26,7 @@
 _dynamic_imports: typing.Dict[str, str] = {
     "BasePrompt": ".base_prompt",
     "ChatMessage": ".chat_message",
+    "ChatMessageType": ".chat_message_type",
     "ChatMessageWithPlaceholders": ".chat_message_with_placeholders",
     "ChatPrompt": ".chat_prompt",
     "CreateChatPromptRequest": ".create_chat_prompt_request",
@@ -32,6 +35,7 @@
     "CreateTextPromptRequest": ".create_text_prompt_request",
     "CreateTextPromptType": ".create_text_prompt_type",
     "PlaceholderMessage": ".placeholder_message",
+    "PlaceholderMessageType": ".placeholder_message_type",
     "Prompt": ".prompt",
     "PromptMeta": ".prompt_meta",
     "PromptMetaListResponse": ".prompt_meta_list_response",
@@ -72,6 +76,7 @@ def __dir__():
 __all__ = [
     "BasePrompt",
     "ChatMessage",
+    "ChatMessageType",
     "ChatMessageWithPlaceholders",
     "ChatPrompt",
     "CreateChatPromptRequest",
@@ -80,6 +85,7 @@ def __dir__():
     "CreateTextPromptRequest",
     "CreateTextPromptType",
     "PlaceholderMessage",
+    "PlaceholderMessageType",
     "Prompt",
     "PromptMeta",
     "PromptMetaListResponse",

langfuse/api/prompts/types/chat_message.py

@@ -4,11 +4,13 @@
 
 import pydantic
 from ...core.pydantic_utilities import UniversalBaseModel
+from .chat_message_type import ChatMessageType
 
 
 class ChatMessage(UniversalBaseModel):
     role: str
     content: str
+    type: ChatMessageType
 
     model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
         extra="allow", frozen=True

langfuse/api/prompts/types/placeholder_message.py

@@ -4,10 +4,12 @@
 
 import pydantic
 from ...core.pydantic_utilities import UniversalBaseModel
+from .placeholder_message_type import PlaceholderMessageType
 
 
 class PlaceholderMessage(UniversalBaseModel):
     name: str
+    type: PlaceholderMessageType
 
     model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
         extra="allow", frozen=True