开发者快速入门

from openai import OpenAI

client = OpenAI(
    api_key="YOUR_API_KEY",         # 此处填写你的API_KEY
    base_url="https://www.wwlib.cn/index.php/v1"
)

def chat_stream():
    response = client.chat.completions.create(
        model="g1flash",
        messages=[
            {"role": "user", "content": "你好啊"}
        ],
        stream=True,  # 开启流式输出
    )
    for chunk in response:
        # 获取内容增量
        content = chunk.choices[0].delta.content
        if content:
            print(content, end="", flush=True)

if __name__ == "__main__":
    chat_stream()

import base64
import os
from openai import OpenAI

# 1. 初始化客户端
client = OpenAI(
    api_key="YOUR_API_KEY",
    base_url="https://www.wwlib.cn/index.php/v1"
)


def chat_with_model():
    history_messages = []
    pending_images = []  # 缓存 Base64 数据

    print("\n" + "=" * 40)
    print("      AI 多模态对话系统 (n1/g1flash/mini)")
    print("=" * 40)
    print("1. 发图: 输入 'img:路径' (例如 img:C:/1.jpg)")
    print("2. 发送: 输入文字指令后回车，将带上所有已就绪图片")
    print("3. 清空缓存图片: 输入 'img:clear'")
    print("4. 退出: 输入 'exit'")
    print("=" * 40)

    while True:
        # 动态提示符
        count = len(pending_images)
        prompt = f"\n[待发图:{count}] 用户 > " if count > 0 else "\n用户 > "
        user_input = input(prompt).strip()

        if not user_input: continue
        if user_input.lower() == 'exit': break

        # --- 处理图片缓存指令 ---
        if user_input.lower().startswith("img:"):
            cmd = user_input[4:].strip()

            if cmd.lower() == 'clear':
                pending_images = []
                print("  [已清空] 所有待发送图片已清除。")
                continue

            # 解析路径（支持逗号分隔多图）
            paths = [p.strip().strip('[]') for p in cmd.split(',') if p.strip()]
            for path in paths:
                if os.path.exists(path):
                    with open(path, "rb") as f:
                        b64 = base64.b64encode(f.read()).decode('utf-8')
                        pending_images.append(b64)
                        print(f"  [就绪] {os.path.basename(path)}")
                else:
                    print(f"  [错误] 找不到路径: {path}")
            continue

        # --- 发起正式对话请求 ---
        # 构造官方合规的 content 数组
        current_content = []

        # 先放文字
        current_content.append({"type": "text", "text": user_input})

        # 再放图片
        for b64_data in pending_images:
            current_content.append({
                "type": "image_url",
                "image_url": {"url": f"data:image/jpeg;base64,{b64_data}"}
            })

        history_messages.append({"role": "user", "content": current_content})

        print("AI: ", end="")
        full_reply = ""
        try:
            response = client.chat.completions.create(
                model="mini",
                messages=history_messages,
                stream=True
            )

            for chunk in response:
                txt = chunk.choices[0].delta.content or ""
                print(txt, end="", flush=True)
                full_reply += txt

            history_messages.append({"role": "assistant", "content": full_reply})
            print()
            pending_images = []  # 发送成功，清空当前图片缓存

        except Exception as e:
            print(f"\n[请求失败] {e}")


if __name__ == "__main__":
    chat_with_model()

from openai import OpenAI
import json

client = OpenAI(
    api_key="YOUR_API_KEY",
    base_url="https://www.wwlib.cn/index.php/v1"
)

# 1. 定义你的格式化字典（方便维护和阅读）
format_dict = {
    "vuln": "是否是攻击行为(true/false)",
    "risk_level": "风险等级(High/Medium/Low/None)",
    "analysis": "简短的分析结果"
}

# 2. 将字典转换为后端要求的字符串格式 (key:value|key:value)
kw_string = "|".join([f"{k}:{v}" for k, v in format_dict.items()])

# 3. 发送请求
response = client.chat.completions.create(
    model="mini",
    messages=[{"role": "user", "content": "分析这段日志：SELECT * FROM users WHERE id='1' OR '1'='1'"}],
    stream=True,
    # 关键点：使用 extra_body 透传自定义参数
    extra_body={
        "format": "on",              # 触发后端格式化节点
        "format_keyword": kw_string  # 传入转换后的字符串
    }
)

