Update version to v0.0.77 (#851)

This commit is contained in:
Deshraj Yadav
2023-10-26 09:51:51 -07:00
committed by GitHub
parent 0f8a2e624a
commit ab9598d00a
3 changed files with 24 additions and 33 deletions

View File

@@ -1,7 +1,7 @@
from typing import Iterable, Optional, Union
from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
from langchain.callbacks.stdout import StdOutCallbackHandler
from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
from embedchain.config import BaseLlmConfig
from embedchain.helper.json_serializable import register_deserializable
@@ -48,11 +48,7 @@ class GPT4ALLLlm(BaseLlm):
if config.top_p:
kwargs["top_p"] = config.top_p
callbacks = None
if config.stream:
callbacks = [StreamingStdOutCallbackHandler()]
else:
callbacks =[StdOutCallbackHandler()]
callbacks = [StreamingStdOutCallbackHandler()] if config.stream else [StdOutCallbackHandler()]
response = self.instance.generate(prompts=messages, callbacks=callbacks, **kwargs)
answer = ""