Skip to content

Commit

Permalink
add basic tests
Browse files Browse the repository at this point in the history
  • Loading branch information
wtfsayo committed Feb 13, 2025
1 parent 43bc95d commit 3b33125
Showing 1 changed file with 79 additions and 0 deletions.
79 changes: 79 additions & 0 deletions packages/plugin-openai/src/index.ts
Original file line number Diff line number Diff line change
Expand Up @@ -291,6 +291,85 @@ export const openaiPlugin: Plugin = {
return data.text;
},
},
tests: [
{
name: "openai_plugin_tests",
tests: [
{
name: 'test_url_and_api_key_validation',
fn: async (runtime) => {
const baseURL =
runtime.getSetting("OPENAI_BASE_URL") ?? "https://api.openai.com/v1";
const response = await fetch(`${baseURL}/models`, {
headers: { Authorization: `Bearer ${runtime.getSetting("OPENAI_API_KEY")}` },
});
console.log("response", await response.json());
if (!response.ok) {
throw new Error(`Failed to validate OpenAI API key: ${response.statusText}`);
}
}
},
{
name: 'test_text_large',
fn: async (runtime) => {
console.log("test_openai_plugin");
try {
const text = await runtime.useModel(ModelClass.TEXT_LARGE, {
context: "Test Mode: Reply with a short answer",
prompt: "What is the nature of reality?",
});
if (text.length === 0) {
throw new Error("Failed to generate text");
}
console.log("generated with test_text_large:", text);
} catch (error) {
console.error("Error in test_text_large:", error);
throw error;
}
}
},
{
name: 'test_text_small',
fn: async (runtime) => {
console.log("test_openai_plugin");
try {
const text = await runtime.useModel(ModelClass.TEXT_SMALL, {
context: "Test Mode:",
prompt: "What is the nature of reality in 10 words?",
});
if (text.length === 0) {
throw new Error("Failed to generate text");
}
console.log("generated with test_text_small:", text);
} catch (error) {
console.error("Error in test_text_small:", error);
throw error;
}
}
},
{
name: 'test_image_generation',
fn: async (runtime) => {
console.log("test_openai_plugin");
try {
const image = await runtime.useModel(ModelClass.IMAGE, {
prompt: "A beautiful sunset over a calm ocean",
n: 1,
size: "1024x1024"
});
if (image.length === 0) {
throw new Error("Failed to generate image");
}
console.log("generated with test_image_generation:", image);
} catch (error) {
console.error("Error in test_image_generation:", error);
throw error;
}
}
}
]
}
],
routes: [
{
path: "/helloworld",
Expand Down

0 comments on commit 3b33125

Please sign in to comment.