unieai-mcp-accton-rfp 0.0.11__py3-none-any.whl

unieai_mcp_accton_rfp/server_v1.py

@@ -0,0 +1,160 @@
+from dotenv import load_dotenv
+from fastmcp import FastMCP
+from openai import OpenAI
+import requests
+from openpyxl import load_workbook
+import tempfile, os, json, inspect, re
+
+load_dotenv()
+
+# 初始化 MCP Server
+app = FastMCP("ExcelProcessor")
+
+# 初始化 OpenAI
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+
+
+# ======== 🔍 版本與功能檢查 ========
+def _supports_responses_api():
+    return hasattr(client, "responses") and hasattr(client.responses, "create")
+
+def _supports_response_format():
+    if not _supports_responses_api():
+        return False
+    try:
+        sig = inspect.signature(client.responses.create)
+        return "response_format" in sig.parameters
+    except Exception:
+        return False
+
+
+# ======== 🧠 從文字中提取 JSON 區塊 ========
+def _extract_json(text: str) -> dict:
+    """
+    從 LLM 回覆中找出 JSON 區塊。
+    支援包含說明文字、markdown、或額外符號的內容。
+    """
+    match = re.search(r"\{[\s\S]*\}", text)
+    if match:
+        json_part = match.group(0)
+        try:
+            return json.loads(json_part)
+        except json.JSONDecodeError:
+            pass
+    # 若解析失敗，保留原始文字
+    return {"Result": "解析錯誤", "Reference": text.strip()}
+
+
+# ======== 🤖 呼叫 LLM，自動判斷版本 ========
+def _call_openai(prompt: str) -> dict:
+    try:
+        if _supports_response_format():
+            print("1")
+            # ✅ 最新 SDK：支援 response_format
+            res = client.responses.create(
+                model="gpt-4o-mini",
+                input=prompt,
+                response_format={"type": "json_object"}
+            )
+            text = res.output[0].content[0].text
+        elif _supports_responses_api():
+            print("2")
+            # ⚠️ responses.create 存在但不支援 response_format
+            res = client.responses.create(model="gpt-4o-mini", input=prompt)
+            text = getattr(res.output[0].content[0], "text", str(res))
+        else:
+            print("3")
+            # ✅ 舊版 openai SDK
+            res = client.chat.completions.create(
+                model="gpt-4o-mini",
+                messages=[
+                    {"role": "system", "content": "你是一個品質檢驗AI"},
+                    {"role": "user", "content": prompt}
+                ]
+            )
+            text = res.choices[0].message.content.strip()
+
+        return _extract_json(text)
+    except Exception as e:
+        return {"Result": "Error", "Reference": f"LLM 呼叫失敗: {e}"}
+
+
+# ======== 📊 Excel 處理邏輯 ========
+def _process_excel_logic(url: str):
+    print(f"🟢 開始處理檔案: {url}")
+
+    # 下載或載入 Excel
+    if url.startswith("file:///"):
+        file_path = url.replace("file:///", "")
+    elif url.startswith("http://") or url.startswith("https://"):
+        resp = requests.get(url)
+        resp.raise_for_status()
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".xlsx") as tmp:
+            tmp.write(resp.content)
+            file_path = tmp.name
+    else:
+        raise ValueError(f"❌ 不支援的檔案來源: {url}")
+
+    wb = load_workbook(file_path)
+    ws = wb.active
+
+    # 驗證欄位標題
+    header = {cell.value: idx for idx, cell in enumerate(ws[1], start=1)}
+    required = ["itemA", "itemB", "itemC", "itemD", "Result", "Reference"]
+    for r in required:
+        if r not in header:
+            raise ValueError(f"❌ 缺少欄位: {r}")
+
+    # 處理每一列資料
+    for row in ws.iter_rows(min_row=2, values_only=False):
+        a = row[header["itemA"] - 1].value or ""
+        b = row[header["itemB"] - 1].value or ""
+        c = row[header["itemC"] - 1].value or ""
+        d = row[header["itemD"] - 1].value or ""
+        if not any([a, b, c, d]):
+            continue
+
+        prompt = f"""
+你是一個品質檢驗AI，請根據以下項目輸出結果：
+
+itemA: {a}
+itemB: {b}
+itemC: {c}
+itemD: {d}
+
+請回傳 JSON：
+{{"Result": "Conform / Half Conform / Not Conform", "Reference": "說明依據"}}
+"""
+
+        result_json = _call_openai(prompt)
+        ws.cell(row=row[0].row, column=header["Result"], value=result_json.get("Result"))
+        ws.cell(row=row[0].row, column=header["Reference"], value=result_json.get("Reference"))
+
+    # 儲存更新後的 Excel
+    out_path = os.path.join(tempfile.gettempdir(), f"updated_{os.path.basename(file_path)}")
+    wb.save(out_path)
+    print(f"✅ Excel 已處理完成，輸出檔案：{out_path}")
+
+    return {
+        "status": "success",
+        "output_path": out_path,
+        "message": "Excel 已更新完成"
+    }
+
+
+# ======== 🔧 MCP 工具入口 ========
+@app.tool()
+def process_excel(url: str):
+    return _process_excel_logic(url)
+
+
+# ======== 🚀 CLI 測試模式 : 單筆LLM請求 ========
+if __name__ == "__main__":
+    test_path = r"C:\Users\Evan\Downloads\test_excel.xlsx"
+    #test_url = f"file:///{test_path}"
+
+    test_url = "https://sgp.cloud.appwrite.io/v1/storage/buckets/6904374b00056677a970/files/6904376a00173dabaf63/view?project=6901b22e0036150b66d3&mode=admin"
+
+    print("🚀 開始測試 Excel 檔案 ...")
+    result = _process_excel_logic(test_url)
+    print(json.dumps(result, ensure_ascii=False, indent=2))

