@delofarag/ai-utils 1.4.2 → 1.4.3

package/dist/chatbot.d.ts

@@ -0,0 +1,33 @@
+import type { Agent } from "./heart/agent";
+import type { MemoryChain } from "./heart/memorychain";
+type SessionStreamable = Agent<any> | MemoryChain<any>;
+type SessionProps = {
+    streamable: SessionStreamable;
+    breakword?: string;
+    numberOfMessages?: number;
+    id?: string;
+};
+/**
+ * die session funktion für ein streamable, wird hauptsächlich für testing eines agenten/memorychains verwendet
+ * (diese 2 sind die einzigen Streamables die es gibt)
+ * @param streamable das streamable welches die session führt
+ * @param breakword das wort welches die session beendet
+ * @param numberOfMessages die maximale anzahl von messages welche die session senden kann
+ * @param id die id der session
+ * @returns void
+ */
+export declare function session({ streamable, breakword, numberOfMessages, id }: SessionProps): Promise<void>;
+type StreamResponseInit = ResponseInit & {
+    encoder?: TextEncoder;
+};
+/**
+ * eine HTTP-Response objekt erweiterung welches einen stream returnbar macht. ein stream kann halt nicht per
+ * json einfach geschickt werden weil das nicht serialisiert werden kann, deswegen muss es einen prozess nochmal
+ * untergehen befor man das "returnen" kann
+ * @param aiStream der stream von der ai
+ * @param init die initialisierung des response objekts
+ * @returns das response objekt
+ */
+export declare function StreamResponse<T>(aiStream: AsyncIterable<T>, init?: StreamResponseInit): Response;
+export {};
+//# sourceMappingURL=chatbot.d.ts.map

package/dist/chatbot.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatbot.d.ts","sourceRoot":"","sources":["../src/chatbot.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,eAAe,CAAA;AAC1C,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAA;AAEtD,KAAK,iBAAiB,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,WAAW,CAAC,GAAG,CAAC,CAAA;AAEtD,KAAK,YAAY,GAAG;IAChB,UAAU,EAAE,iBAAiB,CAAA;IAC7B,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,gBAAgB,CAAC,EAAE,MAAM,CAAA;IACzB,EAAE,CAAC,EAAE,MAAM,CAAA;CACd,CAAA;AAED;;;;;;;;GAQG;AACH,wBAAsB,OAAO,CAAC,EAC1B,UAAU,EACV,SAAkB,EAClB,gBAA2C,EAC3C,EAAoB,EACvB,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC,CA6B9B;AAED,KAAK,kBAAkB,GAAG,YAAY,GAAG;IACrC,OAAO,CAAC,EAAE,WAAW,CAAC;CACvB,CAAC;AAEJ;;;;;;;GAOG;AACH,wBAAgB,cAAc,CAAC,CAAC,EAC5B,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,EAC1B,IAAI,GAAE,kBAAuB,GAC5B,QAAQ,CA4BV"}

package/dist/chatbot.js

@@ -0,0 +1,72 @@
+import { input } from "@delofarag/base-utils/server";
+import { logChunk } from "./helpers";
+/**
+ * die session funktion für ein streamable, wird hauptsächlich für testing eines agenten/memorychains verwendet
+ * (diese 2 sind die einzigen Streamables die es gibt)
+ * @param streamable das streamable welches die session führt
+ * @param breakword das wort welches die session beendet
+ * @param numberOfMessages die maximale anzahl von messages welche die session senden kann
+ * @param id die id der session
+ * @returns void
+ */
+export async function session({ streamable, breakword = "exit", numberOfMessages = Number.POSITIVE_INFINITY, id = `${Date.now()}` }) {
+    let messages = 0;
+    while (true) {
+        try {
+            const message = await input("You: ");
+            if (message === breakword) {
+                break;
+            }
+            const response = streamable.stream({
+                input: message,
+                thread_id: id
+            });
+            console.log("Assistant: ");
+            for await (const chunk of response) {
+                logChunk(chunk);
+            }
+            console.log("");
+        }
+        catch (e) {
+            console.error("Error: ", e);
+        }
+        messages = messages + 2;
+        if (messages > numberOfMessages) {
+            console.log(`Message-limit of ${numberOfMessages} reached, stopping session`);
+            break;
+        }
+    }
+}
+/**
+ * eine HTTP-Response objekt erweiterung welches einen stream returnbar macht. ein stream kann halt nicht per
+ * json einfach geschickt werden weil das nicht serialisiert werden kann, deswegen muss es einen prozess nochmal
+ * untergehen befor man das "returnen" kann
+ * @param aiStream der stream von der ai
+ * @param init die initialisierung des response objekts
+ * @returns das response objekt
+ */
+export function StreamResponse(aiStream, init = {}) {
+    const encoder = init.encoder ?? new TextEncoder();
+    const { encoder: _encoder, headers, ...responseInit } = init;
+    const stream = new ReadableStream({
+        async start(controller) {
+            try {
+                for await (const chunk of aiStream) {
+                    controller.enqueue(encoder.encode(JSON.stringify(chunk) + "\n"));
+                }
+                controller.close();
+            }
+            catch (error) {
+                controller.error(error);
+            }
+        },
+    });
+    return new Response(stream, {
+        ...responseInit,
+        headers: {
+            "Content-Type": "application/x-ndjson",
+            ...headers,
+        },
+    });
+}
+//# sourceMappingURL=chatbot.js.map

