@delofarag/ai-utils 1.3.5 → 1.4.0

package/README.md

@@ -29,25 +29,26 @@ Du kannst jederzeit ein eigenes LLM übergeben (z.B. anderes Modell, anderer Pro
 
 ### 1. Chain
 
-**Was:** Stateless LLM-Chain mit strukturiertem Output (Zod-Schema) und optionalem RAG.
+**Was:** Stateless LLM-Chain mit strukturiertem Output (Zod-Schema) und optionalem RAG. Für `output` (Zod v3) kannst du **z.object()** oder **z.record()** verwenden.
 
 **Wann:** Einmalige Abfragen ohne Konversationsgedächtnis. Ideal für formularähnliche Eingabe → strukturierte Ausgabe.
 
 **Initialisierung:**
 
 ```ts
-import { Chain, DEFAULT_SCHEMA } from "@delofarag/ai-utils"
+import { Chain, DEFAULT_OUTPUT_SCHEMA } from "@delofarag/ai-utils"
 import { z } from "zod/v3"
 
-const schema = z.object({
+const output = z.object({
     output: z.string().describe("Deine Antwort"),
     score: z.number().optional()
 })
+// alternativ: z.record(z.string()) für beliebige Key-Value-Struktur
 
 const chain = new Chain({
     prompt: "Du bist ein hilfreicher Assistent.",
     // llm optional – Default: Groq
-    schema
+    output
 })
 
 const result = await chain.invoke({ input: "Was ist die Hauptstadt von Frankreich?" })
@@ -56,7 +57,7 @@ const result = await chain.invoke({ input: "Was ist die Hauptstadt von Frankreic
 
 **RAG:** Mit `chain.setContext(vectorStore)` und `chain.addContext(["Text 1", "Text 2"])` wird automatisch Retrieval vor dem LLM-Call eingebaut.
 
-**Warum so:** Chain ist die kleinste Einheit – nur Prompt + LLM + Schema. Kein Memory, keine Tools. Einfach zu testen und zu komponieren.
+**Warum so:** Chain ist die kleinste Einheit – nur Prompt + LLM + Output (Zod-Schema für .invoke()). Kein Memory, keine Tools. Einfach zu testen und zu komponieren.
 
 ---
 
@@ -98,24 +99,32 @@ const result2 = await memoryChain.invoke({
 
 ### 3. Agent
 
-**Was:** LLM mit Tools (z.B. Web-Suche, API-Calls). Nutzt LangGraphs `createReactAgent` unter der Haube.
+**Was:** LLM mit Tools (z.B. Web-Suche, API-Calls, RAG). Nutzt LangGraphs `createReactAgent` unter der Haube.
 
 **Wann:** Wenn das LLM externe Aktionen ausführen soll (Suche, Rechner, Datenbank, etc.).
 
 **Initialisierung:**
 
 ```ts
-import { Agent, ToolRegistry, tavilySearchTool, getLLM } from "@delofarag/ai-utils"
+import { Agent, ToolRegistry, createRAGTool, createFaissStore, tavilySearchTool, getLLM } from "@delofarag/ai-utils"
+
+const vectorStore = await createFaissStore(["Dokumenteninhalt..."])
+const ragTool = createRAGTool({
+    vectorStore,
+    name: "search_context",
+    description: "Durchsucht den Kontext nach relevanten Informationen"
+})
 
 const registry = new ToolRegistry([
     { name: "calculator", description: "...", schema: z.object({ a: z.number(), b: z.number() }), func: ({ a, b }) => a + b },
-    tavilySearchTool
-] as const)
+    tavilySearchTool,
+    ragTool
+])
 
 const agent = new Agent({
     prompt: "Du bist ein hilfreicher Assistent mit Zugang zu Tools.",
     tools: registry.allTools,
-    llm: getLLM({ type: "groq", apikey: process.env.CHATGROQ_API_KEY! }),
+    llm: getLLM({ type: "groq" }),
     memory: new SmartCheckpointSaver(new MemorySaver(), { llm }) // optional
 })
 
@@ -123,35 +132,42 @@ const result = await agent.invoke({
     thread_id: "session-1",  // nötig wenn memory gesetzt
     input: "Was steht heute in den Nachrichten zu KI?"
 })
