From e2abcea618e151fb95d2b39efe211049fb12ffcd Mon Sep 17 00:00:00 2001 From: BeatrixCohere <128378696+BeatrixCohere@users.noreply.github.com> Date: Thu, 27 Jun 2024 16:07:19 +0100 Subject: [PATCH] Change --- src/backend/config/deployments.py | 14 +++++++------- .../src/components/Agents/CreateAgentForm.tsx | 8 ++++---- 2 files changed, 11 insertions(+), 11 deletions(-) diff --git a/src/backend/config/deployments.py b/src/backend/config/deployments.py index a972608857..4a17d7eaaf 100644 --- 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 index 2b9a5adcec..e8b375eaa3 100644 --- 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: [], });