- improve: added agent pre selection when it was recently added

dcSpark · Nov 9, 2023 · 77a67f2 · 77a67f2
1 parent 3f8f306
commit 77a67f2
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/apps/shinkai-visor/src/components/add-agent/add-agent.tsx b/apps/shinkai-visor/src/components/add-agent/add-agent.tsx
@@ -60,7 +60,7 @@ export const AddAgent = () => {
   });
   const { mutateAsync: createAgent, isLoading } = useCreateAgent({
     onSuccess: () => {
-      history.replace({ pathname: '/inboxes/create-job' });
+      history.replace({ pathname: '/inboxes/create-job' }, { agentName: form.getValues().agentName });
     },
   });
   const modelOptions: { value: Models; label: string }[] = [

diff --git a/apps/shinkai-visor/src/components/create-job/create-job.tsx b/apps/shinkai-visor/src/components/create-job/create-job.tsx
@@ -44,7 +44,7 @@ type FormSchemaType = z.infer<typeof formSchema>;
 export const CreateJob = () => {
   const history = useHistory();
   const intl = useIntl();
-  const location = useLocation<{ files: File[] }>();
+  const location = useLocation<{ files: File[], agentName: string }>();
   const query = useQuery();
   const auth = useAuth((state) => state.auth);
   const form = useForm<FormSchemaType>({
@@ -84,6 +84,14 @@ export const CreateJob = () => {
   useEffect(() => {
     form.setValue('files', location?.state?.files || []);
   }, [location, form]);
+  useEffect(() => {
+    if (location?.state?.agentName) {
+      const agent = agents.find(agent => (agent.full_identity_name as any)?.subidentity_name === location.state.agentName);
+      if (agent) {
+        form.setValue('agent', agent.id);
+      }
+    }
+  }, [form, location, agents]);
   const submit = (values: FormSchemaType) => {
     if (!auth) return;
     let content = values.content;