Issue #1064: added edit on maching learning

smclab · Aug 7, 2024 · 55251a9 · 55251a9
1 parent 3d1d00c
commit 55251a9
Show file tree

Hide file tree

Showing 8 changed files with 48 additions and 26 deletions.
diff --git a/js-packages/admin-ui/src/App.tsx b/js-packages/admin-ui/src/App.tsx
@@ -487,11 +487,11 @@ export default function App() {
  </Route>
  </Route>
  <Route path="embedding-models" element={<EmbeddingModels />} />
- <Route path="embedding-models/create/new" element={<EmbeddingModelCreate />} />
- <Route path="embedding-models/:embeddingModelsId" element={<EmbeddingModel />} />
+ <Route path="embedding-models/new" element={<EmbeddingModelCreate />} />
+ <Route path="embedding-models/:embeddingModelsId" element={<EmbeddingModelCreate />} />
  <Route path="large-languages-models" element={<LargeLanguageModels />} />
- <Route path="large-languages-models/:largeLanguagesModelsId" element={<LargeLanguageModelE />} />
- <Route path="large-languages-models/create/new" element={<LargeLanguageModel />} />
+ <Route path="large-languages-models/new" element={<LargeLanguageModel />} />
+ <Route path="large-languages-models/:largeLanguageModelId" element={<LargeLanguageModel />} />
  <Route path="wizard">
  <Route path="web-crawler" element={<WebCrawlerWizard />} />
  </Route>

diff --git a/js-packages/admin-ui/src/components/EmbeddingModelCreate.tsx b/js-packages/admin-ui/src/components/EmbeddingModelCreate.tsx
@@ -4,10 +4,12 @@ import { useToast } from "./ToastProvider";
 import { useCreateOrUpdateEmbeddingModelMutation, useEmbeddingModelQuery } from "../graphql-generated";
 import { ContainerFluid, CustomButtom, MainTitle, TextInput, fromFieldValidators, useForm } from "./Form";
 import React from "react";
+import { EmbeddingModelQuery } from "./EmbeddingModel";
+import { EmbeddingModelsQuery } from "./EmbeddingModels";
 
 gql`
- mutation CreateOrUpdateEmbeddingModel($apiKey: String, $apiUrl: String!, $description: String!, $name: String!) {
- embeddingModel(embeddingModelDTO: { name: $name, apiKey: $apiKey, apiUrl: $apiUrl, description: $description }) {
+ mutation CreateOrUpdateEmbeddingModel($id: ID, $apiKey: String, $apiUrl: String!, $description: String!, $name: String!) {
+ embeddingModel(id: $id, embeddingModelDTO: { name: $name, apiKey: $apiKey, apiUrl: $apiUrl, description: $description }) {
  entity {
  id
  name
@@ -25,13 +27,14 @@ export function EmbeddingModelCreate() {
  skip: !embeddingModelsId || embeddingModelsId === "new",
  });
  const [createOrUpdateEmbeddingModelsMutate] = useCreateOrUpdateEmbeddingModelMutation({
- refetchQueries: [],
+ refetchQueries: [EmbeddingModelsQuery, EmbeddingModelQuery],
  onCompleted(data) {
  if (data.embeddingModel?.entity) {
  if (embeddingModelsId === "new") {
  navigate(`/embedding-models/`, { replace: true });
  showToast({ displayType: "success", title: "Models created", content: data.embeddingModel.entity.name ?? "" });
  } else {
+ navigate(`/embedding-models/`, { replace: true });
  showToast({ displayType: "info", title: "Models updated", content: data.embeddingModel.entity.name ?? "" });
  }
  }
@@ -50,7 +53,7 @@ export function EmbeddingModelCreate() {
  originalValues: modelQuery.data?.embeddingModel,
  isLoading: modelQuery.loading || modelQuery.loading,
  onSubmit(data) {
- createOrUpdateEmbeddingModelsMutate({ variables: { ...data } });
+ createOrUpdateEmbeddingModelsMutate({ variables: { id: embeddingModelsId !== "new" ? embeddingModelsId : undefined, ...data } });
  },
  getValidationMessages: fromFieldValidators(null),
  });

diff --git a/js-packages/admin-ui/src/components/EmbeddingModels.tsx b/js-packages/admin-ui/src/components/EmbeddingModels.tsx
@@ -64,7 +64,7 @@ export function EmbeddingModels() {
  field: (data) => data?.embeddingModels,
  }}
  viewAdd={true}
- onCreatePath="create/new/"
+ onCreatePath="new/"
  onDelete={(model) => {
  if (model?.id) deleteModelsMutate({ variables: { id: model.id } });
  }}

diff --git a/js-packages/admin-ui/src/components/LargeLanguageModel.tsx b/js-packages/admin-ui/src/components/LargeLanguageModel.tsx
@@ -5,7 +5,7 @@ import { useLargeLanguageModelQuery } from "../graphql-generated";
 import { ContainerFluid, MainTitle, TextInput } from "./Form";
 import React from "react";
 
-const LargeLanguageModel = gql`
+export const LargeLanguageModelQ = gql`
  query LargeLanguageModel($id: ID!) {
  largeLanguageModel(id: $id) {
  name

diff --git a/js-packages/admin-ui/src/components/LargeLanguageModelCreate.tsx b/js-packages/admin-ui/src/components/LargeLanguageModelCreate.tsx
@@ -5,10 +5,20 @@ import { useCreateOrUpdateLargeLanguageModelMutation, useLargeLanguageModelQuery
 import { ContainerFluid, CustomButtom, MainTitle, TextInput, fromFieldValidators, useForm } from "./Form";
 import React from "react";
 import { CodeInput } from "./CodeInput";
+import { LargeLanguageModelsQuery } from "./LargeLanguageModels";
+import { LargeLanguageModelQ } from "./LargeLanguageModel";
 
-gql`
- mutation CreateOrUpdateLargeLanguageModel($apiKey: String, $apiUrl: String!, $description: String!, $name: String!, $jsonConfig: String) {
+const create = gql`
+ mutation CreateOrUpdateLargeLanguageModel(
+ $id: ID
+ $apiKey: String
+ $apiUrl: String!
+ $description: String!
+ $name: String!
+ $jsonConfig: String
+ ) {
  largeLanguageModel(
+ id: $id
  largeLanguageModelDTO: { name: $name, apiKey: $apiKey, apiUrl: $apiUrl, description: $description, jsonConfig: $jsonConfig }
  ) {
  entity {
@@ -28,13 +38,14 @@ export function LargeLanguageModel() {
  skip: !largeLanguageModelId || largeLanguageModelId === "new",
  });
  const [createOrUpdateEmbeddingModelsMutate] = useCreateOrUpdateLargeLanguageModelMutation({
- refetchQueries: [],
+ refetchQueries: [create, LargeLanguageModelsQuery, LargeLanguageModelQ],
  onCompleted(data) {
  if (data.largeLanguageModel?.entity) {
  if (largeLanguageModelId === "new") {
  navigate(`/large-languages-models/`, { replace: true });
  showToast({ displayType: "success", title: "Models created", content: data.largeLanguageModel.entity.name ?? "" });
  } else {
+ navigate(`/large-languages-models/`, { replace: true });
  showToast({ displayType: "info", title: "Models updated", content: data.largeLanguageModel.entity.name ?? "" });
  }
  }
@@ -54,7 +65,9 @@ export function LargeLanguageModel() {
  originalValues: modelQuery.data?.largeLanguageModel,
  isLoading: modelQuery.loading || modelQuery.loading,
  onSubmit(data) {
- createOrUpdateEmbeddingModelsMutate({ variables: { ...data } });
+ createOrUpdateEmbeddingModelsMutate({
+ variables: { id: largeLanguageModelId !== "new" ? largeLanguageModelId : undefined, ...data },
+ });
  },
  getValidationMessages: fromFieldValidators(null),
  });

diff --git a/js-packages/admin-ui/src/components/LargeLanguageModels.tsx b/js-packages/admin-ui/src/components/LargeLanguageModels.tsx
@@ -72,7 +72,7 @@ export function LargeLanguageModels() {
  queryResult: largeLanguageModels,
  field: (data) => data?.largeLanguageModels,
  }}
- onCreatePath="create/new/"
+ onCreatePath="new/"
  onDelete={(model) => {
  if (model?.id) deleteModelsMutate({ variables: { id: model.id } });
  }}

diff --git a/js-packages/admin-ui/src/components/SideNavigation.tsx b/js-packages/admin-ui/src/components/SideNavigation.tsx
@@ -132,14 +132,14 @@ export function SideNavigation({ isSideMenuOpen }: { isSideMenuOpen: boolean })
  />
  <SideNavigationItem IsChildren={true} setSelect={setSelect} elementSelect={select} label="Logs" path="/logs" />
  </SideNavigationCollapsible>
- <SideNavigationItem
- IsChildren={false}
- setSelect={setSelect}
- elementSelect={select}
- label="Maching Learning"
- path="maching-learning"
- />
- <SideNavigationCollapsible label="Embedding models">
+ <SideNavigationCollapsible label="Machine Learning">
+ <SideNavigationItem
+  IsChildren={false}
+  setSelect={setSelect}
+  elementSelect={select}
+  label="Hugging Face Integration"
+  path="maching-learning"
+  />
  <SideNavigationItem
  IsChildren={true}
  setSelect={setSelect}

diff --git a/js-packages/admin-ui/src/graphql-generated.ts b/js-packages/admin-ui/src/graphql-generated.ts
@@ -4983,6 +4983,7 @@ export type EmbeddingModelQueryVariables = Exact<{
 export type EmbeddingModelQuery = { __typename?: 'Query', embeddingModel?: { __typename?: 'EmbeddingModel', name?: string | null, description?: string | null, apiUrl?: string | null, apiKey?: string | null } | null };
 
 export type CreateOrUpdateEmbeddingModelMutationVariables = Exact<{
+ id?: InputMaybe<Scalars['ID']>;
  apiKey?: InputMaybe<Scalars['String']>;
  apiUrl: Scalars['String'];
  description: Scalars['String'];
@@ -5176,6 +5177,7 @@ export type LargeLanguageModelQueryVariables = Exact<{
 export type LargeLanguageModelQuery = { __typename?: 'Query', largeLanguageModel?: { __typename?: 'LargeLanguageModel', name?: string | null, description?: string | null, apiUrl?: string | null, apiKey?: string | null } | null };
 
 export type CreateOrUpdateLargeLanguageModelMutationVariables = Exact<{
+ id?: InputMaybe<Scalars['ID']>;
  apiKey?: InputMaybe<Scalars['String']>;
  apiUrl: Scalars['String'];
  description: Scalars['String'];
@@ -9919,8 +9921,9 @@ export type EmbeddingModelQueryHookResult = ReturnType<typeof useEmbeddingModelQ
 export type EmbeddingModelLazyQueryHookResult = ReturnType<typeof useEmbeddingModelLazyQuery>;
 export type EmbeddingModelQueryResult = Apollo.QueryResult<EmbeddingModelQuery, EmbeddingModelQueryVariables>;
 export const CreateOrUpdateEmbeddingModelDocument = gql`
- mutation CreateOrUpdateEmbeddingModel($apiKey: String, $apiUrl: String!, $description: String!, $name: String!) {
+ mutation CreateOrUpdateEmbeddingModel($id: ID, $apiKey: String, $apiUrl: String!, $description: String!, $name: String!) {
  embeddingModel(
+ id: $id
  embeddingModelDTO: {name: $name, apiKey: $apiKey, apiUrl: $apiUrl, description: $description}
  ) {
  entity {
@@ -9945,6 +9948,7 @@ export type CreateOrUpdateEmbeddingModelMutationFn = Apollo.MutationFunction<Cre
  * @example
  * const [createOrUpdateEmbeddingModelMutation, { data, loading, error }] = useCreateOrUpdateEmbeddingModelMutation({
  * variables: {
+ * id: // value for 'id'
  * apiKey: // value for 'apiKey'
  * apiUrl: // value for 'apiUrl'
  * description: // value for 'description'
@@ -10946,8 +10950,9 @@ export type LargeLanguageModelQueryHookResult = ReturnType<typeof useLargeLangua
 export type LargeLanguageModelLazyQueryHookResult = ReturnType<typeof useLargeLanguageModelLazyQuery>;
 export type LargeLanguageModelQueryResult = Apollo.QueryResult<LargeLanguageModelQuery, LargeLanguageModelQueryVariables>;
 export const CreateOrUpdateLargeLanguageModelDocument = gql`
- mutation CreateOrUpdateLargeLanguageModel($apiKey: String, $apiUrl: String!, $description: String!, $name: String!, $jsonConfig: String) {
+ mutation CreateOrUpdateLargeLanguageModel($id: ID, $apiKey: String, $apiUrl: String!, $description: String!, $name: String!, $jsonConfig: String) {
  largeLanguageModel(
+ id: $id
  largeLanguageModelDTO: {name: $name, apiKey: $apiKey, apiUrl: $apiUrl, description: $description, jsonConfig: $jsonConfig}
  ) {
  entity {
@@ -10972,6 +10977,7 @@ export type CreateOrUpdateLargeLanguageModelMutationFn = Apollo.MutationFunction
  * @example
  * const [createOrUpdateLargeLanguageModelMutation, { data, loading, error }] = useCreateOrUpdateLargeLanguageModelMutation({
  * variables: {
+ * id: // value for 'id'
  * apiKey: // value for 'apiKey'
  * apiUrl: // value for 'apiUrl'
  * description: // value for 'description'
@@ -14304,4 +14310,4 @@ export function useCreateYouTubeDataSourceMutation(baseOptions?: Apollo.Mutation
 export type CreateYouTubeDataSourceMutationHookResult = ReturnType<typeof useCreateYouTubeDataSourceMutation>;
 export type CreateYouTubeDataSourceMutationResult = Apollo.MutationResult<CreateYouTubeDataSourceMutation>;
 export type CreateYouTubeDataSourceMutationOptions = Apollo.BaseMutationOptions<CreateYouTubeDataSourceMutation, CreateYouTubeDataSourceMutationVariables>;
-// Generated on 2024-08-07T15:32:59+02:00
+// Generated on 2024-08-07T17:25:10+02:00