import sys
sys.path.append('../')

from src.server.get_similarity import QAExt

from langchain_openai import ChatOpenAI


base_llm = ChatOpenAI(
    openai_api_key='xxxxxxxxxxxxx',
    openai_api_base='http://192.168.10.14:8000/v1',
    model_name='Qwen2-7B',
    verbose=True
)

ext = QAExt(base_llm)
question = "明天适合去吗？"
message = [
    ("我们明天去爬山吧", "好呀"),
    ("天气怎么样", "天气晴朗"),
]

 
result =  ext.extend_query(question, message)
print(result.content)