+
+agent.addTool(weiteresTool)  // Tools nachträglich hinzufügen
 ```
 
-**RAG:** `agent.setContext(vectorStore)` erzeugt automatisch ein `search_context`-Tool, mit dem das LLM den Vector Store durchsuchen kann.
+**RAG:** RAG ist ein normales Tool – nutze `createRAGTool({ vectorStore, name, description })` und füge es zu `tools` hinzu oder via `agent.addTool()`.
 
-**Warum so:** Der Agent entscheidet selbst, wann er Tools nutzt. Die Klasse fügt dem System-Prompt Regeln hinzu, damit unnötige Tool-Calls vermieden werden.
+**Warum so:** Der Agent entscheidet selbst, wann er Tools nutzt. RAG wird wie jedes andere Tool behandelt (kein setContext/addContext mehr).
 
 ---
 
 ### 4. Chatbot
 
-**Was:** High-Level Wrapper – je nach Konfiguration entweder ein `MemoryChain` oder ein `Agent`.
+**Was:** High-Level Wrapper – je nach Konfiguration entweder ein `MemoryChain` oder ein `Agent`. Discriminated Union: `tools` → Agent, `vectorStore` → MemoryChain.
 
 **Wann:** Schnell einen chatbasierten Assistenten bauen, mit oder ohne Tools.
 
 **Initialisierung:**
 
 ```ts
-import { Chatbot, tavilySearchTool, getLLM } from "@delofarag/ai-utils"
+import { Chatbot, createRAGTool, createFaissStore, tavilySearchTool, getLLM } from "@delofarag/ai-utils"
 
-// Ohne Tools → nutzt MemoryChain
+// Ohne Tools, mit RAG → MemoryChain + vectorStore
+const vectorStore = await createFaissStore(["Kontextdaten..."])
 const simpleChatbot = new Chatbot({
-    llm: getLLM({ type: "groq", apikey: process.env.CHATGROQ_API_KEY! }),
-    prompt: "Du bist ein freundlicher Assistent."
+    llm: getLLM({ type: "groq" }),
+    prompt: "Du bist ein freundlicher Assistent.",
+    vectorStore
 })
 
-// Mit Tools → nutzt Agent
+// Mit Tools → Agent (RAG als Tool möglich)
 const toolChatbot = new Chatbot({
-    llm: getLLM({ type: "groq", apikey: process.env.CHATGROQ_API_KEY! }),
-    tools: [tavilySearchTool],
+    llm: getLLM({ type: "groq" }),
+    tools: [
+        tavilySearchTool,
+        createRAGTool({ vectorStore, name: "search", description: "Durchsucht den Kontext" })
+    ],
     prompt: "Du bist ein Assistent mit Webzugang."
 })
 
@@ -185,7 +201,7 @@ const registry = new ToolRegistry([
         schema: z.object({ name: z.string() }),
         func: ({ name }) => `Hallo, ${name}!`
     }
-] as const)  // 'as const' für besseren Autocomplete
+])
 
 const tool = registry.getTool("greet")
 const allTools = registry.allTools
@@ -256,22 +272,28 @@ const kurz = await summarize({ data: langerText, maxWords: 50 })
 
 ## RAG-Implementierungen
 
-In `rag.ts`:
+**Vector Stores** (in `rag.ts`):
 
 - **turn_to_docs(data)** – Wandelt Strings/Objekte in LangChain-`Document[]` um
 - **createSupabaseVectoreStore({ supabase, data, table_name?, RPC_function? })** – Supabase Vector Store aus Daten
 - **getSupabaseVectorStore({ supabase, table_name?, RPC_function? })** – Bestehenden Store holen
 - **createFaissStore({ data, save_path?, embeddings? })** – FAISS-Store (lokal, speicherbar)
 - **loadFaissStore({ path, embeddings? })** – FAISS-Store laden
+
+**RAG als Tool** (für Agent):
+
+- **createRAGTool({ vectorStore, name, description })** – Erzeugt ein Tool, mit dem der Agent den Vector Store durchsucht. In `tools` übergeben oder via `agent.addTool()`.
+
+**Retrieval-Chains** (in `rag.ts`):
+
 - **createRAGChain({ vectorStore, llm, prompt?, num_of_results_from_vdb? })** – Retrieval-Chain
 - **createRAGChainFromRetriever({ retriever, llm, prompt? })** – Alternative mit eigenem Retriever
 
 **Typischer Ablauf:**
 
