Telemetry support for ec cli commands (#1030)

This commit is contained in:
Sidharth Mohanty
2023-12-19 19:29:04 +05:30
committed by GitHub
parent b4f3bbbbc9
commit 48c38b5dc3
8 changed files with 47 additions and 21 deletions

View File

@@ -33,14 +33,16 @@ class OpenAILlm(BaseLlm):
if config.top_p:
kwargs["model_kwargs"]["top_p"] = config.top_p
if config.stream:
from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
from langchain.callbacks.streaming_stdout import \
StreamingStdOutCallbackHandler
callbacks = config.callbacks if config.callbacks else [StreamingStdOutCallbackHandler()]
chat = ChatOpenAI(**kwargs, streaming=config.stream, callbacks=callbacks)
else:
chat = ChatOpenAI(**kwargs)
if self.functions is not None:
from langchain.chains.openai_functions import create_openai_fn_runnable
from langchain.chains.openai_functions import \
create_openai_fn_runnable
from langchain.prompts import ChatPromptTemplate
structured_prompt = ChatPromptTemplate.from_messages(messages)