Fix the ci error about the langchain (#341)

Signed-off-by: SimFG <bang.fu@zilliz.com>
zilliztech · May 15, 2023 · 261b416 · 261b416
1 parent ca7f2f8
commit 261b416
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/gptcache/adapter/langchain_models.py b/gptcache/adapter/langchain_models.py
@@ -1,9 +1,9 @@
 from typing import Optional, List, Any
 
 from gptcache.adapter.adapter import adapt
+from gptcache.manager.scalar_data.base import Answer, DataType
 from gptcache.session import Session
 from gptcache.utils import import_pydantic, import_langchain
-from gptcache.manager.scalar_data.base import Answer, DataType
 
 import_pydantic()
 import_langchain()
@@ -82,6 +82,10 @@ class LangChainChat(BaseChatModel, BaseModel):
             chat([HumanMessage(content="Translate this sentence from English to French. I love programming.")])
     """
 
+    @property
+    def _llm_type(self) -> str:
+        return "gptcache_llm_chat"
+
     chat: Any
     session: Session = None