We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent dac0a9c commit 051dc13Copy full SHA for 051dc13
apps/setting/models_provider/impl/base_chat_open_ai.py
@@ -15,8 +15,14 @@
15
from common.config.tokenizer_manage_config import TokenizerManage
16
17
18
+def custom_get_token_ids(text: str):
19
+ tokenizer = TokenizerManage.get_tokenizer()
20
+ return tokenizer.encode(text)
21
+
22
23
class BaseChatOpenAI(ChatOpenAI):
24
usage_metadata: dict = {}
25
+ custom_get_token_ids = custom_get_token_ids
26
27
def get_last_generation_info(self) -> Optional[Dict[str, Any]]:
28
return self.usage_metadata
0 commit comments