Skip to content

Commit 7390dba

Browse files
committed
removed dict info and put in another file
1 parent 7faaaea commit 7390dba

File tree

3 files changed

+19
-14
lines changed

3 files changed

+19
-14
lines changed

scrapegraphai/helpers/__init__.py

+1
Original file line numberDiff line numberDiff line change
@@ -4,3 +4,4 @@
44
"""
55
from .nodes_metadata import nodes_metadata
66
from .schemas import graph_schema
7+
from .models_tokens import models_tokens
+17
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
"""
2+
Models token
3+
"""
4+
models_tokens = {
5+
"gpt-3.5-turbo-0125": 16385,
6+
"gpt-3.5-turbo": 4096,
7+
"gpt-3.5-turbo-1106": 16385,
8+
"gpt-3.5-turbo-instruct": 4096,
9+
"gpt-4-0125-preview": 128000,
10+
"gpt-4-turbo-preview": 128000,
11+
"gpt-4-1106-preview": 128000,
12+
"gpt-4-vision-preview": 128000,
13+
"gpt-4": 8192,
14+
"gpt-4-0613": 8192,
15+
"gpt-4-32k": 32768,
16+
"gpt-4-32k-0613": 32768,
17+
}

scrapegraphai/utils/token_calculator.py

+1-14
Original file line numberDiff line numberDiff line change
@@ -3,21 +3,8 @@
33
"""
44
from typing import List
55
import tiktoken
6+
from ..helpers.models_tokens import models_tokens
67

7-
models_tokens = {
8-
"gpt-3.5-turbo-0125": 16385,
9-
"gpt-3.5-turbo": 4096,
10-
"gpt-3.5-turbo-1106": 16385,
11-
"gpt-3.5-turbo-instruct": 4096,
12-
"gpt-4-0125-preview": 128000,
13-
"gpt-4-turbo-preview": 128000,
14-
"gpt-4-1106-preview": 128000,
15-
"gpt-4-vision-preview": 128000,
16-
"gpt-4": 8192,
17-
"gpt-4-0613": 8192,
18-
"gpt-4-32k": 32768,
19-
"gpt-4-32k-0613": 32768,
20-
}
218

229
def truncate_text_tokens(text: str, model: str, encoding_name: str) -> List[str]:
2310
"""

0 commit comments

Comments
 (0)