community: Fix Baichuan Chat. (#15207)

- **Description:** Baichuan Chat (with both Baichuan-Turbo and
Baichuan-Turbo-192K models) has updated their APIs. There are breaking
changes. For example, BAICHUAN_SECRET_KEY is removed in the latest API
but is still required in Langchain. Baichuan's Langchain integration
needs to be updated to the latest version.
  - **Issue:** #15206
  - **Dependencies:** None,
  - **Twitter handle:** None

@hwchase17.

Co-authored-by: BaiChuanHelper <wintergyc@WinterGYCs-MacBook-Pro.local>

libs/community/tests/integration_tests/chat_models/test_baichuan.py

@@ -2,17 +2,36 @@ from langchain_core.messages import AIMessage, HumanMessage
 
 from langchain_community.chat_models.baichuan import ChatBaichuan
 
+# For testing, run:
+# TEST_FILE=tests/integration_tests/chat_models/test_baichuan.py make test
 
-def test_chat_baichuan() -> None:
+
+def test_chat_baichuan_default() -> None:
+    chat = ChatBaichuan(streaming=True)
+    message = HumanMessage(content="请完整背诵将进酒，背诵5遍")
+    response = chat([message])
+    assert isinstance(response, AIMessage)
+    assert isinstance(response.content, str)
+
+
+def test_chat_baichuan_default_non_streaming() -> None:
     chat = ChatBaichuan()
+    message = HumanMessage(content="请完整背诵将进酒，背诵5遍")
+    response = chat([message])
+    assert isinstance(response, AIMessage)
+    assert isinstance(response.content, str)
+
+
+def test_chat_baichuan_turbo() -> None:
+    chat = ChatBaichuan(model="Baichuan2-Turbo", streaming=True)
     message = HumanMessage(content="Hello")
     response = chat([message])
     assert isinstance(response, AIMessage)
     assert isinstance(response.content, str)
 
 
-def test_chat_baichuan_with_model() -> None:
-    chat = ChatBaichuan(model="Baichuan2-13B")
+def test_chat_baichuan_turbo_non_streaming() -> None:
+    chat = ChatBaichuan(model="Baichuan2-Turbo")
     message = HumanMessage(content="Hello")
     response = chat([message])
     assert isinstance(response, AIMessage)
@@ -20,7 +39,7 @@ def test_chat_baichuan_with_model() -> None:
 
 
 def test_chat_baichuan_with_temperature() -> None:
-    chat = ChatBaichuan(model="Baichuan2-13B", temperature=1.0)
+    chat = ChatBaichuan(temperature=1.0)
     message = HumanMessage(content="Hello")
     response = chat([message])
     assert isinstance(response, AIMessage)
@@ -29,13 +48,15 @@ def test_chat_baichuan_with_temperature() -> None:
 
 def test_chat_baichuan_with_kwargs() -> None:
     chat = ChatBaichuan()
-    message = HumanMessage(content="Hello")
-    response = chat([message], temperature=0.88, top_p=0.7)
+    message = HumanMessage(content="百川192K API是什么时候上线的？")
+    response = chat([message], temperature=0.88, top_p=0.7, with_search_enhance=True)
+    print(response)
     assert isinstance(response, AIMessage)
     assert isinstance(response.content, str)
 
 
 def test_extra_kwargs() -> None:
-    chat = ChatBaichuan(temperature=0.88, top_p=0.7)
+    chat = ChatBaichuan(temperature=0.88, top_p=0.7, with_search_enhance=True)
     assert chat.temperature == 0.88
     assert chat.top_p == 0.7
+    assert chat.with_search_enhance is True

libs/community/tests/unit_tests/chat_models/test_baichuan.py

@@ -18,7 +18,6 @@ from langchain_community.chat_models.baichuan import (
     _convert_delta_to_message_chunk,
     _convert_dict_to_message,
     _convert_message_to_dict,
-    _signature,
 )
 
 
@@ -85,62 +84,33 @@ def test__convert_delta_to_message_human() -> None:
     assert result == expected_output
 
 
-def test__signature() -> None:
-    secret_key = SecretStr("YOUR_SECRET_KEY")
-
-    result = _signature(
-        secret_key=secret_key,
-        payload={
-            "model": "Baichuan2-53B",
-            "messages": [{"role": "user", "content": "Hi"}],
-        },
-        timestamp=1697734335,
-    )
-
-    # The signature was generated by the demo provided by Baichuan.
-    # https://platform.baichuan-ai.com/docs/api#4
-    expected_output = "24a50b2db1648e25a244c67c5ab57d3f"
-    assert result == expected_output
-
-
 def test_baichuan_key_masked_when_passed_from_env(
     monkeypatch: MonkeyPatch, capsys: CaptureFixture
 ) -> None:
     """Test initialization with an API key provided via an env variable"""
     monkeypatch.setenv("BAICHUAN_API_KEY", "test-api-key")
-    monkeypatch.setenv("BAICHUAN_SECRET_KEY", "test-secret-key")
 
     chat = ChatBaichuan()
     print(chat.baichuan_api_key, end="")
     captured = capsys.readouterr()
     assert captured.out == "**********"
 
-    print(chat.baichuan_secret_key, end="")
-    captured = capsys.readouterr()
-    assert captured.out == "**********"
-
 
 def test_baichuan_key_masked_when_passed_via_constructor(
     capsys: CaptureFixture,
 ) -> None:
     """Test initialization with an API key provided via the initializer"""
-    chat = ChatBaichuan(
-        baichuan_api_key="test-api-key", baichuan_secret_key="test-secret-key"
-    )
+    chat = ChatBaichuan(baichuan_api_key="test-api-key")
     print(chat.baichuan_api_key, end="")
     captured = capsys.readouterr()
     assert captured.out == "**********"
 
-    print(chat.baichuan_secret_key, end="")
-    captured = capsys.readouterr()
-
-    assert captured.out == "**********"
-
 
 def test_uses_actual_secret_value_from_secret_str() -> None:
     """Test that actual secret is retrieved using `.get_secret_value()`."""
     chat = ChatBaichuan(
-        baichuan_api_key="test-api-key", baichuan_secret_key="test-secret-key"
+        baichuan_api_key="test-api-key",
+        baichuan_secret_key="test-secret-key",  # For backward compatibility
     )
     assert cast(SecretStr, chat.baichuan_api_key).get_secret_value() == "test-api-key"
     assert (