package/dist/chatbot.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatbot.js","sourceRoot":"","sources":["../src/chatbot.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,8BAA8B,CAAA;AACpD,OAAO,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAA;AAapC;;;;;;;;GAQG;AACH,MAAM,CAAC,KAAK,UAAU,OAAO,CAAC,EAC1B,UAAU,EACV,SAAS,GAAG,MAAM,EAClB,gBAAgB,GAAG,MAAM,CAAC,iBAAiB,EAC3C,EAAE,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,EAAE,EACT;IACX,IAAI,QAAQ,GAAG,CAAC,CAAA;IAEhB,OAAO,IAAI,EAAE,CAAC;QACV,IAAI,CAAC;YACD,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,CAAA;YACpC,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;gBACxB,MAAK;YACT,CAAC;YAED,MAAM,QAAQ,GAAG,UAAU,CAAC,MAAM,CAAC;gBAC/B,KAAK,EAAE,OAAO;gBACd,SAAS,EAAE,EAAE;aAChB,CAAC,CAAA;YACF,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAA;YAC1B,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;gBACjC,QAAQ,CAAC,KAAK,CAAC,CAAA;YACnB,CAAC;YACD,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;QACnB,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACT,OAAO,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,CAAA;QAC/B,CAAC;QAED,QAAQ,GAAG,QAAQ,GAAG,CAAC,CAAA;QACvB,IAAI,QAAQ,GAAG,gBAAgB,EAAE,CAAC;YAC9B,OAAO,CAAC,GAAG,CAAC,oBAAoB,gBAAgB,4BAA4B,CAAC,CAAA;YAC7E,MAAK;QACT,CAAC;IACL,CAAC;AACL,CAAC;AAMD;;;;;;;GAOG;AACH,MAAM,UAAU,cAAc,CAC1B,QAA0B,EAC1B,OAA2B,EAAE;IAE7B,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,IAAI,IAAI,WAAW,EAAE,CAAC;IAElD,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,YAAY,EAAE,GAAG,IAAI,CAAC;IAE7D,MAAM,MAAM,GAAG,IAAI,cAAc,CAAa;QAC5C,KAAK,CAAC,KAAK,CAAC,UAAU;YACpB,IAAI,CAAC;gBACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;oBACnC,UAAU,CAAC,OAAO,CAChB,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,CAC7C,CAAC;gBACJ,CAAC;gBAED,UAAU,CAAC,KAAK,EAAE,CAAC;YACrB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;KACF,CAAC,CAAC;IAEH,OAAO,IAAI,QAAQ,CAAC,MAAM,EAAE;QAC1B,GAAG,YAAY;QACf,OAAO,EAAE;YACP,cAAc,EAAE,sBAAsB;YACtC,GAAG,OAAO;SACX;KACF,CAAC,CAAC;AACL,CAAC"}

package/dist/index.d.ts

@@ -4,7 +4,6 @@ export * from "./rag";
 export { DynamicStructuredTool } from "./imports";
 export * from "./heart/agent";
 export * from "./heart/chain";
-export * from "./heart/chatbot";
 export * from "./heart/memorychain";
 export * from "./heart/tools/ToolRegistry";
 export { tavilySearchTool, TavilySearch } from "./heart/tools/Tavily";
@@ -22,5 +21,5 @@ export * from "./magic-funcs/parsers/rewrite";
 export * from "./magic-funcs/parsers/structure";
 export * from "./magic-funcs/parsers/summarize";
 export * from "./heart/tools/Tavily";
