Make OpenAI function chain docs more readable (langchain-ai#1902)

* Fix docs and Google Vertex integration test * Update functions docs style * Make OpenAI function chain docs more readable
shcallaway · Jul 9, 2023 · 5a2a361 · 5a2a361
1 parent 3635d96
commit 5a2a361
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 22 deletions.
diff --git a/examples/src/chains/openai_functions_extraction.ts b/examples/src/chains/openai_functions_extraction.ts
@@ -2,16 +2,18 @@ import { z } from "zod";
 import { ChatOpenAI } from "langchain/chat_models/openai";
 import { createExtractionChainFromZod } from "langchain/chains";
 
-const chain = createExtractionChainFromZod(
-  z.object({
-    "person-name": z.string().optional(),
-    "person-age": z.number().optional(),
-    "person-hair_color": z.string().optional(),
-    "dog-name": z.string().optional(),
-    "dog-breed": z.string().optional(),
-  }),
-  new ChatOpenAI({ modelName: "gpt-3.5-turbo-0613", temperature: 0 })
-);
+const zodSchema = z.object({
+  "person-name": z.string().optional(),
+  "person-age": z.number().optional(),
+  "person-hair_color": z.string().optional(),
+  "dog-name": z.string().optional(),
+  "dog-breed": z.string().optional(),
+});
+const chatModel = new ChatOpenAI({
+  modelName: "gpt-3.5-turbo-0613",
+  temperature: 0,
+});
+const chain = createExtractionChainFromZod(zodSchema, chatModel);
 
 console.log(
   await chain.run(`Alex is 5 feet tall. Claudia is 4 feet taller Alex and jumps higher than him. Claudia is a brunette and Alex is blonde.

diff --git a/examples/src/chains/openai_functions_openapi_customization.ts b/examples/src/chains/openai_functions_openapi_customization.ts
@@ -1,8 +1,10 @@
 import { createOpenAPIChain } from "langchain/chains";
 import { ChatOpenAI } from "langchain/chat_models/openai";
 
+const chatModel = new ChatOpenAI({ modelName: "gpt-4-0613", temperature: 0 });
+
 const chain = await createOpenAPIChain("https://api.speak.com/openapi.yaml", {
-  llm: new ChatOpenAI({ modelName: "gpt-4-0613", temperature: 0 }),
+  llm: chatModel,
   headers: {
     authorization: "Bearer SOME_TOKEN",
   },

diff --git a/examples/src/chains/openai_functions_tagging.ts b/examples/src/chains/openai_functions_tagging.ts
@@ -1,18 +1,20 @@
 import { createTaggingChain } from "langchain/chains";
 import { ChatOpenAI } from "langchain/chat_models/openai";
+import type { FunctionParameters } from "langchain/output_parsers";
 
-const chain = createTaggingChain(
-  {
-    type: "object",
-    properties: {
-      sentiment: { type: "string" },
-      tone: { type: "string" },
-      language: { type: "string" },
-    },
-    required: ["tone"],
+const schema: FunctionParameters = {
+  type: "object",
+  properties: {
+    sentiment: { type: "string" },
+    tone: { type: "string" },
+    language: { type: "string" },
   },
-  new ChatOpenAI({ modelName: "gpt-4-0613", temperature: 0 })
-);
+  required: ["tone"],
+};
+
+const chatModel = new ChatOpenAI({ modelName: "gpt-4-0613", temperature: 0 });
+
+const chain = createTaggingChain(schema, chatModel);
 
 console.log(
   await chain.run(

diff --git a/langchain/src/output_parsers/index.ts b/langchain/src/output_parsers/index.ts
@@ -12,6 +12,7 @@ export { CombiningOutputParser } from "./combining.js";
 export { RouterOutputParser, RouterOutputParserInput } from "./router.js";
 export { CustomListOutputParser } from "./list.js";
 export {
+  FunctionParameters,
   OutputFunctionsParser,
   JsonOutputFunctionsParser,
   JsonKeyOutputFunctionsParser,