From 09acbb84101bc6df373ca5a1d6c8d212bd3f577f Mon Sep 17 00:00:00 2001 From: Avinash Raj Date: Thu, 6 Jul 2023 22:46:04 +0530 Subject: [PATCH] Modified PromptLayerChatOpenAI class to support function call (#6366) Introduction of newest function calling feature doesn't work properly with PromptLayerChatOpenAI model since on the `_generate` method, functions argument are not even getting passed to the `ChatOpenAI` base class which results in empty `ai_message.additional_kwargs` Fixes #6365 --- langchain/chat_models/promptlayer_openai.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/langchain/chat_models/promptlayer_openai.py b/langchain/chat_models/promptlayer_openai.py index 3889b53558..fd73674336 100644 --- a/langchain/chat_models/promptlayer_openai.py +++ b/langchain/chat_models/promptlayer_openai.py @@ -49,7 +49,7 @@ class PromptLayerChatOpenAI(ChatOpenAI): from promptlayer.utils import get_api_key, promptlayer_api_request request_start_time = datetime.datetime.now().timestamp() - generated_responses = super()._generate(messages, stop, run_manager) + generated_responses = super()._generate(messages, stop, run_manager, **kwargs) request_end_time = datetime.datetime.now().timestamp() message_dicts, params = super()._create_message_dicts(messages, stop) for i, generation in enumerate(generated_responses.generations):