mirror of
https://github.com/hwchase17/langchain.git
synced 2025-08-11 05:45:01 +00:00
6cd452d985
Description:
- Add system templates and user templates in integration testing
- initialize the response id field value to request_id
- Adjust the default model to hunyuan-pro
- Remove the default values of Temperature and TopP
- Add SystemMessage
all the integration tests have passed.
1、Execute integration tests for the first time
<img width="1359" alt="71ca77a2-e9be-4af6-acdc-4d665002bd9b"
src="https://github.com/user-attachments/assets/9298dc3a-aa26-4bfa-968b-c011a4e699c9">
2、Run the integration test a second time
<img width="1501" alt="image"
src="https://github.com/user-attachments/assets/61335416-4a67-4840-bb89-090ba668e237">
Issue: None
Dependencies: None
Twitter handle: None
---------
Co-authored-by: Chester Curme <chester.curme@gmail.com>
84 lines
2.9 KiB
Python
84 lines
2.9 KiB
Python
import uuid
|
|
from operator import itemgetter
|
|
from typing import Any
|
|
|
|
import pytest
|
|
from langchain.prompts.chat import (
|
|
ChatPromptTemplate,
|
|
HumanMessagePromptTemplate,
|
|
SystemMessagePromptTemplate,
|
|
)
|
|
from langchain_core.messages import AIMessage, HumanMessage
|
|
from langchain_core.runnables.base import RunnableSerializable
|
|
|
|
from langchain_community.chat_models.hunyuan import ChatHunyuan
|
|
|
|
|
|
@pytest.mark.requires("tencentcloud-sdk-python")
|
|
def test_chat_hunyuan() -> None:
|
|
chat = ChatHunyuan()
|
|
message = HumanMessage(content="Hello")
|
|
response = chat.invoke([message])
|
|
assert isinstance(response, AIMessage)
|
|
assert isinstance(response.content, str)
|
|
assert response.id is not None, "request_id is empty"
|
|
assert uuid.UUID(response.id), "Invalid UUID"
|
|
|
|
|
|
@pytest.mark.requires("tencentcloud-sdk-python")
|
|
def test_chat_hunyuan_with_temperature() -> None:
|
|
chat = ChatHunyuan(temperature=0.6)
|
|
message = HumanMessage(content="Hello")
|
|
response = chat.invoke([message])
|
|
assert isinstance(response, AIMessage)
|
|
assert isinstance(response.content, str)
|
|
assert response.id is not None, "request_id is empty"
|
|
assert uuid.UUID(response.id), "Invalid UUID"
|
|
|
|
|
|
@pytest.mark.requires("tencentcloud-sdk-python")
|
|
def test_chat_hunyuan_with_model_name() -> None:
|
|
chat = ChatHunyuan(model="hunyuan-standard")
|
|
message = HumanMessage(content="Hello")
|
|
response = chat.invoke([message])
|
|
assert isinstance(response, AIMessage)
|
|
assert isinstance(response.content, str)
|
|
assert response.id is not None, "request_id is empty"
|
|
assert uuid.UUID(response.id), "Invalid UUID"
|
|
|
|
|
|
@pytest.mark.requires("tencentcloud-sdk-python")
|
|
def test_chat_hunyuan_with_stream() -> None:
|
|
chat = ChatHunyuan(streaming=True)
|
|
message = HumanMessage(content="Hello")
|
|
response = chat.invoke([message])
|
|
assert isinstance(response, AIMessage)
|
|
assert isinstance(response.content, str)
|
|
assert response.id is not None, "request_id is empty"
|
|
assert uuid.UUID(response.id), "Invalid UUID"
|
|
|
|
|
|
@pytest.mark.requires("tencentcloud-sdk-python")
|
|
def test_chat_hunyuan_with_prompt_template() -> None:
|
|
system_prompt = SystemMessagePromptTemplate.from_template(
|
|
"You are a helpful assistant! Your name is {name}."
|
|
)
|
|
user_prompt = HumanMessagePromptTemplate.from_template("Question: {query}")
|
|
chat_prompt = ChatPromptTemplate.from_messages([system_prompt, user_prompt])
|
|
chat: RunnableSerializable[Any, Any] = (
|
|
{"query": itemgetter("query"), "name": itemgetter("name")}
|
|
| chat_prompt
|
|
| ChatHunyuan()
|
|
)
|
|
response = chat.invoke({"query": "Hello", "name": "Tom"})
|
|
assert isinstance(response, AIMessage)
|
|
assert isinstance(response.content, str)
|
|
assert response.id is not None, "request_id is empty"
|
|
assert uuid.UUID(response.id), "Invalid UUID"
|
|
|
|
|
|
def test_extra_kwargs() -> None:
|
|
chat = ChatHunyuan(temperature=0.88, top_p=0.7)
|
|
assert chat.temperature == 0.88
|
|
assert chat.top_p == 0.7
|