@luanpoppe/ai 1.0.0

package/.env.example

@@ -0,0 +1,2 @@
+OPENAI_API_KEY="..."
+GOOGLE_GENAI_API_KEY="..."

package/dist/index.d.ts

@@ -0,0 +1,32 @@
+import { AIModelNames } from "./src/@types/model-names";
+import z from "zod";
+import { MessageInput } from "./src/langchain/messages";
+import { AgentMiddleware, BaseMessage } from "langchain";
+import { ClientTool, ServerTool } from "@langchain/core/tools";
+export type CallParams = {
+    aiModel: AIModelNames;
+    messages: MessageInput[];
+    systemPrompt?: string;
+    maxRetries?: number;
+    middleware?: AgentMiddleware[];
+    tools?: (ServerTool | ClientTool)[];
+};
+export type CallReturn = Promise<{
+    text: string;
+    messages: BaseMessage[];
+}>;
+export type CallStructuredOutputParams<T extends z.ZodSchema> = CallParams & {
+    outputSchema: T;
+};
+export type CallStructuredOutputReturn<T> = Promise<{
+    response: z.infer<T>;
+}>;
+export declare class Langchain {
+    constructor();
+    call(params: CallParams): CallReturn;
+    callStructuredOutput<T extends z.ZodSchema>(params: CallStructuredOutputParams<T>): CallStructuredOutputReturn<typeof params.outputSchema>;
+    private getModel;
+    private standardAgent;
+    private standardMiddlewares;
+}
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AACxD,OAAO,CAAC,MAAM,KAAK,CAAC;AACpB,OAAO,EAAqB,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAC3E,OAAO,EACL,eAAe,EACf,WAAW,EAIZ,MAAM,WAAW,CAAC;AACnB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAE/D,MAAM,MAAM,UAAU,GAAG;IACvB,OAAO,EAAE,YAAY,CAAC;IACtB,QAAQ,EAAE,YAAY,EAAE,CAAC;IACzB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,UAAU,CAAC,EAAE,eAAe,EAAE,CAAC;IAC/B,KAAK,CAAC,EAAE,CAAC,UAAU,GAAG,UAAU,CAAC,EAAE,CAAC;CACrC,CAAC;AAEF,MAAM,MAAM,UAAU,GAAG,OAAO,CAAC;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,WAAW,EAAE,CAAC;CACzB,CAAC,CAAC;AAEH,MAAM,MAAM,0BAA0B,CAAC,CAAC,SAAS,CAAC,CAAC,SAAS,IAAI,UAAU,GAAG;IAC3E,YAAY,EAAE,CAAC,CAAC;CACjB,CAAC;AAEF,MAAM,MAAM,0BAA0B,CAAC,CAAC,IAAI,OAAO,CAAC;IAClD,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;CACtB,CAAC,CAAC;AAEH,qBAAa,SAAS;;IAGd,IAAI,CAAC,MAAM,EAAE,UAAU,GAAG,UAAU;IAepC,oBAAoB,CAAC,CAAC,SAAS,CAAC,CAAC,SAAS,EAC9C,MAAM,EAAE,0BAA0B,CAAC,CAAC,CAAC,GACpC,0BAA0B,CAAC,OAAO,MAAM,CAAC,YAAY,CAAC;IAiBzD,OAAO,CAAC,QAAQ;IAgBhB,OAAO,CAAC,aAAa;IAgBrB,OAAO,CAAC,mBAAmB;CAU5B"}

package/dist/index.js

