Skip to content

Commit c592a2d

Browse files
committed
feat(demohouse/video_analyser):newversion
1 parent 36bf9c7 commit c592a2d

File tree

1 file changed

+9
-7
lines changed
  • demohouse/video_analyser/backend/code

1 file changed

+9
-7
lines changed

demohouse/video_analyser/backend/code/main.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -22,15 +22,17 @@
2222
import utils
2323
from config import VISUAL_SUMMARY_ENDPOINT, QUESTION_ANSWER_ENDPOINT, TTS_ACCESS_TOKEN, TTS_APP_ID
2424

25+
from volcenginesdkarkruntime.types.chat.completion_create_params import Thinking
26+
27+
from volcenginesdkarkruntime.types.chat.chat_completion_content_part_text_param import ChatCompletionContentPartTextParam
28+
2529
from arkitect.core.component.llm import BaseChatLanguageModel
26-
from arkitect.core.component.llm.model import (
30+
from arkitect.types.llm.model import (
2731
ArkChatCompletionChunk,
2832
ArkChatParameters,
2933
ArkChatRequest,
3034
ArkChatResponse,
3135
ArkMessage,
32-
ArkThinking,
33-
ChatCompletionMessageTextPart,
3436
Response,
3537
)
3638
from arkitect.core.component.tts import (
@@ -59,7 +61,7 @@ async def get_request_messages_for_llm(
5961
request_messages = await contexts.get_history(context_id)
6062
if isinstance(request.messages[-1].content, list):
6163
assert isinstance(
62-
request.messages[-1].content[0], ChatCompletionMessageTextPart
64+
request.messages[-1].content[0], ChatCompletionContentPartTextParam
6365
)
6466
text = request.messages[-1].content[0].text
6567
else:
@@ -206,11 +208,11 @@ async def default_model_calling(
206208
# Use VLM to summarize the image asynchronously and return immediately
207209
is_image = (
208210
isinstance(request.messages[-1].content, list)
209-
and isinstance(request.messages[-1].content[0], ChatCompletionMessageTextPart)
211+
and isinstance(request.messages[-1].content[0], ChatCompletionContentPartTextParam)
210212
and request.messages[-1].content[0].text == ""
211213
)
212214
parameters = ArkChatParameters(**request.__dict__)
213-
parameters.thinking = ArkThinking(type="disabled")
215+
parameters.thinking = Thinking(type="disabled")
214216
if is_image:
215217
_ = asyncio.create_task(
216218
summarize_image(contexts, request, parameters, context_id)
@@ -250,7 +252,7 @@ async def default_model_calling(
250252
await tts_client.close()
251253
text = ""
252254
if isinstance(request.messages[-1].content, list) and isinstance(
253-
request.messages[-1].content[0], ChatCompletionMessageTextPart
255+
request.messages[-1].content[0], ChatCompletionContentPartTextParam
254256
):
255257
text = request.messages[-1].content[0].text
256258
elif isinstance(request.messages[-1].content, str):

0 commit comments

Comments
 (0)