@delofarag/ai-utils 1.0.2 → 1.1.0

package/dist/heart/agent.d.ts

@@ -0,0 +1,68 @@
+import { z } from "zod/v3";
+import { DynamicStructuredTool } from "../imports";
+import { BaseChatModel } from "../imports";
+import { BaseCheckpointSaver } from "../imports";
+import { VectorStore } from "../imports";
+interface AgentProps<T extends z.ZodObject<any, any>> {
+    prompt?: string | Array<string>;
+    tools: DynamicStructuredTool[];
+    llm: BaseChatModel;
+    schema?: T;
+    memory?: BaseCheckpointSaver;
+}
+/**
+ * CONSTRUCTOR:
+ * @example constructor({
+        prompt = `Du bist ein hilfreicher Assistent.
+            WICHTIG:
+            - Nutze Tools NUR wenn nötig
+            - Nach jedem Tool-Call gib eine finale Antwort
+            - Stoppe nach der Antwort, rufe keine Tools mehr auf
+            - Wenn du die Antwort hast, gib sie direkt zurück`,
+        tools,
+        llm,
+        schema,
+        memory
+    }: AgentProps<T>) {
+        this.prompt = typeof prompt === "string" ? [["system", prompt]] : Array.isArray(prompt) ? prompt.map((p:string)=>{
+            return ["system", p]
+        }) : []
+        this.tools = tools
+        this.llm = llm
+        this.schema = schema
+        this.memory = memory
+    }
+ */
+export declare class Agent<T extends z.ZodObject<any, any>> {
+    private prompt;
+    private tools;
+    private llm;
+    private schema;
+    private agent;
+    private memory;
+    private vectorStore;
+    private rag_tool;
+    private times_of_added_context;
+    private should_use_schema;
+    constructor({ prompt, tools, llm, schema, memory }: AgentProps<T>);
+    invoke(invokeInput: Record<string, any> & {
+        thread_id?: string;
+        debug?: boolean;
+    }): Promise<T extends undefined ? string : z.infer<T>>;
+    stream(invokeInput: Record<string, any> & {
+        thread_id?: string;
+        debug?: boolean;
+        stream_delay?: number;
+    }): AsyncGenerator<string, void, unknown>;
+    setContext(vectorStore: VectorStore, metadata?: {
+        name?: string;
+        description?: string;
+    }): void;
+    addContext(data: Array<any>): Promise<void>;
+    clearContext(): void;
+    hasContext(): boolean;
+    addTool(tool: DynamicStructuredTool): void;
+    get currentTools(): string[];
+}
+export {};
+//# sourceMappingURL=agent.d.ts.map

package/dist/heart/agent.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../../src/heart/agent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,QAAQ,CAAA;AAC1B,OAAO,EAAE,qBAAqB,EAAiC,MAAM,YAAY,CAAA;AACjF,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAA;AAC1C,OAAO,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAA;AAChD,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAA;AASxC,UAAU,UAAU,CAAC,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,GAAG,EAAC,GAAG,CAAC;IAC/C,MAAM,CAAC,EAAE,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,CAAA;IAC/B,KAAK,EAAE,qBAAqB,EAAE,CAAA;IAC9B,GAAG,EAAE,aAAa,CAAA;IAClB,MAAM,CAAC,EAAE,CAAC,CAAA;IACV,MAAM,CAAC,EAAE,mBAAmB,CAAA;CAC/B;AAED;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,qBAAa,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,GAAG,EAAC,GAAG,CAAC;IAC7C,OAAO,CAAC,MAAM,CAA2B;IACzC,OAAO,CAAC,KAAK,CAAyB;IACtC,OAAO,CAAC,GAAG,CAAe;IAC1B,OAAO,CAAC,MAAM,CAAe;IAC7B,OAAO,CAAC,KAAK,CAAK;IAClB,OAAO,CAAC,MAAM,CAAiC;IAC/C,OAAO,CAAC,WAAW,CAAyB;IAC5C,OAAO,CAAC,QAAQ,CAAmC;IACnD,OAAO,CAAC,sBAAsB,CAAY;IAC1C,OAAO,CAAC,iBAAiB,CAAgB;gBAE7B,EACR,MAKsD,EACtD,KAAK,EACL,GAAG,EACH,MAAM,EACN,MAAM,EACT,EAAE,UAAU,CAAC,CAAC,CAAC;IAUH,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG;QAAE,SAAS,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,OAAO,CAAA;KAAE,GAAG,OAAO,CAAC,CAAC,SAAS,SAAS,GAAG,MAAM,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAqCrI,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG;QAAE,SAAS,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,OAAO,CAAC;QAAC,YAAY,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,cAAc,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC;IAevJ,UAAU,CAAC,WAAW,EAAE,WAAW,EAAE,QAAQ,GAAE;QAAE,IAAI,CAAC,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAA;KAAO;IAgBrF,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC;IAUjC,YAAY;IAOZ,UAAU,IAAI,OAAO;IAIrB,OAAO,CAAC,IAAI,EAAC,qBAAqB;IAIzC,IAAW,YAAY,IAAI,MAAM,EAAE,CAMlC;CACJ"}

package/dist/heart/agent.js

