Create Langchain LLM converter to quickly supply it to Langchain's chain (#102)

* Create Langchain LLM converter to quickly supply it to Langchain's chain * Clean up
2023-12-11 14:55:56 +07:00
parent da0ac1d69f
commit 0e30dcbb06
10 changed files with 24 additions and 15 deletions
--- a/tests/test_llms_chat_models.py
+++ b/tests/test_llms_chat_models.py
@@ -48,7 +48,7 @@ def test_azureopenai_model(openai_completion):
        temperature=0,
    )
    assert isinstance(
-        model._obj, AzureChatOpenAILC
+        model.to_langchain_format(), AzureChatOpenAILC
    ), "Agent not wrapped in Langchain's AzureChatOpenAI"

    # test for str input - stream mode