From ca61185ed6914b6695984f1723e2dcecbafc280e Mon Sep 17 00:00:00 2001 From: Douglas Reid Date: Fri, 12 May 2023 12:04:06 -0700 Subject: [PATCH] deps: update to latest langchain (#44) Co-authored-by: Douglas Reid --- requirements.txt | 2 +- src/steamship_langchain/chat_models/openai.py | 5 +++++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 68e39d1..5d49349 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,2 @@ steamship~=2.16.4 -langchain==0.0.152 \ No newline at end of file +langchain==0.0.167 \ No newline at end of file diff --git a/src/steamship_langchain/chat_models/openai.py b/src/steamship_langchain/chat_models/openai.py index 0a7ab2d..50b24b3 100644 --- a/src/steamship_langchain/chat_models/openai.py +++ b/src/steamship_langchain/chat_models/openai.py @@ -280,3 +280,8 @@ def get_num_tokens_from_messages(self, messages: List[BaseMessage]) -> int: # every reply is primed with assistant num_tokens += 3 return num_tokens + + @property + def _llm_type(self) -> str: + """Return type of chat model.""" + return "openai-chat"