@@ -0,0 +1,146 @@
+import { z } from "zod/v3";
+import { DynamicStructuredTool } from "../imports";
+import { turn_to_docs } from "../rag";
+import { createReactAgent } from "../imports";
+import { HumanMessage } from "../imports";
+import { structure, stream } from "../helpers";
+/**
+ * CONSTRUCTOR:
+ * @example constructor({
+        prompt = `Du bist ein hilfreicher Assistent.
+            WICHTIG:
+            - Nutze Tools NUR wenn nötig
+            - Nach jedem Tool-Call gib eine finale Antwort
+            - Stoppe nach der Antwort, rufe keine Tools mehr auf
+            - Wenn du die Antwort hast, gib sie direkt zurück`,
+        tools,
+        llm,
+        schema,
+        memory
+    }: AgentProps<T>) {
+        this.prompt = typeof prompt === "string" ? [["system", prompt]] : Array.isArray(prompt) ? prompt.map((p:string)=>{
+            return ["system", p]
+        }) : []
+        this.tools = tools
+        this.llm = llm
+        this.schema = schema
+        this.memory = memory
+    }
+ */
+export class Agent {
+    prompt;
+    tools;
+    llm;
+    schema;
+    agent;
+    memory;
+    vectorStore;
+    rag_tool;
+    times_of_added_context = 0;
+    should_use_schema = true;
+    constructor({ prompt = `Du bist ein hilfreicher Assistent. 
+            WICHTIG: 
+            - Nutze Tools NUR wenn nötig
+            - Nach jedem Tool-Call gib eine finale Antwort
+            - Stoppe nach der Antwort, rufe keine Tools mehr auf
+            - Wenn du die Antwort hast, gib sie direkt zurück`, tools, llm, schema, memory }) {
+        this.prompt = typeof prompt === "string" ? [["system", prompt]] : Array.isArray(prompt) ? prompt.map((p) => {
+            return ["system", p];
+        }) : [];
+        this.tools = tools;
+        this.llm = llm;
+        this.schema = schema;
+        this.memory = memory;
+    }
+    async invoke(invokeInput) {
+        const { thread_id, debug, ...variables } = invokeInput;
+        if (this.vectorStore && !thread_id)
+            throw new Error("thread_id is required when using a vector store, else no memory is stored");
+        if (!this.vectorStore && thread_id)
+            console.warn("thread_id is provided but no vector store is set, so no memory is stored");
+        const humanMessages = Object.entries(variables).map(([key, value]) => new HumanMessage(`${key}: ${typeof value === "object" ? JSON.stringify(value) : value}`));
+        // Tools für diesen invoke (inkl. RAG falls vorhanden)
+        const activeTools = this.rag_tool
+            ? [...this.tools, this.rag_tool]
+            : this.tools;
+        this.agent = createReactAgent({
+            llm: this.llm,
+            tools: activeTools,
+            checkpointSaver: this.memory,
+            prompt: (state) => [
+                ...this.prompt,
+                ...state.messages
+            ]
+        });
+        const config = thread_id && this.memory ? { configurable: { thread_id } } : undefined;
+        const result = await this.agent.invoke({ messages: humanMessages }, config);
+        if (debug)
+            return result;
+        const lastMessage = result.messages[result.messages.length - 1];
+        const content = lastMessage.content;
+        if (this.schema && this.should_use_schema) {
+            return await structure({ data: content, into: this.schema, llm: this.llm });
+        }
+        return content;
+    }
+    async *stream(invokeInput) {
+        this.should_use_schema = false;
+        try {
+            const { stream_delay = 1, ...rest } = invokeInput;
+            const response = await this.invoke(rest);
+            const responseStr = typeof response === "string" ? response : JSON.stringify(response);
+            const words = responseStr.split(" ");
+            for await (const word of stream(words, stream_delay)) {
+                yield word + " ";
+            }
+        }
+        finally {
+            this.should_use_schema = true;
+        }
+    }
+    setContext(vectorStore, metadata = {}) {
+        this.vectorStore = vectorStore;
+        this.rag_tool = new DynamicStructuredTool({
+            name: metadata.name ?? "search_context",
+            description: metadata.description ?? "Search the knowledge base for relevant information",
+            schema: z.object({
+                query: z.string().describe("The search query")
+            }),
+            func: async ({ query }) => {
+                const docs = await this.vectorStore?.similaritySearch(query, 3);
+                if (!docs || docs.length === 0)
+                    return "No relevant information found.";
+                return docs.map(doc => doc.pageContent).join("\n\n---\n\n");
+            }
+        });
+    }
+    async addContext(data) {
+        if (!this.vectorStore) {
+            throw new Error("Cant add context, no vector store set");
+        }
+        this.times_of_added_context++;
+        const docs = turn_to_docs(data);
+        await this.vectorStore.addDocuments(docs);
+        console.log(`Added context ${this.times_of_added_context} ${this.times_of_added_context === 1 ? "time" : "times"}`);
+    }
+    clearContext() {
+        this.rag_tool = undefined;
+        this.vectorStore = undefined;
+        this.times_of_added_context = 0;
+        console.log("Context cleared");
+    }
+    hasContext() {
+        return this.vectorStore !== undefined && this.rag_tool !== undefined;
+    }
+    addTool(tool) {
+        this.tools.push(tool);
+    }
+    get currentTools() {
+        const tools = [...this.tools];
+        if (this.rag_tool) {
+            tools.push(this.rag_tool);
+        }
+        return tools.map(tool => tool.name);
+    }
+}
+//# sourceMappingURL=agent.js.map

