klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix OpenAI Assistant (#1393)

Browse Source
This commit is contained in:
Dev Khant
2024-06-08 22:37:52 +05:30
committed by GitHub
parent 04e77ef34e
commit 00c1fa1ec7
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
embedchain/store/assistants.py
View File

@@ -8,7 +8,8 @@ from pathlib import Path
from typing import cast from typing import cast
from openai import OpenAI from openai import OpenAI
from openai.types.beta.threads import MessageContentText, ThreadMessage from openai.types.beta.threads import Message
from openai.types.beta.threads.text_content_block import TextContentBlock
from embedchain import Client, Pipeline from embedchain import Client, Pipeline
from embedchain.config import AddConfig from embedchain.config import AddConfig
@@ -130,8 +131,8 @@ class OpenAIAssistant:
@staticmethod @staticmethod
def _format_message(thread_message): def _format_message(thread_message):
thread_message = cast(ThreadMessage, thread_message) thread_message = cast(Message, thread_message)
content = [c.text.value for c in thread_message.content if isinstance(c, MessageContentText)] content = [c.text.value for c in thread_message.content if isinstance(c, TextContentBlock)]
return " ".join(content) return " ".join(content)
@staticmethod @staticmethod