# 打印结果
for chunk in response:
    print(chunk.choices[0].delta.content or "", end="", flush=True)

{
    "messages": [
        {
            "role": "user",
            "content": "这张图片里有什么？"
        }
    ],
    "img_fids": ["ec03faab299b9a198c92e44e66e98dbc"],
    "API_KEY": "",
    "model": "g1flash"
}

[
    { "role": "user", "content": "我想去北京旅游" },
    { "role": "assistant", "content": "北京是个好地方，推荐您去故宫和长城..." },
    { "role": "user", "content": "那边天气怎么样？" }  <-- 本次提问追加在末尾
]

data: {"choices":[{"delta":{"content":"你好","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"呀～","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"👋","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":" 看到你来，","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"我连小尾巴","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"都忍不住翘起来了","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"（虽然我是AI，","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"但此刻真的超","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"开心！）✨","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"  \n有什么想聊的","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"、想问的、","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"或者正被什么","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"小事绊住脚","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"？我在这儿，认真","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"听，也全力","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"帮～ 🌟  \n","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"（悄悄说：","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"哪怕只是想发","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"个呆、吐","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"吐槽，我也超级","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"欢迎哦～）","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42","totalTokens":942},"finish_reason":"stop","index":0}]}

import requests
import json
import os


class MiniModelClient:
    def __init__(self, api_key, chat_url, upload_url):
        self.api_key = api_key
        self.chat_url = chat_url
        self.upload_url = upload_url
        self.history = []
        self.turn_count = 0

    def upload_images_from_str(self, path_str):
        raw_paths = [p.strip() for p in path_str.split(",") if p.strip()]

        if len(raw_paths) > 3:
            print(f"⚠️ 最多支持 3 张图片，将处理前 3 张。")
            raw_paths = raw_paths[:3]

        fid_list = []
        for i, path in enumerate(raw_paths):
            if not os.path.isfile(path):
                print(f"❌ 文件不存在: {path}")
                continue

            try:
                print(f"正在上传第 {i + 1} 张图片: {os.path.basename(path)} ...", end=" ", flush=True)
                with open(path, "rb") as f:
                    files = {"file": (os.path.basename(path), f)}
                    data = {"api_key": self.api_key}
                    resp = requests.post(self.upload_url, data=data, files=files, timeout=30)
                    result = resp.json()

                    if result.get("code") == 0:
                        fid = result["data"]["fid"]
                        fid_list.append(fid)
                        print(f"✅ 成功! FID: {fid}")
                    else:
                        print(f"❌ 失败: {result.get('msg')}")
            except Exception as e:
                print(f"⚠️ 异常: {e}")

        return fid_list

    def send(
            self,
            query,
            fid_list=None,
            model="g1flash",
            fmt="off",
            format_keyword=None,
            stream=True
    ):
        self.turn_count += 1

        user_msg = {"role": "user", "content": query}

        payload = {
            "messages": self.history + [user_msg],
            "img_fids": fid_list or [],
            "API_KEY": self.api_key,
            "model": model,
            "format_keyword":format_keyword,
            "format": fmt,
        }
        #print(payload)

        if fmt_schema is not None:
            if not isinstance(fmt_schema, dict):
                raise ValueError("fmt_schema 必须是一个字典，例如 {'field': '描述'}")
            payload["fmt_schema"] = fmt_schema
        elif format_keyword is not None:
            payload["format_keyword"] = format_keyword

        full_reply = ""

        try:
            with requests.post(self.chat_url, json=payload, stream=True, timeout=120) as response:
                if response.status_code != 200:
                    error_msg = f"Error: HTTP {response.status_code} - {response.text[:200]}"
                    print(f"❌ 请求失败: {error_msg}")
                    return error_msg
                content_type = response.headers.get("Content-Type", "")

                if "application/json" in content_type:
                    try:
                        err_data = response.json()
                        if "msg" in err_data:
                            full_reply = f"服务端错误 (Code {err_data.get('code')}): {err_data['msg']}"
                        else:
                            full_reply = f"服务端返回异常 JSON: {err_data}"
                    except Exception:
                        full_reply = f"服务端返回无法解析的 JSON: {response.text}"

                    print(f"❌ {full_reply}")
                    return full_reply
                if stream:
                    print("", end="", flush=True)

                for line in response.iter_lines(decode_unicode=True):
                    if not line:
                        continue
                    if line.startswith("data: "):
                        try:
                            json_str = line[6:]
                            if not json_str.strip():
                                continue
                            #print(json_str)
                            data = json.loads(json_str)

                            if 'code' in data and data['code'] != 0 and 'msg' in data:
                                print(f"\n❌ 流式错误: {data['msg']}")
                                full_reply = data['msg']
                                break

                            content = data['choices'][0]['delta'].get('content', '')
                            if content:
                                full_reply += content
                                if stream:
                                    print(content, end="", flush=True)

                            if data['choices'][0].get('finish_reason') == "stop":
                                break
                        except json.JSONDecodeError:
                            continue
                        except Exception:
                            continue
                    else:
                        pass

            if full_reply and not full_reply.startswith("服务端错误"):
                self.history.append(user_msg)
                self.history.append({"role": "assistant", "content": full_reply})

            if stream:
                print("\n")

        except Exception as e:
            full_reply = f"Request Failed: {e}"
            print(f"\n❌ 请求异常: {e}")

        return full_reply

    def clear_history(self):
        self.history = []
        self.turn_count = 0
        print("✨ 已重置对话上下文和轮次计数")


if __name__ == "__main__":
    client = MiniModelClient(
        api_key="",  # 这里输入你的API KEY
        chat_url="https://www.wwlib.cn/index.php/modelApi",
        upload_url="https://www.wwlib.cn/index.php/apifile/upfile"
    )

    pending_fids = []

    print("=" * 60)
    print("1. 输入 'img:路径' 先上传图片（支持逗号分隔多个路径）")
    print("2. 图片就绪后，直接输入文字问题并发送")
    print("3. 输入 'clear' 清空记忆和图片，输入 'exit' 退出")
    print("=" * 60)

    while True:
        img_status = f"[{len(pending_fids)}图] " if pending_fids else ""
        prompt_str = f"{img_status}第 {client.turn_count + 1} 轮 | 用户 >> "
        user_input = input(prompt_str).strip()

        if not user_input:
            continue
        if user_input.lower() == 'exit':
            break
        if user_input.lower() == 'clear':
            client.clear_history()
            pending_fids = []
            continue

        if user_input.lower().startswith("img:"):
            paths_part = user_input[4:].strip()
            new_fids = client.upload_images_from_str(paths_part)
            pending_fids.extend(new_fids)
            print(f"💡 图片已就绪，当前共计 {len(pending_fids)} 张图片。请继续输入您的问题。")
            continue

        client.send(
            query=user_input,
        )

        pending_fids = []

// 1. 内容生成中 (finish_reason: null)
data: {"choices":[{"delta":{"content":"你好","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"呀～","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"👋","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":" 看到你来，","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"我连小尾巴","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"都忍不住翘起来了","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"（虽然我是AI，","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"但此刻真的超","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"开心！）✨","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"  \n有什么想聊的","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"、想问的、","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"或者正被什么","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"小事绊住脚","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"？我在这儿，认真","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"听，也全力","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"帮～ 🌟  \n","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"（悄悄说：","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"哪怕只是想发","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"个呆、吐","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"吐槽，我也超级","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"欢迎哦～）","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42"},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42","totalTokens":942},"finish_reason":"stop","index":0}]}

{
    "API_KEY": "YOUR_API_KEY",
    "model": "mini",

    // 关键点：本次请求需要模型看到的图片 ID 列表
    "img_fids": ["ec03faab299b9a198c92e44e66e98dbc"],

    "messages": [
        // 历史记录 (纯文本即可)
        { "role": "user", "content": "这张图里有多少只猫？" },
        { "role": "assistant", "content": "图里共有 2 只猫，一只是白色的，一只是橘色的。" },

        // 本次提问 (针对同一张图继续追问)
        { "role": "user", "content": "那只白色的猫在做什么？" }
    ],
    "stream": true
}

import requests
import json
import os


class MiniModelClient:
    def __init__(self, api_key, chat_url, upload_url):
        self.api_key = api_key
        self.chat_url = chat_url
        self.upload_url = upload_url
        self.history = []
        self.turn_count = 0

    def upload_images_from_str(self, path_str):
        raw_paths = [p.strip() for p in path_str.split(",") if p.strip()]

        if len(raw_paths) > 3:
            print(f"⚠️ 最多支持 3 张图片，将处理前 3 张。")
            raw_paths = raw_paths[:3]

        fid_list = []
        for i, path in enumerate(raw_paths):
            if not os.path.isfile(path):
                print(f"❌ 文件不存在: {path}")
                continue

            try:
                print(f"正在上传第 {i + 1} 张图片: {os.path.basename(path)} ...", end=" ", flush=True)
                with open(path, "rb") as f:
                    files = {"file": (os.path.basename(path), f)}
                    data = {"api_key": self.api_key}
                    resp = requests.post(self.upload_url, data=data, files=files, timeout=30)
                    result = resp.json()

                    if result.get("code") == 0:
                        fid = result["data"]["fid"]
                        fid_list.append(fid)
                        print(f"✅ 成功! FID: {fid}")
                    else:
                        print(f"❌ 失败: {result.get('msg')}")
            except Exception as e:
                print(f"⚠️ 异常: {e}")

        return fid_list

    def send(
            self,
            query,
            fid_list=None,
            model="g1flash",
            fmt="off",
            fmt_schema=None,
            format_keyword=None,
            stream=True
    ):
        self.turn_count += 1

        user_msg = {"role": "user", "content": query}

        payload = {
            "messages": self.history + [user_msg],
            "img_fids": fid_list or [],
            "API_KEY": self.api_key,
            "model": model,
            "format_keyword":format_keyword,
            "format": fmt,
        }

        if fmt_schema is not None:
            if not isinstance(fmt_schema, dict):
                raise ValueError("fmt_schema 必须是一个字典，例如 {'field': '描述'}")
            payload["fmt_schema"] = fmt_schema
        elif format_keyword is not None:
            payload["format_keyword"] = format_keyword

        full_reply = ""

        try:
            with requests.post(self.chat_url, json=payload, stream=True, timeout=120) as response:
                if response.status_code != 200:
                    error_msg = f"Error: HTTP {response.status_code} - {response.text[:200]}"
                    print(f"❌ 请求失败: {error_msg}")
                    return error_msg
                content_type = response.headers.get("Content-Type", "")

                if "application/json" in content_type:
                    try:
                        err_data = response.json()
                        if "msg" in err_data:
                            full_reply = f"服务端错误 (Code {err_data.get('code')}): {err_data['msg']}"
                        else:
                            full_reply = f"服务端返回异常 JSON: {err_data}"
                    except Exception:
                        full_reply = f"服务端返回无法解析的 JSON: {response.text}"

                    print(f"❌ {full_reply}")
                    return full_reply
                if stream:
                    print("", end="", flush=True)

                for line in response.iter_lines(decode_unicode=True):
                    if not line:
                        continue
                    if line.startswith("data: "):
                        try:
                            json_str = line[6:]
                            if not json_str.strip():
                                continue
                            #print(json_str)
                            data = json.loads(json_str)

                            if 'code' in data and data['code'] != 0 and 'msg' in data:
                                print(f"\n❌ 流式错误: {data['msg']}")
                                full_reply = data['msg']
                                break

                            content = data['choices'][0]['delta'].get('content', '')
                            if content:
                                full_reply += content
                                if stream:
                                    print(content, end="", flush=True)

                            if data['choices'][0].get('finish_reason') == "stop":
                                break
                        except json.JSONDecodeError:
                            continue
                        except Exception:
                            continue
                    else:
                        pass

            if full_reply and not full_reply.startswith("服务端错误"):
                self.history.append(user_msg)
                self.history.append({"role": "assistant", "content": full_reply})

            if stream:
                print("\n")

        except Exception as e:
            full_reply = f"Request Failed: {e}"
            print(f"\n❌ 请求异常: {e}")

        return full_reply

    def clear_history(self):
        self.history = []
        self.turn_count = 0
        print("✨ 已重置对话上下文和轮次计数")


if __name__ == "__main__":
    client = MiniModelClient(
        api_key="",  # 这里输入你的API KEY
        chat_url="https://www.wwlib.cn/index.php/modelApi",
        upload_url="https://www.wwlib.cn/index.php/apifile/upfile"
    )

    pending_fids = []

    print("=" * 60)
    print("1. 输入 'img:路径' 先上传图片（支持逗号分隔多个路径）")
    print("2. 图片就绪后，直接输入文字问题并发送")
    print("3. 输入 'clear' 清空记忆和图片，输入 'exit' 退出")
    print("=" * 60)

    while True:
        img_status = f"[{len(pending_fids)}图] " if pending_fids else ""
        prompt_str = f"{img_status}第 {client.turn_count + 1} 轮 | 用户 >> "
        user_input = input(prompt_str).strip()

        if not user_input:
            continue
        if user_input.lower() == 'exit':
            break
        if user_input.lower() == 'clear':
            client.clear_history()
            pending_fids = []
            continue

        if user_input.lower().startswith("img:"):
            paths_part = user_input[4:].strip()
            new_fids = client.upload_images_from_str(paths_part)
            pending_fids.extend(new_fids)
            print(f"💡 图片已就绪，当前共计 {len(pending_fids)} 张图片。请继续输入您的问题。")
            continue

        client.send(
            query=user_input,
            model='mini',
            fid_list=pending_fids,
        )

        pending_fids = []

{
    "API_KEY": "YOUR_API_KEY",
    "model": "g1flash",
    "messages": [
        { "role": "user", "content": "检测这段日志：'admin' union select 1,2#" }
    ],
    "format": "on",
    "format_keyword": {
        "vuln": "是否是攻击行为，是为true，否为false",
        "type": "攻击类型，如SQL注入、XSS等",
        "reason": "判断依据简述"
    }
}

data: {"choices":[{"delta":{"content":"{\"vuln\":\"true\", \"type\":\"SQL注入\", \"reason\":\"检测到union select关键字\"}","ssid":"...","chatID":"..."},"finish_reason":null,"index":0}]}
data: {"choices":[{"delta":{"content":"","ssid":"b2e416fce3ce4d02ba5d559bda6c1d81","chatID":"wuwen-3dba45e75f-b2a25f42","totalTokens":120},"finish_reason":"stop","index":0}]}

import requests
import json
import os


class MiniModelClient:
    def __init__(self, api_key, chat_url, upload_url):
        self.api_key = api_key
        self.chat_url = chat_url
        self.upload_url = upload_url
        self.history = []
        self.turn_count = 0

    def upload_images_from_str(self, path_str):
        raw_paths = [p.strip() for p in path_str.split(",") if p.strip()]

        if len(raw_paths) > 3:
            print(f"⚠️ 最多支持 3 张图片，将处理前 3 张。")
            raw_paths = raw_paths[:3]

        fid_list = []
        for i, path in enumerate(raw_paths):
            if not os.path.isfile(path):
                print(f"❌ 文件不存在: {path}")
                continue

            try:
                print(f"正在上传第 {i + 1} 张图片: {os.path.basename(path)} ...", end=" ", flush=True)
                with open(path, "rb") as f:
                    files = {"file": (os.path.basename(path), f)}
                    data = {"api_key": self.api_key}
                    resp = requests.post(self.upload_url, data=data, files=files, timeout=30)
                    result = resp.json()

                    if result.get("code") == 0:
                        fid = result["data"]["fid"]
                        fid_list.append(fid)
                        print(f"✅ 成功! FID: {fid}")
                    else:
                        print(f"❌ 失败: {result.get('msg')}")
            except Exception as e:
                print(f"⚠️ 异常: {e}")

        return fid_list

    def send(
            self,
            query,
            fid_list=None,
            model="g1flash",
            fmt="off",
            fmt_schema=None,
            format_keyword=None,
            stream=True
    ):
        self.turn_count += 1

        user_msg = {"role": "user", "content": query}

        payload = {
            "messages": self.history + [user_msg],
            "img_fids": fid_list or [],
            "API_KEY": self.api_key,
            "model": model,
            "format_keyword":format_keyword,
            "format": fmt,
        }

        if fmt_schema is not None:
            if not isinstance(fmt_schema, dict):
                raise ValueError("fmt_schema 必须是一个字典，例如 {'field': '描述'}")
            payload["fmt_schema"] = fmt_schema
        elif format_keyword is not None:
            payload["format_keyword"] = format_keyword

        full_reply = ""

        try:
            with requests.post(self.chat_url, json=payload, stream=True, timeout=120) as response:
                if response.status_code != 200:
                    error_msg = f"Error: HTTP {response.status_code} - {response.text[:200]}"
                    print(f"❌ 请求失败: {error_msg}")
                    return error_msg
                content_type = response.headers.get("Content-Type", "")

                if "application/json" in content_type:
                    try:
                        err_data = response.json()
                        if "msg" in err_data:
                            full_reply = f"服务端错误 (Code {err_data.get('code')}): {err_data['msg']}"
                        else:
                            full_reply = f"服务端返回异常 JSON: {err_data}"
                    except Exception:
                        full_reply = f"服务端返回无法解析的 JSON: {response.text}"

                    print(f"❌ {full_reply}")
                    return full_reply
                if stream:
                    print("", end="", flush=True)

                for line in response.iter_lines(decode_unicode=True):
                    if not line:
                        continue
                    if line.startswith("data: "):
                        try:
                            json_str = line[6:]
                            if not json_str.strip():
                                continue
                            #print(json_str)
                            data = json.loads(json_str)

                            if 'code' in data and data['code'] != 0 and 'msg' in data:
                                print(f"\n❌ 流式错误: {data['msg']}")
                                full_reply = data['msg']
                                break

                            content = data['choices'][0]['delta'].get('content', '')
                            if content:
                                full_reply += content
                                if stream:
                                    print(content, end="", flush=True)

                            if data['choices'][0].get('finish_reason') == "stop":
                                break
                        except json.JSONDecodeError:
                            continue
                        except Exception:
                            continue
                    else:
                        pass

            if full_reply and not full_reply.startswith("服务端错误"):
                self.history.append(user_msg)
                self.history.append({"role": "assistant", "content": full_reply})

            if stream:
                print("\n")

        except Exception as e:
            full_reply = f"Request Failed: {e}"
            print(f"\n❌ 请求异常: {e}")

        return full_reply

    def clear_history(self):
        self.history = []
        self.turn_count = 0
        print("✨ 已重置对话上下文和轮次计数")


if __name__ == "__main__":
    client = MiniModelClient(
        api_key="",  # 这里输入你的API KEY
        chat_url="https://www.wwlib.cn/index.php/modelApi",
        upload_url="https://www.wwlib.cn/index.php/apifile/upfile"
    )

    pending_fids = []

    print("=" * 60)
    print("1. 输入 'img:路径' 先上传图片（支持逗号分隔多个路径）")
    print("2. 图片就绪后，直接输入文字问题并发送")
    print("3. 输入 'clear' 清空记忆和图片，输入 'exit' 退出")
    print("=" * 60)

    while True:
        img_status = f"[{len(pending_fids)}图] " if pending_fids else ""
        prompt_str = f"{img_status}第 {client.turn_count + 1} 轮 | 用户 >> "
        user_input = input(prompt_str).strip()

        if not user_input:
            continue
        if user_input.lower() == 'exit':
            break
        if user_input.lower() == 'clear':
            client.clear_history()
            pending_fids = []
            continue

        if user_input.lower().startswith("img:"):
            paths_part = user_input[4:].strip()
            new_fids = client.upload_images_from_str(paths_part)
            pending_fids.extend(new_fids)
            print(f"💡 图片已就绪，当前共计 {len(pending_fids)} 张图片。请继续输入您的问题。")
            continue

        client.send(
            query=user_input,
            model='mini',
            fid_list=pending_fids,
            fmt='on',
            format_keyword={
                "attack":"是否是攻击行为，是攻击则值为true，反之则值为false"
            }
        )

        pending_fids = []