alighasami
3d5eaf9445
Security: Sync from Public / sync-from-public (push) Has been cancelled
Test: Benchmark Nightly / build (push) Has been cancelled
Test: Benchmark Nightly / Notify Cats on failure (push) Has been cancelled
CI: Python / Checks (push) Has been cancelled
Test: Evals Python / Workflow Comparison Python (push) Has been cancelled
Util: Check Docs URLs / check-docs-urls (push) Has been cancelled
Test: Visual Storybook / Cloudflare Pages (push) Has been cancelled
Test: E2E Performance / build-and-test-performance (push) Has been cancelled
Test: Workflows Nightly / Run Workflow Tests (push) Has been cancelled
Util: Cleanup CI Docker Images / Delete stale CI images (push) Has been cancelled
Test: Benchmark Destroy Env / build (push) Has been cancelled
Util: Update Node Popularity / update-popularity (push) Has been cancelled
Test: E2E Coverage Weekly / Coverage Tests (push) Has been cancelled
46 lines
1.3 KiB
TypeScript
46 lines
1.3 KiB
TypeScript
import type { Tool } from '@langchain/core/tools';
|
|
import type { OpenAIClient } from '@langchain/openai';
|
|
import { zodToJsonSchema } from 'zod-to-json-schema';
|
|
|
|
// Copied from langchain(`langchain/src/tools/convert_to_openai.ts`)
|
|
// since these functions are not exported
|
|
|
|
/**
|
|
* Formats a `Tool` instance into a format that is compatible
|
|
* with OpenAI's ChatCompletionFunctions. It uses the `zodToJsonSchema`
|
|
* function to convert the schema of the tool into a JSON
|
|
* schema, which is then used as the parameters for the OpenAI function.
|
|
*/
|
|
export function formatToOpenAIFunction(
|
|
tool: Tool,
|
|
): OpenAIClient.Chat.ChatCompletionCreateParams.Function {
|
|
return {
|
|
name: tool.name,
|
|
description: tool.description,
|
|
parameters: zodToJsonSchema(tool.schema),
|
|
};
|
|
}
|
|
|
|
export function formatToOpenAITool(tool: Tool): OpenAIClient.Chat.ChatCompletionTool {
|
|
const schema = zodToJsonSchema(tool.schema);
|
|
return {
|
|
type: 'function',
|
|
function: {
|
|
name: tool.name,
|
|
description: tool.description,
|
|
parameters: schema,
|
|
},
|
|
};
|
|
}
|
|
|
|
export function formatToOpenAIAssistantTool(tool: Tool): OpenAIClient.Beta.AssistantTool {
|
|
return {
|
|
type: 'function',
|
|
function: {
|
|
name: tool.name,
|
|
description: tool.description,
|
|
parameters: zodToJsonSchema(tool.schema),
|
|
},
|
|
};
|
|
}
|