fix: workaround for refactor on new versions of langchain (#924)

* fix: workaround for refactor on new versions of langchain * build(deps): updated dependency locks
gpt-engineer-org · Dec 21, 2023 · 242f37d · 242f37d
1 parent a74c5eb
commit 242f37d
Show file tree

Hide file tree

Showing 2 changed files with 308 additions and 280 deletions.
diff --git a/gpt_engineer/core/token_usage.py b/gpt_engineer/core/token_usage.py
@@ -5,9 +5,20 @@
 
 import tiktoken
 
-from langchain.callbacks.openai_info import get_openai_token_cost_for_model
 from langchain.schema import AIMessage, HumanMessage, SystemMessage
 
+# workaround for function moved in:
+# https://github.com/langchain-ai/langchain/blob/535db72607c4ae308566ede4af65295967bb33a8/libs/community/langchain_community/callbacks/openai_info.py
+try:
+    from langchain.callbacks.openai_info import (
+        get_openai_token_cost_for_model,  # fmt: skip
+    )
+except ImportError:
+    from langchain_community.callbacks.openai_info import (
+        get_openai_token_cost_for_model,  # fmt: skip
+    )
+
+
 Message = Union[AIMessage, HumanMessage, SystemMessage]
 
 logger = logging.getLogger(__name__)