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"