[Feature] Add support for Groq LLMs (#1284)

2024-02-25 11:58:03 -08:00
parent b4bb4cf053
commit 92dd7edb57
6 changed files with 105 additions and 3 deletions
--- a/embedchain/llm/openai.py
+++ b/embedchain/llm/openai.py
@@ -58,8 +58,7 @@ class OpenAILlm(BaseLlm):
        messages: list[BaseMessage],
    ) -> str:
        from langchain.output_parsers.openai_tools import JsonOutputToolsParser
-        from langchain_core.utils.function_calling import \
-            convert_to_openai_tool
+        from langchain_core.utils.function_calling import convert_to_openai_tool

        openai_tools = [convert_to_openai_tool(tools)]
        chat = chat.bind(tools=openai_tools).pipe(JsonOutputToolsParser())