Update models

e2b-dev · Nov 11, 2023 · ba9fd96 · ba9fd96
1 parent 3b40107
commit ba9fd96
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 5 deletions.
diff --git a/api-service/agent/smol_agent.py b/api-service/agent/smol_agent.py
@@ -38,15 +38,20 @@
         "completion": Decimal(0.00006),
         "max_tokens": 8192,
     },
+    "gpt-4-1106-preview": {
+        "prompt": Decimal(0.00001),
+        "completion": Decimal(0.00003),
+        "max_tokens": 128000,
+    },
     "gpt-4-32k": {
         "prompt": Decimal(0.00006),
         "completion": Decimal(0.00012),
         "max_tokens": 32768,
     },
     "gpt-3.5-turbo": {
-        "prompt": Decimal(0.0000015),
+        "prompt": Decimal(0.000001),
         "completion": Decimal(0.000002),
-        "max_tokens": 4096,
+        "max_tokens": 16384,
     },
     "gpt-3.5-turbo-16k": {
         "prompt": Decimal(0.000003),
@@ -147,7 +152,7 @@ async def create(
             on_interaction_request,
             model,
             model_name,
-            e2b_api_key
+            e2b_api_key,
         )
 
     async def generate_file(
@@ -281,7 +286,11 @@ async def _dev(self, instructions: Any):
             playground = None
             tasks: List[asyncio.Task[Any]] = []
             try:
-                playground = Playground(env_id="PPSrlH5TIvFx", get_envs=self.get_envs, api_key=self.e2b_api_key)
+                playground = Playground(
+                    env_id="PPSrlH5TIvFx",
+                    get_envs=self.get_envs,
+                    api_key=self.e2b_api_key,
+                )
                 rootdir = "/repo"
 
                 async def initialize_playground():

diff --git a/pages/agent/smol-developer/setup.tsx b/pages/agent/smol-developer/setup.tsx
@@ -64,8 +64,8 @@ const steps = [
 const openAIModels = [
   { displayName: 'GPT-4', value: 'gpt-4' },
   { displayName: 'GPT-4 32k', value: 'gpt-4-32k' },
+  { displayName: 'GPT-4 Turbo', value: 'gpt-4-1106-preview' },
   { displayName: 'GPT-3.5 Turbo', value: 'gpt-3.5-turbo' },
-  { displayName: 'GPT-3.5 Turbo 16k', value: 'gpt-3.5-turbo-16k' },
 ]
 
 export const getServerSideProps: GetServerSideProps = async (ctx) => {