@xuda.io/ai_module 1.1.5616 → 1.1.5618

package/gpt.mjs

@@ -0,0 +1,155 @@
+// process.env.NODE_TLS_REJECT_UNAUTHORIZED = '0';
+
+import { Agent, run, tool, user } from '@openai/agents';
+import { z } from 'zod';
+
+import OpenAI from 'openai';
+const client = new OpenAI({ apiKey: process.env['OPENAI_API_KEY'] });
+
+import dotenv from 'dotenv';
+dotenv.config();
+
+// import { setDefaultOpenAIKey } from '@openai/agents';
+// setDefaultOpenAIKey('sk-proj-XmeLaZ7RLn0nCc4nSaJHT3BlbkFJfKFulKrY55ILLnM1CytS'); // sk-...
+
+const test_gpt = async function (req) {
+  try {
+    const spanishAgent = new Agent({
+      name: 'Spanish agent',
+      instructions: 'You only speak Spanish.',
+    });
+
+    const englishAgent = new Agent({
+      name: 'English agent',
+      instructions: 'You only speak English',
+    });
+
+    const triageAgent = new Agent({
+      name: 'Triage agent',
+      instructions: 'Handoff to the appropriate agent based on the language of the request.',
+      handoffs: [spanishAgent, englishAgent],
+    });
+
+    const result = await run(triageAgent, 'how are you?');
+    console.log(result.finalOutput);
+  } catch (err) {
+    console.error(err);
+  }
+};
+
+const test_tool = async function (req) {
+  try {
+    const getWeather = tool({
+      name: 'get_weather',
+      description: 'Return the weather for a given city.',
+      parameters: z.object({ city: z.string() }),
+      async execute({ city }) {
+        return `The weather in ${city} is sunny.`;
+      },
+    });
+
+    const agent = new Agent({
+      name: 'Weather bot',
+      instructions: 'You are a helpful weather bot.',
+      model: 'o4-mini',
+      tools: [getWeather],
+    });
+
+    const result = await run(agent, 'what is the weather in Herzeliya?');
+    console.log(result.finalOutput);
+  } catch (err) {
+    console.error(err);
+  }
+};
+const test_user = async function (req) {
+  try {
+    const agent = new Agent({
+      name: 'Assistant',
+      instructions: 'You are a helpful assistant.',
+    });
+
+    const result = await run(agent, 'hi');
+    console.log(result.finalOutput);
+  } catch (err) {
+    console.error(err);
+  }
+};
+const test_output = async function (req) {
+  try {
+    const refundAgent = new Agent({
+      name: 'Refund Agent',
+      instructions: 'You are a refund agent. You are responsible for refunding customers, refund anyone with $5.',
+      outputType: z.object({
+        refundApproved: z.boolean(),
+      }),
+    });
+
+    const orderAgent = new Agent({
+      name: 'Order Agent',
+      instructions: 'You are an order agent. You are responsible for processing orders.',
+      outputType: z.object({
+        orderId: z.string(),
+      }),
+    });
+
+    const triageAgent = Agent.create({
+      name: 'Triage Agent',
+      instructions: 'You are a triage agent. You are responsible for triaging customer issues.',
+      handoffs: [refundAgent, orderAgent],
+    });
+
+    const result = await run(triageAgent, 'I need to a refund for my order');
+
+    console.log(result.finalOutput);
+  } catch (err) {
+    console.error(err);
+  }
+};
+const test_mcp = async function (req) {
+  debugger;
+  try {
+    const resp = await client.responses.create({
+      model: 'gpt-4.1',
+      tools: [
+        {
+          type: 'mcp',
+          server_label: 'mcp',
+          server_url: 'https://mcp.gotopanel.com/mcp',
+          require_approval: 'never',
+          // allowed_tools: ['say_hello'],
+        },
+      ],
+      input: 'Use the say_hello tool to greet someone named Alice.',
+    });
+
+    console.log(resp.output_text);
+  } catch (err) {
+    console.error(err);
+  }
+};
+
+const test_mcp_tool = async function (req) {
+  try {
+    const agent = new Agent({
+      name: 'Weather bot',
+      instructions: 'You are a helpful weather bot.',
+      model: 'gpt-4.1',
+      tools: [
+        {
+          type: 'mcp',
+          server_label: 'mcp',
+          server_url: 'https://mcp.xuda.dev/mcp',
+          require_approval: 'never',
+        },
+      ],
+    });
+
+    const result = await run(agent, 'Use the say_hello tool to greet someone named Alice.');
+    console.log(result.finalOutput);
+  } catch (err) {
+    console.error(err);
+  }
+};
+
+setTimeout(test_mcp, 10000);
+// test_mcp({ prompt: process.argv[2] });