openai[patch]: fix name param (#19365)

langchain-ai · Mar 20, 2024 · 0b20c09 · 0b20c09
1 parent f6c8700
commit 0b20c09
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 5 deletions.
diff --git a/libs/partners/openai/Makefile b/libs/partners/openai/Makefile
@@ -6,9 +6,9 @@ all: help
 # Define a variable for the test file path.
 TEST_FILE ?= tests/unit_tests/
 
-integration_tests: TEST_FILE=tests/integration_tests/
+integration_test integration_tests: TEST_FILE=tests/integration_tests/
 
-test tests integration_tests:
+test tests integration_test integration_tests:
 	poetry run pytest $(TEST_FILE)
 
 

diff --git a/libs/partners/openai/langchain_openai/chat_models/base.py b/libs/partners/openai/langchain_openai/chat_models/base.py
@@ -140,8 +140,17 @@ def _convert_message_to_dict(message: BaseMessage) -> dict:
     """
     message_dict: Dict[str, Any] = {
         "content": message.content,
-        "name": message.name,
     }
+    if message.name is not None:
+        message_dict["name"] = message.name
+    elif (
+        "name" in message.additional_kwargs
+        and message.additional_kwargs["name"] is not None
+    ):
+        # fall back on additional kwargs for backwards compatibility
+        message_dict["name"] = message.additional_kwargs["name"]
+
+    # populate role and additional message data
     if isinstance(message, ChatMessage):
         message_dict["role"] = message.role
     elif isinstance(message, HumanMessage):
@@ -171,8 +180,6 @@ def _convert_message_to_dict(message: BaseMessage) -> dict:
             del message_dict["name"]
     else:
         raise TypeError(f"Got unknown type {message}")
-    if "name" in message.additional_kwargs:
-        message_dict["name"] = message.additional_kwargs["name"]
     return message_dict