package/dist/heart/agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../src/heart/agent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,QAAQ,CAAA;AAC1B,OAAO,EAAE,qBAAqB,EAAiC,MAAM,YAAY,CAAA;AAIjF,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAA;AAErC,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAA;AAC7C,OAAO,EAAE,YAAY,EAA0B,MAAM,YAAY,CAAA;AACjE,OAAO,EAAE,SAAS,EAAC,MAAM,EAAE,MAAM,YAAY,CAAA;AAY7C;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,MAAM,OAAO,KAAK;IACN,MAAM,CAA2B;IACjC,KAAK,CAAyB;IAC9B,GAAG,CAAe;IAClB,MAAM,CAAe;IACrB,KAAK,CAAK;IACV,MAAM,CAAiC;IACvC,WAAW,CAAyB;IACpC,QAAQ,CAAmC;IAC3C,sBAAsB,GAAW,CAAC,CAAA;IAClC,iBAAiB,GAAY,IAAI,CAAA;IAEzC,YAAY,EACR,MAAM,GAAG;;;;;8DAK6C,EACtD,KAAK,EACL,GAAG,EACH,MAAM,EACN,MAAM,EACM;QACZ,IAAI,CAAC,MAAM,GAAG,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAQ,EAAC,EAAE;YAC7G,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAA;QACxB,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAA;QACP,IAAI,CAAC,KAAK,GAAG,KAAK,CAAA;QAClB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAA;QACd,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;QACpB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;IACxB,CAAC;IAEM,KAAK,CAAC,MAAM,CAAC,WAA0E;QAC1F,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,SAAS,EAAE,GAAG,WAAW,CAAA;QAEtD,IAAG,IAAI,CAAC,WAAW,IAAI,CAAC,SAAS;YAAE,MAAM,IAAI,KAAK,CAAC,2EAA2E,CAAC,CAAA;QAC/H,IAAG,CAAC,IAAI,CAAC,WAAW,IAAI,SAAS;YAAE,OAAO,CAAC,IAAI,CAAC,0EAA0E,CAAC,CAAA;QAE3H,MAAM,aAAa,GAAmB,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,GAAG,CAC/D,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,IAAI,YAAY,CAAC,GAAG,GAAG,KAAK,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAC7G,CAAA;QAED,sDAAsD;QACtD,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ;YAC7B,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC;YAChC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAA;QAEhB,IAAI,CAAC,KAAK,GAAG,gBAAgB,CAAC;YAC1B,GAAG,EAAE,IAAI,CAAC,GAAU;YACpB,KAAK,EAAE,WAAkB;YACzB,eAAe,EAAE,IAAI,CAAC,MAAa;YACnC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC;gBACf,GAAG,IAAI,CAAC,MAAM;gBACd,GAAG,KAAK,CAAC,QAAQ;aACpB;SACJ,CAAC,CAAA;QAEF,MAAM,MAAM,GAAG,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAA;QACrF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,aAAa,EAAS,EAAE,MAAM,CAAC,CAAA;QAClF,IAAG,KAAK;YAAE,OAAO,MAAM,CAAA;QACvB,MAAM,WAAW,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAA;QAC/D,MAAM,OAAO,GAAG,WAAW,CAAC,OAAO,CAAA;QAEnC,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACxC,OAAO,MAAM,SAAS,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,CAAQ,CAAA;QACtF,CAAC;QACD,OAAO,OAAO,CAAA;IAClB,CAAC;IAEM,KAAK,CAAC,CAAC,MAAM,CAAC,WAAiG;QAClH,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAA;QAC9B,IAAG,CAAC;YACA,MAAM,EAAE,YAAY,GAAG,CAAC,EAAE,GAAG,IAAI,EAAE,GAAG,WAAW,CAAA;YACjD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;YACxC,MAAM,WAAW,GAAG,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAA;YACtF,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;YACpC,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,EAAC,YAAY,CAAC,EAAC,CAAC;gBACjD,MAAM,IAAI,GAAG,GAAG,CAAA;YACpB,CAAC;QACL,CAAC;gBAAS,CAAC;YACP,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAA;QACjC,CAAC;IACL,CAAC;IAEM,UAAU,CAAC,WAAwB,EAAE,WAAoD,EAAE;QAC9F,IAAI,CAAC,WAAW,GAAG,WAAW,CAAA;QAC9B,IAAI,CAAC,QAAQ,GAAG,IAAI,qBAAqB,CAAC;YACtC,IAAI,EAAE,QAAQ,CAAC,IAAI,IAAI,gBAAgB;YACvC,WAAW,EAAE,QAAQ,CAAC,WAAW,IAAI,oDAAoD;YACzF,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;gBACb,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;aACjD,CAAC;YACF,IAAI,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;gBACtB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,WAAW,EAAE,gBAAgB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAA;gBAC/D,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;oBAAE,OAAO,gCAAgC,CAAA;gBACvE,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;YAC/D,CAAC;SACJ,CAAC,CAAA;IACN,CAAC;IAEM,KAAK,CAAC,UAAU,CAAC,IAAgB;QACpC,IAAG,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAA;QAC5D,CAAC;QACD,IAAI,CAAC,sBAAsB,EAAE,CAAA;QAC7B,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,CAAC,CAAA;QAC/B,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;QACzC,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,CAAC,sBAAsB,IAAI,IAAI,CAAC,sBAAsB,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAA;IACvH,CAAC;IAEM,YAAY;QACf,IAAI,CAAC,QAAQ,GAAG,SAAS,CAAA;QACzB,IAAI,CAAC,WAAW,GAAG,SAAS,CAAA;QAC5B,IAAI,CAAC,sBAAsB,GAAG,CAAC,CAAA;QAC/B,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAA;IAClC,CAAC;IAEM,UAAU;QACb,OAAO,IAAI,CAAC,WAAW,KAAK,SAAS,IAAI,IAAI,CAAC,QAAQ,KAAK,SAAS,CAAA;IACxE,CAAC;IAEM,OAAO,CAAC,IAA0B;QACrC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACzB,CAAC;IAED,IAAW,YAAY;QACnB,MAAM,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAA;QAC7B,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAChB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAC7B,CAAC;QACD,OAAO,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACvC,CAAC;CACJ"}

package/dist/heart/chain.d.ts

@@ -0,0 +1,50 @@
+import { BaseChatModel, MessagesPlaceholder, VectorStore } from "../imports";
+import { z } from "zod/v3";
+export declare const DEFAULT_SCHEMA: z.ZodObject<{
+    output: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    output: string;
+}, {
+    output: string;
+}>;
+interface ChainProps<T extends z.ZodObject<any, any>> {
+    prompt?: string | Array<string | MessagesPlaceholder<any>>;
+    llm: BaseChatModel;
+    schema?: T;
+}
+/**
+ * CONSTRUCTOR:
+ * @example constructor({prompt = "du bist ein hilfreicher Assistent",llm,schema}:ChainProps<T>){
+        this.prompt = typeof prompt === "string" ? [["system", prompt]] : Array.isArray(prompt) ? prompt.map((p:string | MessagesPlaceholder<any>)=>{
+            if(typeof p === "string"){
+                return ["system", p]
+            } else {
+                return p
+            }
+        }) : []
+        this.llm = llm
+        this.schema = (schema ?? DEFAULT_SCHEMA) as unknown as T
+        this.parser = StructuredOutputParser.fromZodSchema(this.schema)
+    }
+ */
+export declare class Chain<T extends z.ZodObject<any, any> = typeof DEFAULT_SCHEMA> {
+    private prompt;
+    private vectorStore;
+    private times_of_added_context;
+    private parser;
+    private llm;
+    private schema;
+    constructor({ prompt, llm, schema }: ChainProps<T>);
+    invoke(input: Record<string, any> & {
+        debug?: boolean;
+    }): Promise<z.infer<T>>;
+    stream(input: Record<string, any> & {
+        debug?: boolean;
+    }): AsyncGenerator<string, void, unknown>;
+    /** Fügt RAG-Kontext hinzu. Docs werden EINMAL zum VectorStore hinzugefügt. */
+    addContext(data: Array<any>): Promise<void>;
+    setContext(vectorStore: VectorStore): Promise<void>;
+    clearContext(): void;
+}
+export {};
+//# sourceMappingURL=chain.d.ts.map

