From f5fbc9c5e96b8503774598d33508432f8ec96681 Mon Sep 17 00:00:00 2001
From: Fritz Obermeyer <fritz.obermeyer@gmail.com>
Date: Sun, 26 Feb 2023 14:36:53 -0500
Subject: [PATCH] Expose p50k_edit encoding (#32)

---
 tests/test_simple_public.py   | 2 ++
 tiktoken_ext/openai_public.py | 1 +
 2 files changed, 3 insertions(+)

diff --git a/tests/test_simple_public.py b/tests/test_simple_public.py
index 4410923..ce10da4 100644
--- a/tests/test_simple_public.py
+++ b/tests/test_simple_public.py
@@ -24,3 +24,5 @@ def test_encoding_for_model():
     assert enc.name == "gpt2"
     enc = tiktoken.encoding_for_model("text-davinci-003")
     assert enc.name == "p50k_base"
+    enc = tiktoken.encoding_for_model("text-davinci-edit-001")
+    assert enc.name == "p50k_edit"
diff --git a/tiktoken_ext/openai_public.py b/tiktoken_ext/openai_public.py
index a64db9f..522d58f 100644
--- a/tiktoken_ext/openai_public.py
+++ b/tiktoken_ext/openai_public.py
@@ -84,4 +84,5 @@ ENCODING_CONSTRUCTORS = {
     "r50k_base": r50k_base,
     "p50k_base": p50k_base,
     "cl100k_base": cl100k_base,
+    "p50k_edit": p50k_edit,
 }