Merge pull request #901 from KennyDizi/main

Add new GPT-4o models and update tiktoken to 0.7.0
Codium-ai · May 14, 2024 · b4458ff · b4458ff
2 parents e6708fc + 36ad893
commit b4458ff
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py
@@ -11,6 +11,8 @@
  'gpt-4-32k': 32000,
  'gpt-4-1106-preview': 128000, # 128K, but may be limited by config.max_model_tokens
  'gpt-4-0125-preview': 128000, # 128K, but may be limited by config.max_model_tokens
+ 'gpt-4o': 128000, # 128K, but may be limited by config.max_model_tokens
+ 'gpt-4o-2024-05-13': 128000, # 128K, but may be limited by config.max_model_tokens
  'gpt-4-turbo-preview': 128000, # 128K, but may be limited by config.max_model_tokens
  'gpt-4-turbo-2024-04-09': 128000, # 128K, but may be limited by config.max_model_tokens
  'gpt-4-turbo': 128000, # 128K, but may be limited by config.max_model_tokens

diff --git a/requirements.txt b/requirements.txt
@@ -19,7 +19,7 @@ PyYAML==6.0.1
 python-gitlab==3.15.0
 retry==0.9.2
 starlette-context==0.3.6
-tiktoken==0.5.2
+tiktoken==0.7.0
 ujson==5.8.0
 uvicorn==0.22.0
 tenacity==8.2.3