update

jhakulin · May 5, 2024 · 3cd4a39 · 3cd4a39
1 parent 9d4161e
commit 3cd4a39
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 18 deletions.
diff --git a/sdk/azure-ai-assistant/azure/ai/assistant/management/async_conversation_thread_client.py b/sdk/azure-ai-assistant/azure/ai/assistant/management/async_conversation_thread_client.py
@@ -275,15 +275,25 @@ async def create_conversation_thread_message(
             thread_id = self._thread_config.get_thread_id_by_name(thread_name)
             attachments = await self._update_message_attachments(thread_id, attachments) if attachments is not None else []
 
-            # Create the message with the attachments
-            await self._ai_client.beta.threads.messages.create(
-                thread_id,
-                role=role,
-                metadata=metadata,
-                attachments=attachments,
-                content=message,
-                timeout=timeout
-            )
+            if attachments:
+                # Create the message with the attachments
+                await self._ai_client.beta.threads.messages.create(
+                    thread_id,
+                    role=role,
+                    metadata=metadata,
+                    attachments=attachments,
+                    content=message,
+                    timeout=timeout
+                )
+            else:
+                # Create the message without attachments
+                await self._ai_client.beta.threads.messages.create(
+                    thread_id,
+                    role=role,
+                    metadata=metadata,
+                    content=message,
+                    timeout=timeout
+                )
 
             logger.info(f"Created message: {message} in thread: {thread_id}, attachments: {attachments}")
         except Exception as e:

diff --git a/sdk/azure-ai-assistant/azure/ai/assistant/management/conversation_thread_client.py b/sdk/azure-ai-assistant/azure/ai/assistant/management/conversation_thread_client.py
@@ -275,15 +275,25 @@ def create_conversation_thread_message(
             thread_id = self._thread_config.get_thread_id_by_name(thread_name)
             attachments = self._update_message_attachments(thread_id, attachments) if attachments is not None else []
 
-            # Create the message with the attachments
-            self._ai_client.beta.threads.messages.create(
-                thread_id,
-                role=role,
-                metadata=metadata,
-                attachments=attachments,
-                content=message,
-                timeout=timeout
-            )
+            if attachments:
+                # Create the message with the attachments
+                self._ai_client.beta.threads.messages.create(
+                    thread_id,
+                    role=role,
+                    metadata=metadata,
+                    attachments=attachments,
+                    content=message,
+                    timeout=timeout
+                )
+            else:
+                # Create the message without attachments
+                self._ai_client.beta.threads.messages.create(
+                    thread_id,
+                    role=role,
+                    metadata=metadata,
+                    content=message,
+                    timeout=timeout
+                )
 
             logger.info(f"Created message: {message} in thread: {thread_id}, attachments: {attachments}")
         except Exception as e: