from langchain_core.messages import HumanMessage, SystemMessage
from langchain_openai import ChatOpenAI
from common.config import Config

conf = Config()

# ============ 配置llm区域 ============
my_llm = ChatOpenAI(
    api_key="EMPTY",
    base_url="https://a674ab0b6f8d4da580fa259ce6f68937--8000.ap-shanghai2.cloudstudio.club/v1",
    model="export/qwen2.5-merged0809"
)

if __name__ == '__main__':
    # 构造对话消息
    messages = [
        SystemMessage(content="请从以下中医文本中抽取知识图谱结构，包括实体与关系。"),
        HumanMessage(
            content="【方剂名称】斩梦丹\n- 中医百科\n- 斩梦丹\n斩梦丹的药方\n出处\n《普济方》卷三十三\n组成\n知母30克黄柏30克（去皮）滑石90克\n【制法】\n上药为末，白水和丸。\n斩梦丹的功效\n主治\n治梦泄遗精。\n用法\n空腹时温酒送下。")
    ]
    # 调用模型
    result = ""
    for chunk in my_llm.stream(messages):
        result += chunk.content
        print(chunk.content, end="", flush=True)
    print("\n")
    print("*" * 100)
    print(result)