-1. Daten mit `turn_to_docs` vorbereiten
-2. Vector Store erstellen (Supabase oder FAISS)
-3. `chain.setContext(vectorStore)` oder `agent.setContext(vectorStore)` aufrufen
-4. Optional: `addContext(docs)` für weitere Dokumente
+1. Vector Store erstellen: `createFaissStore({ data })` oder `createSupabaseVectoreStore({ data })`
+2. **Chain/MemoryChain:** `chain.setContext(vectorStore)` – RAG wird automatisch eingebaut. Optional `chain.addContext(weitereDaten)` für weitere Docs
+3. **Agent/Chatbot:** `createRAGTool({ vectorStore, name, description })` als Tool übergeben
 
 ---
 
@@ -325,14 +347,14 @@ const chatbot = new Chatbot({ llm, memory })
 
 | Util | Zweck |
 |------|-------|
-| **Chain** | Stateless LLM + Schema, optional RAG |
+| **Chain** | Stateless LLM + Schema, optional RAG (setContext/addContext) |
 | **MemoryChain** | Chain + Konversations-Memory pro thread_id, optional RAG |
-| **Agent** | LLM + Tools, optional Memory + RAG |
-| **Chatbot** | Einfacher Einstieg: MemoryChain oder Agent |
+| **Agent** | LLM + Tools, optional Memory. RAG via createRAGTool + addTool |
+| **Chatbot** | Discriminated Union: tools → Agent, vectorStore → MemoryChain |
 | **ToolRegistry** | Tools registrieren (empfohlen) |
 | **tavilySearchTool** | Web-Suche (TAVILY_API_KEY) |
 | **Magic-Funcs** | ask, websearch, classify, decide, extract, structure, rewrite, summarize, promptify, ragify |
-| **RAG** | Supabase/FAISS Vector Stores, createRAGChain |
+| **RAG** | createRAGTool, Supabase/FAISS Vector Stores, createRAGChain |
 | **SupabaseCheckpointSaver** | Persistente Memory in Supabase |
 | **SmartCheckpointSaver** | Memory mit Auto-Summarization |
 | **getLLM()** | LLM aus Groq, OpenRouter oder Ollama |

package/dist/heart/agent.d.ts

@@ -1,26 +1,31 @@
 import { z } from "zod/v3";
+import type { OutputSchema } from "./chain";
 import { DynamicStructuredTool } from "../imports";
 import { BaseChatModel } from "../imports";
 import { BaseCheckpointSaver } from "../imports";
