from langchain_deepseek import ChatDeepSeek
from http import HTTPStatus
from urllib.parse import urlparse, unquote
from pathlib import PurePosixPath
import requests
from dashscope import ImageSynthesis
import os

from llm.base import MultiModalAICapability


class TongYiAdapter(MultiModalAICapability):
    def __init__(self, api_key, model, **kwargs):
        self.api_key = api_key
        self.model = model
        self.llm = ChatDeepSeek(api_key=api_key, model=model, streaming=True)

    async def chat(self, messages, **kwargs):
        # 兼容 DeepSeek 的调用方式
        return await self.llm.ainvoke(messages)

    async def stream_chat(self, messages, **kwargs):
        async for chunk in self.llm.astream(messages):
            yield chunk

    @staticmethod
    def create_image_task(api_key, model, prompt: str, style='<watercolor>', size='1024*1024', n=1):
        """创建异步图片生成任务"""
        rsp = ImageSynthesis.async_call(
            api_key=api_key,
            model=model,
            prompt=prompt,
            n=n,
            style=style,
            size=size
        )
        if rsp.status_code == HTTPStatus.OK:
            return rsp
        else:
            raise Exception(f"Failed, status_code: {rsp.status_code}, code: {rsp.code}, message: {rsp.message}")

    @staticmethod
    def fetch_image_task_status(task):
        """获取异步图片任务状态"""
        status = ImageSynthesis.fetch(task)
        if status.status_code == HTTPStatus.OK:
            return status.output.task_status
        else:
            raise Exception(f"Failed, status_code: {status.status_code}, code: {status.code}, message: {status.message}")