motia 0.7.2-beta.134 → 0.7.2-beta.135-685562

package/dist/cjs/cloud/build/builders/python/index.d.ts

@@ -6,7 +6,6 @@ export declare class PythonBuilder implements StepBuilder {
     private readonly listener;
     private packager;
     constructor(builder: Builder, listener: BuildListener);
-    private getRequirements;
     buildApiSteps(steps: Step<ApiRouteConfig>[]): Promise<RouterBuildResult>;
     build(step: Step): Promise<void>;
     private generatePackage;

package/dist/cjs/cloud/build/builders/python/index.js

@@ -12,7 +12,7 @@ const archiver_1 = require("../archiver");
 const include_static_files_1 = require("../include-static-files");
 const extract_python_data_1 = require("./python-data/extract-python-data");
 const read_requirements_1 = require("./python-data/read-requirements");
-const resolve_dep_name_1 = require("./python-data/resolve-dep-name");
+const resolve_dep_names_1 = require("./python-data/resolve-dep-names");
 const uv_packager_1 = require("./uv-packager");
 class PythonBuilder {
     constructor(builder, listener) {
@@ -21,17 +21,6 @@ class PythonBuilder {
         (0, activate_python_env_1.activatePythonVenv)({ baseDir: this.builder.projectDir });
         this.packager = new uv_packager_1.UvPackager();
     }
-    getRequirements() {
-        const requirementsFile = path_1.default.join(this.builder.projectDir, 'requirements.txt');
-        const depNames = Object.keys((0, read_requirements_1.readRequirements)(requirementsFile, (name) => ({ name, importName: name })));
-        const sitePackagesPath = (0, activate_python_env_1.getSitePackagesPath)({ baseDir: this.builder.projectDir });
-        const mapper = (0, resolve_dep_name_1.resolveDepNames)(depNames, sitePackagesPath);
-        const describer = (name) => {
-            const [, to] = mapper.find(([from]) => from === name) ?? [];
-            return { name, importName: to ?? name };
-        };
-        return (0, read_requirements_1.readRequirements)(requirementsFile, describer);
-    }
     async buildApiSteps(steps) {
         const zipName = 'router-python.zip';
         const archive = new archiver_1.Archiver(path_1.default.join(constants_1.distDir, zipName));
@@ -79,8 +68,11 @@ class PythonBuilder {
         }
     }
     async generatePackage(bundleDir, entrypointPath, archive, fileContent) {
-        const requirements = this.getRequirements();
-        const { externalDependencies, files } = (0, extract_python_data_1.extractPythonData)(this.builder.projectDir, entrypointPath, requirements, fileContent);
+        const requirementsFile = path_1.default.join(this.builder.projectDir, 'requirements.txt');
+        const requirements = (0, read_requirements_1.readRequirements)(requirementsFile);
+        const sitePackagesPath = (0, activate_python_env_1.getSitePackagesPath)({ baseDir: this.builder.projectDir });
+        const dependenciesMap = (0, resolve_dep_names_1.resolveDepNames)(Object.keys(requirements), sitePackagesPath);
+        const { externalDependencies, files } = (0, extract_python_data_1.extractPythonData)(this.builder.projectDir, entrypointPath, dependenciesMap, fileContent);
         // move files
         for (const file of files) {
             fs_1.default.mkdirSync(path_1.default.dirname(path_1.default.join(bundleDir, file)), { recursive: true });
@@ -94,7 +86,10 @@ class PythonBuilder {
         const dependencies = Object.values(externalDependencies);
         if (dependencies.length > 0) {
             // create requirements.txt
-            fs_1.default.writeFileSync(path_1.default.join(bundleDir, 'requirements.txt'), Object.values(externalDependencies).join('\n'));
+            const requirementsContent = Object.values(externalDependencies)
+                .map((dependency) => requirements[dependency])
+                .join('\n');
+            fs_1.default.writeFileSync(path_1.default.join(bundleDir, 'requirements.txt'), requirementsContent);
             await this.packager.packageDependencies(bundleDir);
         }
         // zip entire folder

package/dist/cjs/cloud/build/builders/python/python-data/__tests__/extract-python-data.test.js

@@ -8,11 +8,10 @@ const glob_1 = require("glob");
 const read_requirements_1 = require("../read-requirements");
 const extract_python_data_1 = require("../extract-python-data");
 const python_errors_1 = require("../python-errors");
-const mockPackageDescriber = (name) => ({ name, importName: name });
 describe('extractPythonData', () => {
     test('extracts python data correctly', () => {
         const rootDir = path_1.default.join(__dirname, './examples/example-1');
-        const requirements = (0, read_requirements_1.readRequirements)(path_1.default.join(rootDir, 'requirements.txt'), mockPackageDescriber);
+        const requirements = (0, read_requirements_1.readRequirements)(path_1.default.join(rootDir, 'requirements.txt'));
         const steps = (0, glob_1.globSync)('**/*_step.py', { absolute: false, cwd: path_1.default.join(rootDir, 'steps') });
         for (const file of steps) {
             const result = (0, extract_python_data_1.extractPythonData)(rootDir, `/steps/${file}`, requirements);
@@ -25,17 +24,17 @@ describe('extractPythonData', () => {
     });
     test('extracts python data correctly with invalid dependency', () => {
         const rootDir = path_1.default.join(__dirname, './examples/invalid-dependency');
-        const requirements = (0, read_requirements_1.readRequirements)(path_1.default.join(rootDir, 'requirements.txt'), mockPackageDescriber);
+        const requirements = (0, read_requirements_1.readRequirements)(path_1.default.join(rootDir, 'requirements.txt'));
         expect(() => (0, extract_python_data_1.extractPythonData)(rootDir, `/steps/api_step.py`, requirements)).toThrow(python_errors_1.PythonImportNotFoundError);
     });
     test('extracts python data correctly with compilation error', () => {
         const rootDir = path_1.default.join(__dirname, './examples/compilation-error');
-        const requirements = (0, read_requirements_1.readRequirements)(path_1.default.join(rootDir, 'requirements.txt'), mockPackageDescriber);
+        const requirements = (0, read_requirements_1.readRequirements)(path_1.default.join(rootDir, 'requirements.txt'));
         expect(() => (0, extract_python_data_1.extractPythonData)(rootDir, `/steps/api_step.py`, requirements)).toThrow(new python_errors_1.PythonError("Compilation error: no viable alternative at input ':' at line 3:10 in /steps/api_step.py", '/steps/api_step.py'));
     });
     test('extracts python data with nested import from requirements', () => {
         const rootDir = path_1.default.join(__dirname, './examples/chessarena');
-        const requirements = (0, read_requirements_1.readRequirements)(path_1.default.join(rootDir, 'requirements.txt'), mockPackageDescriber);
+        const requirements = (0, read_requirements_1.readRequirements)(path_1.default.join(rootDir, 'requirements.txt'));
         const result = (0, extract_python_data_1.extractPythonData)(rootDir, `/steps/evaluate_player_move_step.py`, requirements);
         expect(result.externalDependencies).toEqual({ chess: 'chess>=1.0.0', pydantic: 'pydantic>=2.6.1' });
     });

package/dist/cjs/cloud/build/builders/python/python-data/__tests__/get-dependencies-from-file.test.js

@@ -11,9 +11,7 @@ import requests
 from .local_module import something
 from ..parent_module import other
 `;
-        const requirements = {
-            requests: 'requests==2.25.1',
-        };
+        const requirements = { requests: 'requests' };
         const deps = (0, get_dependencies_from_file_1.getDependenciesFromFile)(pythonCode, 'test.py', requirements);
         expect(Array.from(deps.standardLibDependencies)).toEqual(['os', 'sys', 'collections']);
         expect(Array.from(deps.externalDependencies)).toEqual(['requests']);
@@ -31,9 +29,7 @@ import json
 import numpy 
 import mypackage.submodule
 `;
-        const requirements = {
-            numpy: 'numpy==1.21.0',
-        };
+        const requirements = { numpy: 'numpy' };
         const deps = (0, get_dependencies_from_file_1.getDependenciesFromFile)(pythonCode, 'test.py', requirements);
         expect(Array.from(deps.standardLibDependencies)).toEqual(['json']);
         expect(Array.from(deps.externalDependencies)).toEqual(['numpy']);
@@ -45,9 +41,7 @@ import json
 import numpy as np
 import mypackage.submodule
 `;
-        const requirements = {
-            numpy: 'numpy==1.21.0',
-        };
+        const requirements = { numpy: 'numpy' };
         const deps = (0, get_dependencies_from_file_1.getDependenciesFromFile)(pythonCode, 'test.py', requirements);
         expect(Array.from(deps.standardLibDependencies)).toEqual(['json']);
         expect(Array.from(deps.externalDependencies)).toEqual(['numpy']);

package/dist/cjs/cloud/build/builders/python/python-data/__tests__/read-requirements.test.js

@@ -9,7 +9,6 @@ const path_1 = __importDefault(require("path"));
 const os_1 = require("os");
 describe('readRequirements', () => {
     let tempFilePath;
-    const mockPackageDescriber = jest.fn((name) => ({ name, importName: name }));
     beforeEach(() => {
         // Create a temporary file path
         tempFilePath = path_1.default.join((0, os_1.tmpdir)(), `requirements-${Date.now()}.txt`);
@@ -25,7 +24,7 @@ describe('readRequirements', () => {
 numpy>=1.20.0
 flask~=2.0`;
         fs_1.default.writeFileSync(tempFilePath, content);
-        const requirements = (0, read_requirements_1.readRequirements)(tempFilePath, mockPackageDescriber);
+        const requirements = (0, read_requirements_1.readRequirements)(tempFilePath);
         expect(Object.keys(requirements)).toHaveLength(3);
         expect(requirements.requests).toBe('requests==2.25.1');
         expect(requirements.numpy).toBe('numpy>=1.20.0');
@@ -40,38 +39,17 @@ numpy>=1.20.0
 
 flask~=2.0`;
         fs_1.default.writeFileSync(tempFilePath, content);
-        const requirements = (0, read_requirements_1.readRequirements)(tempFilePath, mockPackageDescriber);
+        const requirements = (0, read_requirements_1.readRequirements)(tempFilePath);
         expect(Object.keys(requirements)).toHaveLength(3);
         expect(requirements.requests).toBe('requests==2.25.1');
         expect(requirements.numpy).toBe('numpy>=1.20.0');
         expect(requirements.flask).toBe('flask~=2.0');
     });
-    test('handles complex package names and version specifiers', () => {
-        const content = `scikit-learn==1.0.2
-python-dateutil==1.4
-Django<4.0.0,>=3.2.0`;
-        fs_1.default.writeFileSync(tempFilePath, content);
-        const describer = jest.fn((name) => {
-            // mocking a describer resolution for these packages
-            // we're going to implement it separately
-            const map = {
-                'scikit-learn': 'sklearn',
-                'python-dateutil': 'dateutil',
-                Django: 'django',
-            };
-            return map[name] ? { name, importName: map[name] } : { name, importName: name };
-        });
-        const requirements = (0, read_requirements_1.readRequirements)(tempFilePath, describer);
-        expect(Object.keys(requirements)).toEqual(['sklearn', 'dateutil', 'django']);
-        expect(requirements['sklearn']).toBe('scikit-learn==1.0.2');
-        expect(requirements['dateutil']).toBe('python-dateutil==1.4');
-        expect(requirements['django']).toBe('Django<4.0.0,>=3.2.0');
-    });
     test('handles requirements with extra dependencies', () => {
         const content = `requests[security]==2.25.1
 flask[async]==2.0.1`;
         fs_1.default.writeFileSync(tempFilePath, content);
-        const requirements = (0, read_requirements_1.readRequirements)(tempFilePath, mockPackageDescriber);
+        const requirements = (0, read_requirements_1.readRequirements)(tempFilePath);
         expect(Object.keys(requirements)).toHaveLength(2);
         expect(requirements['requests']).toBe('requests[security]==2.25.1');
         expect(requirements['flask']).toBe('flask[async]==2.0.1');
@@ -81,7 +59,7 @@ flask[async]==2.0.1`;
 
 # And empty lines`;
         fs_1.default.writeFileSync(tempFilePath, content);
-        const requirements = (0, read_requirements_1.readRequirements)(tempFilePath, mockPackageDescriber);
+        const requirements = (0, read_requirements_1.readRequirements)(tempFilePath);
         expect(Object.keys(requirements)).toHaveLength(0);
     });
     test('handles malformed lines gracefully', () => {
@@ -90,7 +68,7 @@ flask[async]==2.0.1`;
 numpy>=1.20.0
 ==invalid-version-only`;
         fs_1.default.writeFileSync(tempFilePath, content);
-        const requirements = (0, read_requirements_1.readRequirements)(tempFilePath, mockPackageDescriber);
+        const requirements = (0, read_requirements_1.readRequirements)(tempFilePath);
         expect(Object.keys(requirements)).toHaveLength(2);
         expect(requirements.requests).toBe('requests==2.25.1');
         expect(requirements.numpy).toBe('numpy>=1.20.0');
@@ -98,7 +76,7 @@ numpy>=1.20.0
     test('handles mixed line endings', () => {
         const content = 'requests==2.25.1\r\nnumpy>=1.20.0\nflask~=2.0';
         fs_1.default.writeFileSync(tempFilePath, content);
-        const requirements = (0, read_requirements_1.readRequirements)(tempFilePath, mockPackageDescriber);
+        const requirements = (0, read_requirements_1.readRequirements)(tempFilePath);
         expect(Object.keys(requirements)).toHaveLength(3);
         expect(requirements.requests).toBe('requests==2.25.1');
         expect(requirements.numpy).toBe('numpy>=1.20.0');
@@ -107,7 +85,7 @@ numpy>=1.20.0
     test('handles requirements with no versions', () => {
         const content = ['requests', 'numpy', 'flask'].join('\n');
         fs_1.default.writeFileSync(tempFilePath, content);
-        const requirements = (0, read_requirements_1.readRequirements)(tempFilePath, mockPackageDescriber);
+        const requirements = (0, read_requirements_1.readRequirements)(tempFilePath);
         expect(Object.keys(requirements)).toHaveLength(3);
         expect(requirements.requests).toBe('requests');
         expect(requirements.numpy).toBe('numpy');

package/dist/cjs/cloud/build/builders/python/python-data/__tests__/resolve-dep-name.test.js

@@ -4,26 +4,35 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 const path_1 = __importDefault(require("path"));
-const resolve_dep_name_1 = require("../resolve-dep-name");
+const resolve_dep_names_1 = require("../resolve-dep-names");
 const python_errors_1 = require("../python-errors");
 const sitePackagesDir = path_1.default.join(__dirname, 'site-packages');
 describe('resolveDepName', () => {
     test('resolves dependency names correctly', () => {
         const depNames = ['httpx'];
-        const result = (0, resolve_dep_name_1.resolveDepNames)(depNames, sitePackagesDir);
-        expect(result).toEqual([['httpx', 'httpx']]);
+        const result = (0, resolve_dep_names_1.resolveDepNames)(depNames, sitePackagesDir);
+        expect(result).toEqual({ httpx: 'httpx' });
     });
     test('resolves multiple dependency names correctly', () => {
         const depNames = ['httpx', 'scikit-learn', 'opencv-python'];
-        const result = (0, resolve_dep_name_1.resolveDepNames)(depNames, sitePackagesDir);
-        expect(result).toEqual([
-            ['httpx', 'httpx'],
-            ['scikit-learn', 'sklearn'],
-            ['opencv-python', 'cv2'],
-        ]);
+        const result = (0, resolve_dep_names_1.resolveDepNames)(depNames, sitePackagesDir);
+        expect(result).toEqual({
+            httpx: 'httpx',
+            sklearn: 'scikit-learn',
+            cv2: 'opencv-python',
+        });
+    });
+    test('resolves pymongo correctly', () => {
+        const depNames = ['pymongo'];
+        const result = (0, resolve_dep_names_1.resolveDepNames)(depNames, sitePackagesDir);
+        expect(result).toEqual({
+            pymongo: 'pymongo',
+            bson: 'pymongo',
+            gridfs: 'pymongo',
+        });
     });
     test('should throw an error if the dependency is not found', () => {
         const depNames = ['httpx', 'scikit-learn', 'opencv-python', 'pydantic'];
-        expect(() => (0, resolve_dep_name_1.resolveDepNames)(depNames, sitePackagesDir)).toThrow(new python_errors_1.PythonError('Could not find dependency name in site-packages: pydantic', 'pydantic'));
+        expect(() => (0, resolve_dep_names_1.resolveDepNames)(depNames, sitePackagesDir)).toThrow(new python_errors_1.PythonError('Could not find dependency name in site-packages: pydantic', 'pydantic'));
     });
 });

package/dist/cjs/cloud/build/builders/python/python-data/extract-python-data.d.ts

@@ -4,4 +4,4 @@ export type PythonResult = {
     externalDependencies: Requirements;
     files: string[];
 };
-export declare const extractPythonData: (rootDir: string, filePath: string, externalDependencies: Requirements, fileContent?: string) => PythonResult;
+export declare const extractPythonData: (rootDir: string, filePath: string, dependenciesMap: Record<string, string>, fileContent?: string) => PythonResult;

package/dist/cjs/cloud/build/builders/python/python-data/extract-python-data.js

@@ -2,7 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.extractPythonData = void 0;
 const traverse_tree_1 = require("./traverse-tree");
-const extractPythonData = (rootDir, filePath, externalDependencies, 
+const extractPythonData = (rootDir, filePath, dependenciesMap, 
 // optional
 fileContent) => {
     const result = {
@@ -10,10 +10,10 @@ fileContent) => {
         externalDependencies: new Set(),
         files: new Set(),
     };
-    (0, traverse_tree_1.traverseTree)(rootDir, filePath, result, externalDependencies, fileContent);
+    (0, traverse_tree_1.traverseTree)(rootDir, filePath, result, dependenciesMap, fileContent);
     const resultDependencies = {};
     result.externalDependencies.forEach((dependency) => {
-        resultDependencies[dependency] = externalDependencies[dependency];
+        resultDependencies[dependency] = dependenciesMap[dependency];
     });
     return {
         standardLibDependencies: Array.from(result.standardLibDependencies),

package/dist/cjs/cloud/build/builders/python/python-data/get-dependencies-from-file.d.ts

@@ -1,7 +1,6 @@
-import { Requirements } from './read-requirements';
 export type Dependencies = {
     standardLibDependencies: Set<string>;
     externalDependencies: Set<string>;
     projectDependencies: Set<string>;
 };
-export declare const getDependenciesFromFile: (content: string, path: string, requirements: Requirements) => Dependencies;
+export declare const getDependenciesFromFile: (content: string, path: string, externalDependenciesMap: Record<string, string>) => Dependencies;

package/dist/cjs/cloud/build/builders/python/python-data/get-dependencies-from-file.js

@@ -23,7 +23,7 @@ function parse(source, sourceName) {
     parser.addErrorListener(listener);
     return parser.file_input();
 }
-const getDependenciesFromFile = (content, path, requirements) => {
+const getDependenciesFromFile = (content, path, externalDependenciesMap) => {
     const result = parse(content + '\n', path);
     const modulesSet = new Set();
     (0, python_ast_1.createVisitor)({
@@ -60,7 +60,7 @@ const getDependenciesFromFile = (content, path, requirements) => {
         else if (constants_1.STANDARD_LIB_MODULES.has(module)) {
             dependencies.standardLibDependencies.add(module);
         }
-        else if (requirements[module] || requirements[moduleName]) {
+        else if (externalDependenciesMap[module] || externalDependenciesMap[moduleName]) {
             dependencies.externalDependencies.add(module);
         }
         else {

package/dist/cjs/cloud/build/builders/python/python-data/read-requirements.d.ts

@@ -1,15 +1,13 @@
-export type Requirements = {
-    [importName: string]: string;
-};
-type PackageInfo = {
-    name: string;
-    importName: string;
-};
-type PackageDescriber = (packageName: string) => PackageInfo;
+/**
+ * { libraryName: lineFromRequirementsFile }
+ *
+ * Example:
+ * { scikit-learn: 'scikit-learn==1.0.2' }
+ */
+export type Requirements = Record<string, string>;
 /**
  * Read the requirements.txt file and return a set of dependencies
  * @param filePath
  * @returns set of all dependencies names
  */
-export declare const readRequirements: (filePath: string, describer: PackageDescriber) => Requirements;
-export {};
+export declare const readRequirements: (filePath: string) => Requirements;

package/dist/cjs/cloud/build/builders/python/python-data/read-requirements.js

@@ -10,7 +10,7 @@ const fs_1 = __importDefault(require("fs"));
  * @param filePath
  * @returns set of all dependencies names
  */
-const readRequirements = (filePath, describer) => {
+const readRequirements = (filePath) => {
     const content = fs_1.default.readFileSync(filePath, 'utf8');
     const lines = content.split('\n');
     const requirements = {};
@@ -23,8 +23,8 @@ const readRequirements = (filePath, describer) => {
         // Extract package name (everything before version specifiers)
         const packageMatch = trimmedLine.match(/^([a-zA-Z0-9_-]+)/);
         if (packageMatch) {
-            const packageInfo = describer(packageMatch[1]);
-            requirements[packageInfo.importName] = trimmedLine;
+            const packageName = packageMatch[1];
+            requirements[packageName] = trimmedLine;
         }
     }
     return requirements;

package/dist/cjs/cloud/build/builders/python/python-data/{resolve-dep-name.d.ts → resolve-dep-names.d.ts}

@@ -1,6 +1,6 @@
 /**
  * Resolve the name of the import from the name of the package
  * @param depName the package name
- * @returns the import name
+ * @returns a map of the import name to the package name
  */
-export declare const resolveDepNames: (depNames: string[], sitePackagesDir: string) => [string, string][];
+export declare const resolveDepNames: (depNames: string[], sitePackagesDir: string) => Record<string, string>;

package/dist/cjs/cloud/build/builders/python/python-data/{resolve-dep-name.js → resolve-dep-names.js}

@@ -10,13 +10,13 @@ const python_errors_1 = require("./python-errors");
 /**
  * Resolve the name of the import from the name of the package
  * @param depName the package name
- * @returns the import name
+ * @returns a map of the import name to the package name
  */
 const resolveDepNames = (depNames, sitePackagesDir) => {
     const folders = fs_1.default
         .readdirSync(sitePackagesDir)
         .filter((folder) => fs_1.default.statSync(path_1.default.join(sitePackagesDir, folder)).isDirectory());
-    const result = [];
+    const result = {};
     for (const depName of depNames) {
         const regex = new RegExp(`^${depName.replace(/-/g, '[-_]')}-.*\\.dist-info$`);
         const folder = folders.find((folder) => regex.test(folder));
@@ -25,7 +25,7 @@ const resolveDepNames = (depNames, sitePackagesDir) => {
             const recordFile = fs_1.default.existsSync(path_1.default.join(sitePackagesDir, folder, 'RECORD'));
             if (topLevelFile) {
                 const topLevel = fs_1.default.readFileSync(path_1.default.join(sitePackagesDir, folder, 'top_level.txt'), 'utf8').trim();
-                result.push([depName, topLevel]);
+                result[topLevel] = depName;
             }
             else if (recordFile) {
                 const record = fs_1.default.readFileSync(path_1.default.join(sitePackagesDir, folder, 'RECORD'), 'utf8').trim();
@@ -35,8 +35,7 @@ const resolveDepNames = (depNames, sitePackagesDir) => {
                     // RECORD format: path/__init__.py,sha256=hash,size
                     const match = /^([^/]+)\/__init__\.py,/.exec(line);
                     if (match) {
-                        result.push([depName, match[1]]);
-                        break;
+                        result[match[1]] = depName;
                     }
                 }
             }

package/dist/cjs/cloud/build/builders/python/python-data/traverse-tree.d.ts

@@ -1,7 +1,6 @@
-import { Requirements } from './read-requirements';
 export type TraverseTreeResult = {
     standardLibDependencies: Set<string>;
     externalDependencies: Set<string>;
     files: Set<string>;
 };
-export declare const traverseTree: (rootDir: string, filePath: string, result: TraverseTreeResult, externalDependencies: Requirements, fileContent?: string) => void;
+export declare const traverseTree: (rootDir: string, filePath: string, result: TraverseTreeResult, dependenciesMap: Record<string, string>, fileContent?: string) => void;

package/dist/cjs/cloud/build/builders/python/python-data/traverse-tree.js

@@ -9,7 +9,7 @@ const path_1 = __importDefault(require("path"));
 const convert_import_path_1 = require("./convert-import-path");
 const get_dependencies_from_file_1 = require("./get-dependencies-from-file");
 const python_errors_1 = require("./python-errors");
-const traverseTree = (rootDir, filePath, result, externalDependencies, 
+const traverseTree = (rootDir, filePath, result, dependenciesMap, 
 // optional
 fileContent) => {
     const fileAbsolutePath = path_1.default.join(rootDir, filePath);
@@ -21,7 +21,7 @@ fileContent) => {
         result.files.add(initPath.replace(rootDir, ''));
     }
     const content = fileContent || fs_1.default.readFileSync(fileAbsolutePath, 'utf8');
-    const dependencies = (0, get_dependencies_from_file_1.getDependenciesFromFile)(content, filePath, externalDependencies);
+    const dependencies = (0, get_dependencies_from_file_1.getDependenciesFromFile)(content, filePath, dependenciesMap);
     result.files.add(filePath);
     dependencies.externalDependencies.forEach((dependency) => {
         result.externalDependencies.add(dependency);
@@ -36,7 +36,7 @@ fileContent) => {
         const dependencyPath = dependencyFilePath.replace(rootDir, '');
         if (!result.files.has(dependencyPath)) {
             try {
-                (0, exports.traverseTree)(rootDir, dependencyPath, result, externalDependencies);
+                (0, exports.traverseTree)(rootDir, dependencyPath, result, dependenciesMap);
             }
             catch (error) {
                 if (error instanceof python_errors_1.PythonFileNotFoundError) {
@@ -44,7 +44,7 @@ fileContent) => {
                         // try root folder
                         try {
                             const rootDependencyFilePath = path_1.default.resolve(rootDir, `${pythonPath}.py`).replace(rootDir, '');
-                            return (0, exports.traverseTree)(rootDir, rootDependencyFilePath, result, externalDependencies);
+                            return (0, exports.traverseTree)(rootDir, rootDependencyFilePath, result, dependenciesMap);
                         }
                         catch (_error) {
                             // let it throw

package/dist/esm/cloud/build/builders/python/index.d.ts

@@ -6,7 +6,6 @@ export declare class PythonBuilder implements StepBuilder {
     private readonly listener;
     private packager;
     constructor(builder: Builder, listener: BuildListener);
-    private getRequirements;
     buildApiSteps(steps: Step<ApiRouteConfig>[]): Promise<RouterBuildResult>;
     build(step: Step): Promise<void>;
     private generatePackage;

package/dist/esm/cloud/build/builders/python/index.js

@@ -6,7 +6,7 @@ import { Archiver } from '../archiver';
 import { includeStaticFiles } from '../include-static-files';
 import { extractPythonData } from './python-data/extract-python-data';
 import { readRequirements } from './python-data/read-requirements';
-import { resolveDepNames } from './python-data/resolve-dep-name';
+import { resolveDepNames } from './python-data/resolve-dep-names';
 import { UvPackager } from './uv-packager';
 export class PythonBuilder {
     constructor(builder, listener) {
@@ -15,17 +15,6 @@ export class PythonBuilder {
         activatePythonVenv({ baseDir: this.builder.projectDir });
         this.packager = new UvPackager();
     }
-    getRequirements() {
-        const requirementsFile = path.join(this.builder.projectDir, 'requirements.txt');
-        const depNames = Object.keys(readRequirements(requirementsFile, (name) => ({ name, importName: name })));
-        const sitePackagesPath = getSitePackagesPath({ baseDir: this.builder.projectDir });
-        const mapper = resolveDepNames(depNames, sitePackagesPath);
-        const describer = (name) => {
-            const [, to] = mapper.find(([from]) => from === name) ?? [];
-            return { name, importName: to ?? name };
-        };
-        return readRequirements(requirementsFile, describer);
-    }
     async buildApiSteps(steps) {
         const zipName = 'router-python.zip';
         const archive = new Archiver(path.join(distDir, zipName));
@@ -73,8 +62,11 @@ export class PythonBuilder {
         }
     }
     async generatePackage(bundleDir, entrypointPath, archive, fileContent) {
-        const requirements = this.getRequirements();
-        const { externalDependencies, files } = extractPythonData(this.builder.projectDir, entrypointPath, requirements, fileContent);
+        const requirementsFile = path.join(this.builder.projectDir, 'requirements.txt');
+        const requirements = readRequirements(requirementsFile);
+        const sitePackagesPath = getSitePackagesPath({ baseDir: this.builder.projectDir });
+        const dependenciesMap = resolveDepNames(Object.keys(requirements), sitePackagesPath);
+        const { externalDependencies, files } = extractPythonData(this.builder.projectDir, entrypointPath, dependenciesMap, fileContent);
         // move files
         for (const file of files) {
             fs.mkdirSync(path.dirname(path.join(bundleDir, file)), { recursive: true });
@@ -88,7 +80,10 @@ export class PythonBuilder {
         const dependencies = Object.values(externalDependencies);
         if (dependencies.length > 0) {
             // create requirements.txt
-            fs.writeFileSync(path.join(bundleDir, 'requirements.txt'), Object.values(externalDependencies).join('\n'));
+            const requirementsContent = Object.values(externalDependencies)
+                .map((dependency) => requirements[dependency])
+                .join('\n');
+            fs.writeFileSync(path.join(bundleDir, 'requirements.txt'), requirementsContent);
             await this.packager.packageDependencies(bundleDir);
         }
         // zip entire folder

package/dist/esm/cloud/build/builders/python/python-data/__tests__/extract-python-data.test.js

@@ -3,11 +3,10 @@ import { globSync } from 'glob';
 import { readRequirements } from '../read-requirements';
 import { extractPythonData } from '../extract-python-data';
 import { PythonError, PythonImportNotFoundError } from '../python-errors';
-const mockPackageDescriber = (name) => ({ name, importName: name });
 describe('extractPythonData', () => {
     test('extracts python data correctly', () => {
         const rootDir = path.join(__dirname, './examples/example-1');
-        const requirements = readRequirements(path.join(rootDir, 'requirements.txt'), mockPackageDescriber);
+        const requirements = readRequirements(path.join(rootDir, 'requirements.txt'));
         const steps = globSync('**/*_step.py', { absolute: false, cwd: path.join(rootDir, 'steps') });
         for (const file of steps) {
             const result = extractPythonData(rootDir, `/steps/${file}`, requirements);
@@ -20,17 +19,17 @@ describe('extractPythonData', () => {
     });
     test('extracts python data correctly with invalid dependency', () => {
         const rootDir = path.join(__dirname, './examples/invalid-dependency');
-        const requirements = readRequirements(path.join(rootDir, 'requirements.txt'), mockPackageDescriber);
+        const requirements = readRequirements(path.join(rootDir, 'requirements.txt'));
         expect(() => extractPythonData(rootDir, `/steps/api_step.py`, requirements)).toThrow(PythonImportNotFoundError);
     });
     test('extracts python data correctly with compilation error', () => {
         const rootDir = path.join(__dirname, './examples/compilation-error');
-        const requirements = readRequirements(path.join(rootDir, 'requirements.txt'), mockPackageDescriber);
+        const requirements = readRequirements(path.join(rootDir, 'requirements.txt'));
         expect(() => extractPythonData(rootDir, `/steps/api_step.py`, requirements)).toThrow(new PythonError("Compilation error: no viable alternative at input ':' at line 3:10 in /steps/api_step.py", '/steps/api_step.py'));
     });
     test('extracts python data with nested import from requirements', () => {
         const rootDir = path.join(__dirname, './examples/chessarena');
-        const requirements = readRequirements(path.join(rootDir, 'requirements.txt'), mockPackageDescriber);
+        const requirements = readRequirements(path.join(rootDir, 'requirements.txt'));
         const result = extractPythonData(rootDir, `/steps/evaluate_player_move_step.py`, requirements);
         expect(result.externalDependencies).toEqual({ chess: 'chess>=1.0.0', pydantic: 'pydantic>=2.6.1' });
     });

package/dist/esm/cloud/build/builders/python/python-data/__tests__/get-dependencies-from-file.test.js

@@ -9,9 +9,7 @@ import requests
 from .local_module import something
 from ..parent_module import other
 `;
-        const requirements = {
-            requests: 'requests==2.25.1',
-        };
+        const requirements = { requests: 'requests' };
         const deps = getDependenciesFromFile(pythonCode, 'test.py', requirements);
         expect(Array.from(deps.standardLibDependencies)).toEqual(['os', 'sys', 'collections']);
         expect(Array.from(deps.externalDependencies)).toEqual(['requests']);
@@ -29,9 +27,7 @@ import json
 import numpy 
 import mypackage.submodule
 `;
-        const requirements = {
-            numpy: 'numpy==1.21.0',
-        };
+        const requirements = { numpy: 'numpy' };
         const deps = getDependenciesFromFile(pythonCode, 'test.py', requirements);
         expect(Array.from(deps.standardLibDependencies)).toEqual(['json']);
         expect(Array.from(deps.externalDependencies)).toEqual(['numpy']);
@@ -43,9 +39,7 @@ import json
 import numpy as np
 import mypackage.submodule
 `;
-        const requirements = {
-            numpy: 'numpy==1.21.0',
-        };
+        const requirements = { numpy: 'numpy' };
         const deps = getDependenciesFromFile(pythonCode, 'test.py', requirements);
         expect(Array.from(deps.standardLibDependencies)).toEqual(['json']);
         expect(Array.from(deps.externalDependencies)).toEqual(['numpy']);

package/dist/esm/cloud/build/builders/python/python-data/__tests__/read-requirements.test.js

@@ -4,7 +4,6 @@ import path from 'path';
 import { tmpdir } from 'os';
 describe('readRequirements', () => {
     let tempFilePath;
-    const mockPackageDescriber = jest.fn((name) => ({ name, importName: name }));
     beforeEach(() => {
         // Create a temporary file path
         tempFilePath = path.join(tmpdir(), `requirements-${Date.now()}.txt`);
@@ -20,7 +19,7 @@ describe('readRequirements', () => {
 numpy>=1.20.0
 flask~=2.0`;
         fs.writeFileSync(tempFilePath, content);
-        const requirements = readRequirements(tempFilePath, mockPackageDescriber);
+        const requirements = readRequirements(tempFilePath);
         expect(Object.keys(requirements)).toHaveLength(3);
         expect(requirements.requests).toBe('requests==2.25.1');
         expect(requirements.numpy).toBe('numpy>=1.20.0');
@@ -35,38 +34,17 @@ numpy>=1.20.0
 
 flask~=2.0`;
         fs.writeFileSync(tempFilePath, content);
-        const requirements = readRequirements(tempFilePath, mockPackageDescriber);
+        const requirements = readRequirements(tempFilePath);
         expect(Object.keys(requirements)).toHaveLength(3);
         expect(requirements.requests).toBe('requests==2.25.1');
         expect(requirements.numpy).toBe('numpy>=1.20.0');
         expect(requirements.flask).toBe('flask~=2.0');
     });
-    test('handles complex package names and version specifiers', () => {
-        const content = `scikit-learn==1.0.2
-python-dateutil==1.4
-Django<4.0.0,>=3.2.0`;
-        fs.writeFileSync(tempFilePath, content);
-        const describer = jest.fn((name) => {
-            // mocking a describer resolution for these packages
-            // we're going to implement it separately
-            const map = {
-                'scikit-learn': 'sklearn',
-                'python-dateutil': 'dateutil',
-                Django: 'django',
-            };
-            return map[name] ? { name, importName: map[name] } : { name, importName: name };
-        });
-        const requirements = readRequirements(tempFilePath, describer);
-        expect(Object.keys(requirements)).toEqual(['sklearn', 'dateutil', 'django']);
-        expect(requirements['sklearn']).toBe('scikit-learn==1.0.2');
-        expect(requirements['dateutil']).toBe('python-dateutil==1.4');
-        expect(requirements['django']).toBe('Django<4.0.0,>=3.2.0');
-    });
     test('handles requirements with extra dependencies', () => {
         const content = `requests[security]==2.25.1
 flask[async]==2.0.1`;
         fs.writeFileSync(tempFilePath, content);
-        const requirements = readRequirements(tempFilePath, mockPackageDescriber);
+        const requirements = readRequirements(tempFilePath);
         expect(Object.keys(requirements)).toHaveLength(2);
         expect(requirements['requests']).toBe('requests[security]==2.25.1');
         expect(requirements['flask']).toBe('flask[async]==2.0.1');
@@ -76,7 +54,7 @@ flask[async]==2.0.1`;
 
 # And empty lines`;
         fs.writeFileSync(tempFilePath, content);
-        const requirements = readRequirements(tempFilePath, mockPackageDescriber);
+        const requirements = readRequirements(tempFilePath);
         expect(Object.keys(requirements)).toHaveLength(0);
     });
     test('handles malformed lines gracefully', () => {
@@ -85,7 +63,7 @@ flask[async]==2.0.1`;
 numpy>=1.20.0
 ==invalid-version-only`;
         fs.writeFileSync(tempFilePath, content);
-        const requirements = readRequirements(tempFilePath, mockPackageDescriber);
+        const requirements = readRequirements(tempFilePath);
         expect(Object.keys(requirements)).toHaveLength(2);
         expect(requirements.requests).toBe('requests==2.25.1');
         expect(requirements.numpy).toBe('numpy>=1.20.0');
@@ -93,7 +71,7 @@ numpy>=1.20.0
     test('handles mixed line endings', () => {
         const content = 'requests==2.25.1\r\nnumpy>=1.20.0\nflask~=2.0';
         fs.writeFileSync(tempFilePath, content);
-        const requirements = readRequirements(tempFilePath, mockPackageDescriber);
+        const requirements = readRequirements(tempFilePath);
         expect(Object.keys(requirements)).toHaveLength(3);
         expect(requirements.requests).toBe('requests==2.25.1');
         expect(requirements.numpy).toBe('numpy>=1.20.0');
@@ -102,7 +80,7 @@ numpy>=1.20.0
     test('handles requirements with no versions', () => {
         const content = ['requests', 'numpy', 'flask'].join('\n');
         fs.writeFileSync(tempFilePath, content);
-        const requirements = readRequirements(tempFilePath, mockPackageDescriber);
+        const requirements = readRequirements(tempFilePath);
         expect(Object.keys(requirements)).toHaveLength(3);
         expect(requirements.requests).toBe('requests');
         expect(requirements.numpy).toBe('numpy');

package/dist/esm/cloud/build/builders/python/python-data/__tests__/resolve-dep-name.test.js

@@ -1,21 +1,30 @@
 import path from 'path';
-import { resolveDepNames } from '../resolve-dep-name';
+import { resolveDepNames } from '../resolve-dep-names';
 import { PythonError } from '../python-errors';
 const sitePackagesDir = path.join(__dirname, 'site-packages');
 describe('resolveDepName', () => {
     test('resolves dependency names correctly', () => {
         const depNames = ['httpx'];
         const result = resolveDepNames(depNames, sitePackagesDir);
-        expect(result).toEqual([['httpx', 'httpx']]);
+        expect(result).toEqual({ httpx: 'httpx' });
     });
     test('resolves multiple dependency names correctly', () => {
         const depNames = ['httpx', 'scikit-learn', 'opencv-python'];
         const result = resolveDepNames(depNames, sitePackagesDir);
-        expect(result).toEqual([
-            ['httpx', 'httpx'],
-            ['scikit-learn', 'sklearn'],
-            ['opencv-python', 'cv2'],
-        ]);
+        expect(result).toEqual({
+            httpx: 'httpx',
+            sklearn: 'scikit-learn',
+            cv2: 'opencv-python',
+        });
+    });
+    test('resolves pymongo correctly', () => {
+        const depNames = ['pymongo'];
+        const result = resolveDepNames(depNames, sitePackagesDir);
+        expect(result).toEqual({
+            pymongo: 'pymongo',
+            bson: 'pymongo',
+            gridfs: 'pymongo',
+        });
     });
     test('should throw an error if the dependency is not found', () => {
         const depNames = ['httpx', 'scikit-learn', 'opencv-python', 'pydantic'];

package/dist/esm/cloud/build/builders/python/python-data/extract-python-data.d.ts

@@ -4,4 +4,4 @@ export type PythonResult = {
     externalDependencies: Requirements;
     files: string[];
 };
-export declare const extractPythonData: (rootDir: string, filePath: string, externalDependencies: Requirements, fileContent?: string) => PythonResult;
+export declare const extractPythonData: (rootDir: string, filePath: string, dependenciesMap: Record<string, string>, fileContent?: string) => PythonResult;

package/dist/esm/cloud/build/builders/python/python-data/extract-python-data.js

@@ -1,5 +1,5 @@
 import { traverseTree } from './traverse-tree';
-export const extractPythonData = (rootDir, filePath, externalDependencies, 
+export const extractPythonData = (rootDir, filePath, dependenciesMap, 
 // optional
 fileContent) => {
     const result = {
@@ -7,10 +7,10 @@ fileContent) => {
         externalDependencies: new Set(),
         files: new Set(),
     };
-    traverseTree(rootDir, filePath, result, externalDependencies, fileContent);
+    traverseTree(rootDir, filePath, result, dependenciesMap, fileContent);
     const resultDependencies = {};
     result.externalDependencies.forEach((dependency) => {
-        resultDependencies[dependency] = externalDependencies[dependency];
+        resultDependencies[dependency] = dependenciesMap[dependency];
     });
     return {
         standardLibDependencies: Array.from(result.standardLibDependencies),

package/dist/esm/cloud/build/builders/python/python-data/get-dependencies-from-file.d.ts

@@ -1,7 +1,6 @@
-import { Requirements } from './read-requirements';
 export type Dependencies = {
     standardLibDependencies: Set<string>;
     externalDependencies: Set<string>;
     projectDependencies: Set<string>;
 };
-export declare const getDependenciesFromFile: (content: string, path: string, requirements: Requirements) => Dependencies;
+export declare const getDependenciesFromFile: (content: string, path: string, externalDependenciesMap: Record<string, string>) => Dependencies;

package/dist/esm/cloud/build/builders/python/python-data/get-dependencies-from-file.js

@@ -20,7 +20,7 @@ function parse(source, sourceName) {
     parser.addErrorListener(listener);
     return parser.file_input();
 }
-export const getDependenciesFromFile = (content, path, requirements) => {
+export const getDependenciesFromFile = (content, path, externalDependenciesMap) => {
     const result = parse(content + '\n', path);
     const modulesSet = new Set();
     createVisitor({
@@ -57,7 +57,7 @@ export const getDependenciesFromFile = (content, path, requirements) => {
         else if (STANDARD_LIB_MODULES.has(module)) {
             dependencies.standardLibDependencies.add(module);
         }
-        else if (requirements[module] || requirements[moduleName]) {
+        else if (externalDependenciesMap[module] || externalDependenciesMap[moduleName]) {
             dependencies.externalDependencies.add(module);
         }
         else {

package/dist/esm/cloud/build/builders/python/python-data/read-requirements.d.ts

@@ -1,15 +1,13 @@
-export type Requirements = {
-    [importName: string]: string;
-};
-type PackageInfo = {
-    name: string;
-    importName: string;
-};
-type PackageDescriber = (packageName: string) => PackageInfo;
+/**
+ * { libraryName: lineFromRequirementsFile }
+ *
+ * Example:
+ * { scikit-learn: 'scikit-learn==1.0.2' }
+ */
+export type Requirements = Record<string, string>;
 /**
  * Read the requirements.txt file and return a set of dependencies
  * @param filePath
  * @returns set of all dependencies names
  */
-export declare const readRequirements: (filePath: string, describer: PackageDescriber) => Requirements;
-export {};
+export declare const readRequirements: (filePath: string) => Requirements;

package/dist/esm/cloud/build/builders/python/python-data/read-requirements.js

@@ -4,7 +4,7 @@ import fs from 'fs';
  * @param filePath
  * @returns set of all dependencies names
  */
-export const readRequirements = (filePath, describer) => {
+export const readRequirements = (filePath) => {
     const content = fs.readFileSync(filePath, 'utf8');
     const lines = content.split('\n');
     const requirements = {};
@@ -17,8 +17,8 @@ export const readRequirements = (filePath, describer) => {
         // Extract package name (everything before version specifiers)
         const packageMatch = trimmedLine.match(/^([a-zA-Z0-9_-]+)/);
         if (packageMatch) {
-            const packageInfo = describer(packageMatch[1]);
-            requirements[packageInfo.importName] = trimmedLine;
+            const packageName = packageMatch[1];
+            requirements[packageName] = trimmedLine;
         }
     }
     return requirements;

package/dist/{types/cloud/build/builders/python/python-data/resolve-dep-name.d.ts → esm/cloud/build/builders/python/python-data/resolve-dep-names.d.ts}

@@ -1,6 +1,6 @@
 /**
  * Resolve the name of the import from the name of the package
  * @param depName the package name
- * @returns the import name
+ * @returns a map of the import name to the package name
  */
-export declare const resolveDepNames: (depNames: string[], sitePackagesDir: string) => [string, string][];
+export declare const resolveDepNames: (depNames: string[], sitePackagesDir: string) => Record<string, string>;

package/dist/esm/cloud/build/builders/python/python-data/{resolve-dep-name.js → resolve-dep-names.js}

@@ -4,13 +4,13 @@ import { PythonError } from './python-errors';
 /**
  * Resolve the name of the import from the name of the package
  * @param depName the package name
- * @returns the import name
+ * @returns a map of the import name to the package name
  */
 export const resolveDepNames = (depNames, sitePackagesDir) => {
     const folders = fs
         .readdirSync(sitePackagesDir)
         .filter((folder) => fs.statSync(path.join(sitePackagesDir, folder)).isDirectory());
-    const result = [];
+    const result = {};
     for (const depName of depNames) {
         const regex = new RegExp(`^${depName.replace(/-/g, '[-_]')}-.*\\.dist-info$`);
         const folder = folders.find((folder) => regex.test(folder));
@@ -19,7 +19,7 @@ export const resolveDepNames = (depNames, sitePackagesDir) => {
             const recordFile = fs.existsSync(path.join(sitePackagesDir, folder, 'RECORD'));
             if (topLevelFile) {
                 const topLevel = fs.readFileSync(path.join(sitePackagesDir, folder, 'top_level.txt'), 'utf8').trim();
-                result.push([depName, topLevel]);
+                result[topLevel] = depName;
             }
             else if (recordFile) {
                 const record = fs.readFileSync(path.join(sitePackagesDir, folder, 'RECORD'), 'utf8').trim();
@@ -29,8 +29,7 @@ export const resolveDepNames = (depNames, sitePackagesDir) => {
                     // RECORD format: path/__init__.py,sha256=hash,size
                     const match = /^([^/]+)\/__init__\.py,/.exec(line);
                     if (match) {
-                        result.push([depName, match[1]]);
-                        break;
+                        result[match[1]] = depName;
                     }
                 }
             }

package/dist/esm/cloud/build/builders/python/python-data/traverse-tree.d.ts

@@ -1,7 +1,6 @@
-import { Requirements } from './read-requirements';
 export type TraverseTreeResult = {
     standardLibDependencies: Set<string>;
     externalDependencies: Set<string>;
     files: Set<string>;
 };
-export declare const traverseTree: (rootDir: string, filePath: string, result: TraverseTreeResult, externalDependencies: Requirements, fileContent?: string) => void;
+export declare const traverseTree: (rootDir: string, filePath: string, result: TraverseTreeResult, dependenciesMap: Record<string, string>, fileContent?: string) => void;

package/dist/esm/cloud/build/builders/python/python-data/traverse-tree.js

@@ -3,7 +3,7 @@ import path from 'path';
 import { convertImportToPath } from './convert-import-path';
 import { getDependenciesFromFile } from './get-dependencies-from-file';
 import { PythonFileNotFoundError, PythonImportNotFoundError } from './python-errors';
-export const traverseTree = (rootDir, filePath, result, externalDependencies, 
+export const traverseTree = (rootDir, filePath, result, dependenciesMap, 
 // optional
 fileContent) => {
     const fileAbsolutePath = path.join(rootDir, filePath);
@@ -15,7 +15,7 @@ fileContent) => {
         result.files.add(initPath.replace(rootDir, ''));
     }
     const content = fileContent || fs.readFileSync(fileAbsolutePath, 'utf8');
-    const dependencies = getDependenciesFromFile(content, filePath, externalDependencies);
+    const dependencies = getDependenciesFromFile(content, filePath, dependenciesMap);
     result.files.add(filePath);
     dependencies.externalDependencies.forEach((dependency) => {
         result.externalDependencies.add(dependency);
@@ -30,7 +30,7 @@ fileContent) => {
         const dependencyPath = dependencyFilePath.replace(rootDir, '');
         if (!result.files.has(dependencyPath)) {
             try {
-                traverseTree(rootDir, dependencyPath, result, externalDependencies);
+                traverseTree(rootDir, dependencyPath, result, dependenciesMap);
             }
             catch (error) {
                 if (error instanceof PythonFileNotFoundError) {
@@ -38,7 +38,7 @@ fileContent) => {
                         // try root folder
                         try {
                             const rootDependencyFilePath = path.resolve(rootDir, `${pythonPath}.py`).replace(rootDir, '');
-                            return traverseTree(rootDir, rootDependencyFilePath, result, externalDependencies);
+                            return traverseTree(rootDir, rootDependencyFilePath, result, dependenciesMap);
                         }
                         catch (_error) {
                             // let it throw

package/dist/types/cloud/build/builders/python/index.d.ts

@@ -6,7 +6,6 @@ export declare class PythonBuilder implements StepBuilder {
     private readonly listener;
     private packager;
     constructor(builder: Builder, listener: BuildListener);
-    private getRequirements;
     buildApiSteps(steps: Step<ApiRouteConfig>[]): Promise<RouterBuildResult>;
     build(step: Step): Promise<void>;
     private generatePackage;

package/dist/types/cloud/build/builders/python/python-data/extract-python-data.d.ts

@@ -4,4 +4,4 @@ export type PythonResult = {
     externalDependencies: Requirements;
     files: string[];
 };
-export declare const extractPythonData: (rootDir: string, filePath: string, externalDependencies: Requirements, fileContent?: string) => PythonResult;
+export declare const extractPythonData: (rootDir: string, filePath: string, dependenciesMap: Record<string, string>, fileContent?: string) => PythonResult;

package/dist/types/cloud/build/builders/python/python-data/get-dependencies-from-file.d.ts

@@ -1,7 +1,6 @@
-import { Requirements } from './read-requirements';
 export type Dependencies = {
     standardLibDependencies: Set<string>;
     externalDependencies: Set<string>;
     projectDependencies: Set<string>;
 };
-export declare const getDependenciesFromFile: (content: string, path: string, requirements: Requirements) => Dependencies;
+export declare const getDependenciesFromFile: (content: string, path: string, externalDependenciesMap: Record<string, string>) => Dependencies;

package/dist/types/cloud/build/builders/python/python-data/read-requirements.d.ts

@@ -1,15 +1,13 @@
-export type Requirements = {
-    [importName: string]: string;
-};
-type PackageInfo = {
-    name: string;
-    importName: string;
-};
-type PackageDescriber = (packageName: string) => PackageInfo;
+/**
+ * { libraryName: lineFromRequirementsFile }
+ *
+ * Example:
+ * { scikit-learn: 'scikit-learn==1.0.2' }
+ */
+export type Requirements = Record<string, string>;
 /**
  * Read the requirements.txt file and return a set of dependencies
  * @param filePath
  * @returns set of all dependencies names
  */
-export declare const readRequirements: (filePath: string, describer: PackageDescriber) => Requirements;
-export {};
+export declare const readRequirements: (filePath: string) => Requirements;

package/dist/{esm/cloud/build/builders/python/python-data/resolve-dep-name.d.ts → types/cloud/build/builders/python/python-data/resolve-dep-names.d.ts}

@@ -1,6 +1,6 @@
 /**
  * Resolve the name of the import from the name of the package
  * @param depName the package name
- * @returns the import name
+ * @returns a map of the import name to the package name
  */
-export declare const resolveDepNames: (depNames: string[], sitePackagesDir: string) => [string, string][];
+export declare const resolveDepNames: (depNames: string[], sitePackagesDir: string) => Record<string, string>;

package/dist/types/cloud/build/builders/python/python-data/traverse-tree.d.ts

@@ -1,7 +1,6 @@
-import { Requirements } from './read-requirements';
 export type TraverseTreeResult = {
     standardLibDependencies: Set<string>;
     externalDependencies: Set<string>;
     files: Set<string>;
 };
-export declare const traverseTree: (rootDir: string, filePath: string, result: TraverseTreeResult, externalDependencies: Requirements, fileContent?: string) => void;
+export declare const traverseTree: (rootDir: string, filePath: string, result: TraverseTreeResult, dependenciesMap: Record<string, string>, fileContent?: string) => void;

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "motia",
   "description": "A Modern Unified Backend Framework for APIs, Events and Agents",
-  "version": "0.7.2-beta.134",
+  "version": "0.7.2-beta.135-685562",
   "license": "MIT",
   "repository": {
     "type": "git",
@@ -46,9 +46,9 @@
     "python-ast": "^0.1.0",
     "table": "^6.9.0",
     "ts-node": "^10.9.2",
-    "@motiadev/workbench": "0.7.2-beta.134",
-    "@motiadev/stream-client-node": "0.7.2-beta.134",
-    "@motiadev/core": "0.7.2-beta.134"
+    "@motiadev/core": "0.7.2-beta.135-685562",
+    "@motiadev/stream-client-node": "0.7.2-beta.135-685562",
+    "@motiadev/workbench": "0.7.2-beta.135-685562"
   },
   "devDependencies": {
     "@amplitude/analytics-types": "^2.9.2",