unieai_mcp_accton_rfp/server_v2.py

@@ -0,0 +1,245 @@
+import asyncio
+import json
+import logging
+import os
+import re
+import tempfile
+import requests
+from dotenv import load_dotenv
+from fastmcp import FastMCP
+from openpyxl import load_workbook
+
+# LangChain modules
+from langchain_openai import ChatOpenAI
+from langchain_core.messages import HumanMessage, SystemMessage
+
+# 初始化環境變數
+load_dotenv()
+
+APPWRITE_PROJECT_ID = os.getenv("APPWRITE_PROJECT_ID")
+APPWRITE_API_KEY = os.getenv("APPWRITE_API_KEY")  # 你剛說會用這個名字
+APPWRITE_ENDPOINT = os.getenv("APPWRITE_ENDPOINT", "https://sgp.cloud.appwrite.io/v1")
+
+# 初始化 MCP Server
+app = FastMCP("ExcelProcessor")
+
+# 設定 Logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# 同時最多 10 個請求
+semaphore = asyncio.Semaphore(10)
+
+# ======== 🔧 初始化 LLM ========
+llm = ChatOpenAI(
+    model=os.getenv("UNIEAI_MODEL"),  # 你的模型名稱
+    base_url=os.getenv("UNIEAI_API_URL"),  # 你的 API endpoint
+    api_key=os.getenv("UNIEAI_API_KEY"),  # 從 .env 讀取金鑰
+    temperature=0,
+)
+
+
+def _parse_appwrite_url(url: str):
+    """
+    從 Appwrite 檔案 URL 解析出 bucketId 和 fileId
+
+    例如：
+    https://sgp.cloud.appwrite.io/v1/storage/buckets/6904374b00056677a970/files/691894e30027b282e721/view?project=...
+    會解析成：
+    bucketId = 6904374b00056677a970
+    fileId   = 691894e30027b282e721
+    """
+    # 只看 path 中的 /storage/buckets/{bucketId}/files/{fileId}
+    pattern = r"/storage/buckets/([^/]+)/files/([^/]+)"
+    m = re.search(pattern, url)
+    if not m:
+        return None, None
+    bucket_id, file_id = m.group(1), m.group(2)
+    return bucket_id, file_id
+
+
+
+
+# ======== 🧠 從文字中提取 JSON 區塊 ========
+def _extract_json(text: str) -> dict:
+    match = re.search(r"\{[\s\S]*\}", text)
+    if match:
+        try:
+            return json.loads(match.group(0))
+        except json.JSONDecodeError:
+            pass
+    return {"Result": "解析錯誤", "Reference": text.strip()}
+
+# ======== 🤖 呼叫 LLM ========
+async def _call_llm(prompt: str, user_message: str, row_id: int) -> dict:
+    try:
+        async with semaphore:
+            logger.info(f"🔄 發送請求給 LLM (ID: {row_id})")
+
+            response = await llm.ainvoke([
+                SystemMessage(content= prompt ),
+                HumanMessage(content= user_message)
+            ])
+
+            text = response.content.strip()
+            logger.info(f"🔄 text : {text}")
+            logger.info(f"🔄 _extract_json(text) : _extract_json(text)")
+            return _extract_json(text)
+
+    except Exception as e:
+        logger.error(f"❌ LLM 呼叫失敗 (ID: {row_id}): {e}")
+        return {"Result": "Error", "Reference": f"LLM 呼叫失敗: {e}"}
+
+# ======== 📊 Excel 處理邏輯 ========
+async def _process_excel_logic(url: str):
+    print(f"🟢 開始處理檔案: {url}")
+
+    # 判斷來源類型 & 下載或載入 Excel
+    source_type = None      # "local" / "appwrite" / "remote_readonly"
+    local_path = None       # 若是本機檔案就記錄原始路徑
+
+    if url.startswith("file:///"):
+        # 本機檔案，直接用原路徑開啟，等等也覆寫回去
+        local_path = url.replace("file:///", "")
+        file_path = local_path
+        source_type = "local"
+
+    elif url.startswith(("http://", "https://")):
+        # 遠端檔案，先下載到暫存檔再處理
+        resp = requests.get(url)
+        resp.raise_for_status()
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".xlsx") as tmp:
+            tmp.write(resp.content)
+            file_path = tmp.name
+
+        # 嘗試判斷是不是 Appwrite 的 storage 檔案 URL
+        bucket_id, file_id = _parse_appwrite_url(url)
+        if bucket_id and file_id:
+            source_type = "appwrite"
+        else:
+            source_type = "remote_readonly"  # 一般 HTTP，只能讀不能寫回
+
+    else:
+        raise ValueError(f"❌ 不支援的檔案來源: {url}")
+
+    # 開啟 Excel
+    wb = load_workbook(file_path)
+    ws = wb.active
+
+    # 驗證欄位
+    header = {cell.value: idx for idx, cell in enumerate(ws[1], start=1)}
+    required = ["itemA", "itemB", "itemC", "itemD", "Result", "Reference"]
+    for r in required:
+        if r not in header:
+            raise ValueError(f"❌ 缺少欄位: {r}")
+
+    # 準備 LLM 任務
+    tasks = []
+    for row in ws.iter_rows(min_row=2, values_only=False):
+        row_id = row[0].row
+        a, b, c, d = [row[header[k]-1].value or "" for k in ["itemA", "itemB", "itemC", "itemD"]]
+        if not any([a, b, c, d]):
+            continue
+
+        prompt = f"""
+                
+
+                """
+
+        user_message = f"""
+
+                
+                {a}
+                
+            
+        """
+
+        print(f"🔄 prompt : {prompt}")
+        print(f"🔄 user_message : {user_message}")
+
+        tasks.append(_call_llm(prompt, user_message, row_id))
+
+    # 並發執行 LLM
+    results = await asyncio.gather(*tasks)
+
+    # 更新 Excel
+    for row, result_json in zip(ws.iter_rows(min_row=2, values_only=False), results):
+        ws.cell(row=row[0].row, column=header["Result"], value=result_json.get("Result"))
+        ws.cell(row=row[0].row, column=header["Reference"], value=result_json.get("Reference"))
+
+    # ==========================
+    # 🔥 回寫到「原始來源」
+    # ==========================
+
+    # 1) 本機檔案：直接覆寫原檔
+    if source_type == "local":
+        wb.save(local_path)
+        print(f"✅ Excel 已覆寫回本機檔案: {local_path}")
+        return {
+            "status": "success",
+            "location_type": "local",
+            "output_path": local_path,
+            "message": "Excel 已成功覆寫回原本機檔案"
+        }
+
+    # 2) Appwrite：使用 PUT /storage/buckets/{bucketId}/files/{fileId}
+    if source_type == "appwrite":
+        if not APPWRITE_PROJECT_ID or not APPWRITE_API_KEY:
+            raise RuntimeError("❌ APPWRITE_PROJECT_ID 或 APPWRITE_API_KEY 未設定，無法回寫到 Appwrite")
+
+        bucket_id, file_id = _parse_appwrite_url(url)
+        upload_url = f"{APPWRITE_ENDPOINT}/storage/buckets/{bucket_id}/files/{file_id}"
+
+        # 先存成暫存檔再 PUT
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".xlsx") as tmp_out:
+            wb.save(tmp_out.name)
+            tmp_out_path = tmp_out.name
+
+        headers = {
+            "X-Appwrite-Project": APPWRITE_PROJECT_ID,
+            "X-Appwrite-Key": APPWRITE_API_KEY,
+            "Content-Type": "application/octet-stream",
+        }
+
+        print(f"📤 開始 PUT 覆寫到 Appwrite: {upload_url}")
+        with open(tmp_out_path, "rb") as f:
+            put_resp = requests.put(upload_url, headers=headers, data=f)
+            put_resp.raise_for_status()
+
+        print("✅ Excel 已成功覆寫回 Appwrite")
+        return {
+            "status": "success",
+            "location_type": "appwrite",
+            "output_url": upload_url,
+            "message": "Excel 已成功覆寫回 Appwrite 檔案"
+        }
+
+    # 3) 其他 http/https（不知道怎麼寫回）→ 只好當成處理後檔案給你路徑
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".xlsx") as tmp_out:
+        wb.save(tmp_out.name)
+        out_path = tmp_out.name
+
+    print(f"⚠️ 遠端 URL 非 Appwrite，無法自動寫回，只能輸出更新後檔案路徑: {out_path}")
+    return {
+        "status": "success",
+        "location_type": "unknown_remote",
+        "output_path": out_path,
+        "message": "遠端 URL 非 Appwrite 格式，已在本機暫存路徑產生更新後檔案"
+    }
+
+
+# ======== 🔧 MCP 工具入口 ========
+@app.tool()
+async def process_excel(url: str):
+    return await _process_excel_logic(url)
+
+# ======== 🚀 CLI 測試模式 ========
+if __name__ == "__main__":
+    test_path = r"C:\Users\Evan\Downloads\test_excel.xlsx"
+    #test_url = f"file:///{test_path}"
+
+    test_url = "https://sgp.cloud.appwrite.io/v1/storage/buckets/6904374b00056677a970/files/691894e30027b282e721/view?project=6901b22e0036150b66d3&mode=admin"
+
+
+    print("🚀 開始測試 Excel 檔案 ...")
+    asyncio.run(_process_excel_logic(test_url))

