import os
from openai import OpenAI

# 初始化 OpenAI 客戶端，設定 API 金鑰跟自定義 base URL
client = OpenAI(
    api_key="sk-olPkgbB0-qQbgW5kO-ovidK4cR-WTmLAUco",  # 直接寫入你的金鑰
    base_url="http://192.168.88.82:4000"  # 指定自定義 API 端點
)

# 發送聊天完成請求，啟用串流模式
stream = client.chat.completions.create(
    model="ollama_chat/qwen3:8b-q8_0",  # 指定模型
    messages=[
        {"role": "user", "content": "用一句話解釋量子糾纏"}  # 訊息內容
    ],
    stream=True  # 啟用串流模式
)

# 逐步處理回應並即時顯示每個 chunk
for chunk in stream:
    content = chunk.choices[0].delta.content or ""
    if content:
        print(content, end="", flush=True)  # 即時印出每個片段，不換行，強制刷新緩衝

print()  # 最後換行