-export * from "./session";
+export * from "./chatbot";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,WAAW,CAAA;AACzB,cAAc,UAAU,CAAA;AACxB,cAAc,OAAO,CAAA;AACrB,OAAO,EAAE,qBAAqB,EAAE,MAAM,WAAW,CAAA;AACjD,cAAc,eAAe,CAAA;AAC7B,cAAc,eAAe,CAAA;AAC7B,cAAc,iBAAiB,CAAA;AAC/B,cAAc,qBAAqB,CAAA;AACnC,cAAc,4BAA4B,CAAA;AAC1C,OAAO,EAAE,gBAAgB,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAA;AACrE,cAAc,uBAAuB,CAAA;AACrC,cAAc,uCAAuC,CAAA;AACrD,cAAc,yCAAyC,CAAA;AACvD,cAAc,6BAA6B,CAAA;AAC3C,cAAc,mCAAmC,CAAA;AACjD,cAAc,mCAAmC,CAAA;AACjD,cAAc,iCAAiC,CAAA;AAC/C,cAAc,oCAAoC,CAAA;AAClD,cAAc,iCAAiC,CAAA;AAC/C,cAAc,+BAA+B,CAAA;AAC7C,cAAc,+BAA+B,CAAA;AAC7C,cAAc,iCAAiC,CAAA;AAC/C,cAAc,iCAAiC,CAAA;AAC/C,cAAc,sBAAsB,CAAA;AACpC,cAAc,WAAW,CAAA"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,WAAW,CAAA;AACzB,cAAc,UAAU,CAAA;AACxB,cAAc,OAAO,CAAA;AACrB,OAAO,EAAE,qBAAqB,EAAE,MAAM,WAAW,CAAA;AACjD,cAAc,eAAe,CAAA;AAC7B,cAAc,eAAe,CAAA;AAC7B,cAAc,qBAAqB,CAAA;AACnC,cAAc,4BAA4B,CAAA;AAC1C,OAAO,EAAE,gBAAgB,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAA;AACrE,cAAc,uBAAuB,CAAA;AACrC,cAAc,uCAAuC,CAAA;AACrD,cAAc,yCAAyC,CAAA;AACvD,cAAc,6BAA6B,CAAA;AAC3C,cAAc,mCAAmC,CAAA;AACjD,cAAc,mCAAmC,CAAA;AACjD,cAAc,iCAAiC,CAAA;AAC/C,cAAc,oCAAoC,CAAA;AAClD,cAAc,iCAAiC,CAAA;AAC/C,cAAc,+BAA+B,CAAA;AAC7C,cAAc,+BAA+B,CAAA;AAC7C,cAAc,iCAAiC,CAAA;AAC/C,cAAc,iCAAiC,CAAA;AAC/C,cAAc,sBAAsB,CAAA;AACpC,cAAc,WAAW,CAAA"}

package/dist/index.js

@@ -4,7 +4,6 @@ export * from "./rag";
 export { DynamicStructuredTool } from "./imports";
 export * from "./heart/agent";
 export * from "./heart/chain";
-export * from "./heart/chatbot";
 export * from "./heart/memorychain";
 export * from "./heart/tools/ToolRegistry";
 export { tavilySearchTool, TavilySearch } from "./heart/tools/Tavily";
@@ -22,5 +21,5 @@ export * from "./magic-funcs/parsers/rewrite";
 export * from "./magic-funcs/parsers/structure";
 export * from "./magic-funcs/parsers/summarize";
 export * from "./heart/tools/Tavily";
-export * from "./session";
+export * from "./chatbot";
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,WAAW,CAAA;AACzB,cAAc,UAAU,CAAA;AACxB,cAAc,OAAO,CAAA;AACrB,OAAO,EAAE,qBAAqB,EAAE,MAAM,WAAW,CAAA;AACjD,cAAc,eAAe,CAAA;AAC7B,cAAc,eAAe,CAAA;AAC7B,cAAc,iBAAiB,CAAA;AAC/B,cAAc,qBAAqB,CAAA;AACnC,cAAc,4BAA4B,CAAA;AAC1C,OAAO,EAAE,gBAAgB,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAA;AACrE,cAAc,uBAAuB,CAAA;AACrC,cAAc,uCAAuC,CAAA;AACrD,cAAc,yCAAyC,CAAA;AACvD,cAAc,6BAA6B,CAAA;AAC3C,cAAc,mCAAmC,CAAA;AACjD,cAAc,mCAAmC,CAAA;AACjD,cAAc,iCAAiC,CAAA;AAC/C,cAAc,oCAAoC,CAAA;AAClD,cAAc,iCAAiC,CAAA;AAC/C,cAAc,+BAA+B,CAAA;AAC7C,cAAc,+BAA+B,CAAA;AAC7C,cAAc,iCAAiC,CAAA;AAC/C,cAAc,iCAAiC,CAAA;AAC/C,cAAc,sBAAsB,CAAA;AACpC,cAAc,WAAW,CAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,WAAW,CAAA;AACzB,cAAc,UAAU,CAAA;AACxB,cAAc,OAAO,CAAA;AACrB,OAAO,EAAE,qBAAqB,EAAE,MAAM,WAAW,CAAA;AACjD,cAAc,eAAe,CAAA;AAC7B,cAAc,eAAe,CAAA;AAC7B,cAAc,qBAAqB,CAAA;AACnC,cAAc,4BAA4B,CAAA;AAC1C,OAAO,EAAE,gBAAgB,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAA;AACrE,cAAc,uBAAuB,CAAA;AACrC,cAAc,uCAAuC,CAAA;AACrD,cAAc,yCAAyC,CAAA;AACvD,cAAc,6BAA6B,CAAA;AAC3C,cAAc,mCAAmC,CAAA;AACjD,cAAc,mCAAmC,CAAA;AACjD,cAAc,iCAAiC,CAAA;AAC/C,cAAc,oCAAoC,CAAA;AAClD,cAAc,iCAAiC,CAAA;AAC/C,cAAc,+BAA+B,CAAA;AAC7C,cAAc,+BAA+B,CAAA;AAC7C,cAAc,iCAAiC,CAAA;AAC/C,cAAc,iCAAiC,CAAA;AAC/C,cAAc,sBAAsB,CAAA;AACpC,cAAc,WAAW,CAAA"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@delofarag/ai-utils",
-  "version": "1.4.2",
+  "version": "1.4.3",
   "type": "module",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",