Fix types

dust-tt · Nov 15, 2023 · f499573 · f499573
1 parent a95d384
commit f499573
Show file tree

Hide file tree

Showing 4 changed files with 22 additions and 25 deletions.
diff --git a/front/lib/api/assistant/types.ts b/front/lib/api/assistant/types.ts
diff --git a/front/pages/api/v1/w/[wId]/assistant/conversations/[cId]/content_fragments.ts b/front/pages/api/v1/w/[wId]/assistant/conversations/[cId]/content_fragments.ts
@@ -16,8 +16,6 @@ export type PostContentFragmentsResponseBody = {
   contentFragment: ContentFragmentType;
 };
 
-
-
 export const PostContentFragmentRequestBodySchema = t.type({
   title: t.string,
   content: t.string,
@@ -39,7 +37,6 @@ export type PostContentFragmentRequestBody = t.TypeOf<
   typeof PostContentFragmentRequestBodySchema
 >;
 
-
 async function handler(
   req: NextApiRequest,
   res: NextApiResponse<PostContentFragmentsResponseBody | ReturnedAPIErrorType>
@@ -93,7 +90,8 @@ async function handler(
         });
       }
 
-      const { content, title, url, contentType, context } = bodyValidation.right;
+      const { content, title, url, contentType, context } =
+        bodyValidation.right;
 
       if (content.length === 0 || content.length > 64 * 1024) {
         return apiError(req, res, {

diff --git a/front/pages/api/w/[wId]/assistant/conversations/[cId]/content_fragment/index.ts b/front/pages/api/w/[wId]/assistant/conversations/[cId]/content_fragment/index.ts
@@ -7,12 +7,29 @@ import {
   getConversation,
   postNewContentFragment,
 } from "@app/lib/api/assistant/conversation";
-import { PostContentFragmentRequestBodySchema } from "@app/lib/api/assistant/types";
 import { Authenticator, getSession } from "@app/lib/auth";
 import { ReturnedAPIErrorType } from "@app/lib/error";
 import { apiError, withLogging } from "@app/logger/withlogging";
 import { ContentFragmentType } from "@app/types/assistant/conversation";
 
+export const PostContentFragmentRequestBodySchema = t.type({
+  title: t.string,
+  content: t.string,
+  url: t.union([t.string, t.null]),
+  contentType: t.union([
+    t.literal("slack_thread_content"),
+    t.literal("file_attachment"),
+  ]),
+  context: t.type({
+    timezone: t.string,
+    profilePictureUrl: t.union([t.string, t.null]),
+  }),
+});
+
+export type PostContentFragmentRequestBody = t.TypeOf<
+  typeof PostContentFragmentRequestBodySchema
+>;
+
 export const PostMessagesRequestBodySchema = t.type({
   content: t.string,
   mentions: t.array(

diff --git a/front/pages/api/w/[wId]/assistant/conversations/index.ts b/front/pages/api/w/[wId]/assistant/conversations/index.ts
@@ -10,7 +10,6 @@ import {
   postNewContentFragment,
 } from "@app/lib/api/assistant/conversation";
 import { postUserMessageWithPubSub } from "@app/lib/api/assistant/pubsub";
-import { PostContentFragmentRequestBodySchema } from "@app/lib/api/assistant/types";
 import { Authenticator, getSession } from "@app/lib/auth";
 import { ReturnedAPIErrorType } from "@app/lib/error";
 import { apiError, withLogging } from "@app/logger/withlogging";
@@ -22,6 +21,8 @@ import {
   UserMessageType,
 } from "@app/types/assistant/conversation";
 
+import { PostContentFragmentRequestBodySchema } from "./[cId]/content_fragment";
+
 export const PostConversationsRequestBodySchema = t.type({
   title: t.union([t.string, t.null]),
   visibility: t.union([