Migrating to use @deco/deco from jsr

Signed-off-by: Marcos Candeia <[email protected]>
deco-cx · Sep 11, 2024 · d208494 · d208494
1 parent 475157f
commit d208494
Show file tree

Hide file tree

Showing 163 changed files with 1,132 additions and 2,829 deletions.
diff --git a/admin/types.ts b/admin/types.ts
@@ -1,48 +1,41 @@
-import { type Resolvable } from "deco/engine/core/resolver.ts";
 import { type fjp } from "./deps.ts";
-
+import { type Resolvable } from "@deco/deco";
 export interface Pagination<T> {
   data: T[];
   page: number;
   pageSize: number;
   total: number;
 }
-
 export interface PatchState {
   type: "patch-state";
   payload: fjp.Operation[];
   revision: string;
 }
-
 export interface FetchState {
   type: "fetch-state";
 }
-
 export interface StatePatched {
   type: "state-patched";
   payload: fjp.Operation[];
   revision: string;
   // Maybe add data and user info in here
   metadata?: unknown;
 }
-
 export interface StateFetched {
   type: "state-fetched";
   payload: State;
 }
-
 export interface OperationFailed {
   type: "operation-failed";
   code: "UNAUTHORIZED" | "INTERNAL_SERVER_ERROR";
   reason: string;
 }
-
-export type Acked<T> = T & { ack: string };
-
+export type Acked<T> = T & {
+  ack: string;
+};
 export interface State {
   decofile: Record<string, Resolvable>;
   revision: string;
 }
-
 export type Commands = PatchState | FetchState;
 export type Events = StatePatched | StateFetched | OperationFailed;
diff --git a/ai-assistants/actions/awsUploadImage.ts b/ai-assistants/actions/awsUploadImage.ts
@@ -1,47 +1,38 @@
-import { logger } from "deco/observability/otel/config.ts";
-import { meter } from "deco/observability/otel/metrics.ts";
 import base64ToBlob from "../utils/blobConversion.ts";
 import { AssistantIds } from "../types.ts";
-import { ValueType } from "deco/deps.ts";
 import { AppContext } from "../mod.ts";
-
+import { logger, meter, ValueType } from "@deco/deco/o11y";
 const stats = {
   awsUploadImageError: meter.createCounter("assistant_aws_upload_error", {
     unit: "1",
     valueType: ValueType.INT,
   }),
 };
-
 export interface AWSUploadImageProps {
   file: string | ArrayBuffer | null;
   assistantIds?: AssistantIds;
 }
-
 // TODO(ItamarRocha): Check if possible to upload straight to bucket instead of using presigned url
 async function getSignedUrl(
   mimetype: string,
   ctx: AppContext,
 ): Promise<string> {
   const randomID = crypto.randomUUID();
   const name = `${randomID}.${mimetype.split("/")[1]}`;
-
   // Get signed URL from S3
   const s3Params = {
     Bucket: ctx.assistantAwsProps?.assistantBucketName.get?.() ?? "",
     Key: name,
     ContentType: mimetype,
     ACL: "public-read",
   };
-
   const uploadURL = await ctx.s3?.getSignedUrlPromise("putObject", s3Params);
   return uploadURL as string;
 }
-
 async function uploadFileToS3(presignedUrl: string, data: Blob) {
   const response = await fetch(presignedUrl, { method: "PUT", body: data });
   return response;
 }
