mirror of
https://github.com/RYDE-WORK/Langchain-Chatchat.git
synced 2026-01-30 18:56:23 +08:00
35a7ca74c0
* 支持了agentlm * 支持了agentlm和相关提示词 * 修改了Agent的一些功能，加入了Embed方面的一个优化 * 修改了部分Agent的工具 * 增加一些Langchain的自带工具 * 修复一些兼容性的bug * 恢复知识库 * 恢复知识库 * 修复Azure问题
26 lines
990 B
Python
26 lines
990 B
Python
import sys
|
|
from pathlib import Path
|
|
|
|
root_path = Path(__file__).parent.parent.parent
|
|
sys.path.append(str(root_path))
|
|
from server.model_workers.azure import request_azure_api
|
|
import pytest
|
|
from configs import TEMPERATURE, ONLINE_LLM_MODEL
|
|
@pytest.mark.parametrize("version", ["azure-api"])
|
|
def test_azure(version):
|
|
messages = [{"role": "system", "content": "You are a helpful assistant."},{"role": "user", "content": "介绍一下自己"}]
|
|
for resp in request_azure_api(
|
|
messages=messages,
|
|
api_base_url=ONLINE_LLM_MODEL["azure-api"]["api_base_url"],
|
|
api_key=ONLINE_LLM_MODEL["azure-api"]["api_key"],
|
|
deployment_name=ONLINE_LLM_MODEL["azure-api"]["deployment_name"],
|
|
api_version=ONLINE_LLM_MODEL["azure-api"]["api_version"],
|
|
temperature=TEMPERATURE,
|
|
max_tokens=1024,
|
|
model_name="azure-api"
|
|
):
|
|
assert resp["code"] == 200
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_azure("azure-api") |