[Improvement] Fix deprecation warnings (#1288)

embedchain/helpers/callbacks.py

@@ -56,7 +56,7 @@ def generate(rq: queue.Queue):
     ```
     def askQuestion(callback_fn: StreamingStdOutCallbackHandlerYield):
         llm = OpenAI(streaming=True, callbacks=[callback_fn])
-        return llm(prompt="Write a poem about a tree.")
+        return llm.invoke(prompt="Write a poem about a tree.")
 
     @app.route("/", methods=["GET"])
     def generate_output():

embedchain/llm/aws_bedrock.py

@@ -38,12 +38,11 @@ class AWSBedrockLlm(BaseLlm):
         }
 
         if config.stream:
-            from langchain.callbacks.streaming_stdout import \
-                StreamingStdOutCallbackHandler
+            from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
 
             callbacks = [StreamingStdOutCallbackHandler()]
             llm = Bedrock(**kwargs, streaming=config.stream, callbacks=callbacks)
         else:
             llm = Bedrock(**kwargs)
 
-        return llm(prompt)
+        return llm.invoke(prompt)

embedchain/llm/cohere.py

@@ -40,4 +40,4 @@ class CohereLlm(BaseLlm):
             p=config.top_p,
         )
 
-        return llm(prompt)
+        return llm.invoke(prompt)

embedchain/llm/llama2.py

@@ -48,4 +48,4 @@ class Llama2Llm(BaseLlm):
                 "top_p": self.config.top_p,
             },
         )
-        return llm(prompt)
+        return llm.invoke(prompt)

embedchain/llm/ollama.py

@@ -33,4 +33,4 @@ class OllamaLlm(BaseLlm):
             callback_manager=CallbackManager(callback_manager),
         )
 
-        return llm(prompt)
+        return llm.invoke(prompt)

embedchain/llm/together.py

@@ -40,4 +40,4 @@ class TogetherLlm(BaseLlm):
             top_p=config.top_p,
         )
 
-        return llm(prompt)
+        return llm.invoke(prompt)

embedchain/llm/vllm.py

@@ -37,4 +37,4 @@ class VLLM(BaseLlm):
             llm_args.update(config.model_kwargs)
 
         llm = BaseVLLM(**llm_args)
-        return llm(prompt)
+        return llm.invoke(prompt)