Skip to main content

AI-enabled time and memory profiler for Python applications

Project description

Zero-Hassle AI Enabled Python Profiler for Time & Memory

AIPyMemTimeProfiler is a zero-configuration, AI-assisted Python profiler that automatically captures function-level memory and time usage — without any code changes.

Powered by LLMs (like Ollama), it not only measures performance but also analyzes your functions for inefficiencies and memory leaks.

Key Features

Zero code modifications — No decorators or wrappers needed

AI-powered analysis of selected functions

Function-level metrics for time, memory, arguments, return object size

Smart project detection — Profiles only your code, not external libraries

Structured JSON output for automation or visualization

Optional profiling of third-party libraries

✅ Supports Flask apps, CLI scripts, nested folder hierarchies

Installation

pip install aipymemtimeprofiler

CLI entry point

profile_code

Quick Start

1. Set Environment Variables

export PROFILER_FILE_PATH="/absolute/path/to/your_script.py"
export PROFILER_DIR_PATH="/absolute/path/to/your/project/root"

If you just want to try, you can test it with sample_project.

export PROFILER_FILE_PATH="$(pwd)/sample_project/inside/app.py": The Python file to be profiled
export PROFILER_DIR_PATH="$(pwd)/sample_project/inside": Root of your project for accurate filtering

2. If you want to verify 3rd party libraries

export INCLUDE_LIBRARIES=True

LLM Environment Setup

Download Ollama from Ollama

ollama run <yout_model>

If you don't know which model to use.

ollama run deepseek-r1:1.5b: It is preferable as it is light weight.

Set your model env variable.

export AGENT_NAME="<your_model>"
export AGENTIC_PROFILER=True

TL;DR

pip install aipymemtimeprofiler export PROFILER_FILE_PATH=./app.py export PROFILER_DIR_PATH=./ profile_code

You’ll be prompted to select a function to analyze. Example:

Index Function File Path
0 calculate_data /project/app/compute.py
1 main /project/app/main.py
N Skip Analysis -

Time & Memory Metrics

Automatically captures:

  • Max execution time (CPU)
  • Peak memory usage
  • RSS memory growth
  • Return object size
  • Arguments passed

Project-Aware Analysis

Only your code is profiled.
System libraries and external modules are ignored using intelligent project-path detection.

Profiler Metrics

The following table describes the metrics collected by the profiler:

Metric Description Key
Function Name The name of the function being profiled. function
File Path The absolute path to the file where the function is defined. file
Line Number The line number where the function starts in the file. line
Execution Time Maximum time taken by each function in milliseconds (ms). max_time_ms
CPU Time Time spent by the CPU on this function (ms). cpu_time_ms
Peak Memory Usage Maximum memory usage during function execution in kilobytes (KB). max_mem
RSS Memory Growth Growth in Resident Set Size (RSS) memory in kilobytes (KB), helps spot memory leaks. mem_growth_rss_kb
Arguments The arguments passed to the function being profiled. args
Possible Memory Leak Indicates if a potential memory leak is detected (if any). possible_memory_leak
Notes Any additional notes related to the profiling data. note
Returned Object Size The size of the returned object in bytes. return_obj

Option to select a function for analysis, which is analysed by the Ollama model installed and configured. This is the table providing the options for analysis.

Available Functions for Analysis

Index Function Name File Path
0 function_one /path/to/file_one.py
1 function_two /path/to/file_two.py
2 function_three /path/to/file_three.py
3 function_four /path/to/file_four.py
4 function_five /path/to/file_five.py
... ... ...
N function_n /path/to/file_n.py
N+1 Skip Analysis -

Structured JSON Reports

{ "function": "calculate_total", "file": "/app/logic.py", "line": 23, "cpu_time_ms": 15.3, "max_mem": 1024, "mem_growth_rss_kb": 300, "args": ["x", "y"], "return_obj": 124, "possible_memory_leak": false, "note": "" }

Works with Any Project Structure

Handles nested folder hierarchies easily — just point to your project root and go.


Supported Use Cases

  • Pure Python projects
  • Flask APIs and apps
  • Any directory layout

Want More?

  • Console table toggle
  • HTML report output

Pull requests are welcome!

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

aipymemtimeprofiler-1.0.2.tar.gz (15.3 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

aipymemtimeprofiler-1.0.2-py3-none-any.whl (17.5 kB view details)

Uploaded Python 3

File details

Details for the file aipymemtimeprofiler-1.0.2.tar.gz.

File metadata

  • Download URL: aipymemtimeprofiler-1.0.2.tar.gz
  • Upload date:
  • Size: 15.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.9.6

File hashes

Hashes for aipymemtimeprofiler-1.0.2.tar.gz
Algorithm Hash digest
SHA256 20198a995b118631c0704e73088f3f8153af510f8ac0c19d2ba1a26dfdde8d1c
MD5 a05ec2a4662e380d092de7aa49ef33ac
BLAKE2b-256 10decc8be2d1d65a488b205227d437d446dc848f4e0e71019fe6bb8d2b1c9765

See more details on using hashes here.

File details

Details for the file aipymemtimeprofiler-1.0.2-py3-none-any.whl.

File metadata

File hashes

Hashes for aipymemtimeprofiler-1.0.2-py3-none-any.whl
Algorithm Hash digest
SHA256 aab498ab8bb452c7d54e1aba941601e20c50d9580d619b0671d674035a7aceca
MD5 08823c770ee209f9b5e137a597e4306f
BLAKE2b-256 f6bcab10a0f9d46834f344169c90f63281aa35a28db23a1249ef1cab968d568d

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page