seekrai 0.5.25__tar.gz → 0.5.26__tar.gz

{seekrai-0.5.25 → seekrai-0.5.26}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: seekrai
-Version: 0.5.25
+Version: 0.5.26
 Summary: Python client for SeekrAI
 License: Apache-2.0
 License-File: LICENSE

{seekrai-0.5.25 → seekrai-0.5.26}/pyproject.toml

@@ -13,7 +13,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "seekrai"
-version = "0.5.25"
+version = "0.5.26"
 authors = [
     "SeekrFlow <support@seekr.com>"
 ]

{seekrai-0.5.25 → seekrai-0.5.26}/src/seekrai/resources/chat/completions.py

@@ -20,7 +20,7 @@ class ChatCompletions:
     def create(
         self,
         *,
-        messages: List[Dict[str, str]],
+        messages: List[Dict[str, Any]],
         model: str,
         max_completion_tokens: int | None = None,
         max_tokens: int | None = 512,
@@ -43,8 +43,11 @@ class ChatCompletions:
         Method to generate completions based on a given prompt using a specified model.
 
         Args:
-            messages (List[Dict[str, str]]): A list of messages in the format
-                `[{"role": seekrai.types.chat_completions.MessageRole, "content": TEXT}, ...]`
+            messages (List[Dict[str, Any]]): A list of messages in the format
+                `[{"role": seekrai.types.chat_completions.MessageRole, "content": TEXT}, ...]` or
+                `[{"role": seekrai.types.chat_completions.MessageRole, "content": PARTS}, ...]`
+                where PARTS is a list of content dicts, e.g. {"type": "text", "text": "..."} or
+                {"type": "image_url", "image_url": {"url": "data:image/jpeg;base64,..."}}
             model (str): The name of the model to query.
             max_completion_tokens (int, optional): The maximum number of tokens the output can contain.
             max_tokens (int, optional): The maximum number of tokens to generate.
@@ -141,7 +144,7 @@ class AsyncChatCompletions:
     async def create(
         self,
         *,
-        messages: List[Dict[str, str]],
+        messages: List[Dict[str, Any]],
         model: str,
         max_completion_tokens: int | None = None,
         max_tokens: int | None = 512,
@@ -164,8 +167,11 @@ class AsyncChatCompletions:
         Async method to generate completions based on a given prompt using a specified model.
 
         Args:
-            messages (List[Dict[str, str]]): A list of messages in the format
-                `[{"role": seekrai.types.chat_completions.MessageRole, "content": TEXT}, ...]`
+            messages (List[Dict[str, Any]]): A list of messages in the format
+                `[{"role": seekrai.types.chat_completions.MessageRole, "content": TEXT}, ...]` or
+                `[{"role": seekrai.types.chat_completions.MessageRole, "content": PARTS}, ...]`
+                where PARTS is a list of content dicts, e.g. {"type": "text", "text": "..."} or
+                {"type": "image_url", "image_url": {"url": "data:image/jpeg;base64,..."}}
             model (str): The name of the model to query.
             max_completion_tokens (int, optional): The maximum number of tokens the output can contain.
             max_tokens (int, optional): The maximum number of tokens to generate.

{seekrai-0.5.25 → seekrai-0.5.26}/src/seekrai/types/chat_completions.py

@@ -40,7 +40,7 @@ class ToolCalls(BaseModel):
 
 class ChatCompletionMessage(BaseModel):
     role: MessageRole
-    content: str | None = None
+    content: str | List[Dict[str, Any]] | None = None
     # tool_calls: List[ToolCalls] | None = None