klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Feature] OpenAI Function Calling (#1224)

Browse Source
This commit is contained in:
UnMonsieur
2024-02-12 02:58:11 +01:00
committed by GitHub
parent 38e212c721
commit 41bd258b93
31 changed files with 259 additions and 213 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/llm/test_azure_openai.py
View File

@@ -28,7 +28,7 @@ def test_get_llm_model_answer(azure_openai_llm):
def test_get_answer(azure_openai_llm):
with patch("langchain.chat_models.AzureChatOpenAI") as mock_chat:
with patch("langchain_community.chat_models.AzureChatOpenAI") as mock_chat:
mock_chat_instance = mock_chat.return_value
mock_chat_instance.return_value = MagicMock(content="Test Response")
@@ -60,7 +60,7 @@ def test_get_messages(azure_openai_llm):
def test_get_answer_top_p_is_provided(azure_openai_llm, caplog):
with patch("langchain.chat_models.AzureChatOpenAI") as mock_chat:
with patch("langchain_community.chat_models.AzureChatOpenAI") as mock_chat:
mock_chat_instance = mock_chat.return_value
mock_chat_instance.return_value = MagicMock(content="Test Response")