@@ -0,0 +1,92 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Langchain = void 0;
+const models_1 = require("./src/langchain/models");
+const zod_1 = __importDefault(require("zod"));
+const messages_1 = require("./src/langchain/messages");
+const langchain_1 = require("langchain");
+class Langchain {
+    constructor() { }
+    async call(params) {
+        const { messages } = params;
+        const agent = (0, langchain_1.createAgent)({
+            ...this.standardAgent(params),
+        });
+        const response = await agent.invoke({ messages });
+        return {
+            text: response.messages[0]?.text ?? "Empty response from the model",
+            messages: response.messages,
+        };
+    }
+    async callStructuredOutput(params) {
+        const { outputSchema, messages } = params;
+        const agent = (0, langchain_1.createAgent)({
+            ...this.standardAgent(params),
+            responseFormat: outputSchema,
+        });
+        const response = await agent.invoke({
+            messages,
+        });
+        const parsedResponse = outputSchema.parse(response?.structuredResponse);
+        return { response: parsedResponse };
+    }
+    getModel(aiModel) {
+        if (aiModel.startsWith("gpt")) {
+            return models_1.LangchainModels.gpt({
+                modelName: aiModel,
+            });
+        }
+        if (aiModel.startsWith("gemini")) {
+            return models_1.LangchainModels.gemini({
+                model: aiModel,
+            });
+        }
+        throw new Error("Model not supported");
+    }
+    standardAgent(params) {
+        const { aiModel, systemPrompt, maxRetries = 3, middleware, tools } = params;
+        const model = this.getModel(aiModel);
+        return {
+            model,
+            systemPrompt: systemPrompt ?? "",
+            middleware: [
+                ...this.standardMiddlewares(maxRetries),
+                ...(middleware ?? []),
+            ],
+            tools: tools ?? [],
+            responseFormat: undefined,
+        };
+    }
+    standardMiddlewares(maxRetries) {
+        return [
+            (0, langchain_1.modelRetryMiddleware)({
+                maxRetries,
+                backoffFactor: 2.0,
+                initialDelayMs: 1000,
+            }),
+            (0, langchain_1.modelFallbackMiddleware)("gemini-2.5-flash", "gpt-4o-mini"),
+        ];
+    }
+}
+exports.Langchain = Langchain;
+const example = async () => {
+    const langchain = new Langchain();
+    const outputSchema = zod_1.default.object({
+        answer: zod_1.default.string(),
+    });
+    const messages = [
+        messages_1.LangchainMessages.system("You are a helpful assistant."),
+        messages_1.LangchainMessages.human("What is the capital of France?"),
+    ];
+    const { response } = await langchain.callStructuredOutput({
+        aiModel: "gemini-2.5-flash",
+        outputSchema,
+        messages,
+    });
+    console.log({ response });
+};
+example();
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../index.ts"],"names":[],"mappings":";;;;;;AAAA,mDAAyD;AAEzD,8CAAoB;AACpB,uDAA2E;AAC3E,yCAMmB;AAyBnB,MAAa,SAAS;IACpB,gBAAe,CAAC;IAEhB,KAAK,CAAC,IAAI,CAAC,MAAkB;QAC3B,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC;QAE5B,MAAM,KAAK,GAAG,IAAA,uBAAW,EAAC;YACxB,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC;SAC9B,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;QAElD,OAAO;YACL,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,+BAA+B;YACnE,QAAQ,EAAE,QAAQ,CAAC,QAAQ;SAC5B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,oBAAoB,CACxB,MAAqC;QAErC,MAAM,EAAE,YAAY,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC;QAE1C,MAAM,KAAK,GAAG,IAAA,uBAAW,EAAC;YACxB,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC;YAC7B,cAAc,EAAE,YAAmB;SACpC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC;YAClC,QAAQ;SACT,CAAC,CAAC;QAEH,MAAM,cAAc,GAAG,YAAY,CAAC,KAAK,CAAC,QAAQ,EAAE,kBAAkB,CAAC,CAAC;QAExE,OAAO,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC;IACtC,CAAC;IAEO,QAAQ,CAAC,OAAqB;QACpC,IAAI,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YAC9B,OAAO,wBAAe,CAAC,GAAG,CAAC;gBACzB,SAAS,EAAE,OAAO;aACnB,CAAC,CAAC;QACL,CAAC;QAED,IAAI,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YACjC,OAAO,wBAAe,CAAC,MAAM,CAAC;gBAC5B,KAAK,EAAE,OAAO;aACf,CAAC,CAAC;QACL,CAAC;QAED,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;IACzC,CAAC;IAEO,aAAa,CAAC,MAAkB;QACtC,MAAM,EAAE,OAAO,EAAE,YAAY,EAAE,UAAU,GAAG,CAAC,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;QAE5E,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QACrC,OAAO;YACL,KAAK;YACL,YAAY,EAAE,YAAY,IAAI,EAAE;YAChC,UAAU,EAAE;gBACV,GAAG,IAAI,CAAC,mBAAmB,CAAC,UAAU,CAAC;gBACvC,GAAG,CAAC,UAAU,IAAI,EAAE,CAAC;aACtB;YACD,KAAK,EAAE,KAAK,IAAI,EAAE;YAClB,cAAc,EAAE,SAAgB;SACjC,CAAC;IACJ,CAAC;IAEO,mBAAmB,CAAC,UAAkB;QAC5C,OAAO;YACL,IAAA,gCAAoB,EAAC;gBACnB,UAAU;gBACV,aAAa,EAAE,GAAG;gBAClB,cAAc,EAAE,IAAI;aACrB,CAAC;YACF,IAAA,mCAAuB,EAAC,kBAAkB,EAAE,aAAa,CAAC;SAC3D,CAAC;IACJ,CAAC;CACF;AA/ED,8BA+EC;AAED,MAAM,OAAO,GAAG,KAAK,IAAI,EAAE;IACzB,MAAM,SAAS,GAAG,IAAI,SAAS,EAAE,CAAC;IAElC,MAAM,YAAY,GAAG,aAAC,CAAC,MAAM,CAAC;QAC5B,MAAM,EAAE,aAAC,CAAC,MAAM,EAAE;KACnB,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG;QACf,4BAAiB,CAAC,MAAM,CAAC,8BAA8B,CAAC;QACxD,4BAAiB,CAAC,KAAK,CAAC,gCAAgC,CAAC;KAC1D,CAAC;IAEF,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,SAAS,CAAC,oBAAoB,CAAC;QACxD,OAAO,EAAE,kBAAkB;QAC3B,YAAY;QACZ,QAAQ;KACT,CAAC,CAAC;IAEH,OAAO,CAAC,GAAG,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;AAC5B,CAAC,CAAC;AAEF,OAAO,EAAE,CAAC"}

