Change

cohere-ai · Jun 27, 2024 · e2abcea · e2abcea
1 parent 5adcf9d
commit e2abcea
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 11 deletions.
diff --git a/src/backend/config/deployments.py b/src/backend/config/deployments.py
@@ -30,13 +30,13 @@ class ModelDeploymentName(StrEnum):
 
 
 ALL_MODEL_DEPLOYMENTS = {
-#     ModelDeploymentName.CoherePlatform: Deployment(
-#         name=ModelDeploymentName.CoherePlatform,
-#         deployment_class=CohereDeployment,
-#         models=CohereDeployment.list_models(),
-#         is_available=CohereDeployment.is_available(),
-#         env_vars=COHERE_ENV_VARS,
-#     ),
+    ModelDeploymentName.CoherePlatform: Deployment(
+        name=ModelDeploymentName.CoherePlatform,
+        deployment_class=CohereDeployment,
+        models=CohereDeployment.list_models(),
+        is_available=CohereDeployment.is_available(),
+        env_vars=COHERE_ENV_VARS,
+    ),
     ModelDeploymentName.SingleContainer: Deployment(
         name=ModelDeploymentName.SingleContainer,
         deployment_class=SingleContainerDeployment,

diff --git a/src/interfaces/coral_web/src/components/Agents/CreateAgentForm.tsx b/src/interfaces/coral_web/src/components/Agents/CreateAgentForm.tsx
@@ -3,7 +3,7 @@ import React, { useContext, useState } from 'react';
 
 import { AgentForm, AgentFormFieldKeys, AgentFormFields } from '@/components/Agents/AgentForm';
 import { Button, Text } from '@/components/Shared';
-import { DEFAULT_AGENT_MODEL, DEPLOYMENT_SINGLE_CONTAINER } from '@/constants';
+import { DEFAULT_AGENT_MODEL, DEPLOYMENT_COHERE_PLATFORM } from '@/constants';
 import { ModalContext } from '@/context/ModalContext';
 import { useCreateAgent, useIsAgentNameUnique, useRecentAgents } from '@/hooks/agents';
 import { useNotify } from '@/hooks/toast';
@@ -23,8 +23,8 @@ export const CreateAgentForm: React.FC = () => {
     name: '',
     description: '',
     preamble: '',
-    deployment: DEPLOYMENT_SINGLE_CONTAINER,
-    model: '',
+    deployment: DEPLOYMENT_COHERE_PLATFORM,
+    model: DEFAULT_AGENT_MODEL,
     tools: [],
   });
 
@@ -77,7 +77,7 @@ export const CreateAgentForm: React.FC = () => {
         name: '',
         description: '',
         preamble: '',
-        deployment: DEPLOYMENT_SINGLE_CONTAINER,
+        deployment: DEPLOYMENT_COHERE_PLATFORM,
         model: DEFAULT_AGENT_MODEL,
         tools: [],
       });