-import { VectorStore } from "../imports";
-interface AgentProps<T extends z.ZodType<any, any, any>> {
-    prompt?: string | Array<string>;
+interface AgentProps<T extends OutputSchema> {
     tools: DynamicStructuredTool[];
-    llm: BaseChatModel;
-    schema?: T;
+    prompt?: string | Array<string>;
+    llm?: BaseChatModel;
+    output?: T;
     memory?: BaseCheckpointSaver;
 }
 /**
  * CONSTRUCTOR:
- * @example constructor({
+ * @example
+ * constructor({
         prompt = `Du bist ein hilfreicher Assistent.`,
-        llm = getLLM({type:"groq", apikey: process.env.CHATGROQ_API_KEY ?? ""}),
+        llm = getLLM({ type:"groq" }),
         tools,
-        schema,
-        memory
+        output,
+        memory,
     }: AgentProps<T>) {
         this.prompt = typeof prompt === "string" ? [["system", prompt]] : Array.isArray(prompt) ? prompt.map((p:string)=>{
-            return ["system", p]
+            if(typeof p === "string"){
+                return ["system", p]
+            } else {
+                return p // weil wenn es kein string ist muss es ein MessagePlaceholder sein
+            }
         }) : []
         this.prompt.push(["system",`WICHTIG:
             - Nutze Tools NUR wenn nötig
@@ -29,38 +34,30 @@ interface AgentProps<T extends z.ZodType<any, any, any>> {
             - Vermeide unnötige Tool-Calls, die dem user nichts bringen`])
         this.tools = tools
         this.llm = llm
-        this.schema = schema
+        this.output = output
         this.memory = memory
     }
+ * @param output - Zod-Schema: beschreibt die Struktur des Rückgabewerts von .invoke()
  */
-export declare class Agent<T extends z.ZodObject<any, any>> {
+export declare class Agent<T extends OutputSchema> {
     private prompt;
     private tools;
     private llm;
-    private schema;
+    private output;
     private agent;
     private memory;
-    private vectorStore;
-    private rag_tool;
-    private times_of_added_context;
-    private should_use_schema;
-    constructor({ prompt, llm, tools, schema, memory }: AgentProps<T>);
+    private should_use_output;
+    constructor({ prompt, llm, tools, output, memory, }: AgentProps<T>);
     invoke(invokeInput: Record<string, any> & {
         thread_id?: string;
         debug?: boolean;
     }): Promise<T extends undefined ? string : z.infer<T>>;
+    /** bro nutzt später vielleicht intern mal die native .stream() von createReactAgent */
     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[];
 }

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

@@ -1 +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,EAAC,MAAM,YAAY,CAAA;AACjD,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAA;AAC1C,OAAO,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAA;AAChD,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAA;AAOxC,UAAU,UAAU,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;IACnD,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,MAA6C,EAC7C,GAAuE,EACvE,KAAK,EACL,MAAM,EACN,MAAM,EACT,EAAE,UAAU,CAAC,CAAC,CAAC;IAeH,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"}
+{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../../src/heart/agent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,QAAQ,CAAA;AAC1B,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,SAAS,CAAA;AAC3C,OAAO,EAAE,qBAAqB,EAAC,MAAM,YAAY,CAAA;AACjD,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAA;AAC1C,OAAO,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAA;AAchD,UAAU,UAAU,CAAC,CAAC,SAAS,YAAY;IACvC,KAAK,EAAE,qBAAqB,EAAE,CAAA;IAC9B,MAAM,CAAC,EAAE,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,CAAA;IAC/B,GAAG,CAAC,EAAE,aAAa,CAAA;IACnB,MAAM,CAAC,EAAE,CAAC,CAAA;IACV,MAAM,CAAC,EAAE,mBAAmB,CAAA;CAC/B;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,qBAAa,KAAK,CAAC,CAAC,SAAS,YAAY;IACrC,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,iBAAiB,CAAgB;gBAE7B,EACR,MAA6C,EAC7C,GAA6B,EAC7B,KAAK,EACL,MAAM,EACN,MAAM,GACT,EAAE,UAAU,CAAC,CAAC,CAAC;IAmBH,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;IAoCnJ,uFAAuF;IACzE,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,OAAO,CAAC,IAAI,EAAC,qBAAqB;IAIzC,IAAW,YAAY,IAAI,MAAM,EAAE,CAElC;CACJ"}

package/dist/heart/agent.js

@@ -1,21 +1,23 @@
-import { z } from "zod/v3";
-import { DynamicStructuredTool } from "../imports";
-import { turn_to_docs } from "../rag";
 import { createReactAgent } from "../imports";
 import { HumanMessage } from "../imports";
 import { getLLM, stream } from "../helpers";
 import { structure } from "../magic-funcs/parsers/structure";
 /**
  * CONSTRUCTOR:
- * @example constructor({
+ * @example
+ * constructor({
         prompt = `Du bist ein hilfreicher Assistent.`,
-        llm = getLLM({type:"groq", apikey: process.env.CHATGROQ_API_KEY ?? ""}),
+        llm = getLLM({ type:"groq" }),
         tools,
-        schema,
-        memory
+        output,
+        memory,
     }: AgentProps<T>) {
         this.prompt = typeof prompt === "string" ? [["system", prompt]] : Array.isArray(prompt) ? prompt.map((p:string)=>{
-            return ["system", p]
+            if(typeof p === "string"){
+                return ["system", p]
+            } else {
+                return p // weil wenn es kein string ist muss es ein MessagePlaceholder sein
+            }
         }) : []
         this.prompt.push(["system",`WICHTIG:
             - Nutze Tools NUR wenn nötig
@@ -24,24 +26,27 @@ import { structure } from "../magic-funcs/parsers/structure";
             - Vermeide unnötige Tool-Calls, die dem user nichts bringen`])
         this.tools = tools
         this.llm = llm
-        this.schema = schema
+        this.output = output
         this.memory = memory
     }
+ * @param output - Zod-Schema: beschreibt die Struktur des Rückgabewerts von .invoke()
  */
 export class Agent {
     prompt;
     tools;
     llm;
-    schema;
+    output;
     agent;
     memory;
-    vectorStore;
-    rag_tool;
-    times_of_added_context = 0;
-    should_use_schema = true;
-    constructor({ prompt = `Du bist ein hilfreicher Assistent.`, llm = getLLM({ type: "groq", apikey: process.env.CHATGROQ_API_KEY ?? "" }), tools, schema, memory }) {
+    should_use_output = true;
+    constructor({ prompt = `Du bist ein hilfreicher Assistent.`, llm = getLLM({ type: "groq" }), tools, output, memory, }) {
         this.prompt = typeof prompt === "string" ? [["system", prompt]] : Array.isArray(prompt) ? prompt.map((p) => {
-            return ["system", p];
+            if (typeof p === "string") {
+                return ["system", p];
+            }
+            else {
+                return p; // weil wenn es kein string ist muss es ein MessagePlaceholder sein
+            }
         }) : [];
         this.prompt.push(["system", `WICHTIG: 
             - Nutze Tools NUR wenn nötig
@@ -50,7 +55,7 @@ export class Agent {
             - Vermeide unnötige Tool-Calls, die dem user nichts bringen`]);
         this.tools = tools;
         this.llm = llm;
-        this.schema = schema;
+        this.output = output;
         this.memory = memory;
     }
     async invoke(invokeInput) {
@@ -60,13 +65,9 @@ export class Agent {
         if (!this.memory && thread_id)
             console.warn("WARN: thread_id is provided but no memory 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,
+            tools: this.tools,
             checkpointSaver: this.memory,
             prompt: (state) => [
                 ...this.prompt,
@@ -74,20 +75,26 @@ export class Agent {
             ]
         });
         const config = thread_id && this.memory ? { configurable: { thread_id } } : undefined;
+        if (thread_id && !this.memory)
+            console.warn("thread_id wurde beim invoke des agenten mitgegeben aber keine memory, somit wird nichts gespeichert und die thread_id ist gleichgültig");
         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 });
+        const raw = lastMessage?.content;
+        const content = typeof raw === "string" ? raw : (Array.isArray(raw) ? raw.map((c) => c?.text ?? c).join("") : String(raw ?? ""));
+        if (this.output && this.should_use_output) {
+            return await structure({ data: content, into: this.output, llm: this.llm });
+        }
+        else {
+            return content;
         }
-        return content;
     }
+    /** bro nutzt später vielleicht intern mal die native .stream() von createReactAgent */
     async *stream(invokeInput) {
-        this.should_use_schema = false;
+        this.should_use_output = false;
         try {
-            const { stream_delay = 1, ...rest } = invokeInput;
+            const { stream_delay = 50, ...rest } = invokeInput;
             const response = await this.invoke(rest);
             const responseStr = typeof response === "string" ? response : JSON.stringify(response);
             const words = responseStr.split(" ");
@@ -96,52 +103,14 @@ export class Agent {
             }
         }
         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.should_use_output = true;
         }
-        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);
+        return this.tools.map(tool => tool.name);
     }
 }
 //# sourceMappingURL=agent.js.map

package/dist/heart/agent.js.map

@@ -1 +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,EAAC,MAAM,YAAY,CAAA;AAIjD,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAA;AACrC,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAA;AAC7C,OAAO,EAAE,YAAY,EAAC,MAAM,YAAY,CAAA;AACxC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,YAAY,CAAA;AAC3C,OAAO,EAAE,SAAS,EAAE,MAAM,kCAAkC,CAAA;AAU5D;;;;;;;;;;;;;;;;;;;;;;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,oCAAoC,EAC7C,GAAG,GAAG,MAAM,CAAC,EAAC,IAAI,EAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,EAAE,EAAC,CAAC,EACvE,KAAK,EACL,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,MAAM,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAC;;;;wEAIqC,CAAC,CAAC,CAAA;QAClE,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,MAAM,IAAI,CAAC,SAAS;YAAE,MAAM,IAAI,KAAK,CAAC,mEAAmE,CAAC,CAAA;QAClH,IAAG,CAAC,IAAI,CAAC,MAAM,IAAI,SAAS;YAAE,OAAO,CAAC,IAAI,CAAC,0EAA0E,CAAC,CAAA;QAEtH,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,CAAC;QAEF,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"}
+{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../src/heart/agent.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAA;AAC7C,OAAO,EAAE,YAAY,EAAC,MAAM,YAAY,CAAA;AACxC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,YAAY,CAAA;AAC3C,OAAO,EAAE,SAAS,EAAE,MAAM,kCAAkC,CAAA;AAgB5D;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,MAAM,OAAO,KAAK;IACN,MAAM,CAA2B;IACjC,KAAK,CAAyB;IAC9B,GAAG,CAAe;IAClB,MAAM,CAAe;IACrB,KAAK,CAAK;IACV,MAAM,CAAiC;IACvC,iBAAiB,GAAY,IAAI,CAAA;IAEzC,YAAY,EACR,MAAM,GAAG,oCAAoC,EAC7C,GAAG,GAAG,MAAM,CAAC,EAAE,IAAI,EAAC,MAAM,EAAE,CAAC,EAC7B,KAAK,EACL,MAAM,EACN,MAAM,GACM;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,IAAG,OAAO,CAAC,KAAK,QAAQ,EAAC,CAAC;gBACtB,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAA;YACxB,CAAC;iBAAM,CAAC;gBACJ,OAAO,CAAC,CAAA,CAAC,mEAAmE;YAChF,CAAC;QACL,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAA;QACP,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAC;;;;wEAIqC,CAAC,CAAC,CAAA;QAClE,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,MAAM,IAAI,CAAC,SAAS;YAAE,MAAM,IAAI,KAAK,CAAC,mEAAmE,CAAC,CAAA;QAClH,IAAG,CAAC,IAAI,CAAC,MAAM,IAAI,SAAS;YAAE,OAAO,CAAC,IAAI,CAAC,0EAA0E,CAAC,CAAA;QAEtH,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,CAAC;QAGF,IAAI,CAAC,KAAK,GAAG,gBAAgB,CAAC;YAC1B,GAAG,EAAE,IAAI,CAAC,GAAU;YACpB,KAAK,EAAE,IAAI,CAAC,KAAY;YACxB,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,IAAG,SAAS,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO,CAAC,IAAI,CAAC,wIAAwI,CAAC,CAAA;QACpL,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,GAAG,GAAG,WAAW,EAAE,OAAO,CAAA;QAChC,MAAM,OAAO,GAAG,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,IAAI,EAAE,CAAC,CAAC,CAAA;QAErI,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;aAAM,CAAC;YACJ,OAAO,OAAc,CAAA;QACzB,CAAC;IACL,CAAC;IAED,uFAAuF;IAChF,KAAK,CAAC,CAAC,MAAM,CAAC,WAAiG;QAClH,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAA;QAC9B,IAAG,CAAC;YACA,MAAM,EAAE,YAAY,GAAG,EAAE,EAAE,GAAG,IAAI,EAAE,GAAG,WAAW,CAAA;YAClD,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,OAAO,CAAC,IAA0B;QACrC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACzB,CAAC;IAED,IAAW,YAAY;QACnB,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAC5C,CAAC;CACJ"}

package/dist/heart/chain.d.ts

@@ -1,44 +1,51 @@
 import { BaseChatModel, MessagesPlaceholder, VectorStore } from "../imports";
 import { z } from "zod/v3";
-export declare const DEFAULT_SCHEMA: z.ZodObject<{
+/** Output-Schema für .invoke(): z.object() oder z.record() (Zod v3). */
+export type OutputSchema = z.ZodObject<any, any, any> | z.ZodRecord<any, any>;
+export declare const DEFAULT_OUTPUT_SCHEMA: z.ZodObject<{
     output: z.ZodString;
 }, "strip", z.ZodTypeAny, {
     output: string;
 }, {
     output: string;
 }>;
-interface ChainProps<T extends z.ZodType<any, any, any>> {
+interface ChainProps<T extends OutputSchema> {
     prompt?: string | Array<string | MessagesPlaceholder<any>>;
     llm?: BaseChatModel;
-    schema?: T;
+    output?: T;
+    vectorStore?: VectorStore;
 }
 /**
  * CONSTRUCTOR:
- * @example constructor({
+ * @example
+ * constructor({
         prompt = "du bist ein hilfreicher Assistent",
         llm = getLLM({type:"groq", apikey: process.env.CHATGROQ_API_KEY ?? ""}),
-        schema = DEFAULT_SCHEMA as unknown as T as T
+        output = DEFAULT_OUTPUT_SCHEMA as unknown as T as T
+        vectorStore = undefined
     }: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
+                return p // weil wenn es kein string ist dann ist es ein MessagePlaceholder
             }
         }) : []
         this.llm = llm
-        this.schema = schema
-        this.parser = StructuredOutputParser.fromZodSchema(this.schema)
+        this.vectorStore = vectorStore
+        this.output = output
+        this.parser = StructuredOutputParser.fromZodSchema(this.output)
     }
+ * @param output - Zod-Schema: beschreibt die Struktur des Rückgabewerts von .invoke()
  */
-export declare class Chain<T extends z.ZodType<any, any, any> = typeof DEFAULT_SCHEMA> {
+export declare class Chain<T extends OutputSchema = typeof DEFAULT_OUTPUT_SCHEMA> {
     private prompt;
     private vectorStore;
     private times_of_added_context;
     private parser;
     private llm;
-    private schema;
-    constructor({ prompt, llm, schema }: ChainProps<T>);
+    private output;
+    constructor({ prompt, llm, output, vectorStore }?: ChainProps<T>);
     invoke(input: Record<string, any> & {
         debug?: boolean;
     }): Promise<z.infer<T>>;
@@ -47,7 +54,7 @@ export declare class Chain<T extends z.ZodType<any, any, any> = typeof DEFAULT_S
     }): 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>;
+    setContext(vectorStore: VectorStore): void;
     clearContext(): void;
 }
 export {};

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

@@ -1 +1 @@
-{"version":3,"file":"chain.d.ts","sourceRoot":"","sources":["../../src/heart/chain.ts"],"names":[],"mappings":"AACA,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,OAAO,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;IACnD,MAAM,CAAC,EAAC,MAAM,GAAG,KAAK,CAAC,MAAM,GAAG,mBAAmB,CAAC,GAAG,CAAC,CAAC,CAAA;IACzD,GAAG,CAAC,EAAC,aAAa,CAAA;IAClB,MAAM,CAAC,EAAC,CAAC,CAAA;CACZ;AAED;;;;;;;;;;;;;;;;;;GAkBG;AACH,qBAAa,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,OAAO,cAAc;IACzE,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,EACR,MAA4C,EAC5C,GAAuE,EACvE,MAA4C,EAC/C,EAAC,UAAU,CAAC,CAAC,CAAC;IAaF,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"}
+{"version":3,"file":"chain.d.ts","sourceRoot":"","sources":["../../src/heart/chain.ts"],"names":[],"mappings":"AACA,OAAO,EACH,aAAa,EAGb,mBAAmB,EACnB,WAAW,EAGd,MAAM,YAAY,CAAA;AAEnB,OAAO,EAAE,CAAC,EAAE,MAAM,QAAQ,CAAA;AAE1B,wEAAwE;AACxE,MAAM,MAAM,YAAY,GAAG,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE,GAAG,CAAC,CAAA;AAE7E,eAAO,MAAM,qBAAqB;;;;;;EAEhC,CAAA;AAEF,UAAU,UAAU,CAAC,CAAC,SAAS,YAAY;IACvC,MAAM,CAAC,EAAC,MAAM,GAAG,KAAK,CAAC,MAAM,GAAG,mBAAmB,CAAC,GAAG,CAAC,CAAC,CAAA;IACzD,GAAG,CAAC,EAAC,aAAa,CAAA;IAClB,MAAM,CAAC,EAAC,CAAC,CAAC;IACV,WAAW,CAAC,EAAE,WAAW,CAAA;CAC5B;AAED;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,qBAAa,KAAK,CAAC,CAAC,SAAS,YAAY,GAAG,OAAO,qBAAqB;IACpE,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,EACR,MAA4C,EAC5C,GAA6B,EAC7B,MAAmD,EACnD,WAAuB,EAC1B,GAAC,UAAU,CAAC,CAAC,CAAM;IAcP,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;IAWjC,UAAU,CAAC,WAAW,EAAE,WAAW;IAKnC,YAAY;CAKtB"}