package/dist/src/@types/model-names.d.ts

@@ -0,0 +1,5 @@
+type ChatGPTModels = "gpt-4" | "gpt-4o" | "gpt-4.1" | "gpt-5" | "gpt-5.1" | "gpt-5-mini" | "gpt-5-nano";
+type GeminiModels = "gemini-2.5-flash" | "gemini-2.5-pro" | "gemini-3-flash" | "gemini-3-pro";
+export type AIModelNames = ChatGPTModels | GeminiModels;
+export {};
+//# sourceMappingURL=model-names.d.ts.map

package/dist/src/@types/model-names.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-names.d.ts","sourceRoot":"","sources":["../../../src/@types/model-names.ts"],"names":[],"mappings":"AAAA,KAAK,aAAa,GACd,OAAO,GACP,QAAQ,GACR,SAAS,GACT,OAAO,GACP,SAAS,GACT,YAAY,GACZ,YAAY,CAAC;AAEjB,KAAK,YAAY,GACb,kBAAkB,GAClB,gBAAgB,GAChB,gBAAgB,GAChB,cAAc,CAAC;AAEnB,MAAM,MAAM,YAAY,GAAG,aAAa,GAAG,YAAY,CAAC"}

package/dist/src/@types/model-names.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=model-names.js.map

package/dist/src/@types/model-names.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-names.js","sourceRoot":"","sources":["../../../src/@types/model-names.ts"],"names":[],"mappings":""}

package/dist/src/core/env.d.ts

@@ -0,0 +1,6 @@
+import "dotenv/config";
+export declare const env: {
+    OPENAI_API_KEY: string;
+    GOOGLE_GENAI_API_KEY: string;
+};
+//# sourceMappingURL=env.d.ts.map

package/dist/src/core/env.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"env.d.ts","sourceRoot":"","sources":["../../../src/core/env.ts"],"names":[],"mappings":"AAAA,OAAO,eAAe,CAAC;AAYvB,eAAO,MAAM,GAAG;;;CAAO,CAAC"}

package/dist/src/core/env.js

@@ -0,0 +1,17 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.env = void 0;
+require("dotenv/config");
+const zod_1 = __importDefault(require("zod"));
+const envSchema = zod_1.default.object({
+    OPENAI_API_KEY: zod_1.default.string().optional().default(""),
+    GOOGLE_GENAI_API_KEY: zod_1.default.string().optional().default(""),
+});
+const { error, data } = envSchema.safeParse(process.env);
+if (error)
+    throw new Error(`Invalid environment variables:\n${error.message}`);
+exports.env = data;
+//# sourceMappingURL=env.js.map