package/dist/heart/chain.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chain.d.ts","sourceRoot":"","sources":["../../src/heart/chain.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,aAAa,EAGb,mBAAmB,EACnB,WAAW,EAGd,MAAM,YAAY,CAAA;AAEnB,OAAO,EAAE,CAAC,EAAE,MAAM,QAAQ,CAAA;AAE1B,eAAO,MAAM,cAAc;;;;;;EAEzB,CAAA;AAEF,UAAU,UAAU,CAAC,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,GAAG,EAAC,GAAG,CAAC;IAC/C,MAAM,CAAC,EAAC,MAAM,GAAG,KAAK,CAAC,MAAM,GAAG,mBAAmB,CAAC,GAAG,CAAC,CAAC,CAAA;IACzD,GAAG,EAAC,aAAa,CAAA;IACjB,MAAM,CAAC,EAAC,CAAC,CAAA;CACZ;AAED;;;;;;;;;;;;;;GAcG;AACH,qBAAa,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,GAAG,EAAC,GAAG,CAAC,GAAG,OAAO,cAAc;IACrE,OAAO,CAAC,MAAM,CAA+D;IAC7E,OAAO,CAAC,WAAW,CAAyB;IAC5C,OAAO,CAAC,sBAAsB,CAAY;IAC1C,OAAO,CAAC,MAAM,CAA0B;IACxC,OAAO,CAAC,GAAG,CAAc;IACzB,OAAO,CAAC,MAAM,CAAE;gBAEJ,EAAC,MAA4C,EAAC,GAAG,EAAC,MAAM,EAAC,EAAC,UAAU,CAAC,CAAC,CAAC;IAatE,MAAM,CAAC,KAAK,EAAC,MAAM,CAAC,MAAM,EAAC,GAAG,CAAC,GAAG;QAAC,KAAK,CAAC,EAAE,OAAO,CAAA;KAAC,GAAE,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IA2CvE,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG;QAAE,KAAK,CAAC,EAAE,OAAO,CAAA;KAAE,GAAG,cAAc,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC;IAuD7G,8EAA8E;IACjE,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC;IAU3B,UAAU,CAAC,WAAW,EAAE,WAAW;IAKzC,YAAY;CAKtB"}

package/dist/heart/chain.js

