fix ai gen

2024-05-14 00:18:27 -07:00
parent ce5e55bdf6
commit f3e9f81722
5 changed files with 10 additions and 19 deletions
--- a/frontend/components/editor/generate.tsx
+++ b/frontend/components/editor/generate.tsx
@ -71,20 +71,13 @@ export default function GenerateInput({
      data.code,
      data.line,
      regenerate ? currentPrompt : input,
-      (res: {
-        result: {
-          response: string;
-        };
-        success: boolean;
-        errors: any[];
-        messages: any[];
-      }) => {
+      (res: { response: string; success: boolean }) => {
        if (!res.success) {
-          console.error(res.errors);
+          toast.error("Failed to generate code.");
          return;
        }

-        setCode(res.result.response);
+        setCode(res.response);
        router.refresh();
      }
    );