package/dist/src/core/env.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"env.js","sourceRoot":"","sources":["../../../src/core/env.ts"],"names":[],"mappings":";;;;;;AAAA,yBAAuB;AACvB,8CAAoB;AAEpB,MAAM,SAAS,GAAG,aAAC,CAAC,MAAM,CAAC;IACzB,cAAc,EAAE,aAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;IACjD,oBAAoB,EAAE,aAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;CACxD,CAAC,CAAC;AAEH,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,SAAS,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;AAEzD,IAAI,KAAK;IAAE,MAAM,IAAI,KAAK,CAAC,mCAAmC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;AAElE,QAAA,GAAG,GAAG,IAAI,CAAC"}

package/dist/src/langchain/messages.d.ts

@@ -0,0 +1,8 @@
+import { SystemMessage, HumanMessage, AIMessage } from "langchain";
+export type MessageInput = SystemMessage | HumanMessage | AIMessage;
+export declare class LangchainMessages {
+    static system(message: string): SystemMessage;
+    static human(message: string): HumanMessage;
+    static ai(message: string): AIMessage;
+}
+//# sourceMappingURL=messages.d.ts.map

package/dist/src/langchain/messages.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"messages.d.ts","sourceRoot":"","sources":["../../../src/langchain/messages.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAEnE,MAAM,MAAM,YAAY,GAAG,aAAa,GAAG,YAAY,GAAG,SAAS,CAAC;AAEpE,qBAAa,iBAAiB;IAC5B,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,GAAG,aAAa;IAG7C,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,GAAG,YAAY;IAG3C,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,GAAG,SAAS;CAGtC"}

package/dist/src/langchain/messages.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LangchainMessages = void 0;
+const langchain_1 = require("langchain");
+class LangchainMessages {
+    static system(message) {
+        return new langchain_1.SystemMessage(message);
+    }
+    static human(message) {
+        return new langchain_1.HumanMessage(message);
+    }
+    static ai(message) {
+        return new langchain_1.AIMessage(message);
+    }
+}
+exports.LangchainMessages = LangchainMessages;
+//# sourceMappingURL=messages.js.map

package/dist/src/langchain/messages.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"messages.js","sourceRoot":"","sources":["../../../src/langchain/messages.ts"],"names":[],"mappings":";;;AAAA,yCAAmE;AAInE,MAAa,iBAAiB;IAC5B,MAAM,CAAC,MAAM,CAAC,OAAe;QAC3B,OAAO,IAAI,yBAAa,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC;IACD,MAAM,CAAC,KAAK,CAAC,OAAe;QAC1B,OAAO,IAAI,wBAAY,CAAC,OAAO,CAAC,CAAC;IACnC,CAAC;IACD,MAAM,CAAC,EAAE,CAAC,OAAe;QACvB,OAAO,IAAI,qBAAS,CAAC,OAAO,CAAC,CAAC;IAChC,CAAC;CACF;AAVD,8CAUC"}

package/dist/src/langchain/models.d.ts

@@ -0,0 +1,7 @@
+import { ChatGoogleGenerativeAI, GoogleGenerativeAIChatInput } from "@langchain/google-genai";
+import { ChatOpenAI, ChatOpenAIFields } from "@langchain/openai";
+export declare class LangchainModels {
+    static gpt(params: ChatOpenAIFields): ChatOpenAI<import("@langchain/openai").ChatOpenAICallOptions>;
+    static gemini(params: GoogleGenerativeAIChatInput): ChatGoogleGenerativeAI;
+}
+//# sourceMappingURL=models.d.ts.map

package/dist/src/langchain/models.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../../../src/langchain/models.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,sBAAsB,EACtB,2BAA2B,EAC5B,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AAGjE,qBAAa,eAAe;IAC1B,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,gBAAgB;IAUnC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,2BAA2B;CAOlD"}

package/dist/src/langchain/models.js