-
 // TODO(ItamarRocha): Rate limit
 export default async function awsUploadImage(
   awsUploadImageProps: AWSUploadImageProps,
@@ -57,7 +48,6 @@ export default async function awsUploadImage(
   );
   const uploadURL = await getSignedUrl(blobData.type, ctx);
   const uploadResponse = await uploadFileToS3(uploadURL, blobData);
-
   if (!uploadResponse.ok) {
     stats.awsUploadImageError.add(1, {
       assistantId,

diff --git a/ai-assistants/actions/chat.ts b/ai-assistants/actions/chat.ts
@@ -1,15 +1,12 @@
 import { AppContext } from "../mod.ts";
-
-import { badRequest, notFound } from "deco/mod.ts";
 import { messageProcessorFor } from "../chat/messages.ts";
 import { Notify, Queue } from "../deps.ts";
-
+import { badRequest, notFound } from "@deco/deco";
 export interface Props {
   thread?: string;
   assistant: string;
   message?: string;
 }
-
 /**
  * Processes messages from the message queue.
  * @param {Queue<ChatMessage>} q - The message queue.
@@ -35,35 +32,29 @@ const process = async (
     ]);
   }
 };
-
 export interface MessageContentText {
   type: "text";
   value: string;
   options?: string[];
 }
-
 export interface MessageContentFile {
   type: "file";
   fileId: string;
 }
-
 export interface ReplyMessage {
   threadId: string;
   messageId: string;
   type: "message" | "error";
   content: Array<MessageContentText | MessageContentFile>;
   role: "user" | "assistant";
 }
-
 export interface FunctionCall {
   name: string;
   props: unknown;
 }
-
 export interface FunctionCallReply<T> extends FunctionCall {
   response: T;
 }
-
 export interface ReplyStartFunctionCall {
   threadId: string;
   messageId: string;
@@ -76,17 +67,14 @@ export interface ReplyFunctionCalls<T> {
   type: "function_calls";
   content: FunctionCallReply<T>[];
 }
-
 export type Reply<T> =
   | ReplyMessage
   | ReplyFunctionCalls<T>
   | ReplyStartFunctionCall;
-
 export interface ChatMessage {
   text: string;
   reply: <T = unknown>(reply: Reply<T>) => void;
 }
-
 /**
  * Initializes a WebSocket chat connection and processes incoming messages.
  * @param {Props} props - The properties for the chat session.
@@ -98,21 +86,24 @@ export default async function openChat(
   props: Props,
   req: Request,
   ctx: AppContext,
-): Promise<Response | { replies: Reply<unknown>[]; thread: string }> {
+): Promise<
+  Response | {
+    replies: Reply<unknown>[];
+    thread: string;
+  }
+> {
   if (!props.assistant) {
     notFound();
   }
   const assistant = ctx.assistants[props.assistant];
   if (!assistant) {
     notFound();
   }
-
   const threads = ctx.openAI.beta.threads;
   const threadId = props.thread;
   const threadPromise = threadId
     ? threads.retrieve(threadId)
     : threads.create();
-
   const processorPromise = assistant.then(async (aiAssistant) =>
     messageProcessorFor(aiAssistant, ctx, await threadPromise)
   );
@@ -128,7 +119,6 @@ export default async function openChat(
     });
     return { replies, thread: (await threadPromise).id };
   }
-
   const { socket, response } = Deno.upgradeWebSocket(req);
   const abort = new Notify();
   const messagesQ = new Queue<ChatMessage>();
@@ -138,7 +128,6 @@ export default async function openChat(
       reply: (replyMsg) => socket.send(JSON.stringify(replyMsg)),
     });
   }
-
   /**
    * Handles the WebSocket connection on open event.
    */
@@ -156,19 +145,17 @@ export default async function openChat(
       }),
     );
     assistant.then((aiAssistant) => {
-      socket.send(
-        JSON.stringify({
-          isWelcomeMessage: true,
-          threadId: aiAssistant.threadId,
-          assistantId: aiAssistant.id,
-          type: "message",
-          content: [{
-            type: "text",
-            value: aiAssistant.welcomeMessage ?? "Welcome to the chat!",
-          }],
-          role: "assistant",
-        }),
-      );
+      socket.send(JSON.stringify({
+        isWelcomeMessage: true,
+        threadId: aiAssistant.threadId,
+        assistantId: aiAssistant.id,
+        type: "message",
+        content: [{
+          type: "text",
+          value: aiAssistant.welcomeMessage ?? "Welcome to the chat!",
+        }],
+        role: "assistant",
+      }));
     });
   };
   /**
@@ -177,7 +164,6 @@ export default async function openChat(
   socket.onclose = () => {
     abort.notifyAll();
   };
-
   /**
    * Handles the WebSocket connection on message event.
    * @param {MessageEvent} event - The WebSocket message event.

diff --git a/ai-assistants/actions/describeImage.ts b/ai-assistants/actions/describeImage.ts
@@ -1,10 +1,7 @@
-import { logger } from "deco/observability/otel/config.ts";
-import { meter } from "deco/observability/otel/metrics.ts";
 import { AssistantIds } from "../types.ts";
-import { ValueType } from "deco/deps.ts";
 import { AppContext } from "../mod.ts";
-import { shortcircuit } from "deco/engine/errors.ts";
-
+import { logger, meter, ValueType } from "@deco/deco/o11y";
+import { shortcircuit } from "@deco/deco";
 const stats = {
   promptTokens: meter.createHistogram("assistant_image_prompt_tokens", {
     description: "Tokens used in Sales Assistant Describe Image Input - OpenAI",
@@ -20,13 +17,11 @@ const stats = {
     valueType: ValueType.INT,
   }),
 };
-
 export interface DescribeImageProps {
   uploadURL: string;
   userPrompt: string;
   assistantIds?: AssistantIds;
 }
-
 // TODO(ItamarRocha): Rate limit
 // TODO(@ItamarRocha): Refactor to use https://github.com/deco-cx/apps/blob/main/openai/loaders/vision.ts
 export default async function describeImage(

diff --git a/ai-assistants/actions/transcribeAudio.ts b/ai-assistants/actions/transcribeAudio.ts
@@ -1,10 +1,7 @@
-import { logger } from "deco/observability/otel/config.ts";
 import base64ToBlob from "../utils/blobConversion.ts";
-import { meter } from "deco/observability/otel/metrics.ts";
 import { AssistantIds } from "../types.ts";
-import { ValueType } from "deco/deps.ts";
 import { AppContext } from "../mod.ts";
-
+import { logger, meter, ValueType } from "@deco/deco/o11y";
 const stats = {
   audioSize: meter.createHistogram("assistant_transcribe_audio_size", {
     description:
@@ -20,13 +17,11 @@ const stats = {
     },
   ),
 };
-
 export interface TranscribeAudioProps {
   file: string | ArrayBuffer | null;
   assistantIds?: AssistantIds;
   audioDuration: number;
 }
-
 // TODO(ItamarRocha): Rate limit
 export default async function transcribeAudio(
   transcribeAudioProps: TranscribeAudioProps,
@@ -41,14 +36,12 @@ export default async function transcribeAudio(
     });
     throw new Error("Audio file is empty");
   }
-
   const blobData = base64ToBlob(
     transcribeAudioProps.file,
     "audio",
     transcribeAudioProps.assistantIds,
   );
   const file = new File([blobData], "input.wav", { type: "audio/wav" });
-
   stats.audioSize.record(transcribeAudioProps.audioDuration, {
     assistant_id: assistantId,
   });