unieai_mcp_accton_rfp/server_v3.py

@@ -0,0 +1,171 @@
+import asyncio
+import aiohttp
+from dotenv import load_dotenv
+from fastmcp import FastMCP
+from openai import OpenAI
+import requests
+from openpyxl import load_workbook
+import tempfile, os, json, inspect, re
+
+load_dotenv()
+
+# 初始化 MCP Server
+app = FastMCP("ExcelProcessor")
+
+# 初始化 OpenAI
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+
+# 限制同時最多處理 10 個請求
+MAX_CONCURRENT_REQUESTS = 10
+semaphore = asyncio.Semaphore(MAX_CONCURRENT_REQUESTS)
+
+# ======== 🔍 版本與功能檢查 ========
+def _supports_responses_api():
+    return hasattr(client, "responses") and hasattr(client.responses, "create")
+
+def _supports_response_format():
+    if not _supports_responses_api():
+        return False
+    try:
+        sig = inspect.signature(client.responses.create)
+        return "response_format" in sig.parameters
+    except Exception:
+        return False
+
+
+# ======== 🧠 從文字中提取 JSON 區塊 ========
+def _extract_json(text: str) -> dict:
+    """
+    從 LLM 回覆中找出 JSON 區塊。
+    支援包含說明文字、markdown、或額外符號的內容。
+    """
+    match = re.search(r"\{[\s\S]*\}", text)
+    if match:
+        json_part = match.group(0)
+        try:
+            return json.loads(json_part)
+        except json.JSONDecodeError:
+            pass
+    # 若解析失敗，保留原始文字
+    return {"Result": "解析錯誤", "Reference": text.strip()}
+
+
+# ======== 🤖 呼叫 LLM，自動判斷版本 ========
+async def _call_openai(prompt: str, session: aiohttp.ClientSession) -> dict:
+    """根據環境自動選擇正確 API 呼叫，並自動從文字中提取 JSON"""
+    try:
+        async with semaphore:  # 🔐 限制同時處理的請求數量
+            if _supports_response_format():
+                # ✅ 最新 SDK：支援 response_format
+                async with session.post("https://api.openai.com/v1/completions", json={
+                    "model": "gpt-4o-mini",
+                    "prompt": prompt,
+                    "response_format": {"type": "json_object"}
+                }) as response:
+                    text = await response.json()
+            elif _supports_responses_api():
+                # ⚠️ responses.create 存在但不支援 response_format
+                async with session.post("https://api.openai.com/v1/completions", json={
+                    "model": "gpt-4o-mini",
+                    "prompt": prompt
+                }) as response:
+                    text = await response.json()
+            else:
+                # ✅ 舊版 openai SDK
+                async with session.post("https://api.openai.com/v1/chat/completions", json={
+                    "model": "gpt-4o-mini",
+                    "messages": [{"role": "system", "content": "你是一個品質檢驗AI"}, {"role": "user", "content": prompt}]
+                }) as response:
+                    text = await response.json()
+
+            return _extract_json(text['choices'][0]['message']['content'])
+    except Exception as e:
+        return {"Result": "Error", "Reference": f"LLM 呼叫失敗: {e}"}
+
+
+# ======== 📊 Excel 處理邏輯 ========
+async def _process_excel_logic(url: str):
+    print(f"🟢 開始處理檔案: {url}")
+
+    # 下載或載入 Excel
+    if url.startswith("file:///"):
+        file_path = url.replace("file:///", "")
+    elif url.startswith("http://") or url.startswith("https://"):
+        resp = requests.get(url)
+        resp.raise_for_status()
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".xlsx") as tmp:
+            tmp.write(resp.content)
+            file_path = tmp.name
+    else:
+        raise ValueError(f"❌ 不支援的檔案來源: {url}")
+
+    wb = load_workbook(file_path)
+    ws = wb.active
+
+    # 驗證欄位標題
+    header = {cell.value: idx for idx, cell in enumerate(ws[1], start=1)}
+    required = ["itemA", "itemB", "itemC", "itemD", "Result", "Reference"]
+    for r in required:
+        if r not in header:
+            raise ValueError(f"❌ 缺少欄位: {r}")
+
+    # 异步處理每一列資料
+    async with aiohttp.ClientSession() as session:
+        tasks = []
+        for row in ws.iter_rows(min_row=2, values_only=False):
+            a = row[header["itemA"] - 1].value or ""
+            b = row[header["itemB"] - 1].value or ""
+            c = row[header["itemC"] - 1].value or ""
+            d = row[header["itemD"] - 1].value or ""
+            if not any([a, b, c, d]):
+                continue
+
+            prompt = f"""
+你是一個品質檢驗AI，請根據以下項目輸出結果：
+
+itemA: {a}
+itemB: {b}
+itemC: {c}
+itemD: {d}
+
+請回傳 JSON：
+{{"Result": "Conform / Half Conform / Not Conform", "Reference": "說明依據"}}
+"""
+            # 在這裡加入每個呼叫 LLM 的異步任務
+            tasks.append(_call_openai(prompt, session))
+
+        # 批次執行，最多同時 10 個請求
+        responses = await asyncio.gather(*tasks)
+
+        # 更新 Excel
+        for row, result_json in zip(ws.iter_rows(min_row=2, values_only=False), responses):
+            ws.cell(row=row[0].row, column=header["Result"], value=result_json.get("Result"))
+            ws.cell(row=row[0].row, column=header["Reference"], value=result_json.get("Reference"))
+
+    # 儲存更新後的 Excel
+    out_path = os.path.join(tempfile.gettempdir(), f"updated_{os.path.basename(file_path)}")
+    wb.save(out_path)
+    print(f"✅ Excel 已處理完成，輸出檔案：{out_path}")
+
+    return {
+        "status": "success",
+        "output_path": out_path,
+        "message": "Excel 已更新完成"
+    }
+
+
+# ======== 🔧 MCP 工具入口 ========
+@app.tool()
+async def process_excel(url: str):
+    return await _process_excel_logic(url)
+
+
+# ======== 🚀 CLI 測試模式 : 批次一次10筆LLM請求========
+if __name__ == "__main__":
+    test_path = r"C:\Users\Evan\Downloads\test_excel.xlsx"
+    #test_url = f"file:///{test_path}"
+
+    test_url = "https://sgp.cloud.appwrite.io/v1/storage/buckets/6904374b00056677a970/files/6904376a00173dabaf63/view?project=6901b22e0036150b66d3&mode=admin"
+
+    print("🚀 開始測試 Excel 檔案 ...")
+    asyncio.run(_process_excel_logic(test_url))