@@ -0,0 +1,26 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LangchainModels = void 0;
+const google_genai_1 = require("@langchain/google-genai");
+const openai_1 = require("@langchain/openai");
+const env_1 = require("../core/env");
+class LangchainModels {
+    static gpt(params) {
+        console.log({ env: env_1.env });
+        return new openai_1.ChatOpenAI({
+            temperature: 0,
+            maxTokens: 2048,
+            apiKey: env_1.env.OPENAI_API_KEY,
+            ...params,
+        });
+    }
+    static gemini(params) {
+        return new google_genai_1.ChatGoogleGenerativeAI({
+            maxOutputTokens: 2048,
+            apiKey: env_1.env.GOOGLE_GENAI_API_KEY,
+            ...params,
+        });
+    }
+}
+exports.LangchainModels = LangchainModels;
+//# sourceMappingURL=models.js.map

package/dist/src/langchain/models.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"models.js","sourceRoot":"","sources":["../../../src/langchain/models.ts"],"names":[],"mappings":";;;AAAA,0DAGiC;AACjC,8CAAiE;AACjE,qCAAkC;AAElC,MAAa,eAAe;IAC1B,MAAM,CAAC,GAAG,CAAC,MAAwB;QACjC,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAH,SAAG,EAAE,CAAC,CAAC;QACrB,OAAO,IAAI,mBAAU,CAAC;YACpB,WAAW,EAAE,CAAC;YACd,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,SAAG,CAAC,cAAc;YAC1B,GAAG,MAAM;SACV,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CAAC,MAAM,CAAC,MAAmC;QAC/C,OAAO,IAAI,qCAAsB,CAAC;YAChC,eAAe,EAAE,IAAI;YACrB,MAAM,EAAE,SAAG,CAAC,oBAAoB;YAChC,GAAG,MAAM;SACV,CAAC,CAAC;IACL,CAAC;CACF;AAlBD,0CAkBC"}

package/dist/src/langchain/tools.d.ts

@@ -0,0 +1,12 @@
+import { tool } from "langchain";
+import z from "zod";
+export type CreateToolParams = {
+    toolFunction: Parameters<typeof tool>[0];
+    name: string;
+    description: string;
+    schema: z.ZodSchema;
+};
+export declare class LangchainTools {
+    createTool(params: CreateToolParams): import("langchain").DynamicStructuredTool<z.ZodType<unknown, unknown, z.core.$ZodTypeInternals<unknown, unknown>>, unknown, unknown, unknown>;
+}
+//# sourceMappingURL=tools.d.ts.map

package/dist/src/langchain/tools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.d.ts","sourceRoot":"","sources":["../../../src/langchain/tools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,CAAC,MAAM,KAAK,CAAC;AAEpB,MAAM,MAAM,gBAAgB,GAAG;IAC7B,YAAY,EAAE,UAAU,CAAC,OAAO,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IACzC,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,CAAC,CAAC,SAAS,CAAC;CACrB,CAAC;AAEF,qBAAa,cAAc;IACzB,UAAU,CAAC,MAAM,EAAE,gBAAgB;CAUpC"}

package/dist/src/langchain/tools.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LangchainTools = void 0;
+const langchain_1 = require("langchain");
+class LangchainTools {
+    createTool(params) {
+        const { toolFunction } = params;
+        const { name, description, schema } = params;
+        return (0, langchain_1.tool)(toolFunction, {
+            name,
+            description,
+            schema,
+        });
+    }
+}
+exports.LangchainTools = LangchainTools;
+//# sourceMappingURL=tools.js.map

package/dist/src/langchain/tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.js","sourceRoot":"","sources":["../../../src/langchain/tools.ts"],"names":[],"mappings":";;;AAAA,yCAAiC;AAUjC,MAAa,cAAc;IACzB,UAAU,CAAC,MAAwB;QACjC,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,CAAC;QAChC,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;QAE7C,OAAO,IAAA,gBAAI,EAAC,YAAY,EAAE;YACxB,IAAI;YACJ,WAAW;YACX,MAAM;SACP,CAAC,CAAC;IACL,CAAC;CACF;AAXD,wCAWC"}

package/index.ts