@@ -0,0 +1,165 @@
+import { StructuredOutputParser, ChatPromptTemplate, createRetrievalChain, createStuffDocumentsChain, } from "../imports";
+import { turn_to_docs } from "../rag";
+import { z } from "zod/v3";
+export const DEFAULT_SCHEMA = z.object({
+    output: z.string().describe("Dein Output zur anfrage des Users")
+});
+/**
+ * CONSTRUCTOR:
+ * @example constructor({prompt = "du bist ein hilfreicher Assistent",llm,schema}:ChainProps<T>){
+        this.prompt = typeof prompt === "string" ? [["system", prompt]] : Array.isArray(prompt) ? prompt.map((p:string | MessagesPlaceholder<any>)=>{
+            if(typeof p === "string"){
+                return ["system", p]
+            } else {
+                return p
+            }
+        }) : []
+        this.llm = llm
+        this.schema = (schema ?? DEFAULT_SCHEMA) as unknown as T
+        this.parser = StructuredOutputParser.fromZodSchema(this.schema)
+    }
+ */
+export class Chain {
+    prompt;
+    vectorStore;
+    times_of_added_context = 0;
+    parser;
+    llm;
+    schema;
+    constructor({ prompt = "du bist ein hilfreicher Assistent", llm, schema }) {
+        this.prompt = typeof prompt === "string" ? [["system", prompt]] : Array.isArray(prompt) ? prompt.map((p) => {
+            if (typeof p === "string") {
+                return ["system", p];
+            }
+            else {
+                return p;
+            }
+        }) : [];
+        this.llm = llm;
+        this.schema = (schema ?? DEFAULT_SCHEMA);
+        this.parser = StructuredOutputParser.fromZodSchema(this.schema);
+    }
+    async invoke(input) {
+        const messagesArray = [...this.prompt];
+        messagesArray.push(["system", "You MUST respond ONLY with valid JSON matching this exact schema:\n{format_instructions}\n\nIMPORTANT: \n- Output ONLY valid JSON, no markdown code blocks\n- No backslashes or line breaks in strings\n- All strings must be on single lines\n- Do NOT wrap in ```json``` blocks\n- Return the JSON object DIRECTLY"]);
+        if (this.vectorStore)
+            messagesArray.push(["system", "Hier ist relevanter Kontext:\n{context}"]);
+        for (const key in input) {
+            if (key === "debug")
+                continue;
+            if (key === "thread_id") {
+                console.error("eine normale chain hat keine memory, deswegen wird thread_id ignoriert");
+                continue;
+            }
+            if (typeof input[key] !== "string") {
+                messagesArray.push(["human", `${key}:${JSON.stringify(input[key])}`]);
+            }
+            else {
+                messagesArray.push(["human", `${key}:{${key}}`]);
+            }
+        }
+        const true_prompt = ChatPromptTemplate.fromMessages(messagesArray);
+        if (input.debug)
+            console.log("Prompt: ", true_prompt);
+        const invokeInput = { ...input, format_instructions: this.parser.getFormatInstructions() };
+        if (this.vectorStore) {
+            const retriever = this.vectorStore.asRetriever();
+            const stuff_chain = await createStuffDocumentsChain({
+                llm: this.llm,
+                prompt: true_prompt,
+                outputParser: this.parser
+            });
+            const chain = await createRetrievalChain({
+                combineDocsChain: stuff_chain,
+                retriever: retriever
+            });
+            if (input.debug)
+                console.log("created retrieval chain");
+            const respo = await chain.invoke({ input: JSON.stringify(input), ...invokeInput });
+            return this.schema.parse(respo.answer);
+        }
+        const chain = true_prompt.pipe(this.llm).pipe(this.parser);
+        if (input.debug)
+            console.log("created normal chain");
+        const respo = await chain.invoke(invokeInput);
+        return this.schema.parse(respo);
+    }
+    async *stream(input) {
+        const messagesArray = [...this.prompt];
+        // Beim Streamen KEIN Schema-Prompt - nur reiner Text
+        if (this.vectorStore)
+            messagesArray.push(["system", "Hier ist relevanter Kontext:\n{context}"]);
+        for (const key in input) {
+            if (key === "debug")
+                continue;
+            if (key === "thread_id") {
+                console.error("eine normale chain hat keine memory, deswegen wird thread_id ignoriert");
+                continue;
+            }
+            if (typeof input[key] !== "string") {
+                messagesArray.push(["human", `${key}:${JSON.stringify(input[key])}`]);
+            }
+            else {
+                messagesArray.push(["human", `${key}:{${key}}`]);
+            }
+        }
+        const true_prompt = ChatPromptTemplate.fromMessages(messagesArray);
+        if (input.debug)
+            console.log("Prompt: ", true_prompt);
+        const invokeInput = { ...input };
+        if (this.vectorStore) {
+            const retriever = this.vectorStore.asRetriever();
+            if (input.debug)
+                console.log("created retrieval chain (streaming)");
+            // Für RAG: Hole Context und stream dann die LLM-Antwort
+            const contextDocs = await retriever.invoke(JSON.stringify(input));
+            const contextText = contextDocs.map((doc) => doc.pageContent).join("\n\n");
+            // Stream die LLM-Antwort mit Context
+            const streamChain = true_prompt.pipe(this.llm);
+            const stream = await streamChain.stream({ ...invokeInput, context: contextText });
+            for await (const chunk of stream) {
+                if (chunk && typeof chunk === 'object' && 'content' in chunk) {
+                    yield chunk.content;
+                }
+                else if (typeof chunk === 'string') {
+                    yield chunk;
+                }
+            }
+        }
+        else {
+            // Ohne RAG: Stream direkt
+            const streamChain = true_prompt.pipe(this.llm);
+            if (input.debug)
+                console.log("created normal chain (streaming)");
+            const stream = await streamChain.stream(invokeInput);
+            for await (const chunk of stream) {
+                if (chunk && typeof chunk === 'object' && 'content' in chunk) {
+                    yield chunk.content;
+                }
+                else if (typeof chunk === 'string') {
+                    yield chunk;
+                }
+            }
+        }
+    }
+    /** Fügt RAG-Kontext hinzu. Docs werden EINMAL zum VectorStore hinzugefügt. */
+    async addContext(data) {
+        if (!this.vectorStore) {
+            throw new Error("Cant add context, no vector store set");
+        }
+        this.times_of_added_context++;
+        const docs = turn_to_docs(data);
+        await this.vectorStore.addDocuments(docs);
+        console.log(`Added context ${this.times_of_added_context} ${this.times_of_added_context === 1 ? "time" : "times"}`);
+    }
+    async setContext(vectorStore) {
+        console.log("Setting context");
+        this.vectorStore = vectorStore;
+    }
+    clearContext() {
+        this.vectorStore = undefined;
+        this.times_of_added_context = 0;
+        console.log("Context cleared");
+    }
+}
+//# sourceMappingURL=chain.js.map

