From 5de2e79b3b99ac8cd84cbedd7c09ef54c7dbba6d Mon Sep 17 00:00:00 2001
From: duckdoom4 <60387522+duckdoom4@users.noreply.github.com>
Date: Fri, 27 Jan 2023 13:48:20 +0100
Subject: [PATCH] Add missing models

---
 models/openai_model.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/models/openai_model.py b/models/openai_model.py
index 83f7510..aa5a9af 100644
--- a/models/openai_model.py
+++ b/models/openai_model.py
@@ -23,9 +23,20 @@ class Mode:
 
 
 class Models:
+    # Text models
     DAVINCI = "text-davinci-003"
     CURIE = "text-curie-001"
+    BABBAGE = "text-babbage-001"
+    ADA = "text-ada-001"
+    
+    # Code models
+    CODE_DAVINCI = "code-davinci-002"
+    CODE_CUSHMAN = "code-cushman-001"
+    
+    # Embedding models
     EMBEDDINGS = "text-embedding-ada-002"
+    
+    # Edit models
     EDIT = "text-davinci-edit-001"
     CODE_EDIT = "code-davinci-edit-001"