@@ -0,0 +1,138 @@
+import { LangchainModels } from "./src/langchain/models";
+import { AIModelNames } from "./src/@types/model-names";
+import z from "zod";
+import { LangchainMessages, MessageInput } from "./src/langchain/messages";
+import {
+  AgentMiddleware,
+  BaseMessage,
+  createAgent,
+  modelFallbackMiddleware,
+  modelRetryMiddleware,
+} from "langchain";
+import { ClientTool, ServerTool } from "@langchain/core/tools";
+
+export type CallParams = {
+  aiModel: AIModelNames;
+  messages: MessageInput[];
+  systemPrompt?: string;
+  maxRetries?: number;
+  middleware?: AgentMiddleware[];
+  tools?: (ServerTool | ClientTool)[];
+};
+
+export type CallReturn = Promise<{
+  text: string;
+  messages: BaseMessage[];
+}>;
+
+export type CallStructuredOutputParams<T extends z.ZodSchema> = CallParams & {
+  outputSchema: T;
+};
+
+export type CallStructuredOutputReturn<T> = Promise<{
+  response: z.infer<T>;
+}>;
+
+export class Langchain {
+  constructor() {}
+
+  async call(params: CallParams): CallReturn {
+    const { messages } = params;
+
+    const agent = createAgent({
+      ...this.standardAgent(params),
+    });
+
+    const response = await agent.invoke({ messages });
+
+    return {
+      text: response.messages[0]?.text ?? "Empty response from the model",
+      messages: response.messages,
+    };
+  }
+
+  async callStructuredOutput<T extends z.ZodSchema>(
+    params: CallStructuredOutputParams<T>
+  ): CallStructuredOutputReturn<typeof params.outputSchema> {
+    const { outputSchema, messages } = params;
+
+    const agent = createAgent({
+      ...this.standardAgent(params),
+      responseFormat: outputSchema as any,
+    });
+
+    const response = await agent.invoke({
+      messages,
+    });
+
+    const parsedResponse = outputSchema.parse(response?.structuredResponse);
+
+    return { response: parsedResponse };
+  }
+
+  private getModel(aiModel: AIModelNames) {
+    if (aiModel.startsWith("gpt")) {
+      return LangchainModels.gpt({
+        modelName: aiModel,
+      });
+    }
+
+    if (aiModel.startsWith("gemini")) {
+      return LangchainModels.gemini({
+        model: aiModel,
+      });
+    }
+
+    throw new Error("Model not supported");
+  }
+
+  private standardAgent(params: CallParams): Parameters<typeof createAgent>[0] {
+    const { aiModel, systemPrompt, maxRetries = 3, middleware, tools } = params;
+
+    const model = this.getModel(aiModel);
+    return {
+      model,
+      systemPrompt: systemPrompt ?? "",
+      middleware: [
+        ...this.standardMiddlewares(maxRetries),
+        ...(middleware ?? []),
+      ],
+      tools: tools ?? [],
+      responseFormat: undefined as any,
+    };
+  }
+
+  private standardMiddlewares(maxRetries: number) {
+    return [
+      modelRetryMiddleware({
+        maxRetries,
+        backoffFactor: 2.0,
+        initialDelayMs: 1000,
+      }),
+      modelFallbackMiddleware("gemini-2.5-flash", "gpt-4o-mini"),
+    ];
+  }
+}
+
+const example = async () => {
+  const langchain = new Langchain();
+
+  const outputSchema = z.object({
+    answer: z.string(),
+  });
+
+  const messages = [
+    LangchainMessages.system("You are a helpful assistant."),
+    LangchainMessages.human("What is the capital of France?"),
+  ];
+
+  const { response } = await langchain.callStructuredOutput({
+    aiModel: "gemini-2.5-flash",
+    outputSchema,
+    messages,
+  });
+
+  console.log({ response });
+};
+
+example();

package/package.json

@@ -0,0 +1,25 @@
+{
+  "name": "@luanpoppe/ai",
+  "version": "1.0.0",
+  "description": "",
+  "main": "dist/index.js",
+  "keywords": [],
+  "author": "",
+  "license": "ISC",
+  "dependencies": {
+    "@langchain/core": "^1.1.8",
+    "@langchain/google-genai": "^2.1.3",
+    "@langchain/openai": "^1.2.0",
+    "dotenv": "^17.2.3",
+    "langchain": "^1.2.3",
+    "typescript": "^5.9.3",
+    "zod": "^4.2.1"
+  },
+  "devDependencies": {
+    "@types/node": "^25.0.3",
+    "tsx": "^4.21.0"
+  },
+  "scripts": {
+    "build": "tsc"
+  }
+}

package/src/@types/model-names.ts