package/dist/heart/chain.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chain.js","sourceRoot":"","sources":["../../src/heart/chain.ts"],"names":[],"mappings":"AAAA,OAAO,EAEH,sBAAsB,EACtB,kBAAkB,EAGlB,oBAAoB,EACpB,yBAAyB,GAC5B,MAAM,YAAY,CAAA;AACnB,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAA;AACrC,OAAO,EAAE,CAAC,EAAE,MAAM,QAAQ,CAAA;AAE1B,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,CAAC;IACnC,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,mCAAmC,CAAC;CACnE,CAAC,CAAA;AAQF;;;;;;;;;;;;;;GAcG;AACH,MAAM,OAAO,KAAK;IACN,MAAM,CAA+D;IACrE,WAAW,CAAyB;IACpC,sBAAsB,GAAW,CAAC,CAAA;IAClC,MAAM,CAA0B;IAChC,GAAG,CAAc;IACjB,MAAM,CAAE;IAEhB,YAAY,EAAC,MAAM,GAAG,mCAAmC,EAAC,GAAG,EAAC,MAAM,EAAe;QAC/E,IAAI,CAAC,MAAM,GAAG,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAmC,EAAC,EAAE;YACxI,IAAG,OAAO,CAAC,KAAK,QAAQ,EAAC,CAAC;gBACtB,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAA;YACxB,CAAC;iBAAM,CAAC;gBACJ,OAAO,CAAC,CAAA;YACZ,CAAC;QACL,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAA;QACP,IAAI,CAAC,GAAG,GAAG,GAAG,CAAA;QACd,IAAI,CAAC,MAAM,GAAG,CAAC,MAAM,IAAI,cAAc,CAAiB,CAAA;QACxD,IAAI,CAAC,MAAM,GAAG,sBAAsB,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;IACnE,CAAC;IAEM,KAAK,CAAC,MAAM,CAAC,KAA4C;QAC5D,MAAM,aAAa,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,CAAA;QACtC,aAAa,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,sTAAsT,CAAC,CAAC,CAAA;QACtV,IAAG,IAAI,CAAC,WAAW;YAAE,aAAa,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,yCAAyC,CAAC,CAAC,CAAA;QAC9F,KAAI,MAAM,GAAG,IAAI,KAAK,EAAC,CAAC;YACpB,IAAG,GAAG,KAAK,OAAO;gBAAE,SAAQ;YAC5B,IAAG,GAAG,KAAK,WAAW,EAAC,CAAC;gBACpB,OAAO,CAAC,KAAK,CAAC,wEAAwE,CAAC,CAAA;gBACvF,SAAQ;YACZ,CAAC;YACD,IAAG,OAAO,KAAK,CAAC,GAAG,CAAC,KAAK,QAAQ,EAAC,CAAC;gBAC/B,aAAa,CAAC,IAAI,CAAC,CAAC,OAAO,EAAC,GAAG,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;YACxE,CAAC;iBAAM,CAAC;gBACJ,aAAa,CAAC,IAAI,CAAC,CAAC,OAAO,EAAC,GAAG,GAAG,KAAK,GAAG,GAAG,CAAC,CAAC,CAAA;YACnD,CAAC;QACL,CAAC;QACD,MAAM,WAAW,GAAG,kBAAkB,CAAC,YAAY,CAAC,aAAa,CAAC,CAAA;QAClE,IAAG,KAAK,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,UAAU,EAAE,WAAW,CAAC,CAAA;QACpD,MAAM,WAAW,GAAG,EAAC,GAAG,KAAK,EAAE,mBAAmB,EAAE,IAAI,CAAC,MAAM,CAAC,qBAAqB,EAAE,EAAC,CAAA;QAExF,IAAG,IAAI,CAAC,WAAW,EAAC,CAAC;YACjB,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,CAAA;YAChD,MAAM,WAAW,GAAG,MAAM,yBAAyB,CAAC;gBAChD,GAAG,EAAE,IAAI,CAAC,GAAG;gBACb,MAAM,EAAE,WAAW;gBACnB,YAAY,EAAE,IAAI,CAAC,MAAM;aAC5B,CAAC,CAAA;YACF,MAAM,KAAK,GAAG,MAAM,oBAAoB,CAAC;gBACrC,gBAAgB,EAAE,WAAW;gBAC7B,SAAS,EAAE,SAAS;aACvB,CAAC,CAAA;YACF,IAAI,KAAK,CAAC,KAAK;gBAAE,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAA;YACvD,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,EAAC,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,GAAG,WAAW,EAAC,CAAC,CAAA;YAChF,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;QAC1C,CAAC;QAED,MAAM,KAAK,GAAG,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;QAC1D,IAAI,KAAK,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAA;QACpD,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,CAAA;QAC7C,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;IACnC,CAAC;IAGM,KAAK,CAAC,CAAC,MAAM,CAAC,KAAgD;QACjE,MAAM,aAAa,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,CAAA;QACtC,qDAAqD;QACrD,IAAG,IAAI,CAAC,WAAW;YAAE,aAAa,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,yCAAyC,CAAC,CAAC,CAAA;QAC1F,KAAI,MAAM,GAAG,IAAI,KAAK,EAAC,CAAC;YACpB,IAAG,GAAG,KAAK,OAAO;gBAAE,SAAQ;YAC5B,IAAG,GAAG,KAAK,WAAW,EAAC,CAAC;gBACpB,OAAO,CAAC,KAAK,CAAC,wEAAwE,CAAC,CAAA;gBACvF,SAAQ;YACZ,CAAC;YACD,IAAG,OAAO,KAAK,CAAC,GAAG,CAAC,KAAK,QAAQ,EAAC,CAAC;gBAC/B,aAAa,CAAC,IAAI,CAAC,CAAC,OAAO,EAAC,GAAG,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;YACxE,CAAC;iBAAM,CAAC;gBACJ,aAAa,CAAC,IAAI,CAAC,CAAC,OAAO,EAAC,GAAG,GAAG,KAAK,GAAG,GAAG,CAAC,CAAC,CAAA;YACnD,CAAC;QACL,CAAC;QACL,MAAM,WAAW,GAAG,kBAAkB,CAAC,YAAY,CAAC,aAAa,CAAC,CAAA;QAClE,IAAG,KAAK,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,UAAU,EAAE,WAAW,CAAC,CAAA;QACpD,MAAM,WAAW,GAAG,EAAC,GAAG,KAAK,EAAC,CAAA;QAE9B,IAAG,IAAI,CAAC,WAAW,EAAC,CAAC;YACjB,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,CAAA;YAChD,IAAG,KAAK,CAAC,KAAK;gBAAE,OAAO,CAAC,GAAG,CAAC,qCAAqC,CAAC,CAAA;YAElE,wDAAwD;YACxD,MAAM,WAAW,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAA;YACjE,MAAM,WAAW,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,GAAQ,EAAE,EAAE,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;YAE/E,qCAAqC;YACrC,MAAM,WAAW,GAAG,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;YAC9C,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,EAAC,GAAG,WAAW,EAAE,OAAO,EAAE,WAAW,EAAC,CAAC,CAAA;YAE/E,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBAC/B,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,SAAS,IAAI,KAAK,EAAE,CAAC;oBAC3D,MAAM,KAAK,CAAC,OAAiB,CAAA;gBACjC,CAAC;qBAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;oBACnC,MAAM,KAAK,CAAA;gBACf,CAAC;YACL,CAAC;QACL,CAAC;aAAM,CAAC;YACJ,0BAA0B;YAC1B,MAAM,WAAW,GAAG,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;YAC9C,IAAG,KAAK,CAAC,KAAK;gBAAE,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAA;YAE/D,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,WAAW,CAAC,CAAA;YACpD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBAC/B,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,SAAS,IAAI,KAAK,EAAE,CAAC;oBAC3D,MAAM,KAAK,CAAC,OAAiB,CAAA;gBACjC,CAAC;qBAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;oBACnC,MAAM,KAAK,CAAA;gBACf,CAAC;YACL,CAAC;QACL,CAAC;IACL,CAAC;IAED,8EAA8E;IACvE,KAAK,CAAC,UAAU,CAAC,IAAgB;QACpC,IAAG,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAA;QAC5D,CAAC;QACD,IAAI,CAAC,sBAAsB,EAAE,CAAA;QAC7B,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,CAAC,CAAA;QAC/B,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;QACzC,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,CAAC,sBAAsB,IAAI,IAAI,CAAC,sBAAsB,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAA;IACvH,CAAC;IAEM,KAAK,CAAC,UAAU,CAAC,WAAwB;QAC5C,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAA;QAC9B,IAAI,CAAC,WAAW,GAAG,WAAW,CAAA;IAClC,CAAC;IAEM,YAAY;QACf,IAAI,CAAC,WAAW,GAAG,SAAS,CAAA;QAC5B,IAAI,CAAC,sBAAsB,GAAG,CAAC,CAAA;QAC/B,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAA;IAClC,CAAC;CACJ"}

