bump version to 1.0.32 and extend Content type with Media and MediaVa…

…riable
MagnivOrg · Dec 24, 2024 · 1c7a0b4 · 1c7a0b4
1 parent 6b76200
commit 1c7a0b4
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 3 deletions.
diff --git a/promptlayer/__init__.py b/promptlayer/__init__.py
@@ -1,4 +1,4 @@
 from .promptlayer import AsyncPromptLayer, PromptLayer
 
-__version__ = "1.0.31"
+__version__ = "1.0.32"
 __all__ = ["PromptLayer", "AsyncPromptLayer", "__version__"]
diff --git a/promptlayer/types/prompt_template.py b/promptlayer/types/prompt_template.py
@@ -28,7 +28,23 @@ class ImageContent(TypedDict, total=False):
     image_url: ImageUrl
 
 
-Content = Union[TextContent, ImageContent]
+class Media(TypedDict, total=False):
+    title: str
+    type: str
+    url: str
+
+
+class MediaContnt(TypedDict, total=False):
+    type: Literal["media"]
+    media: Media
+
+
+class MediaVariable(TypedDict, total=False):
+    type: Literal["media_variable"]
+    name: str
+
+
+Content = Union[TextContent, ImageContent, MediaContnt, MediaVariable]
 
 
 class Function(TypedDict, total=False):

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "promptlayer"
-version = "1.0.31"
+version = "1.0.32"
 description = "PromptLayer is a platform for prompt engineering and tracks your LLM requests."
 authors = ["Magniv <[email protected]>"]
 license = "Apache-2.0"