core[minor]: Allow for simple tool schema to be passed to bindTools

furwellness · Jul 29, 2024 · a56b661 · a56b661
1 parent c374a01
commit a56b661
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/langchain-core/src/language_models/chat_models.ts b/langchain-core/src/language_models/chat_models.ts
@@ -33,7 +33,7 @@ import {
 } from "../callbacks/manager.js";
 import type { RunnableConfig } from "../runnables/config.js";
 import type { BaseCache } from "../caches/base.js";
-import { StructuredToolInterface } from "../tools/index.js";
+import { StructuredToolInterface, ToolSchema } from "../tools/index.js";
 import {
   Runnable,
   RunnableLambda,
@@ -184,6 +184,7 @@ export abstract class BaseChatModel<
       | Record<string, unknown>
       | ToolDefinition
       | RunnableToolLike
+      | ToolSchema
     )[],
     kwargs?: Partial<CallOptions>
   ): Runnable<BaseLanguageModelInput, OutputMessageType, CallOptions>;

diff --git a/langchain-core/src/tools/index.ts b/langchain-core/src/tools/index.ts
@@ -47,6 +47,21 @@ export interface ToolParams extends BaseLangChainParams {
   responseFormat?: ResponseFormat;
 }
 
+export interface ToolSchema {
+  /**
+   * The name of the tool to pass to the model.
+   */
+  name: string;
+  /**
+   * An optional description of the tool to pass to the model.
+   */
+  description?: string;
+  /**
+   * A Zod schema representing the parameters of the tool.
+   */
+  schema: ZodObjectAny;
+}
+
 export interface StructuredToolInterface<T extends ZodObjectAny = ZodObjectAny>
   extends RunnableInterface<
     (z.output<T> extends string ? string : never) | z.input<T> | ToolCall,