@luanpoppe/ai 1.0.2 → 1.0.4

package/dist/index.d.ts

@@ -1,40 +1,47 @@
-import { LangchainModels } from "./langchain/models";
+import { LangchainModels, LLMModelConfig } from "./langchain/models";
 import { AIModelNames } from "./@types/model-names";
 import z from "zod";
 import { MessageInput } from "./langchain/messages";
 import { AgentMiddleware, BaseMessage } from "langchain";
 import { ClientTool, ServerTool } from "@langchain/core/tools";
 import { LangchainMessages } from "./langchain/messages";
+import { LangchainTools } from "./langchain/tools";
 type LangchainConstructor = {
     googleGeminiToken?: string;
     openAIApiKey?: string;
 };
-export type CallParams = {
+export type LangchainCallParams = {
+    agent: {
+        middleware?: AgentMiddleware[];
+        tools?: (ServerTool | ClientTool)[];
+    };
+    model: Omit<LLMModelConfig, "apiKey" | "model">;
     aiModel: AIModelNames;
     messages: MessageInput[];
     systemPrompt?: string;
     maxRetries?: number;
-    middleware?: AgentMiddleware[];
-    tools?: (ServerTool | ClientTool)[];
 };
-export type CallReturn = Promise<{
+export type LangchainCallReturn = Promise<{
     text: string;
     messages: BaseMessage[];
 }>;
-export type CallStructuredOutputParams<T extends z.ZodSchema> = CallParams & {
+export type LangchainCallStructuredOutputParams<T extends z.ZodSchema> = LangchainCallParams & {
     outputSchema: T;
 };
-export type CallStructuredOutputReturn<T> = Promise<{
+export type LangchainCallStructuredOutputReturn<T> = Promise<{
     response: z.infer<T>;
 }>;
 export declare class Langchain {
     private tokens;
     constructor(tokens: LangchainConstructor);
-    call(params: CallParams): CallReturn;
-    callStructuredOutput<T extends z.ZodSchema>(params: CallStructuredOutputParams<T>): CallStructuredOutputReturn<typeof params.outputSchema>;
+    call(params: LangchainCallParams): LangchainCallReturn;
+    callStructuredOutput<T extends z.ZodSchema>(params: LangchainCallStructuredOutputParams<T>): LangchainCallStructuredOutputReturn<typeof params.outputSchema>;
+    getRawAgent(params: LangchainCallParams, outputSchema?: z.ZodSchema | undefined): {
+        agent: import("langchain").ReactAgent<Record<string, any>, import("@langchain/core/utils/types").InteropZodObject | import("langchain").AnyAnnotationRoot | undefined, import("@langchain/core/utils/types").InteropZodObject | import("langchain").AnyAnnotationRoot, readonly AgentMiddleware<any, any, any>[]>;
+    };
     private getModel;
     private standardAgent;
     private standardMiddlewares;
 }
-export { LangchainModels, LangchainMessages };
+export { LangchainModels, LangchainMessages, LangchainTools };
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,CAAC,MAAM,KAAK,CAAC;AACpB,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,EACL,eAAe,EACf,WAAW,EAIZ,MAAM,WAAW,CAAC;AACnB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAC/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAEzD,KAAK,oBAAoB,GAAG;IAC1B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,CAAC;AAEF,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;IACR,OAAO,CAAC,MAAM;gBAAN,MAAM,EAAE,oBAAoB;IAE1C,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;IAkBhB,OAAO,CAAC,aAAa;IAgBrB,OAAO,CAAC,mBAAmB;CAU5B;AAED,OAAO,EAAE,eAAe,EAAE,iBAAiB,EAAE,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACrE,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,CAAC,MAAM,KAAK,CAAC;AACpB,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,EACL,eAAe,EACf,WAAW,EAIZ,MAAM,WAAW,CAAC;AACnB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAC/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AACzD,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAEnD,KAAK,oBAAoB,GAAG;IAC1B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG;IAChC,KAAK,EAAE;QACL,UAAU,CAAC,EAAE,eAAe,EAAE,CAAC;QAC/B,KAAK,CAAC,EAAE,CAAC,UAAU,GAAG,UAAU,CAAC,EAAE,CAAC;KACrC,CAAC;IAEF,KAAK,EAAE,IAAI,CAAC,cAAc,EAAE,QAAQ,GAAG,OAAO,CAAC,CAAC;IAEhD,OAAO,EAAE,YAAY,CAAC;IACtB,QAAQ,EAAE,YAAY,EAAE,CAAC;IACzB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG,OAAO,CAAC;IACxC,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,WAAW,EAAE,CAAC;CACzB,CAAC,CAAC;AAEH,MAAM,MAAM,mCAAmC,CAAC,CAAC,SAAS,CAAC,CAAC,SAAS,IACnE,mBAAmB,GAAG;IACpB,YAAY,EAAE,CAAC,CAAC;CACjB,CAAC;AAEJ,MAAM,MAAM,mCAAmC,CAAC,CAAC,IAAI,OAAO,CAAC;IAC3D,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;CACtB,CAAC,CAAC;AAEH,qBAAa,SAAS;IACR,OAAO,CAAC,MAAM;gBAAN,MAAM,EAAE,oBAAoB;IAE1C,IAAI,CAAC,MAAM,EAAE,mBAAmB,GAAG,mBAAmB;IAiBtD,oBAAoB,CAAC,CAAC,SAAS,CAAC,CAAC,SAAS,EAC9C,MAAM,EAAE,mCAAmC,CAAC,CAAC,CAAC,GAC7C,mCAAmC,CAAC,OAAO,MAAM,CAAC,YAAY,CAAC;IAiBlE,WAAW,CACT,MAAM,EAAE,mBAAmB,EAC3B,YAAY,CAAC,EAAE,CAAC,CAAC,SAAS,GAAG,SAAS;;;IAUxC,OAAO,CAAC,QAAQ;IAyBhB,OAAO,CAAC,aAAa;IAmBrB,OAAO,CAAC,mBAAmB;CAU5B;AAED,OAAO,EAAE,eAAe,EAAE,iBAAiB,EAAE,cAAc,EAAE,CAAC"}

package/dist/index.js

@@ -1,11 +1,13 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.LangchainMessages = exports.LangchainModels = exports.Langchain = void 0;
+exports.LangchainTools = exports.LangchainMessages = exports.LangchainModels = exports.Langchain = void 0;
 const models_1 = require("./langchain/models");
 Object.defineProperty(exports, "LangchainModels", { enumerable: true, get: function () { return models_1.LangchainModels; } });
 const langchain_1 = require("langchain");
 const messages_1 = require("./langchain/messages");
 Object.defineProperty(exports, "LangchainMessages", { enumerable: true, get: function () { return messages_1.LangchainMessages; } });
+const tools_1 = require("./langchain/tools");
+Object.defineProperty(exports, "LangchainTools", { enumerable: true, get: function () { return tools_1.LangchainTools; } });
 class Langchain {
     tokens;
     constructor(tokens) {
@@ -18,7 +20,8 @@ class Langchain {
         });
         const response = await agent.invoke({ messages });
         return {
-            text: response.messages[0]?.text ?? "Empty response from the model",
+            text: response.messages.at(-1)?.content ??
+                "Empty response from the model",
             messages: response.messages,
         };
     }
@@ -34,24 +37,35 @@ class Langchain {
         const parsedResponse = outputSchema.parse(response?.structuredResponse);
         return { response: parsedResponse };
     }
-    getModel(aiModel) {
+    getRawAgent(params, outputSchema) {
+        const agent = (0, langchain_1.createAgent)({
+            ...this.standardAgent(params),
+            responseFormat: outputSchema,
+        });
+        return { agent };
+    }
+    getModel(params) {
+        const { aiModel, model } = params;
+        const { maxTokens, temperature } = model;
+        const config = {
+            model: aiModel,
+            maxTokens: maxTokens,
+            temperature: temperature,
+        };
         if (aiModel.startsWith("gpt")) {
-            return models_1.LangchainModels.gpt({
-                modelName: aiModel,
-                apiKey: this.tokens.openAIApiKey,
-            });
+            config.apiKey = this.tokens.openAIApiKey;
+            return models_1.LangchainModels.gpt(config);
         }
         if (aiModel.startsWith("gemini")) {
-            return models_1.LangchainModels.gemini({
-                model: aiModel,
-                apiKey: this.tokens.googleGeminiToken ?? "",
-            });
+            config.apiKey = this.tokens.googleGeminiToken;
+            return models_1.LangchainModels.gemini(config);
         }
         throw new Error("Model not supported");
     }
     standardAgent(params) {
-        const { aiModel, systemPrompt, maxRetries = 3, middleware, tools } = params;
-        const model = this.getModel(aiModel);
+        const { systemPrompt, maxRetries = 3 } = params;
+        const { middleware, tools } = params.agent;
+        const model = this.getModel(params);
         return {
             model,
             systemPrompt: systemPrompt ?? "",

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,+CAAqD;AA4H5C,gGA5HA,wBAAe,OA4HA;AAxHxB,yCAMmB;AAEnB,mDAAyD;AAgH/B,kGAhHjB,4BAAiB,OAgHiB;AAnF3C,MAAa,SAAS;IACA;IAApB,YAAoB,MAA4B;QAA5B,WAAM,GAAN,MAAM,CAAsB;IAAG,CAAC;IAEpD,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;gBAClB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;aACjC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YACjC,OAAO,wBAAe,CAAC,MAAM,CAAC;gBAC5B,KAAK,EAAE,OAAO;gBACd,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,iBAAiB,IAAI,EAAE;aAC5C,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;AAjFD,8BAiFC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,+CAAqE;AA2J5D,gGA3JA,wBAAe,OA2JA;AAvJxB,yCAMmB;AAEnB,mDAAyD;AA+I/B,kGA/IjB,4BAAiB,OA+IiB;AA9I3C,6CAAmD;AA8IN,+FA9IpC,sBAAc,OA8IoC;AA3G3D,MAAa,SAAS;IACA;IAApB,YAAoB,MAA4B;QAA5B,WAAM,GAAN,MAAM,CAAsB;IAAG,CAAC;IAEpD,KAAK,CAAC,IAAI,CAAC,MAA2B;QACpC,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,EACD,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,OAAkB;gBAC7C,+BAA+B;YACjC,QAAQ,EAAE,QAAQ,CAAC,QAAQ;SAC5B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,oBAAoB,CACxB,MAA8C;QAE9C,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;IAED,WAAW,CACT,MAA2B,EAC3B,YAAsC;QAEtC,MAAM,KAAK,GAAG,IAAA,uBAAW,EAAC;YACxB,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC;YAC7B,cAAc,EAAE,YAAmB;SACpC,CAAC,CAAC;QAEH,OAAO,EAAE,KAAK,EAAE,CAAC;IACnB,CAAC;IAEO,QAAQ,CAAC,MAA2B;QAC1C,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;QAClC,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,GAAG,KAAK,CAAC;QAEzC,MAAM,MAAM,GAAmB;YAC7B,KAAK,EAAE,OAAO;YACd,SAAS,EAAE,SAAS;YACpB,WAAW,EAAE,WAAW;SACzB,CAAC;QAEF,IAAI,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YAC9B,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;YAEzC,OAAO,wBAAe,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACrC,CAAC;QAED,IAAI,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YACjC,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC;YAE9C,OAAO,wBAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACxC,CAAC;QAED,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;IACzC,CAAC;IAEO,aAAa,CACnB,MAA2B;QAE3B,MAAM,EAAE,YAAY,EAAE,UAAU,GAAG,CAAC,EAAE,GAAG,MAAM,CAAC;QAChD,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC,KAAK,CAAC;QAE3C,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACpC,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;AAzGD,8BAyGC"}

package/dist/langchain/models.d.ts

@@ -1,7 +1,13 @@
-import { ChatGoogleGenerativeAI, GoogleGenerativeAIChatInput } from "@langchain/google-genai";
-import { ChatOpenAI, ChatOpenAIFields } from "@langchain/openai";
+import { ChatGoogleGenerativeAI } from "@langchain/google-genai";
+import { ChatOpenAI } from "@langchain/openai";
+export type LLMModelConfig = {
+    model: string;
+    apiKey?: string | undefined;
+    maxTokens?: number | undefined;
+    temperature?: number | undefined;
+};
 export declare class LangchainModels {
-    static gpt(params: ChatOpenAIFields): ChatOpenAI<import("@langchain/openai").ChatOpenAICallOptions>;
-    static gemini(params: GoogleGenerativeAIChatInput): ChatGoogleGenerativeAI;
+    static gpt(params: LLMModelConfig): ChatOpenAI<import("@langchain/openai").ChatOpenAICallOptions>;
+    static gemini(params: LLMModelConfig): ChatGoogleGenerativeAI;
 }
 //# sourceMappingURL=models.d.ts.map

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

@@ -1 +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;AAEjE,qBAAa,eAAe;IAC1B,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,gBAAgB;IAUnC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,2BAA2B;CAWlD"}
+{"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../../src/langchain/models.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,sBAAsB,EAEvB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,UAAU,EAAoB,MAAM,mBAAmB,CAAC;AAEjE,MAAM,MAAM,cAAc,GAAG;IAC3B,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC5B,SAAS,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC/B,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAClC,CAAC;AAEF,qBAAa,eAAe;IAC1B,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,cAAc;IAgBjC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,cAAc;CAkBrC"}

package/dist/langchain/models.js

@@ -5,20 +5,32 @@ const google_genai_1 = require("@langchain/google-genai");
 const openai_1 = require("@langchain/openai");
 class LangchainModels {
     static gpt(params) {
-        if (!params.apiKey)
+        const { model, apiKey, maxTokens, temperature } = params;
+        if (!apiKey)
             throw new Error("OpenAI API key is not passed in the model parameters");
-        return new openai_1.ChatOpenAI({
-            maxTokens: 2048,
-            ...params,
-        });
+        const options = {
+            model,
+            apiKey,
+        };
+        if (maxTokens)
+            options.maxTokens = maxTokens;
+        if (temperature)
+            options.temperature = temperature;
+        return new openai_1.ChatOpenAI(options);
     }
     static gemini(params) {
-        if (!params.apiKey)
+        const { apiKey, maxTokens, model, temperature } = params;
+        if (!apiKey)
             throw new Error("Google Gemini API key is not passed in the model parameters");
-        return new google_genai_1.ChatGoogleGenerativeAI({
-            maxOutputTokens: 2048,
-            ...params,
-        });
+        const options = {
+            model,
+            apiKey,
+        };
+        if (maxTokens)
+            options.maxOutputTokens = maxTokens;
+        if (temperature)
+            options.temperature = temperature;
+        return new google_genai_1.ChatGoogleGenerativeAI(options);
     }
 }
 exports.LangchainModels = LangchainModels;

package/dist/langchain/models.js.map

@@ -1 +1 @@
-{"version":3,"file":"models.js","sourceRoot":"","sources":["../../src/langchain/models.ts"],"names":[],"mappings":";;;AAAA,0DAGiC;AACjC,8CAAiE;AAEjE,MAAa,eAAe;IAC1B,MAAM,CAAC,GAAG,CAAC,MAAwB;QACjC,IAAI,CAAC,MAAM,CAAC,MAAM;YAChB,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;QAE1E,OAAO,IAAI,mBAAU,CAAC;YACpB,SAAS,EAAE,IAAI;YACf,GAAG,MAAM;SACV,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CAAC,MAAM,CAAC,MAAmC;QAC/C,IAAI,CAAC,MAAM,CAAC,MAAM;YAChB,MAAM,IAAI,KAAK,CACb,6DAA6D,CAC9D,CAAC;QAEJ,OAAO,IAAI,qCAAsB,CAAC;YAChC,eAAe,EAAE,IAAI;YACrB,GAAG,MAAM;SACV,CAAC,CAAC;IACL,CAAC;CACF;AAtBD,0CAsBC"}
+{"version":3,"file":"models.js","sourceRoot":"","sources":["../../src/langchain/models.ts"],"names":[],"mappings":";;;AAAA,0DAGiC;AACjC,8CAAiE;AASjE,MAAa,eAAe;IAC1B,MAAM,CAAC,GAAG,CAAC,MAAsB;QAC/B,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,GAAG,MAAM,CAAC;QACzD,IAAI,CAAC,MAAM;YACT,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;QAE1E,MAAM,OAAO,GAAqB;YAChC,KAAK;YACL,MAAM;SACP,CAAC;QAEF,IAAI,SAAS;YAAE,OAAO,CAAC,SAAS,GAAG,SAAS,CAAC;QAC7C,IAAI,WAAW;YAAE,OAAO,CAAC,WAAW,GAAG,WAAW,CAAC;QAEnD,OAAO,IAAI,mBAAU,CAAC,OAAO,CAAC,CAAC;IACjC,CAAC;IAED,MAAM,CAAC,MAAM,CAAC,MAAsB;QAClC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,WAAW,EAAE,GAAG,MAAM,CAAC;QAEzD,IAAI,CAAC,MAAM;YACT,MAAM,IAAI,KAAK,CACb,6DAA6D,CAC9D,CAAC;QAEJ,MAAM,OAAO,GAAgC;YAC3C,KAAK;YACL,MAAM;SACP,CAAC;QAEF,IAAI,SAAS;YAAE,OAAO,CAAC,eAAe,GAAG,SAAS,CAAC;QACnD,IAAI,WAAW;YAAE,OAAO,CAAC,WAAW,GAAG,WAAW,CAAC;QAEnD,OAAO,IAAI,qCAAsB,CAAC,OAAO,CAAC,CAAC;IAC7C,CAAC;CACF;AAnCD,0CAmCC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@luanpoppe/ai",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "description": "",
   "main": "dist/index.js",
   "keywords": [],
@@ -19,6 +19,7 @@
     "typescript": "^5.9.3"
   },
   "scripts": {
-    "build": "tsc"
+    "build": "tsc",
+    "pub": "pnpm build && pnpm publish --access=public"
   }
 }

package/src/index.ts

@@ -1,4 +1,4 @@
-import { LangchainModels } from "./langchain/models";
+import { LangchainModels, LLMModelConfig } from "./langchain/models";
 import { AIModelNames } from "./@types/model-names";
 import z from "zod";
 import { MessageInput } from "./langchain/messages";
@@ -19,12 +19,17 @@ type LangchainConstructor = {
 };
 
 export type LangchainCallParams = {
+  agent: {
+    middleware?: AgentMiddleware[];
+    tools?: (ServerTool | ClientTool)[];
+  };
+
+  model: Omit<LLMModelConfig, "apiKey" | "model">;
+
   aiModel: AIModelNames;
   messages: MessageInput[];
   systemPrompt?: string;
   maxRetries?: number;
-  middleware?: AgentMiddleware[];
-  tools?: (ServerTool | ClientTool)[];
 };
 
 export type LangchainCallReturn = Promise<{
@@ -54,7 +59,9 @@ export class Langchain {
     const response = await agent.invoke({ messages });
 
     return {
-      text: response.messages.at(-1)?.text ?? "Empty response from the model",
+      text:
+        (response.messages.at(-1)?.content as string) ??
+        "Empty response from the model",
       messages: response.messages,
     };
   }
@@ -78,19 +85,38 @@ export class Langchain {
     return { response: parsedResponse };
   }
 
-  private getModel(aiModel: AIModelNames) {
+  getRawAgent(
+    params: LangchainCallParams,
+    outputSchema?: z.ZodSchema | undefined
+  ) {
+    const agent = createAgent({
+      ...this.standardAgent(params),
+      responseFormat: outputSchema as any,
+    });
+
+    return { agent };
+  }
+
+  private getModel(params: LangchainCallParams) {
+    const { aiModel, model } = params;
+    const { maxTokens, temperature } = model;
+
+    const config: LLMModelConfig = {
+      model: aiModel,
+      maxTokens: maxTokens,
+      temperature: temperature,
+    };
+
     if (aiModel.startsWith("gpt")) {
-      return LangchainModels.gpt({
-        modelName: aiModel,
-        apiKey: this.tokens.openAIApiKey,
-      });
+      config.apiKey = this.tokens.openAIApiKey;
+
+      return LangchainModels.gpt(config);
     }
 
     if (aiModel.startsWith("gemini")) {
-      return LangchainModels.gemini({
-        model: aiModel,
-        apiKey: this.tokens.googleGeminiToken ?? "",
-      });
+      config.apiKey = this.tokens.googleGeminiToken;
+
+      return LangchainModels.gemini(config);
     }
 
     throw new Error("Model not supported");
@@ -99,9 +125,10 @@ export class Langchain {
   private standardAgent(
     params: LangchainCallParams
   ): Parameters<typeof createAgent>[0] {
-    const { aiModel, systemPrompt, maxRetries = 3, middleware, tools } = params;
+    const { systemPrompt, maxRetries = 3 } = params;
+    const { middleware, tools } = params.agent;
 
-    const model = this.getModel(aiModel);
+    const model = this.getModel(params);
     return {
       model,
       systemPrompt: systemPrompt ?? "",

package/src/langchain/models.ts

@@ -4,26 +4,46 @@ import {
 } from "@langchain/google-genai";
 import { ChatOpenAI, ChatOpenAIFields } from "@langchain/openai";
 
+export type LLMModelConfig = {
+  model: string;
+  apiKey?: string | undefined;
+  maxTokens?: number | undefined;
+  temperature?: number | undefined;
+};
+
 export class LangchainModels {
-  static gpt(params: ChatOpenAIFields) {
-    if (!params.apiKey)
+  static gpt(params: LLMModelConfig) {
+    const { model, apiKey, maxTokens, temperature } = params;
+    if (!apiKey)
       throw new Error("OpenAI API key is not passed in the model parameters");
 
-    return new ChatOpenAI({
-      maxTokens: 2048,
-      ...params,
-    });
+    const options: ChatOpenAIFields = {
+      model,
+      apiKey,
+    };
+
+    if (maxTokens) options.maxTokens = maxTokens;
+    if (temperature) options.temperature = temperature;
+
+    return new ChatOpenAI(options);
   }
 
-  static gemini(params: GoogleGenerativeAIChatInput) {
-    if (!params.apiKey)
+  static gemini(params: LLMModelConfig) {
+    const { apiKey, maxTokens, model, temperature } = params;
+
+    if (!apiKey)
       throw new Error(
         "Google Gemini API key is not passed in the model parameters"
       );
 
-    return new ChatGoogleGenerativeAI({
-      maxOutputTokens: 2048,
-      ...params,
-    });
+    const options: GoogleGenerativeAIChatInput = {
+      model,
+      apiKey,
+    };
+
+    if (maxTokens) options.maxOutputTokens = maxTokens;
+    if (temperature) options.temperature = temperature;
+
+    return new ChatGoogleGenerativeAI(options);
   }
 }