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 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.config import AddConfig
@@ -130,8 +131,8 @@ class OpenAIAssistant:
@staticmethod
def _format_message(thread_message):
thread_message = cast(ThreadMessage, thread_message)
content = [c.text.value for c in thread_message.content if isinstance(c, MessageContentText)]
thread_message = cast(Message, thread_message)
content = [c.text.value for c in thread_message.content if isinstance(c, TextContentBlock)]
return " ".join(content)
@staticmethod