package/dist/heart/chatbot.d.ts

@@ -0,0 +1,45 @@
+import { BaseCheckpointSaver, VectorStore, DynamicStructuredTool } from "../imports";
+import { BaseChatModel } from "../imports";
+type ChatbotProps = {
+    llm: BaseChatModel;
+    prompt?: string | Array<string>;
+    tools?: DynamicStructuredTool[];
+    memory?: BaseCheckpointSaver;
+};
+/**
+ * CONSTRUCTOR:
+ * @example constructor({llm,prompt = "Du bist ein hilfreicher chatbot der mit dem User ein höffliches und hilfreiches Gespräch führt",tools,memory}:ChatbotProps){
+        if(tools){
+            this.agent = new Agent({
+                memory: memory ?? new SmartCheckpointSaver(new MemorySaver(),{ llm }),
+                tools : tools,
+                prompt: prompt,
+                llm: llm
+            })
+        } else {
+            this.chain = new MemoryChain({
+                memory: memory ?? new SmartCheckpointSaver(new MemorySaver(),{ llm }),
+                prompt: prompt,
+                llm: llm
+            })
+        }
+ */
+export declare class Chatbot {
+    private chain;
+    private agent;
+    constructor({ llm, prompt, tools, memory }: ChatbotProps);
+    chat({ input, thread_id }: {
+        input: string;
+        thread_id: string;
+    }): AsyncGenerator<string, string, unknown>;
+    session({ breakword, numberOfMessages, id }?: {
+        breakword?: string;
+        numberOfMessages?: number;
+        id?: string;
+    }): Promise<void>;
+    addContext(data: Array<any>): Promise<void>;
+    setContext(vectorStore: VectorStore): Promise<void>;
+    clearContext(): Promise<void>;
+}
+export {};
+//# sourceMappingURL=chatbot.d.ts.map

package/dist/heart/chatbot.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatbot.d.ts","sourceRoot":"","sources":["../../src/heart/chatbot.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAqF,WAAW,EAAE,qBAAqB,EAAuB,MAAM,YAAY,CAAA;AAC5L,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAA;AAQ1C,KAAK,YAAY,GAAG;IAChB,GAAG,EAAE,aAAa,CAAA;IAClB,MAAM,CAAC,EAAE,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,CAAA;IAC/B,KAAK,CAAC,EAAE,qBAAqB,EAAE,CAAA;IAC/B,MAAM,CAAC,EAAE,mBAAmB,CAAA;CAC/B,CAAA;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,qBAAa,OAAO;IAChB,OAAO,CAAC,KAAK,CAAyB;IACtC,OAAO,CAAC,KAAK,CAAwB;gBAEzB,EAAC,GAAG,EAAC,MAAyG,EAAC,KAAK,EAAC,MAAM,EAAC,EAAC,YAAY;IAiBvI,IAAI,CAAC,EAAC,KAAK,EAAC,SAAS,EAAC,EAAC;QAAC,KAAK,EAAC,MAAM,CAAC;QAAC,SAAS,EAAC,MAAM,CAAA;KAAC,GAAG,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC;IAWlG,OAAO,CAAC,EACjB,SAAkB,EAClB,gBAA2C,EAC3C,EAAoB,EACvB,GAAC;QACE,SAAS,CAAC,EAAC,MAAM,CAAC;QAClB,gBAAgB,CAAC,EAAC,MAAM,CAAC;QACzB,EAAE,CAAC,EAAC,MAAM,CAAA;KACR;IA2BO,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC;IAU3B,UAAU,CAAC,WAAW,EAAE,WAAW;IAUnC,YAAY;CAS5B"}

package/dist/heart/chatbot.js

@@ -0,0 +1,117 @@
+import { SmartCheckpointSaver } from "../memory";
+import { MemorySaver } from "../imports";
+import { logChunk } from "../helpers";
+import { input } from "@delofarag/base-utils";
+import { MemoryChain } from "./memorychain";
+import { Agent } from "./agent";
+/**
+ * CONSTRUCTOR:
+ * @example constructor({llm,prompt = "Du bist ein hilfreicher chatbot der mit dem User ein höffliches und hilfreiches Gespräch führt",tools,memory}:ChatbotProps){
+        if(tools){
+            this.agent = new Agent({
+                memory: memory ?? new SmartCheckpointSaver(new MemorySaver(),{ llm }),
+                tools : tools,
+                prompt: prompt,
+                llm: llm
+            })
+        } else {
+            this.chain = new MemoryChain({
+                memory: memory ?? new SmartCheckpointSaver(new MemorySaver(),{ llm }),
+                prompt: prompt,
+                llm: llm
+            })
+        }
+ */
+export class Chatbot {
+    chain;
+    agent;
+    constructor({ llm, prompt = "Du bist ein hilfreicher chatbot der mit dem User ein höffliches und hilfreiches Gespräch führt", tools, memory }) {
+        if (tools) {
+            this.agent = new Agent({
+                memory: memory ?? new SmartCheckpointSaver(new MemorySaver(), { llm }),
+                tools: tools,
+                prompt: prompt,
+                llm: llm
+            });
+        }
+        else {
+            this.chain = new MemoryChain({
+                memory: memory ?? new SmartCheckpointSaver(new MemorySaver(), { llm }),
+                prompt: prompt,
+                llm: llm
+            });
+        }
+    }
+    async *chat({ input, thread_id }) {
+        const streamable = this.chain ? this.chain : this.agent;
+        if (!streamable)
+            throw new Error("kein streamable!");
+        let fullResponse = "";
+        for await (const chunk of streamable.stream({ input, thread_id })) {
+            fullResponse += chunk;
+            yield chunk;
+        }
+        return fullResponse;
+    }
+    async session({ 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 = this.chat({
+                    input: message,
+                    thread_id: id,
+                });
+                console.log("Assistant: ");
+                for await (const chunk of response) {
+                    logChunk(chunk);
+                }
+                console.log(""); // Zeilenumbruch nach dem Streamen, damit Output nicht überschrieben wird
+            }
+            catch (e) {
+                console.error("Error: ", e);
+            }
+            messages = messages + 2;
+            if (messages > numberOfMessages) {
+                break;
+            }
+        }
+    }
+    async addContext(data) {
+        if (this.chain) {
+            await this.chain.addContext(data);
+        }
+        else if (this.agent) {
+            await this.agent.addContext(data);
+        }
+        else {
+            throw Error("weder agent noch chain, kein addContext möglich");
+        }
+    }
+    async setContext(vectorStore) {
+        if (this.chain) {
+            this.chain.setContext(vectorStore);
+        }
+        else if (this.agent) {
+            this.agent.setContext(vectorStore);
+        }
+        else {
+            throw Error("weder agent noch chain, kein setContext möglich");
+        }
+    }
+    async clearContext() {
+        if (this.chain) {
+            this.chain.clearContext();
+        }
+        else if (this.agent) {
+            this.agent.clearContext();
+        }
+        else {
+            throw Error("weder agent noch chain, kein clearContext möglich");
+        }
+    }
+}
+//# sourceMappingURL=chatbot.js.map

