chore: import upstream snapshot with attribution

test/prompts/processors/python.utils.test.ts

@@ -0,0 +1,91 @@
+import { PythonShell } from 'python-shell';
+import { describe, expect, it, vi } from 'vitest';
+import logger from '../../../src/logger';
+import {
+  pythonPromptFunction,
+  pythonPromptFunctionLegacy,
+} from '../../../src/prompts/processors/python';
+import { runPython } from '../../../src/python/pythonUtils';
+
+import type { ApiProvider } from '../../../src/types/index';
+
+vi.mock('fs');
+vi.mock('python-shell');
+vi.mock('../../../src/python/pythonUtils');
+vi.mock('../../../src/logger', () => ({
+  default: {
+    debug: vi.fn(),
+    error: vi.fn(),
+    warn: vi.fn(),
+    info: vi.fn(),
+  },
+}));
+
+describe('pythonPromptFunction', () => {
+  interface PythonContext {
+    vars: Record<string, string | object>;
+    provider: ApiProvider;
+  }
+
+  it('should call python wrapper function with correct arguments', async () => {
+    const filePath = 'path/to/script.py';
+    const functionName = 'testFunction';
+    const context = {
+      vars: { key: 'value' },
+      provider: {
+        id: () => 'providerId',
+        label: 'providerLabel',
+        callApi: vi.fn(),
+      } as ApiProvider,
+    } as PythonContext;
+
+    const mockRunPython = vi.mocked(runPython);
+    mockRunPython.mockResolvedValue('mocked result');
+
+    await expect(pythonPromptFunction(filePath, functionName, context)).resolves.toBe(
+      'mocked result',
+    );
+    expect(mockRunPython).toHaveBeenCalledWith(filePath, functionName, [
+      {
+        ...context,
+        provider: {
+          id: 'providerId',
+          label: 'providerLabel',
+        },
+        config: {},
+      },
+    ]);
+  });
+
+  it('should call legacy function with correct arguments', async () => {
+    const filePath = 'path/to/script.py';
+    const context = {
+      vars: { key: 'value' },
+      provider: { id: () => 'providerId', label: 'providerLabel' } as ApiProvider,
+    } as PythonContext;
+    const mockPythonShellRun = vi.mocked(PythonShell.run);
+    const mockLoggerDebug = vi.mocked(logger.debug);
+    mockPythonShellRun.mockImplementation(() => {
+      return Promise.resolve(['mocked result']);
+    });
+    await expect(pythonPromptFunctionLegacy(filePath, context)).resolves.toBe('mocked result');
+    expect(mockPythonShellRun).toHaveBeenCalledWith(filePath, {
+      mode: 'text',
+      pythonPath: process.env.PROMPTFOO_PYTHON || 'python',
+      args: [
+        JSON.stringify({
+          vars: context.vars,
+          provider: {
+            id: context.provider.id(),
+            label: context.provider.label,
+          },
+          config: {},
+        }),
+      ],
+    });
+    expect(mockLoggerDebug).toHaveBeenCalledWith(`Executing python prompt script ${filePath}`);
+    expect(mockLoggerDebug).toHaveBeenCalledWith(
+      `Python prompt script ${filePath} returned: mocked result`,
+    );
+  });
+});