@@ -0,0 +1,16 @@
+type ChatGPTModels =
+  | "gpt-4"
+  | "gpt-4o"
+  | "gpt-4.1"
+  | "gpt-5"
+  | "gpt-5.1"
+  | "gpt-5-mini"
+  | "gpt-5-nano";
+
+type GeminiModels =
+  | "gemini-2.5-flash"
+  | "gemini-2.5-pro"
+  | "gemini-3-flash"
+  | "gemini-3-pro";
+
+export type AIModelNames = ChatGPTModels | GeminiModels;

package/src/core/env.ts

@@ -0,0 +1,13 @@
+import "dotenv/config";
+import z from "zod";
+
+const envSchema = z.object({
+  OPENAI_API_KEY: z.string().optional().default(""),
+  GOOGLE_GENAI_API_KEY: z.string().optional().default(""),
+});
+
+const { error, data } = envSchema.safeParse(process.env);
+
+if (error) throw new Error(`Invalid environment variables:\n${error.message}`);
+
+export const env = data;

package/src/langchain/messages.ts

@@ -0,0 +1,15 @@
+import { SystemMessage, HumanMessage, AIMessage } from "langchain";
+
+export type MessageInput = SystemMessage | HumanMessage | AIMessage;
+
+export class LangchainMessages {
+  static system(message: string): SystemMessage {
+    return new SystemMessage(message);
+  }
+  static human(message: string): HumanMessage {
+    return new HumanMessage(message);
+  }
+  static ai(message: string): AIMessage {
+    return new AIMessage(message);
+  }
+}

package/src/langchain/models.ts

@@ -0,0 +1,26 @@
+import {
+  ChatGoogleGenerativeAI,
+  GoogleGenerativeAIChatInput,
+} from "@langchain/google-genai";
+import { ChatOpenAI, ChatOpenAIFields } from "@langchain/openai";
+import { env } from "../core/env";
+
+export class LangchainModels {
+  static gpt(params: ChatOpenAIFields) {
+    console.log({ env });
+    return new ChatOpenAI({
+      temperature: 0,
+      maxTokens: 2048,
+      apiKey: env.OPENAI_API_KEY,
+      ...params,
+    });
+  }
+
+  static gemini(params: GoogleGenerativeAIChatInput) {
+    return new ChatGoogleGenerativeAI({
+      maxOutputTokens: 2048,
+      apiKey: env.GOOGLE_GENAI_API_KEY,
+      ...params,
+    });
+  }
+}

package/src/langchain/tools.ts

@@ -0,0 +1,22 @@
+import { tool } from "langchain";
+import z from "zod";
+
+export type CreateToolParams = {
+  toolFunction: Parameters<typeof tool>[0];
+  name: string;
+  description: string;
+  schema: z.ZodSchema;
+};
+
+export class LangchainTools {
+  createTool(params: CreateToolParams) {
+    const { toolFunction } = params;
+    const { name, description, schema } = params;
+
+    return tool(toolFunction, {
+      name,
+      description,
+      schema,
+    });
+  }
+}

package/tsconfig.json

@@ -0,0 +1,43 @@
+{
+  // Visit https://aka.ms/tsconfig to read more about this file
+  "compilerOptions": {
+    // File Layout
+    // "rootDir": "./src",
+    // "outDir": "./dist",
+
+    // Environment Settings
+    // See also https://aka.ms/tsconfig/module
+    "module": "nodenext",
+    "target": "esnext",
+    "types": ["@types/node"],
+    // For nodejs:
+    // "lib": ["esnext"],
+    // "types": ["node"],
+    // and npm install -D @types/node
+
+    // Other Outputs
+    "sourceMap": true,
+    "declaration": true,
+    "declarationMap": true,
+
+    // Stricter Typechecking Options
+    "noUncheckedIndexedAccess": true,
+    "exactOptionalPropertyTypes": true,
+
+    // Style Options
+    // "noImplicitReturns": true,
+    // "noImplicitOverride": true,
+    // "noUnusedLocals": true,
+    // "noUnusedParameters": true,
+    // "noFallthroughCasesInSwitch": true,
+    // "noPropertyAccessFromIndexSignature": true,
+
+    // Recommended Options
+    "strict": true,
+    "verbatimModuleSyntax": false,
+    "isolatedModules": true,
+    "noUncheckedSideEffectImports": true,
+    "moduleDetection": "force",
+    "skipLibCheck": true
+  }
+}