package/dist/heart/chatbot.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatbot.js","sourceRoot":"","sources":["../../src/heart/chatbot.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,oBAAoB,EAAE,MAAM,WAAW,CAAA;AAChD,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAA;AACxC,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAA;AACrC,OAAO,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAA;AAC7C,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAA;AAC3C,OAAO,EAAE,KAAK,EAAE,MAAM,SAAS,CAAA;AAS/B;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,OAAO,OAAO;IACR,KAAK,CAAyB;IAC9B,KAAK,CAAwB;IAErC,YAAY,EAAC,GAAG,EAAC,MAAM,GAAG,gGAAgG,EAAC,KAAK,EAAC,MAAM,EAAc;QACjJ,IAAG,KAAK,EAAC,CAAC;YACN,IAAI,CAAC,KAAK,GAAG,IAAI,KAAK,CAAC;gBACnB,MAAM,EAAE,MAAM,IAAI,IAAI,oBAAoB,CAAC,IAAI,WAAW,EAAE,EAAC,EAAE,GAAG,EAAE,CAAC;gBACrE,KAAK,EAAG,KAAK;gBACb,MAAM,EAAE,MAAM;gBACd,GAAG,EAAE,GAAG;aACX,CAAC,CAAA;QACN,CAAC;aAAM,CAAC;YACJ,IAAI,CAAC,KAAK,GAAG,IAAI,WAAW,CAAC;gBACzB,MAAM,EAAE,MAAM,IAAI,IAAI,oBAAoB,CAAC,IAAI,WAAW,EAAE,EAAC,EAAE,GAAG,EAAE,CAAC;gBACrE,MAAM,EAAE,MAAM;gBACd,GAAG,EAAE,GAAG;aACX,CAAC,CAAA;QACN,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,CAAC,IAAI,CAAC,EAAC,KAAK,EAAC,SAAS,EAAkC;QACjE,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAA;QACvD,IAAI,CAAC,UAAU;YAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAA;QACpD,IAAI,YAAY,GAAG,EAAE,CAAA;QACrB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,UAAU,CAAC,MAAM,CAAC,EAAC,KAAK,EAAC,SAAS,EAAC,CAAC,EAAC,CAAC;YAC5D,YAAY,IAAI,KAAK,CAAA;YACrB,MAAM,KAAK,CAAA;QACf,CAAC;QACD,OAAO,YAAY,CAAA;IACvB,CAAC;IAEM,KAAK,CAAC,OAAO,CAAC,EACjB,SAAS,GAAG,MAAM,EAClB,gBAAgB,GAAG,MAAM,CAAC,iBAAiB,EAC3C,EAAE,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,EAAE,KAKpB,EAAE;QACF,IAAI,QAAQ,GAAG,CAAC,CAAA;QAChB,OAAM,IAAI,EAAC,CAAC;YACR,IAAG,CAAC;gBACA,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,CAAA;gBACpC,IAAG,OAAO,KAAK,SAAS,EAAC,CAAC;oBACtB,MAAK;gBACT,CAAC;gBACD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC;oBACvB,KAAK,EAAE,OAAO;oBACd,SAAS,EAAE,EAAE;iBAChB,CAAC,CAAA;gBACF,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAA;gBAC1B,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;oBACjC,QAAQ,CAAC,KAAK,CAAC,CAAA;gBACnB,CAAC;gBACD,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA,CAAC,yEAAyE;YAC7F,CAAC;YAAC,OAAM,CAAC,EAAC,CAAC;gBACP,OAAO,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,CAAA;YAC/B,CAAC;YACD,QAAQ,GAAG,QAAQ,GAAG,CAAC,CAAA;YACvB,IAAG,QAAQ,GAAG,gBAAgB,EAAC,CAAC;gBAC5B,MAAK;YACT,CAAC;QACL,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,UAAU,CAAC,IAAgB;QACpC,IAAI,IAAI,CAAC,KAAK,EAAC,CAAC;YACZ,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAA;QACrC,CAAC;aAAM,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACpB,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAA;QACrC,CAAC;aAAM,CAAC;YACJ,MAAM,KAAK,CAAC,iDAAiD,CAAC,CAAA;QAClE,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,UAAU,CAAC,WAAwB;QAC5C,IAAI,IAAI,CAAC,KAAK,EAAC,CAAC;YACZ,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,CAAA;QACtC,CAAC;aAAM,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACpB,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,CAAA;QACtC,CAAC;aAAM,CAAC;YACJ,MAAM,KAAK,CAAC,iDAAiD,CAAC,CAAA;QAClE,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,YAAY;QACrB,IAAI,IAAI,CAAC,KAAK,EAAC,CAAC;YACZ,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,CAAA;QAC7B,CAAC;aAAM,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACpB,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,CAAA;QAC7B,CAAC;aAAM,CAAC;YACJ,MAAM,KAAK,CAAC,mDAAmD,CAAC,CAAA;QACpE,CAAC;IACL,CAAC;CACJ"}